Make compatible with multi-db configuration

ONSdigital · Nov 27, 2024 · 2a0fc49 · 2a0fc49
1 parent 9f125fb
commit 2a0fc49
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/cms/settings/functional_test.py b/cms/settings/functional_test.py
@@ -1,3 +1,5 @@
+import copy
+
 from .base import *  # noqa: F403  # pylint: disable=wildcard-import,unused-wildcard-import
 
 # Debugging to be enabled locally only
@@ -33,5 +35,9 @@
 # We don't need such a strong algorithm in tests, so use MD5
 PASSWORD_HASHERS = ["django.contrib.auth.hashers.MD5PasswordHasher"]
 
-DATABASES = {"default": dj_database_url.config(default="postgres://ons:ons@localhost:15432/ons")}  # noqa: F405
+DATABASES = {
+    "default": dj_database_url.config(default="postgres://ons:ons@localhost:15432/ons"),  # noqa: F405
+}
+DATABASES["read_replica"] = copy.deepcopy(DATABASES["default"])
+
 REDIS_URL = "redis://localhost:16379"
diff --git a/functional_tests/behave_fixtures.py b/functional_tests/behave_fixtures.py
@@ -51,6 +51,9 @@ def django_test_case(context: Context) -> None:
     # use of whitenoise for serving static content.
     context.test_case_class = LiveServerTestCase
 
+    # Ensure the test case matches our multi DB configuration
+    context.test_case_class.databases = frozenset({"default", "read_replica"})
+
     # The LiveServerTestCase setup will start a test server for our app, allowing us to test against a live app
     context.test_case_class.setUpClass()