Agent: Move MockPuppet to unit test suite

The MockPuppet is now only used by the unit tests.
This commit is contained in:
Mike Salvatore 2022-03-30 07:39:34 -04:00
parent b5f65b16d7
commit ee0561a061
3 changed files with 3 additions and 4 deletions

View File

@ -5,10 +5,10 @@ from typing import Iterable
from unittest.mock import MagicMock
import pytest
from tests.unit_tests.infection_monkey.master.mock_puppet import MockPuppet
from infection_monkey.master import Exploiter
from infection_monkey.model import VictimHost
from infection_monkey.puppet.mock_puppet import MockPuppet
logger = logging.getLogger()

View File

@ -3,11 +3,11 @@ from typing import Set
from unittest.mock import MagicMock
import pytest
from tests.unit_tests.infection_monkey.master.mock_puppet import MockPuppet
from infection_monkey.i_puppet import FingerprintData, PortScanData, PortStatus
from infection_monkey.master import IPScanner
from infection_monkey.network import NetworkAddress
from infection_monkey.puppet.mock_puppet import MockPuppet
WINDOWS_OS = "windows"
LINUX_OS = "linux"
@ -34,8 +34,7 @@ def scan_config():
{"name": "HTTPFinger", "options": {}},
{"name": "SMBFinger", "options": {}},
{"name": "SSHFinger", "options": {}},
]
],
}