From ea8be28a72dbcf015174cdc842e965e9c665b895 Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Tue, 23 Nov 2021 11:51:09 -0500 Subject: [PATCH] Agent: Log a message in MockMaster.terminate() --- monkey/infection_monkey/master/mock_master.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/monkey/infection_monkey/master/mock_master.py b/monkey/infection_monkey/master/mock_master.py index a322cacd3..e3592841b 100644 --- a/monkey/infection_monkey/master/mock_master.py +++ b/monkey/infection_monkey/master/mock_master.py @@ -1,3 +1,5 @@ +import logging + from infection_monkey.i_master import IMaster from infection_monkey.i_puppet import IPuppet, PortScanData, PortStatus 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.system_info_telem import SystemInfoTelem +logger = logging.getLogger() + class MockMaster(IMaster): def __init__(self, puppet: IPuppet, telemetry_messenger: ITelemetryMessenger): @@ -56,7 +60,7 @@ class MockMaster(IMaster): self._telemetry_messenger.send_telemetry(ScanTelem(h)) def terminate(self) -> None: - pass + logger.info("Terminating MockMaster") def cleanup(self) -> None: pass