ben.carrasco/prowler
1
0
Fork 0
mirror of https://github.com/prowler-cloud/prowler.git synced 2026-07-23 20:42:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(tests): adapt unit tests to db connectors changes

Browse Source
This commit is contained in:
Víctor Fernández Poyatos
2025-03-12 16:14:21 +01:00
parent 178d398aa1
commit d9bc3fbf3c
2 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
api/src/backend/api/tests/test_database.py
+7 -5
View File
@@ -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"
api/src/backend/conftest.py
+1
View File
@@ -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):