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

View File

@ -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"),