mirror
/
auto-archiver
kopia lustrzana https://github.com/bellingcat/auto-archiver
1
0
Forkuj 0
Kod Zgłoszenia Packages Projekty Wydania Wiki Aktywność

Fix unit tests - make caplog checks more robust, having added a new logger/debug call

pull/249/head
Patrick Robertson 2025-03-13 16:07:49 +00:00
rodzic 10ceb7aa15
commit 2e25e59fa6
1 zmienionych plików z 3 dodań i 5 usunięć
Pokaż statystyki Ściągnij plik aktualizacji Ściągnij plik porównania Expand all files Collapse all files

8
tests/extractors/test_tiktok_tikwm_extractor.py
Wyświetl plik

@ -37,9 +37,8 @@ class TestTiktokTikwmExtractor(TestExtractorBase):
assert self.extractor.download(make_item(self.VALID_EXAMPLE_URL)) == False assert self.extractor.download(make_item(self.VALID_EXAMPLE_URL)) == False
mock_get.assert_called_once() mock_get.assert_called_once()
mock_get.return_value.json.assert_called_once() mock_get.return_value.json.assert_called_once()
assert len(caplog.records) == 2
# first message is just the 'Skipping using ytdlp to download files for TikTok' message # first message is just the 'Skipping using ytdlp to download files for TikTok' message
assert "failed to parse JSON response from tikwm.com for url='https://www.tiktok.com/@example/video/1234'" in caplog.records[1].message assert "failed to parse JSON response from tikwm.com for url='https://www.tiktok.com/@example/video/1234'" in caplog.text
mock_get.return_value.json.side_effect = Exception mock_get.return_value.json.side_effect = Exception
with caplog.at_level('ERROR'): with caplog.at_level('ERROR'):
@ -47,8 +46,7 @@ class TestTiktokTikwmExtractor(TestExtractorBase):
mock_get.assert_called() mock_get.assert_called()
assert mock_get.call_count == 2 assert mock_get.call_count == 2
assert mock_get.return_value.json.call_count == 2 assert mock_get.return_value.json.call_count == 2
assert len(caplog.records) == 2 assert "failed to parse JSON response from tikwm.com for url='https://www.tiktok.com/@example/video/1234'" in caplog.text
assert "failed to parse JSON response from tikwm.com for url='https://www.tiktok.com/@example/video/1234'" in caplog.records[1].message
@pytest.mark.parametrize("response", [ @pytest.mark.parametrize("response", [
({"msg": "failure"}), ({"msg": "failure"}),
@ -61,7 +59,7 @@ class TestTiktokTikwmExtractor(TestExtractorBase):
assert self.extractor.download(make_item(self.VALID_EXAMPLE_URL)) == False assert self.extractor.download(make_item(self.VALID_EXAMPLE_URL)) == False
mock_get.assert_called_once() mock_get.assert_called_once()
mock_get.return_value.json.assert_called_once() mock_get.return_value.json.assert_called_once()
assert "failed to get a valid response from tikwm.com" in caplog.records[1].message assert "failed to get a valid response from tikwm.com" in caplog.text
@pytest.mark.parametrize("response,has_vid", [ @pytest.mark.parametrize("response,has_vid", [
({"data": {"id": 123}}, False), ({"data": {"id": 123}}, False),