Move database router to flagsmith-common

Author: emyller

tests/unit/task_processor/test_unit_task_processor_routers.py

@@ -1 +1,55 @@
-# TODO
+@pytest.mark.parametrize(
+    "given_settings, expected",
+    [
+        ({"TASK_PROCESSOR_DATABASE_URL": "postgres://localhost:5432/db"}, True),
+        ({"TASK_PROCESSOR_DATABASE_URL": None}, False),
+        ({"TASK_PROCESSOR_DATABASE_NAME": "db"}, True),
+        ({"TASK_PROCESSOR_DATABASE_NAME": None}, False),
+    ],
+)
+def test_TaskProcessorRouter__checks_whether_is_enabled(
+    settings: SettingsWrapper,
+    given_settings: dict[str, None | str],
+    expected: bool,
+) -> None:
+    # Given
+    for key, value in given_settings.items():
+        setattr(settings, key, value)
+
+    # When
+    router = TaskProcessorRouter()
+
+    # Then
+    assert router.is_enabled is expected
+
+
+@pytest.mark.parametrize("model", apps.get_app_config("task_processor").get_models())
+def test_TaskProcessorRouter__enabled__routes_queries_to_task_processor_database(
+    mocker: MockerFixture,
+    model: type[Model],
+) -> None:
+    # Given
+    mocker.patch.object(TaskProcessorRouter, "is_enabled", new=True)
+
+    # When
+    router = TaskProcessorRouter()
+    result = router.db_for_read(model)
+
+    # Then
+    assert result == "task_processor"
+
+
+@pytest.mark.parametrize("model", apps.get_app_config("task_processor").get_models())
+def test_TaskProcessorRouter__disabled__routes_queries_to_default_database(
+    mocker: MockerFixture,
+    model: type[Model],
+) -> None:
+    # Given
+    mocker.patch.object(TaskProcessorRouter, "is_enabled", new=False)
+
+    # When
+    router = TaskProcessorRouter()
+    result = router.db_for_read(model)
+
+    # Then
+    assert result is None