From a3737c9892a71813a6ab22a6ab0db714b432c10c Mon Sep 17 00:00:00 2001
From: Fabio Caccamo <fabio.caccamo@gmail.com>
Date: Thu, 2 Mar 2023 14:09:38 +0100
Subject: [PATCH] Fix multidb tests.

---
 tests/test_multidb.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/tests/test_multidb.py b/tests/test_multidb.py
index e315d7d..456422e 100644
--- a/tests/test_multidb.py
+++ b/tests/test_multidb.py
@@ -1,10 +1,14 @@
 from django.test import TestCase
-from django.test.testcases import DatabaseOperationForbidden
 
 from admin_interface.models import Theme
 
 from .routers import DatabaseAppsRouter
 
+try:
+    from django.test.testcases import DatabaseOperationForbidden
+except ImportError:
+    DatabaseOperationForbidden = AssertionError
+
 
 class AdminInterfaceModelsWithDBRoutingTestCase(TestCase):
     databases = ["replica"]
@@ -12,7 +16,7 @@ class AdminInterfaceModelsWithDBRoutingTestCase(TestCase):
     def test_standard_dbrouter(self):
         router = DatabaseAppsRouter()
         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):
         DATABASE_APPS_MAPPING = {
@@ -20,7 +24,7 @@ class AdminInterfaceModelsWithDBRoutingTestCase(TestCase):
         }
         router = DatabaseAppsRouter(db_map=DATABASE_APPS_MAPPING)
         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):
         with self.assertRaises(DatabaseOperationForbidden):