UT: Add machines and node to report tests

This commit is contained in:
vakarisz 2022-09-29 11:52:57 +03:00 committed by Kekoa Kaaikala
parent bbd606501e
commit dd0c504743
1 changed files with 34 additions and 0 deletions

View File

@ -0,0 +1,34 @@
from ipaddress import IPv4Interface
from monkey_island.cc.models import CommunicationType, Machine, Node
ISLAND_MACHINE = Machine(
id=0,
island=True,
hardware_id=5,
network_interfaces=[IPv4Interface("10.10.10.99/24")],
)
MACHINE_A = Machine(
id=1,
hardware_id=9,
network_interfaces=[IPv4Interface("10.10.10.1/24")],
)
MACHINE_B = Machine(
id=2,
hardware_id=9,
network_interfaces=[IPv4Interface("10.10.10.2/24")],
)
MACHINE_C = Machine(
id=3,
hardware_id=9,
network_interfaces=[IPv4Interface("10.10.10.3/24")],
)
NODES = [
Node(id=1, connections={"2", CommunicationType.EXPLOITED}),
Node(id=0, connections={"1", CommunicationType.SCANNED}),
Node(id=3, connections={"0", CommunicationType.CC}),
]