def setUp(self):
from django.contrib.auth import get_user_model
self.sut = ShardedRouter()
self.user = get_user_model().objects.create_user(
username='******',
password='******',
email='*****@*****.**')
def test_migrate_sharded_model_with_specific_database_will_not_work(self):
sut = ShardedRouter()
with patch.object(TestModel,
"django_sharding__database",
'blah',
create=True):
with patch.object(TestModel,
"django_sharding__is_sharded",
True,
create=True):
with self.assertRaises(InvalidMigrationException):
sut.allow_migrate(db='default',
app_label='tests',
model_name='TestModel')
with patch.object(TestModel,
"django_sharding__database",
'blah',
create=True):
with patch.object(TestModel,
"django_sharding__is_sharded",
False,
create=True):
sut.allow_migrate(db='default',
app_label='tests',
model_name='TestModel')
with patch.object(TestModel,
"django_sharding__database",
None,
create=True):
with patch.object(TestModel,
"django_sharding__is_sharded",
True,
create=True):
sut.allow_migrate(db='default',
app_label='tests',
model_name='TestModel')
sut.allow_migrate(db='default',
app_label='tests',
model_name='TestModel')
def test_migrate_sharded_model_with_specific_database_will_not_work(self):
sut = ShardedRouter()
sut.get_specific_database_or_none = lambda self: 'default'
sut.get_shard_group_if_sharded_or_none = lambda self: 'default'
with self.assertRaises(InvalidMigrationException):
sut.allow_migrate(db='default',
app_label='tests',
model_name='TestModel')
sut.get_specific_database_or_none = lambda self: None
sut.allow_migrate(db='default',
app_label='tests',
model_name='TestModel')
sut.get_specific_database_or_none = lambda self: 'default'
sut.get_shard_group_if_sharded_or_none = lambda self: None
sut.allow_migrate(db='default',
app_label='tests',
model_name='TestModel')
def setUp(self):
self.sut = ShardedRouter()
self.user = PostgresShardUser.objects.create_user(
username='******',
password='******',
email='*****@*****.**')
def setUp(self):
self.sut = ShardedRouter()
def setUp(self):
self.sut = ShardedRouter()
self.user = get_user_model().objects.create_user(
username='******',
password='******',
email='*****@*****.**')
