From e00fd64530d9a73a9aebd4007a6fe6f4e57e050c Mon Sep 17 00:00:00 2001 From: Shreya Malviya Date: Wed, 24 Nov 2021 13:47:33 +0530 Subject: [PATCH] Agent: Fix PBA return value --- monkey/infection_monkey/puppet/mock_puppet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/monkey/infection_monkey/puppet/mock_puppet.py b/monkey/infection_monkey/puppet/mock_puppet.py index e203d9cec..6996d4d7c 100644 --- a/monkey/infection_monkey/puppet/mock_puppet.py +++ b/monkey/infection_monkey/puppet/mock_puppet.py @@ -151,9 +151,9 @@ class MockPuppet(IPuppet): logger.debug(f"run_pba({name}, {options})") if name == "AccountDiscovery": - return PostBreachData("pba command 1", "pba result 1") + return PostBreachData("pba command 1", ["pba result 1", True]) else: - return PostBreachData("pba command 2", "pba result 2") + return PostBreachData("pba command 2", ["pba result 2", False]) def ping(self, host: str) -> Tuple[bool, Optional[str]]: logger.debug(f"run_ping({host})")