47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
from unittest.mock import Mock
|
|
from scrapers.cses import scrape, scrape_all_problems
|
|
from scrapers.models import ProblemSummary
|
|
|
|
|
|
def test_scrape_success(mocker, mock_cses_html):
|
|
mock_response = Mock()
|
|
mock_response.text = mock_cses_html
|
|
|
|
mocker.patch("scrapers.cses.requests.get", return_value=mock_response)
|
|
|
|
result = scrape("https://cses.fi/problemset/task/1068")
|
|
|
|
assert len(result) == 1
|
|
assert result[0].input == "3\n1 2 3"
|
|
assert result[0].expected == "6"
|
|
|
|
|
|
def test_scrape_all_problems(mocker):
|
|
mock_response = Mock()
|
|
mock_response.text = """
|
|
<h1>Introductory Problems</h1>
|
|
<a href="/problemset/task/1068">Weird Algorithm</a>
|
|
<a href="/problemset/task/1083">Missing Number</a>
|
|
<h1>Sorting and Searching</h1>
|
|
<a href="/problemset/task/1084">Apartments</a>
|
|
"""
|
|
|
|
mocker.patch("scrapers.cses.requests.get", return_value=mock_response)
|
|
|
|
result = scrape_all_problems()
|
|
|
|
assert "Introductory Problems" in result
|
|
assert "Sorting and Searching" in result
|
|
assert len(result["Introductory Problems"]) == 2
|
|
assert result["Introductory Problems"][0] == ProblemSummary(
|
|
id="1068",
|
|
name="Weird Algorithm",
|
|
)
|
|
|
|
|
|
def test_scrape_network_error(mocker):
|
|
mocker.patch("scrapers.cses.requests.get", side_effect=Exception("Network error"))
|
|
|
|
result = scrape("https://cses.fi/problemset/task/1068")
|
|
|
|
assert result == []
|