summaryrefslogtreecommitdiff
path: root/tests/migrations/routers.py
blob: 211c189772cb3184308c24af236e8e9268d1228f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
class DefaultOtherRouter:
    def allow_migrate(self, db, app_label, model_name=None, **hints):
        return db in {"default", "other"}


class TestRouter:
    def allow_migrate(self, db, app_label, model_name=None, **hints):
        """
        The Tribble model should be the only one to appear in the 'other' db.
        """
        if model_name == "tribble":
            return db == "other"
        elif db != "default":
            return False