feat(ci): pytest
This commit is contained in:
parent
ca6f8417c0
commit
51fd6e3676
8 changed files with 289 additions and 2 deletions
54
tests/scrapers/test_codeforces.py
Normal file
54
tests/scrapers/test_codeforces.py
Normal file
|
|
@ -0,0 +1,54 @@
|
|||
import json
|
||||
from unittest.mock import Mock
|
||||
import pytest
|
||||
from scrapers.codeforces import scrape, scrape_contest_problems
|
||||
|
||||
|
||||
def test_scrape_success(mocker, mock_codeforces_html):
|
||||
mock_scraper = Mock()
|
||||
mock_response = Mock()
|
||||
mock_response.text = mock_codeforces_html
|
||||
mock_scraper.get.return_value = mock_response
|
||||
|
||||
mocker.patch(
|
||||
"scrapers.codeforces.cloudscraper.create_scraper", return_value=mock_scraper
|
||||
)
|
||||
|
||||
result = scrape("https://codeforces.com/contest/1900/problem/A")
|
||||
|
||||
assert len(result) == 1
|
||||
assert result[0][0] == "1\n3\n1 2 3"
|
||||
assert result[0][1] == "6"
|
||||
|
||||
|
||||
def test_scrape_contest_problems(mocker):
|
||||
mock_scraper = Mock()
|
||||
mock_response = Mock()
|
||||
mock_response.text = """
|
||||
<a href="/contest/1900/problem/A">A. Problem A</a>
|
||||
<a href="/contest/1900/problem/B">B. Problem B</a>
|
||||
"""
|
||||
mock_scraper.get.return_value = mock_response
|
||||
|
||||
mocker.patch(
|
||||
"scrapers.codeforces.cloudscraper.create_scraper", return_value=mock_scraper
|
||||
)
|
||||
|
||||
result = scrape_contest_problems("1900")
|
||||
|
||||
assert len(result) == 2
|
||||
assert result[0] == {"id": "a", "name": "A. Problem A"}
|
||||
assert result[1] == {"id": "b", "name": "B. Problem B"}
|
||||
|
||||
|
||||
def test_scrape_network_error(mocker):
|
||||
mock_scraper = Mock()
|
||||
mock_scraper.get.side_effect = Exception("Network error")
|
||||
|
||||
mocker.patch(
|
||||
"scrapers.codeforces.cloudscraper.create_scraper", return_value=mock_scraper
|
||||
)
|
||||
|
||||
result = scrape("https://codeforces.com/contest/1900/problem/A")
|
||||
|
||||
assert result == []
|
||||
Loading…
Add table
Add a link
Reference in a new issue