forked from p34709852/monkey
60 lines
1.9 KiB
Python
60 lines
1.9 KiB
Python
from ipaddress import IPv4Address
|
|
from typing import Collection
|
|
|
|
import pytest
|
|
|
|
from envs.monkey_zoo.blackbox.island_client.monkey_island_client import MonkeyIslandClient
|
|
|
|
machine_list = {
|
|
"10.0.0.36": "centos_6",
|
|
"10.0.0.37": "centos_7",
|
|
"10.0.0.38": "centos_8",
|
|
"10.0.0.42": "suse_12",
|
|
"10.0.0.41": "suse_11",
|
|
"10.0.0.99": "kali_2019",
|
|
"10.0.0.86": "rhel_6",
|
|
"10.0.0.87": "rhel_7",
|
|
"10.0.0.88": "rhel_8",
|
|
"10.0.0.77": "debian_7",
|
|
"10.0.0.78": "debian_8",
|
|
"10.0.0.79": "debian_9",
|
|
"10.0.0.66": "oracle_6",
|
|
"10.0.0.67": "oracle_7",
|
|
"10.0.0.22": "ubuntu_12",
|
|
"10.0.0.24": "ubuntu_14",
|
|
"10.0.0.29": "ubuntu_19",
|
|
"10.0.0.4": "windows_2003_r2_32",
|
|
"10.0.0.5": "windows_2003",
|
|
"10.0.0.8": "windows_2008",
|
|
"10.0.0.6": "windows_2008_32",
|
|
"10.0.0.12": "windows_2012",
|
|
"10.0.0.11": "windows_2012_r2",
|
|
"10.0.0.116": "windows_2016",
|
|
"10.0.0.119": "windows_2019",
|
|
}
|
|
|
|
|
|
@pytest.fixture(scope="class")
|
|
def island_client(island):
|
|
island_client_object = MonkeyIslandClient(island)
|
|
yield island_client_object
|
|
|
|
|
|
@pytest.mark.usefixtures("island_client")
|
|
# noinspection PyUnresolvedReferences
|
|
class TestOSCompatibility(object):
|
|
def test_os_compat(self, island_client: MonkeyIslandClient):
|
|
print()
|
|
ips_that_communicated = self._get_agent_ips(island_client)
|
|
for ip, os in machine_list.items():
|
|
if IPv4Address(ip) not in ips_that_communicated:
|
|
print("{} didn't communicate to island".format(os))
|
|
|
|
if len(ips_that_communicated) < len(machine_list):
|
|
assert False
|
|
|
|
def _get_agent_ips(self, island_client: MonkeyIslandClient) -> Collection[IPv4Address]:
|
|
agents = island_client.get_agents()
|
|
machines = island_client.get_machines()
|
|
return {i.ip for a in agents for i in machines[a.machine_id].network_interfaces}
|