from unittest import skipIf

from django.test import TestCase

from admin_interface.models import Theme

from .routers import DatabaseAppsRouter


class AdminInterfaceModelsWithDBRoutingTestCase(TestCase):
    databases = ["replica"]

    def test_standard_dbrouter(self):
        router = DatabaseAppsRouter()
        db_for_theme = router.db_for_read(Theme)
        assert db_for_theme == "default"

    def test_dbrouter_selects_correct_db(self):
        DATABASE_APPS_MAPPING = {
            "admin_interface": "replica",
        }
        router = DatabaseAppsRouter(db_map=DATABASE_APPS_MAPPING)
        db_for_theme = router.db_for_read(Theme)
        assert db_for_theme == "replica"

    def test_dbrouter_errors_when_fetching_from_default(self):
        with self.assertRaises(Exception):
            Theme.get_active_theme()

    def test_dbrouter_fetches_db(self):
        DATABASE_APPS_MAPPING = {
            "admin_interface": "replica",
        }
        router = DatabaseAppsRouter(db_map=DATABASE_APPS_MAPPING)
        with self.settings(DATABASE_ROUTERS=[router]):
            Theme.get_active_theme()