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

UT: Fix UnknownRecordError message in InMemoryAgentLogRepository

This commit is contained in:
Mike Salvatore 2022-09-28 14:10:30 -04:00
parent 3202404e46
commit 93b0fe0f6e
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
monkey/tests/monkey_island/in_memory_agent_log_repository.py
View File

@ -12,7 +12,8 @@ class InMemoryAgentLogRepository(IAgentLogRepository):
def get_agent_log(self, agent_id: AgentID) -> str: def get_agent_log(self, agent_id: AgentID) -> str:
if agent_id not in self._agent_logs: if agent_id not in self._agent_logs:
raise UnknownRecordError("Error occured while getting agent") raise UnknownRecordError(f"Unknown agent {agent_id}")
return self._agent_logs[agent_id] return self._agent_logs[agent_id]
def reset(self): def reset(self):