mirror of
https://github.com/bellingcat/auto-archiver-api.git
synced 2026-06-11 13:08:34 +03:00
7e48f706df
* Update pyproject.toml * add pre-commit * Create .pre-commit-config.yaml * Comment out ruff * Update .pre-commit-config.yaml * General formatting * Create format-and-fail.yml * Update ci.yml * Add pre-commit to dev dependencies * Update pyproject.toml
86 lines
3.4 KiB
Python
86 lines
3.4 KiB
Python
from datetime import datetime, timedelta
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
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"
|
|
|
|
def test_group_not_found(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} 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()
|
|
group.permissions = {"max_archive_lifespan_months": -1}
|
|
mock_get_group.return_value = group
|
|
|
|
result = get_store_archive_until(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(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)
|
|
|
|
|
|
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)
|