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)