p20319658
/
monkey
forked from p34709852/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

UT: Handle ValueError in test_construct_invalid_field__value_error()

This commit is contained in:
Mike Salvatore 2022-09-28 15:52:35 -04:00 committed by Shreya Malviya
parent d3c9691dfe
commit 6390993875
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
monkey/tests/unit_tests/common/agent_events/test_ping_scan_event.py
View File

@ -2,7 +2,6 @@ from ipaddress import IPv4Address
from uuid import UUID from uuid import UUID
import pytest import pytest
from pydantic.errors import IPv4AddressError, UUIDError
from tests.unit_tests.monkey_island.cc.models.test_agent import AGENT_ID from tests.unit_tests.monkey_island.cc.models.test_agent import AGENT_ID
from common import OperatingSystem from common import OperatingSystem
@ -53,6 +52,7 @@ def test_to_dict():
"key, value", "key, value",
[ [
("source", "not-an-uuid"), ("source", "not-an-uuid"),
("source", -1),
("timestamp", "not-a-timestamp"), ("timestamp", "not-a-timestamp"),
("response_received", "not-a-bool"), ("response_received", "not-a-bool"),
("os", 2.1), ("os", 2.1),
@ -68,17 +68,16 @@ def test_construct_invalid_field__type_error(key, value):
@pytest.mark.parametrize( @pytest.mark.parametrize(
"key, value, expected_error", "key, value",
[ [
("source", -1, UUIDError), ("target", "not-a-IPv4Address"),
("target", "not-a-IPv4Address", IPv4AddressError),
], ],
) )
def test_construct_invalid_field__value_error(key, value, expected_error): def test_construct_invalid_field__value_error(key, value):
invalid_type_dict = PING_SIMPLE_DICT.copy() invalid_type_dict = PING_SIMPLE_DICT.copy()
invalid_type_dict[key] = value invalid_type_dict[key] = value
with pytest.raises(expected_error): with pytest.raises(ValueError):
PingScanEvent(**invalid_type_dict) PingScanEvent(**invalid_type_dict)