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

UT: Add _get_ping_scan_event()to test_ping_scanner.py

This commit is contained in:
Shreya Malviya 2022-09-29 15:30:12 +05:30
parent 9f15bea5bd
commit d78615fa92
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
monkey/tests/unit_tests/infection_monkey/network_scanning/test_ping_scanner.py
View File

@ -9,6 +9,7 @@ import infection_monkey.network_scanning.ping_scanner # noqa: F401
from common import OperatingSystem from common import OperatingSystem
from common.agent_events import PingScanEvent from common.agent_events import PingScanEvent
from common.event_queue import IAgentEventQueue from common.event_queue import IAgentEventQueue
from common.types import PingScanData
from infection_monkey.network_scanning import ping from infection_monkey.network_scanning import ping
from infection_monkey.network_scanning.ping_scanner import EMPTY_PING_SCAN from infection_monkey.network_scanning.ping_scanner import EMPTY_PING_SCAN
from infection_monkey.utils.ids import get_agent_id from infection_monkey.utils.ids import get_agent_id
@ -112,12 +113,9 @@ def mock_agent_event_queue():
HOST_IP = "192.168.1.1" HOST_IP = "192.168.1.1"
TIMEOUT = 1.0 TIMEOUT = 1.0
@pytest.mark.usefixtures("set_os_linux")
def test_linux_ping_success(patch_subprocess_running_ping_with_ping_output, mock_agent_event_queue):
patch_subprocess_running_ping_with_ping_output(LINUX_SUCCESS_OUTPUT)
result = ping(HOST_IP, TIMEOUT, mock_agent_event_queue)
event = PingScanEvent( def _get_ping_scan_event(result: PingScanData):
return PingScanEvent(
source=get_agent_id(), source=get_agent_id(),
target=HOST_IP, target=HOST_IP,
timestamp=TIMESTAMP, timestamp=TIMESTAMP,
@ -126,6 +124,13 @@ def test_linux_ping_success(patch_subprocess_running_ping_with_ping_output, mock
os=result.os, os=result.os,
) )
@pytest.mark.usefixtures("set_os_linux")
def test_linux_ping_success(patch_subprocess_running_ping_with_ping_output, mock_agent_event_queue):
patch_subprocess_running_ping_with_ping_output(LINUX_SUCCESS_OUTPUT)
result = ping(HOST_IP, TIMEOUT, mock_agent_event_queue)
event = _get_ping_scan_event(result)
assert result.response_received assert result.response_received
assert result.os == OperatingSystem.LINUX assert result.os == OperatingSystem.LINUX
assert mock_agent_event_queue.publish.call_count == 1 assert mock_agent_event_queue.publish.call_count == 1