From d87cf5a9f52ce2df30c1c3151353763661470ea2 Mon Sep 17 00:00:00 2001 From: Kekoa Kaaikala Date: Mon, 26 Sep 2022 18:54:26 +0000 Subject: [PATCH] Agent: Fix mypy issues in test_propagator.py --- .../unit_tests/infection_monkey/master/test_propagator.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/monkey/tests/unit_tests/infection_monkey/master/test_propagator.py b/monkey/tests/unit_tests/infection_monkey/master/test_propagator.py index b662462ac..174296738 100644 --- a/monkey/tests/unit_tests/infection_monkey/master/test_propagator.py +++ b/monkey/tests/unit_tests/infection_monkey/master/test_propagator.py @@ -4,6 +4,7 @@ from unittest.mock import MagicMock import pytest +from common import OperatingSystem from common.agent_configuration.agent_sub_configurations import ( NetworkScanConfiguration, PropagationConfiguration, @@ -33,7 +34,7 @@ def mock_victim_host_factory(): empty_fingerprint_data = FingerprintData(None, None, {}) dot_1_scan_results = IPScanResults( - PingScanData(True, "windows"), + PingScanData(True, OperatingSystem.WINDOWS), { 22: PortScanData(22, PortStatus.CLOSED, None, None), 445: PortScanData(445, PortStatus.OPEN, "SMB BANNER", "tcp-445"), @@ -47,7 +48,7 @@ dot_1_scan_results = IPScanResults( ) dot_3_scan_results = IPScanResults( - PingScanData(True, "linux"), + PingScanData(True, OperatingSystem.LINUX), { 22: PortScanData(22, PortStatus.OPEN, "SSH BANNER", "tcp-22"), 443: PortScanData(443, PortStatus.OPEN, "HTTPS BANNER", "tcp-443"),