def test_adds_sharded_state_to_dict_when_sharded(self): result = database_config( environment_variable='TEST_ENV_VAR', default_database_url=self.default_database_url, shard_group='default') expected_result = {'SHARD_GROUP': 'default', 'TEST': {}} expected_result.update(self.dj_database_config) self.assertEqual(result, expected_result)
def test_no_extra_args_equivalent_to_dj_database_with_non_shard_status_added( self): result = database_config( environment_variable='TEST_ENV_VAR', default_database_url=self.default_database_url) expected_result = {'SHARD_GROUP': None, 'TEST': {}} expected_result.update(self.dj_database_config) self.assertEqual(result, expected_result)
def test_adds_replica_state_to_dict(self): result = database_config( environment_variable='TEST_ENV_VAR', default_database_url=self.default_database_url, is_replica_of='some_database') expected_result = { 'PRIMARY': 'some_database', 'SHARD_GROUP': None, 'TEST': { 'MIRROR': 'some_database' } } expected_result.update(self.dj_database_config) self.assertEqual(result, expected_result)
def test_returns_none_when_no_env_var_and_no_url(self): result = database_config(environment_variable='TEST_ENV_VAR', default_database_url=None, shard_group='default') expected_result = {} self.assertEqual(result, expected_result)
def test_adds_replica_state_to_dict_also_when_sharded(self): result = database_config(environment_variable='TEST_ENV_VAR', default_database_url=self.default_database_url, shard_group='default_set_of_shards', is_replica_of='some_database') expected_result = {'PRIMARY': 'some_database', 'SHARD_GROUP': 'default_set_of_shards', 'TEST': {'MIRROR': 'some_database'}} expected_result.update(self.dj_database_config) self.assertEqual(result, expected_result)
def test_adds_sharded_state_to_dict_when_sharded(self): result = database_config(environment_variable='TEST_ENV_VAR', default_database_url=self.default_database_url, shard_group='default') expected_result = {'SHARD_GROUP': 'default', 'TEST': {}} expected_result.update(self.dj_database_config) self.assertEqual(result, expected_result)
def test_no_extra_args_equivalent_to_dj_database_with_non_shard_status_added(self): result = database_config(environment_variable='TEST_ENV_VAR', default_database_url=self.default_database_url) expected_result = {'SHARD_GROUP': None, 'TEST': {}} expected_result.update(self.dj_database_config) self.assertEqual(result, expected_result)
def test_returns_none_when_no_env_var_and_no_url(self): result = database_config(environment_variable='TEST_ENV_VAR', default_database_url=None, shard_group='default') expected_result = {} self.assertEqual(result, expected_result)
def test_adds_replica_state_to_dict_also_when_sharded(self): result = database_config(environment_variable='TEST_ENV_VAR', default_database_url=self.default_database_url, shard_group='default_set_of_shards', is_replica_of='some_database') expected_result = {'PRIMARY': 'some_database', 'SHARD_GROUP': 'default_set_of_shards', 'TEST': {'MIRROR': 'some_database'}} expected_result.update(self.db_config) self.assertEqual(result, expected_result)