p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 Projects Releases Wiki Activity

Common: Add AgentConfiguration.to_mapping()

This commit is contained in:
Mike Salvatore 2022-07-22 12:03:55 -04:00
parent b6703becbc
commit d62d81a795
2 changed files with 13 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
monkey/common/configuration/agent_configuration.py
View File

@ -79,6 +79,16 @@ class AgentConfiguration:
except MarshmallowError as err: except MarshmallowError as err:
raise InvalidConfigurationError(str(err)) raise InvalidConfigurationError(str(err))
@staticmethod
def to_mapping(config: AgentConfiguration) -> Mapping[str, Any]:
"""
Serialize an AgentConfiguration to a Mapping
:param config: An AgentConfiguration
:return: A Mapping that represents the AgentConfiguration
"""
return AgentConfigurationSchema().dump(config)
@staticmethod @staticmethod
def to_json(config: AgentConfiguration) -> str: def to_json(config: AgentConfiguration) -> str:
""" """

9
monkey/tests/unit_tests/common/configuration/test_agent_configuration.py
View File

@ -178,13 +178,10 @@ def test_incorrect_type():
AgentConfiguration(**valid_config_dict) AgentConfiguration(**valid_config_dict)
def test_from_dict(): def test_to_from_mapping():
schema = AgentConfigurationSchema() config = AgentConfiguration.from_mapping(AGENT_CONFIGURATION)
dict_ = deepcopy(AGENT_CONFIGURATION)
config = AgentConfiguration.from_mapping(dict_) assert AgentConfiguration.to_mapping(config) == AGENT_CONFIGURATION
assert schema.dump(config) == dict_
def test_from_dict__invalid_data(): def test_from_dict__invalid_data():