diff --git a/api/src/backend/api/tests/test_database.py b/api/src/backend/api/tests/test_database.py index 46d3203414..daaa3a3b1e 100644 --- a/api/src/backend/api/tests/test_database.py +++ b/api/src/backend/api/tests/test_database.py @@ -1,15 +1,17 @@ +from unittest.mock import patch + import pytest +from config.django.base import DATABASE_ROUTERS as PROD_DATABASE_ROUTERS from django.conf import settings from django.db.migrations.recorder import MigrationRecorder from django.db.utils import ConnectionRouter from api.db_router import MainRouter from api.rls import Tenant -from config.django.base import DATABASE_ROUTERS as PROD_DATABASE_ROUTERS -from unittest.mock import patch @patch("api.db_router.MainRouter.admin_db", new="admin") +@patch("api.db_router.MainRouter.admin_read", new="admin_read") class TestMainDatabaseRouter: @pytest.fixture(scope="module") def router(self): @@ -20,12 +22,12 @@ class TestMainDatabaseRouter: @pytest.mark.parametrize("api_model", [Tenant]) def test_router_api_models(self, api_model, router): - assert router.db_for_read(api_model) == "default" + assert router.db_for_read(api_model) == "prowler_user_read" assert router.db_for_write(api_model) == "default" assert router.allow_migrate_model(MainRouter.admin_db, api_model) assert not router.allow_migrate_model("default", api_model) def test_router_django_models(self, router): - assert router.db_for_read(MigrationRecorder.Migration) == MainRouter.admin_db - assert not router.db_for_read(MigrationRecorder.Migration) == "default" + assert router.db_for_read(MigrationRecorder.Migration) == MainRouter.admin_read + assert router.db_for_read(MigrationRecorder.Migration) != "default" diff --git a/api/src/backend/conftest.py b/api/src/backend/conftest.py index c967ff8c63..0bffdf4f75 100644 --- a/api/src/backend/conftest.py +++ b/api/src/backend/conftest.py @@ -890,6 +890,7 @@ def pytest_configure(config): # Apply the mock before the test session starts. This is necessary to avoid admin error when running the # 0004_rbac_missing_admin_roles migration patch("api.db_router.MainRouter.admin_db", new="default").start() + patch("api.db_router.MainRouter.admin_read", new="default").start() def pytest_unconfigure(config):