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

Agent: Log a message in MockMaster.terminate()

This commit is contained in:
Mike Salvatore 2021-11-23 11:51:09 -05:00 committed by Shreya Malviya
parent 357f749557
commit ea8be28a72
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
monkey/infection_monkey/master/mock_master.py
View File

@ -1,3 +1,5 @@
import logging
from infection_monkey.i_master import IMaster from infection_monkey.i_master import IMaster
from infection_monkey.i_puppet import IPuppet, PortScanData, PortStatus from infection_monkey.i_puppet import IPuppet, PortScanData, PortStatus
from infection_monkey.model.host import VictimHost from infection_monkey.model.host import VictimHost
@ -5,6 +7,8 @@ from infection_monkey.telemetry.messengers.i_telemetry_messenger import ITelemet
from infection_monkey.telemetry.scan_telem import ScanTelem from infection_monkey.telemetry.scan_telem import ScanTelem
from infection_monkey.telemetry.system_info_telem import SystemInfoTelem from infection_monkey.telemetry.system_info_telem import SystemInfoTelem
logger = logging.getLogger()
class MockMaster(IMaster): class MockMaster(IMaster):
def __init__(self, puppet: IPuppet, telemetry_messenger: ITelemetryMessenger): def __init__(self, puppet: IPuppet, telemetry_messenger: ITelemetryMessenger):
@ -56,7 +60,7 @@ class MockMaster(IMaster):
self._telemetry_messenger.send_telemetry(ScanTelem(h)) self._telemetry_messenger.send_telemetry(ScanTelem(h))
def terminate(self) -> None: def terminate(self) -> None:
pass logger.info("Terminating MockMaster")
def cleanup(self) -> None: def cleanup(self) -> None:
pass pass