def test_deep_merge_conflict(self):
dict1 = {
'tssc-config': {
'step-foo': {
'implementer': 'foo1',
'config': {
'test0': 'foo',
'test1': 'foo'
}
}
}
}
dict2 = {
'tssc-config': {
'step-foo': {
'implementer': 'foo1',
'config': {
'test1': 'bar',
'test2': 'bar'
}
}
}
}
with self.assertRaisesRegex(
ValueError,
r"Conflict at tssc-config.step-foo.config.test1"):
deep_merge(dict1, dict2)
def merge_sub_step_env_config(self, new_sub_step_env_config):
"""Merge new sub step environment configuration into the existing
sub step environment configuration.
Parameters
----------
new_sub_step_env_config : dict
New sub step environment configuration to merge into
the existing sub step environment configuration.
Raises
------
ValueError
If new sub step environment configuration has duplicative leaf keys to
existing sub step environment configuration.
"""
if new_sub_step_env_config is not None:
try:
self.__sub_step_env_config = deep_merge(
self.__sub_step_env_config,
copy.deepcopy(new_sub_step_env_config))
except ValueError as error:
raise ValueError(
"Error merging new sub step environment configuration" +
" into existing sub step environment configuration" +
f" for sub step ({self.sub_step_name}) of step ({self.step_name}): {error}"
) from error
def test_deep_merge_no_conflict(self):
dict1 = {
'tssc-config': {
'step-foo': {
'implementer': 'foo1',
'config': {
'test0': 'foo',
'test1': 'foo'
}
}
}
}
dict2 = {
'tssc-config': {
'step-foo': {
'implementer': 'foo1',
'config': {
'test2': 'bar'
}
}
}
}
result = deep_merge(dict1, dict2)
# assert that it is in an inplace deep merge of the first param
self.assertEqual(result, dict1)
# assert expected merge result
self.assertEqual(result, {
'tssc-config': {
'step-foo': {
'implementer': 'foo1',
'config': {
'test0': 'foo',
'test1': 'foo',
'test2': 'bar'
}
}
}
})
def __add_config_dict(self, config_dict, source_file_path=None): # pylint: disable=too-many-locals, too-many-branches
"""Add a TSSC configuration dictionary to the list of TSSC configuration dictionaries.
Parameters
----------
config_dict : dict
A dictionary to validate as a TSSC configuration and to add to this Config.
source_file_path : str, optional
File path to the file from which the given config_dict came from.
Raises
------
AssertionError
If the given config_dict is not a valid TSSC configuration dictionary.
If attempt to update an existing sub step and new and existing sub step implementers
do not match.
If sub step does not define a step implementer.
ValueError
If duplicative leaf keys when merging global defaults
If duplicative leaf keys when merging global env defaults
If step config is not of type dict or list
If new sub step configuration has duplicative leaf keys to
existing sub step configuration.
If new sub step environment configuration has duplicative leaf keys to
existing sub step environment configuration.
"""
assert Config.TSSC_CONFIG_KEY in config_dict, \
"Failed to add invalid TSSC config. " + \
f"Missing expected top level key ({Config.TSSC_CONFIG_KEY}): " + \
f"{config_dict}"
# if file path given use that as the source when creating ConfigValue objects
# else use a copy of the given configuration dictionary
if source_file_path is not None:
parent_source = source_file_path
else:
parent_source = copy.deepcopy(config_dict)
# convert all the leaves of the configuration dictionary under
# the Config.TSSC_CONFIG_KEY to ConfigValue objects
tssc_config_values = ConfigValue.convert_leaves_to_config_values(
values=copy.deepcopy(config_dict[Config.TSSC_CONFIG_KEY]),
parent_source=parent_source,
path_parts=[Config.TSSC_CONFIG_KEY]
)
for key, value in tssc_config_values.items():
# if global default key
# else if global env defaults key
# else assume step config
if key == Config.TSSC_CONFIG_KEY_GLOBAL_DEFAULTS:
try:
self.__global_defaults = deep_merge(
copy.deepcopy(self.__global_defaults),
copy.deepcopy(value)
)
except ValueError as error:
raise ValueError(
f"Error merging global defaults: {error}"
) from error
elif key == Config.TSSC_CONFIG_KEY_GLOBAL_ENVIRONMENT_DEFAULTS:
for env, env_config in value.items():
if env not in self.__global_environment_defaults:
self.__global_environment_defaults[env] = {
Config.TSSC_CONFIG_KEY_ENVIRONMENT_NAME: env
}
try:
self.__global_environment_defaults[env] = deep_merge(
copy.deepcopy(self.__global_environment_defaults[env]),
copy.deepcopy(env_config)
)
except ValueError as error:
raise ValueError(
f"Error merging global environment ({env}) defaults: {error}"
) from error
elif key == Config.TSSC_CONFIG_KEY_DECRYPTORS:
config_decryptor_definitions = ConfigValue.convert_leaves_to_values(value)
Config.parse_and_register_decryptors_definitions(config_decryptor_definitions)
else:
step_name = key
step_config = value
# if step_config is dict then assume step with single sub step
if isinstance(step_config, dict):
sub_steps = [step_config]
elif isinstance(step_config, list):
sub_steps = step_config
else:
raise ValueError(
f"Expected step ({step_name}) to have have step config ({step_config})" +
f" of type dict or list but got: {type(step_config)}"
)
for sub_step in sub_steps:
assert Config.TSSC_CONFIG_KEY_STEP_IMPLEMENTER in sub_step, \
f"Step ({step_name}) defines a single sub step with values " + \
f"({sub_step}) but is missing value for key: " + \
f"{Config.TSSC_CONFIG_KEY_STEP_IMPLEMENTER}"
sub_step_implementer_name = \
sub_step[Config.TSSC_CONFIG_KEY_STEP_IMPLEMENTER].value
# if sub step name given
# else if no sub step name given use step implementer as sub step name
if Config.TSSC_CONFIG_KEY_SUB_STEP_NAME in sub_step:
sub_step_name = sub_step[Config.TSSC_CONFIG_KEY_SUB_STEP_NAME].value
else:
sub_step_name = sub_step_implementer_name
if Config.TSSC_CONFIG_KEY_SUB_STEP_CONFIG in sub_step:
sub_step_config = copy.deepcopy(
sub_step[Config.TSSC_CONFIG_KEY_SUB_STEP_CONFIG])
else:
sub_step_config = {}
if Config.TSSC_CONFIG_KEY_SUB_STEP_ENVIRONMENT_CONFIG in sub_step:
sub_step_env_config = copy.deepcopy(
sub_step[Config.TSSC_CONFIG_KEY_SUB_STEP_ENVIRONMENT_CONFIG])
else:
sub_step_env_config = {}
self.add_or_update_step_config(
step_name=step_name,
sub_step_name=sub_step_name,
sub_step_implementer_name=sub_step_implementer_name,
sub_step_config=sub_step_config,
sub_step_env_config=sub_step_env_config
)