Update Atlos tests

tests/databases/test_atlos_db.py

@@ -2,7 +2,7 @@ import pytest
 from datetime import datetime
 
 from auto_archiver.core import Metadata
-from auto_archiver.modules.atlos_db import AtlosDb
+from auto_archiver.modules.atlos_feeder_db_storage import AtlosFeederDbStorage as AtlosDb
 
 
 class FakeAPIResponse:
@@ -12,19 +12,28 @@ class FakeAPIResponse:
         self._data = data
         self.raise_error = raise_error
 
+    def json(self) -> dict:
+        return self._data
+
     def raise_for_status(self) -> None:
         if self.raise_error:
             raise Exception("HTTP error")
 
 
 @pytest.fixture
-def atlos_db(setup_module) -> AtlosDb:
+def atlos_db(setup_module, mocker) -> AtlosDb:
     """Fixture for AtlosDb."""
     configs: dict = {
         "api_token": "abc123",
         "atlos_url": "https://platform.atlos.org",
     }
-    return setup_module("atlos_db", configs)
+    mocker.patch("requests.Session")
+    atlos_feeder = setup_module("atlos_feeder_db_storage", configs)
+    fake_session = mocker.MagicMock()
+    # Configure the default response to have no results so that __iter__ terminates
+    fake_session.get.return_value = FakeAPIResponse({"next": None, "results": []})
+    atlos_feeder.session = fake_session
+    return atlos_feeder
 
 
 def test_failed_no_atlos_id(atlos_db, metadata, mocker):
@@ -38,25 +47,20 @@ def test_failed_with_atlos_id(atlos_db, metadata, mocker):
     """Test failed() posts failure when atlos_id is present."""
     metadata.set("atlos_id", 42)
     fake_resp = FakeAPIResponse({}, raise_error=False)
-    post_mock = mocker.patch("requests.post", return_value=fake_resp)
+    post_mock = mocker.patch.object(atlos_db, "_post", return_value=fake_resp)
     atlos_db.failed(metadata, "failure reason")
-    expected_url = (
-        f"{atlos_db.atlos_url}/api/v2/source_material/metadata/42/auto_archiver"
-    )
-    expected_headers = {"Authorization": f"Bearer {atlos_db.api_token}"}
+    expected_endpoint = f"/api/v2/source_material/metadata/42/auto_archiver"
     expected_json = {
         "metadata": {"processed": True, "status": "error", "error": "failure reason"}
     }
-    post_mock.assert_called_once_with(
-        expected_url, headers=expected_headers, json=expected_json
-    )
+    post_mock.assert_called_once_with(expected_endpoint, json=expected_json)
 
 
 def test_failed_http_error(atlos_db, metadata, mocker):
     """Test failed() raises exception on HTTP error."""
     metadata.set("atlos_id", 42)
-    fake_resp = FakeAPIResponse({}, raise_error=True)
-    mocker.patch("requests.post", return_value=fake_resp)
+    # Patch _post to raise an exception instead of returning a fake response.
+    mocker.patch.object(atlos_db, "_post", side_effect=Exception("HTTP error"))
     with pytest.raises(Exception, match="HTTP error"):
         atlos_db.failed(metadata, "failure reason")
 
@@ -81,12 +85,9 @@ def test_done_with_atlos_id(atlos_db, metadata, mocker):
     now = datetime.now()
     metadata.set("timestamp", now)
     fake_resp = FakeAPIResponse({}, raise_error=False)
-    post_mock = mocker.patch("requests.post", return_value=fake_resp)
+    post_mock = mocker.patch.object(atlos_db, "_post", return_value=fake_resp)
     atlos_db.done(metadata)
-    expected_url = (
-        f"{atlos_db.atlos_url}/api/v2/source_material/metadata/99/auto_archiver"
-    )
-    expected_headers = {"Authorization": f"Bearer {atlos_db.api_token}"}
+    expected_endpoint = f"/api/v2/source_material/metadata/99/auto_archiver"
     expected_results = metadata.metadata.copy()
     expected_results["timestamp"] = now.isoformat()
     expected_json = {
@@ -96,15 +97,13 @@ def test_done_with_atlos_id(atlos_db, metadata, mocker):
             "results": expected_results,
         }
     }
-    post_mock.assert_called_once_with(
-        expected_url, headers=expected_headers, json=expected_json
-    )
+    post_mock.assert_called_once_with(expected_endpoint, json=expected_json)
 
 
 def test_done_http_error(atlos_db, metadata, mocker):
-    """Test done() raises exception on HTTP error."""
+    """Test done() raises an exception on HTTP error."""
     metadata.set("atlos_id", 123)
-    fake_resp = FakeAPIResponse({}, raise_error=True)
-    mocker.patch("requests.post", return_value=fake_resp)
+    # Patch _post to raise an exception.
+    mocker.patch.object(atlos_db, "_post", side_effect=Exception("HTTP error"))
     with pytest.raises(Exception, match="HTTP error"):
         atlos_db.done(metadata)