summaryrefslogtreecommitdiff
path: root/tests/migrations/routers.py
blob: bc036382a731c089037fabc96a71477276c13298 (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