Squashed 'receiptwitness/' content from commit e8d374a

git-subtree-dir: receiptwitness
git-subtree-split: e8d374a89ed8978f429598e02d31b1c5963efe22

tests/test_session/test_manager.py

@@ -0,0 +1,102 @@
+"""Tests for session manager logic."""
+
+from datetime import UTC, datetime, timedelta
+from unittest.mock import AsyncMock, patch
+
+import pytest
+from cryptography.fernet import Fernet
+
+from receiptwitness.scrapers.base import SessionData
+from receiptwitness.session.manager import (
+    get_valid_session,
+    session_from_db_record,
+    session_to_db_value,
+)
+
+TEST_KEY = Fernet.generate_key().decode()
+
+
+@pytest.fixture(autouse=True)
+def _mock_encryption_key():
+    with patch("receiptwitness.session.encryption.settings") as mock_settings:
+        mock_settings.session_encryption_key = TEST_KEY
+        yield
+
+
+def _make_session(hours_until_expire: int = 4) -> SessionData:
+    now = datetime.now(UTC)
+    return SessionData(
+        cookies=[{"name": "sid", "value": "test", "domain": ".meijer.com"}],
+        user_agent="Mozilla/5.0",
+        created_at=now,
+        expires_at=now + timedelta(hours=hours_until_expire),
+    )
+
+
+class TestSessionSerialization:
+    def test_roundtrip(self):
+        session = _make_session()
+        db_value = session_to_db_value(session)
+        restored = session_from_db_record(db_value)
+
+        assert restored is not None
+        assert restored.cookies == session.cookies
+        assert restored.user_agent == session.user_agent
+
+    def test_none_returns_none(self):
+        assert session_from_db_record(None) is None
+
+    def test_invalid_encrypted_returns_none(self):
+        assert session_from_db_record("garbage-data") is None
+
+
+class TestGetValidSession:
+    @pytest.mark.asyncio
+    async def test_valid_existing_session(self):
+        session = _make_session()
+        db_value = session_to_db_value(session)
+
+        scraper = AsyncMock()
+        scraper.check_session.return_value = True
+
+        result, was_refreshed = await get_valid_session(scraper, db_value, "user", "pass")
+        assert not was_refreshed
+        assert result.cookies == session.cookies
+        scraper.login.assert_not_called()
+
+    @pytest.mark.asyncio
+    async def test_expired_session_triggers_login(self):
+        session = _make_session(hours_until_expire=-1)  # already expired
+        db_value = session_to_db_value(session)
+
+        new_session = _make_session()
+        scraper = AsyncMock()
+        scraper.login.return_value = new_session
+
+        result, was_refreshed = await get_valid_session(scraper, db_value, "user", "pass")
+        assert was_refreshed
+        scraper.login.assert_called_once_with("user", "pass")
+
+    @pytest.mark.asyncio
+    async def test_no_existing_session_triggers_login(self):
+        new_session = _make_session()
+        scraper = AsyncMock()
+        scraper.login.return_value = new_session
+
+        result, was_refreshed = await get_valid_session(scraper, None, "user", "pass")
+        assert was_refreshed
+        scraper.login.assert_called_once()
+
+    @pytest.mark.asyncio
+    async def test_failed_session_check_triggers_login(self):
+        session = _make_session()
+        db_value = session_to_db_value(session)
+
+        new_session = _make_session()
+        scraper = AsyncMock()
+        scraper.check_session.return_value = False
+        scraper.login.return_value = new_session
+
+        result, was_refreshed = await get_valid_session(scraper, db_value, "user", "pass")
+        assert was_refreshed
+        scraper.login.assert_called_once()