p34709852
/
monkey
forked from p15670423/monkey
1
0
Fork 1
Code Issues Pull Requests Projects Releases Wiki Activity

Common: Add AgentConfiguration.to_json()

This commit is contained in:
Mike Salvatore 2022-06-24 12:19:31 -04:00
parent 1f9a056b0b
commit 28250daffe
2 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,3 +1,5 @@
from __future__ import annotations
from dataclasses import dataclass from dataclasses import dataclass
from typing import List from typing import List
@ -32,6 +34,10 @@ class AgentConfiguration:
def from_json(config_json: dict): def from_json(config_json: dict):
return AgentConfigurationSchema().loads(config_json) return AgentConfigurationSchema().loads(config_json)
@staticmethod
def to_json(config: AgentConfiguration) -> str:
return AgentConfigurationSchema().dumps(config)
class AgentConfigurationSchema(Schema): class AgentConfigurationSchema(Schema):
keep_tunnel_open_time = fields.Float() keep_tunnel_open_time = fields.Float()

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

@ -198,3 +198,11 @@ def test_from_json():
config = AgentConfiguration.from_json(DEFAULT_AGENT_CONFIGURATION_JSON) config = AgentConfiguration.from_json(DEFAULT_AGENT_CONFIGURATION_JSON)
assert schema.dump(config) == dict_ assert schema.dump(config) == dict_
def test_to_json():
config = AgentConfiguration.from_json(DEFAULT_AGENT_CONFIGURATION_JSON)
assert json.loads(AgentConfiguration.to_json(config)) == json.loads(
DEFAULT_AGENT_CONFIGURATION_JSON
)