auto-archiver-api/app/tests/shared/test_business_logic.py

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 TestGetStoreArchiveUntil:
    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 TestGetStoreArchiveUntilOrNever:
    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)