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

Agent: Fix mypy issues in test_propagator.py

This commit is contained in:
Kekoa Kaaikala 2022-09-26 18:54:26 +00:00
parent aeb6630ebc
commit d87cf5a9f5
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
monkey/tests/unit_tests/infection_monkey/master/test_propagator.py
View File

@ -4,6 +4,7 @@ from unittest.mock import MagicMock
import pytest import pytest
from common import OperatingSystem
from common.agent_configuration.agent_sub_configurations import ( from common.agent_configuration.agent_sub_configurations import (
NetworkScanConfiguration, NetworkScanConfiguration,
PropagationConfiguration, PropagationConfiguration,
@ -33,7 +34,7 @@ def mock_victim_host_factory():
empty_fingerprint_data = FingerprintData(None, None, {}) empty_fingerprint_data = FingerprintData(None, None, {})
dot_1_scan_results = IPScanResults( dot_1_scan_results = IPScanResults(
PingScanData(True, "windows"), PingScanData(True, OperatingSystem.WINDOWS),
{ {
22: PortScanData(22, PortStatus.CLOSED, None, None), 22: PortScanData(22, PortStatus.CLOSED, None, None),
445: PortScanData(445, PortStatus.OPEN, "SMB BANNER", "tcp-445"), 445: PortScanData(445, PortStatus.OPEN, "SMB BANNER", "tcp-445"),
@ -47,7 +48,7 @@ dot_1_scan_results = IPScanResults(
) )
dot_3_scan_results = IPScanResults( dot_3_scan_results = IPScanResults(
PingScanData(True, "linux"), PingScanData(True, OperatingSystem.LINUX),
{ {
22: PortScanData(22, PortStatus.OPEN, "SSH BANNER", "tcp-22"), 22: PortScanData(22, PortStatus.OPEN, "SSH BANNER", "tcp-22"),
443: PortScanData(443, PortStatus.OPEN, "HTTPS BANNER", "tcp-443"), 443: PortScanData(443, PortStatus.OPEN, "HTTPS BANNER", "tcp-443"),