from unittest.mock import Mock from scrapers.atcoder import scrape, scrape_contest_problems, scrape_contests from scrapers.models import ContestSummary, ProblemSummary def test_scrape_success(mocker, mock_atcoder_html): mock_response = Mock() mock_response.text = mock_atcoder_html mocker.patch("scrapers.atcoder.requests.get", return_value=mock_response) result = scrape("https://atcoder.jp/contests/abc350/tasks/abc350_a") assert len(result) == 1 assert result[0].input == "3\n1 2 3" assert result[0].expected == "6" def test_scrape_contest_problems(mocker): mock_response = Mock() mock_response.text = """

Task	Name
	A - Water Tank
	B - Dentist Aoki

""" mocker.patch("scrapers.atcoder.requests.get", return_value=mock_response) result = scrape_contest_problems("abc350") assert len(result) == 2 assert result[0] == ProblemSummary(id="a", name="A - Water Tank") assert result[1] == ProblemSummary(id="b", name="B - Dentist Aoki") def test_scrape_network_error(mocker): mocker.patch( "scrapers.atcoder.requests.get", side_effect=Exception("Network error") ) result = scrape("https://atcoder.jp/contests/abc350/tasks/abc350_a") assert result == [] def test_scrape_contests_success(mocker): def mock_get_side_effect(url, **kwargs): if "page=1" in url: mock_response = Mock() mock_response.text = """

Start Time	Contest Name	Duration	Rated Range
2025-01-15 21:00:00+0900	AtCoder Beginner Contest 350	01:40	- 1999
2025-01-14 21:00:00+0900	AtCoder Regular Contest 170	02:00	1000 - 2799

""" return mock_response else: # Return empty page for all other pages mock_response = Mock() mock_response.text = "No table found" return mock_response mocker.patch("scrapers.atcoder.requests.get", side_effect=mock_get_side_effect) mocker.patch("scrapers.atcoder.time.sleep") result = scrape_contests() assert len(result) == 2 assert result[0] == ContestSummary( id="abc350", name="AtCoder Beginner Contest 350", display_name="Beginner Contest 350 (ABC)", ) assert result[1] == ContestSummary( id="arc170", name="AtCoder Regular Contest 170", display_name="Regular Contest 170 (ARC)", ) def test_scrape_contests_no_table(mocker): mock_response = Mock() mock_response.text = "No table found" mocker.patch("scrapers.atcoder.requests.get", return_value=mock_response) mocker.patch("scrapers.atcoder.time.sleep") result = scrape_contests() assert result == [] def test_scrape_contests_network_error(mocker): mocker.patch( "scrapers.atcoder.requests.get", side_effect=Exception("Network error") ) mocker.patch("scrapers.atcoder.time.sleep") result = scrape_contests() assert result == []