From d9bc3fbf3c6f89635aa5e8b8ee7f156cf2047f5f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?V=C3=ADctor=20Fern=C3=A1ndez=20Poyatos?= Date: Wed, 12 Mar 2025 16:14:21 +0100 Subject: [PATCH] fix(tests): adapt unit tests to db connectors changes --- api/src/backend/api/tests/test_database.py | 12 +++++++----- api/src/backend/conftest.py | 1 + 2 files changed, 8 insertions(+), 5 deletions(-) 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):