def test_get_shard_method_defined_but_unimplimented(self):
from django.contrib.auth import get_user_model
User = get_user_model()
sut = BaseBucketingStrategy(shard_group='default')
with self.assertRaises(NotImplementedError):
sut.get_shard(User)
def test_get_shard_method_defined_but_unimplemented(self):
from django.contrib.auth import get_user_model
User = get_user_model()
sut = BaseBucketingStrategy(shard_group='default')
with self.assertRaises(NotImplementedError):
sut.get_shard(User)
def test_get_shards(self):
sut = BaseBucketingStrategy(shard_group='default')
result = sut.get_shards(settings.DATABASES)
expected_result = ['app_shard_001', 'app_shard_002']
self.assertEqual(sorted(result), expected_result)
def test_get_shards(self):
sut = BaseBucketingStrategy(shard_group='default')
result = sut.get_shards(settings.DATABASES)
expected_result = ['app_shard_001', 'app_shard_002']
self.assertEqual(sorted(result), expected_result)
