From 91ce43e529d067f1e35cdd314c1f6cd4fa507be6 Mon Sep 17 00:00:00 2001 From: Barrett Ruth Date: Tue, 30 Sep 2025 22:37:59 -0400 Subject: [PATCH] fix(test): fix mock --- tests/scrapers/test_codeforces.py | 28 ++++++++++------------------ 1 file changed, 10 insertions(+), 18 deletions(-) diff --git a/tests/scrapers/test_codeforces.py b/tests/scrapers/test_codeforces.py index e1e555f..6971ed6 100644 --- a/tests/scrapers/test_codeforces.py +++ b/tests/scrapers/test_codeforces.py @@ -4,18 +4,10 @@ from scrapers.codeforces import CodeforcesScraper from scrapers.models import ContestSummary, ProblemSummary -def make_mock_session(html: str): - """Return a mock StealthySession that yields the given HTML.""" - mock_session = Mock() - mock_session.fetch.return_value.html_content = html - mock_session.__enter__ = lambda s: s - mock_session.__exit__ = lambda s, exc_type, exc_val, exc_tb: None - return mock_session - - def test_scrape_success(mocker, mock_codeforces_html): - mock_session = make_mock_session(mock_codeforces_html) - mocker.patch("scrapers.codeforces.StealthySession", return_value=mock_session) + mock_page = Mock() + mock_page.html_content = mock_codeforces_html + mocker.patch("scrapers.codeforces.StealthyFetcher.fetch", return_value=mock_page) scraper = CodeforcesScraper() result = scraper.scrape_problem_tests("1900", "A") @@ -31,8 +23,9 @@ def test_scrape_contest_problems(mocker): A. Problem A B. Problem B """ - mock_session = make_mock_session(html) - mocker.patch("scrapers.codeforces.StealthySession", return_value=mock_session) + mock_page = Mock() + mock_page.html_content = html + mocker.patch("scrapers.codeforces.StealthyFetcher.fetch", return_value=mock_page) scraper = CodeforcesScraper() result = scraper.scrape_contest_metadata("1900") @@ -44,11 +37,10 @@ def test_scrape_contest_problems(mocker): def test_scrape_network_error(mocker): - mock_session = Mock() - mock_session.fetch.side_effect = Exception("Network error") - mock_session.__enter__ = lambda s: s - mock_session.__exit__ = lambda s, exc_type, exc_val, exc_tb: None - mocker.patch("scrapers.codeforces.StealthySession", return_value=mock_session) + mocker.patch( + "scrapers.codeforces.StealthyFetcher.fetch", + side_effect=Exception("Network error"), + ) scraper = CodeforcesScraper() result = scraper.scrape_problem_tests("1900", "A")