class TestRouter(TestCase):
def setUp(self):
self.router = FIASRouter()
self.models = (getattr(fias.models, attr) for attr in fias_models)
self.rels = (getattr(fias.models, attr)
for attr in FIASRouter.ALLOWED_REL)
def test_read_db(self):
for model in self.models:
if issubclass(model, models.Model):
self.assertEquals(DATABASE_ALIAS,
self.router.db_for_read(model))
self.assertIsNone(self.router.db_for_read(FakeModel))
def test_write_db(self):
for model in self.models:
if issubclass(model, models.Model):
self.assertEquals(DATABASE_ALIAS,
self.router.db_for_write(model))
self.assertIsNone(self.router.db_for_write(FakeModel))
def test_relation(self):
for m1 in self.models:
for m2 in self.models:
self.assertTrue(self.router.allow_relation(m1, m2))
for m in self.rels:
self.assertFalse(self.router.allow_relation(FakeModel, m))
self.assertTrue(self.router.allow_relation(m, FakeModel))
self.assertIsNone(self.router.allow_relation(FakeModel, FakeModel2))
self.assertIsNone(self.router.allow_relation(FakeModel2, FakeModel))
def test_migrate(self):
method = self.router.allow_migrate
for model in self.models:
if issubclass(model, models.Model):
self.assertTrue(
method(DATABASE_ALIAS, model._meta.app_label, model))
self.assertFalse(
method(DEFAULT_DB_ALIAS, model._meta.app_label, model))
self.assertFalse(
method(DATABASE_ALIAS, FakeModel._meta.app_label, FakeModel))
self.assertIsNone(
method(DEFAULT_DB_ALIAS, FakeModel._meta.app_label, FakeModel))
class TestRouter(TestCase):
def setUp(self):
self.router = FIASRouter()
self.models = (getattr(fias.models, attr) for attr in fias_models)
self.rels = (getattr(fias.models, attr) for attr in FIASRouter.ALLOWED_REL)
def test_read_db(self):
for model in self.models:
if issubclass(model, Model):
self.assertEquals(FIAS_DATABASE_ALIAS, self.router.db_for_read(model))
self.assertIsNone(self.router.db_for_read(TestModel))
def test_write_db(self):
for model in self.models:
if issubclass(model, Model):
self.assertEquals(FIAS_DATABASE_ALIAS, self.router.db_for_write(model))
self.assertIsNone(self.router.db_for_write(TestModel))
def test_relation(self):
for m1 in self.models:
for m2 in self.models:
self.assertTrue(self.router.allow_relation(m1, m2))
for m in self.rels:
self.assertFalse(self.router.allow_relation(TestModel, m))
self.assertTrue(self.router.allow_relation(m, TestModel))
self.assertIsNone(self.router.allow_relation(TestModel, TestModel2))
self.assertIsNone(self.router.allow_relation(TestModel2, TestModel))
def test_syncdb(self):
if DJ_VERSION < 18:
method = self.router.allow_syncdb
else:
method = self.router.allow_migrate
for model in self.models:
if issubclass(model, Model):
self.assertTrue(method(FIAS_DATABASE_ALIAS, model))
self.assertFalse(method('default', model))
self.assertFalse(method(FIAS_DATABASE_ALIAS, TestModel))
self.assertIsNone(method('default', TestModel))