mirror of
https://github.com/bellingcat/auto-archiver-api.git
synced 2026-06-11 04:58:33 +03:00
fix to interoperability endpoint
This commit is contained in:
@@ -1,7 +1,8 @@
|
||||
from datetime import datetime, timedelta
|
||||
from unittest.mock import MagicMock, patch
|
||||
import pytest
|
||||
from app.shared.business_logic import get_store_archive_until
|
||||
from app.shared.business_logic import get_store_archive_until, get_store_archive_until_or_never
|
||||
|
||||
|
||||
class Test_get_store_archive_until:
|
||||
GROUP_ID = "test-group"
|
||||
@@ -11,6 +12,12 @@ class Test_get_store_archive_until:
|
||||
get_store_archive_until(db_session, self.GROUP_ID)
|
||||
assert str(exc.value) == f"Group {self.GROUP_ID} not found."
|
||||
|
||||
@patch("app.shared.db.worker_crud.get_group", return_value=MagicMock(permissions=None))
|
||||
def test_group_no_permissions(self, db_session):
|
||||
with pytest.raises(AssertionError) as exc:
|
||||
get_store_archive_until(db_session, self.GROUP_ID)
|
||||
assert str(exc.value) == f"Group {self.GROUP_ID} has no permissions."
|
||||
|
||||
@patch("app.shared.db.worker_crud.get_group")
|
||||
def test_no_max_lifespan(self, mock_get_group, db_session):
|
||||
group = MagicMock()
|
||||
@@ -29,8 +36,45 @@ class Test_get_store_archive_until:
|
||||
|
||||
result = get_store_archive_until(db_session, self.GROUP_ID)
|
||||
expected = datetime.now() + timedelta(days=180) # 6 months
|
||||
|
||||
|
||||
assert isinstance(result, datetime)
|
||||
# Allow 1 second difference due to execution time
|
||||
assert abs(result - expected) < timedelta(seconds=1)
|
||||
mock_get_group.assert_called_once_with(db_session, self.GROUP_ID)
|
||||
mock_get_group.assert_called_once_with(db_session, self.GROUP_ID)
|
||||
|
||||
|
||||
class Test_get_store_archive_until_or_never:
|
||||
GROUP_ID = "test-group"
|
||||
|
||||
def test_group_not_found(self, db_session):
|
||||
result = get_store_archive_until_or_never(db_session, self.GROUP_ID)
|
||||
assert result is None
|
||||
|
||||
@patch("app.shared.db.worker_crud.get_group", return_value=MagicMock(permissions=None))
|
||||
def test_group_no_permissions(self, db_session):
|
||||
result = get_store_archive_until_or_never(db_session, self.GROUP_ID)
|
||||
assert result is None
|
||||
|
||||
@patch("app.shared.db.worker_crud.get_group")
|
||||
def test_no_max_lifespan(self, mock_get_group, db_session):
|
||||
group = MagicMock()
|
||||
group.permissions = {"max_archive_lifespan_months": -1}
|
||||
mock_get_group.return_value = group
|
||||
|
||||
result = get_store_archive_until_or_never(db_session, self.GROUP_ID)
|
||||
assert result is None
|
||||
mock_get_group.assert_called_once_with(db_session, self.GROUP_ID)
|
||||
|
||||
@patch("app.shared.db.worker_crud.get_group")
|
||||
def test_with_max_lifespan(self, mock_get_group, db_session):
|
||||
group = MagicMock()
|
||||
group.permissions = {"max_archive_lifespan_months": 6}
|
||||
mock_get_group.return_value = group
|
||||
|
||||
result = get_store_archive_until_or_never(db_session, self.GROUP_ID)
|
||||
expected = datetime.now() + timedelta(days=180) # 6 months
|
||||
|
||||
assert isinstance(result, datetime)
|
||||
# Allow 1 second difference due to execution time
|
||||
assert abs(result - expected) < timedelta(seconds=5)
|
||||
mock_get_group.assert_called_once_with(db_session, self.GROUP_ID)
|
||||
|
||||
Reference in New Issue
Block a user