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

UT: Assert mock_agent_event_queue.publish's call counts in test_tcp_scanner.py

This commit is contained in:
Shreya Malviya 2022-09-30 13:48:35 +05:30
parent bab4ebc2bc
commit 0bf9309e07
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
monkey/tests/unit_tests/infection_monkey/network_scanning/test_tcp_scanner.py
View File

@ -38,6 +38,8 @@ def test_tcp_successful(
assert port_scan_data[port].status == PortStatus.CLOSED assert port_scan_data[port].status == PortStatus.CLOSED
assert port_scan_data[port].banner is None assert port_scan_data[port].banner is None
assert mock_agent_event_queue.publish.call_count == 1
@pytest.mark.parametrize("open_ports_data", [{}]) @pytest.mark.parametrize("open_ports_data", [{}])
def test_tcp_empty_response( def test_tcp_empty_response(
@ -51,6 +53,8 @@ def test_tcp_empty_response(
assert port_scan_data[port].status == PortStatus.CLOSED assert port_scan_data[port].status == PortStatus.CLOSED
assert port_scan_data[port].banner is None assert port_scan_data[port].banner is None
assert mock_agent_event_queue.publish.call_count == 1
@pytest.mark.parametrize("open_ports_data", [OPEN_PORTS_DATA]) @pytest.mark.parametrize("open_ports_data", [OPEN_PORTS_DATA])
def test_tcp_no_ports_to_scan( def test_tcp_no_ports_to_scan(
@ -59,6 +63,7 @@ def test_tcp_no_ports_to_scan(
port_scan_data = scan_tcp_ports("127.0.0.1", [], 0, mock_agent_event_queue) port_scan_data = scan_tcp_ports("127.0.0.1", [], 0, mock_agent_event_queue)
assert len(port_scan_data) == 0 assert len(port_scan_data) == 0
assert mock_agent_event_queue.publish.call_count == 1
def test_exception_handling(monkeypatch, mock_agent_event_queue): def test_exception_handling(monkeypatch, mock_agent_event_queue):
@ -67,3 +72,4 @@ def test_exception_handling(monkeypatch, mock_agent_event_queue):
MagicMock(side_effect=Exception), MagicMock(side_effect=Exception),
) )
assert scan_tcp_ports("abc", [123], 123, mock_agent_event_queue) == EMPTY_PORT_SCAN assert scan_tcp_ports("abc", [123], 123, mock_agent_event_queue) == EMPTY_PORT_SCAN
assert mock_agent_event_queue.publish.call_count == 0