From 9754b4731ce62f2e79d2ded754216ea05722d428 Mon Sep 17 00:00:00 2001 From: Shreya Malviya Date: Fri, 30 Sep 2022 17:27:17 +0530 Subject: [PATCH] UT: Mock AbstractAgentEvent's timestamp (time.time()) in test_tcp_scanner.py --- .../network_scanning/test_tcp_scanner.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/monkey/tests/unit_tests/infection_monkey/network_scanning/test_tcp_scanner.py b/monkey/tests/unit_tests/infection_monkey/network_scanning/test_tcp_scanner.py index c96728658..e47febc36 100644 --- a/monkey/tests/unit_tests/infection_monkey/network_scanning/test_tcp_scanner.py +++ b/monkey/tests/unit_tests/infection_monkey/network_scanning/test_tcp_scanner.py @@ -1,3 +1,4 @@ +import time from unittest.mock import MagicMock import pytest @@ -13,6 +14,17 @@ PORTS_TO_SCAN = [22, 80, 8080, 143, 445, 2222] OPEN_PORTS_DATA = {22: "SSH-banner", 80: "", 2222: "SSH2-banner"} +TIMESTAMP = 123.321 + + +@pytest.fixture(scope="module") +def patch_timestamp(monkeypatch): + monkeypatch.setattr( + time, + "time", + TIMESTAMP, + ) + @pytest.fixture def patch_check_tcp_ports(monkeypatch, open_ports_data):