diff --git a/tests/environment/__init__.py b/tests/environment/__init__.py index 7bc9cb3..3c62614 100644 --- a/tests/environment/__init__.py +++ b/tests/environment/__init__.py @@ -37,7 +37,9 @@ class Agent(object): self._clienturl = clienturl self._dockerurl = dockerurl r = requests.post(f"{self._clienturl}/add-agent", json={"url": f"{self._dockerurl}", "meshid": self.meshid}) - self.nodeid = r.json()["id"] + agent_json = r.json() + self.nodeid = agent_json["id"] + self.nodehex = agent_json["hex"] def __enter__(self): return self diff --git a/tests/environment/scripts/client/agent_server.py b/tests/environment/scripts/client/agent_server.py index 1ee8639..52e42e8 100644 --- a/tests/environment/scripts/client/agent_server.py +++ b/tests/environment/scripts/client/agent_server.py @@ -53,7 +53,7 @@ def add_agent(): time.sleep(.1) else: raise Exception(f"Failed to start agent: {text}") - return {"id": agent_id} + return {"id": agent_id, "hex": agent_hex} @api.route('/remove-agent/', methods=['POST']) def remove_agent(agentid):