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)
