update unit tests

climatepolicyradar · Oct 19, 2023 · 2830ab4 · 2830ab4
1 parent fcfb586
commit 2830ab4
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/unit_tests/mocks/services/config_service.py b/unit_tests/mocks/services/config_service.py
@@ -1,7 +1,7 @@
 from pytest import MonkeyPatch
 from app.errors import RepositoryError
 
-from app.model.config import ConfigReadDTO
+from app.model.config import ConfigReadDTO, DocumentConfig, EventConfig
 
 
 def mock_config_service(config_service, monkeypatch: MonkeyPatch, mocker):
@@ -13,7 +13,13 @@ def maybe_throw():
 
     def mock_get_config() -> ConfigReadDTO:
         maybe_throw()
-        return ConfigReadDTO(geographies=[], taxonomies={}, languages={})
+        return ConfigReadDTO(
+            geographies=[],
+            taxonomies={},
+            languages={},
+            document=DocumentConfig(roles=[], types=[], variants=[]),
+            event=EventConfig(types=[]),
+        )
 
     monkeypatch.setattr(config_service, "get", mock_get_config)
     mocker.spy(config_service, "get")
diff --git a/unit_tests/routers/test_config.py b/unit_tests/routers/test_config.py
@@ -10,6 +10,8 @@ def test_get_when_ok(client: TestClient, user_header_token, config_service_mock)
     assert "geographies" in keys
     assert "taxonomies" in keys
     assert "languages" in keys
+    assert "event" in keys
+    assert "document" in keys
     assert config_service_mock.get.call_count == 1