Updated allow_migrate() signature in check framework tests

This commit is contained in:
Markus Holtermann 2016-02-10 15:44:04 +11:00
parent 228427ab1a
commit 18afd50a2b
2 changed files with 3 additions and 3 deletions

View File

@ -313,7 +313,7 @@ class Field(RegisterLookupMixin):
def _check_backend_specific_checks(self, **kwargs):
app_label = self.model._meta.app_label
for db in connections:
if router.allow_migrate(db, app_label, model=self.model):
if router.allow_migrate(db, app_label, model_name=self.model._meta.model_name):
return connections[db].validation.check_field(self, **kwargs)
return []

View File

@ -7,8 +7,8 @@ class TestRouter(object):
"""
Routes to the 'other' database if the model name starts with 'Other'.
"""
def allow_migrate(self, db, app_label, model=None, **hints):
return db == ('other' if model._meta.verbose_name.startswith('other') else 'default')
def allow_migrate(self, db, app_label, model_name=None, **hints):
return db == ('other' if model_name.startswith('other') else 'default')
@override_settings(DATABASE_ROUTERS=[TestRouter()])