bellingcat/auto-archiver
1
0
Fork 0
mirror of https://github.com/bellingcat/auto-archiver.git synced 2026-06-07 19:08:30 +03:00
Code Issues Packages Projects Releases Wiki Activity

closes #330

Browse Source
This commit is contained in:
msramalho
2025-06-18 16:40:55 +01:00
parent 4a36e6f6b0
commit 592dc30415
3 changed files with 66 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
tests/extractors/test_instagram_api_extractor.py
View File

@@ -1,4 +1,5 @@
from datetime import datetime
import math
import pytest
@@ -147,14 +148,14 @@ class TestInstagramAPIExtractor(TestExtractorBase):
self.extractor.full_profile = True
mock_call.side_effect = [mock_user_response, mock_story_response]
mock_highlights.return_value = None
mock_highlights.return_value = 1
mock_stories.return_value = mock_story_response
mock_posts.return_value = None
mock_tagged.return_value = None
mock_posts.return_value = 2
mock_tagged.return_value = 3
result = self.extractor.download_profile(metadata, "test_user")
assert result.get("#stories") == len(mock_story_response)
mock_posts.assert_called_once_with(result, "123")
mock_posts.assert_called_once_with(result, "123", max_to_download=math.inf)
assert "errors" not in result.metadata
def test_download_profile_not_found(self, metadata, mocker):
@@ -175,10 +176,10 @@ class TestInstagramAPIExtractor(TestExtractorBase):
self.extractor.full_profile = True
mock_call.side_effect = [mock_user_response, Exception("Stories API failed"), Exception("Posts API failed")]
mock_highlights.return_value = None
mock_tagged.return_value = None
mock_highlights.return_value = 1
mock_tagged.return_value = 2
stories_tagged.return_value = None
mock_posts.return_value = None
mock_posts.return_value = 4
result = self.extractor.download_profile(metadata, "test_user")
assert result.is_success()