Briq
/
django-admin-interface
6
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix multidb tests.

master
Fabio Caccamo 2023-03-02 14:09:38 +01:00
parent 4deefc20b7
commit a3737c9892
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_multidb.py
View File

@ -1,10 +1,14 @@
from django.test import TestCase from django.test import TestCase
from django.test.testcases import DatabaseOperationForbidden
from admin_interface.models import Theme from admin_interface.models import Theme
from .routers import DatabaseAppsRouter from .routers import DatabaseAppsRouter
try:
from django.test.testcases import DatabaseOperationForbidden
except ImportError:
DatabaseOperationForbidden = AssertionError
class AdminInterfaceModelsWithDBRoutingTestCase(TestCase): class AdminInterfaceModelsWithDBRoutingTestCase(TestCase):
databases = ["replica"] databases = ["replica"]
@ -12,7 +16,7 @@ class AdminInterfaceModelsWithDBRoutingTestCase(TestCase):
def test_standard_dbrouter(self): def test_standard_dbrouter(self):
router = DatabaseAppsRouter() router = DatabaseAppsRouter()
db_for_theme = router.db_for_read(Theme) db_for_theme = router.db_for_read(Theme)
assert db_for_theme == "default" self.assertEqual(db_for_theme, "default")
def test_dbrouter_selects_correct_db(self): def test_dbrouter_selects_correct_db(self):
DATABASE_APPS_MAPPING = { DATABASE_APPS_MAPPING = {
@ -20,7 +24,7 @@ class AdminInterfaceModelsWithDBRoutingTestCase(TestCase):
} }
router = DatabaseAppsRouter(db_map=DATABASE_APPS_MAPPING) router = DatabaseAppsRouter(db_map=DATABASE_APPS_MAPPING)
db_for_theme = router.db_for_read(Theme) db_for_theme = router.db_for_read(Theme)
assert db_for_theme == "replica" self.assertEqual(db_for_theme, "replica")
def test_dbrouter_errors_when_fetching_from_default(self): def test_dbrouter_errors_when_fetching_from_default(self):
with self.assertRaises(DatabaseOperationForbidden): with self.assertRaises(DatabaseOperationForbidden):