BB: Fix credentials object serialization

This commit is contained in:
Mike Salvatore 2022-07-22 13:21:25 -04:00
parent 5c60656f56
commit e1691b108f
1 changed files with 5 additions and 3 deletions

View File

@ -6,6 +6,7 @@ from typing import Union
from bson import json_util from bson import json_util
from common.configuration import AgentConfiguration from common.configuration import AgentConfiguration
from common.credentials import Credentials
from envs.monkey_zoo.blackbox.island_client.monkey_island_requests import MonkeyIslandRequests from envs.monkey_zoo.blackbox.island_client.monkey_island_requests import MonkeyIslandRequests
from envs.monkey_zoo.blackbox.test_configurations.test_configuration import TestConfiguration from envs.monkey_zoo.blackbox.test_configurations.test_configuration import TestConfiguration
@ -37,9 +38,10 @@ class MonkeyIslandClient(object):
"api/agent-configuration", "api/agent-configuration",
data=AgentConfiguration.to_mapping(test_configuration.agent_configuration), data=AgentConfiguration.to_mapping(test_configuration.agent_configuration),
) )
serialized_propagation_credentials = json.dumps( serialized_propagation_credentials = [
[credential.to_json() for credential in test_configuration.propagation_credentials] Credentials.to_mapping(credentials)
) for credentials in test_configuration.propagation_credentials
]
self.requests.post( self.requests.post(
"/api/propagation-credentials/configured-credentials", "/api/propagation-credentials/configured-credentials",
data=serialized_propagation_credentials, data=serialized_propagation_credentials,