Compare commits

...

1 Commits

Author SHA1 Message Date
Kekoa Kaaikala 63fcea39ed Agent: Fix VictimHost mypy issues 2022-09-22 14:23:45 +00:00
5 changed files with 11 additions and 9 deletions

View File

@ -10,7 +10,7 @@ class VictimHost(object):
self.os: Dict[str, Any] = {} self.os: Dict[str, Any] = {}
self.services: Dict[str, Any] = {} self.services: Dict[str, Any] = {}
self.icmp = False self.icmp = False
self.default_server = None self.default_server = ""
def as_dict(self): def as_dict(self):
return self.__dict__ return self.__dict__

View File

@ -13,7 +13,7 @@ def mock_get_interface_to_target(monkeypatch):
def test_factory_no_tunnel(): def test_factory_no_tunnel():
factory = VictimHostFactory(island_ip="192.168.56.1", island_port="5000", on_island=False) factory = VictimHostFactory(island_ip="192.168.56.1", island_port="5000", on_island=False)
network_address = NetworkAddress("192.168.56.2", None) network_address = NetworkAddress("192.168.56.2", "")
victim = factory.build_victim_host(network_address) victim = factory.build_victim_host(network_address)
@ -49,4 +49,4 @@ def test_factory_no_default_server():
victim = factory.build_victim_host(network_address) victim = factory.build_victim_host(network_address)
assert victim.default_server is None assert not victim.default_server

View File

@ -1,11 +1,12 @@
import json import json
from typing import Iterable
import pytest import pytest
from infection_monkey.exploit.sshexec import SSHExploiter from infection_monkey.exploit.sshexec import SSHExploiter
from infection_monkey.i_puppet.i_puppet import ExploiterResultData
from infection_monkey.model.host import VictimHost from infection_monkey.model.host import VictimHost
from infection_monkey.telemetry.exploit_telem import ExploitTelem from infection_monkey.telemetry.exploit_telem import ExploitTelem
from monkey.infection_monkey.i_puppet.i_puppet import ExploiterResultData
DOMAIN_NAME = "domain-name" DOMAIN_NAME = "domain-name"
IP = "0.0.0.0" IP = "0.0.0.0"
@ -16,7 +17,7 @@ HOST_AS_DICT = {
"os": {}, "os": {},
"services": {}, "services": {},
"icmp": False, "icmp": False,
"default_server": None, "default_server": "",
} }
EXPLOITER_NAME = "SSHExploiter" EXPLOITER_NAME = "SSHExploiter"
EXPLOITER_INFO = { EXPLOITER_INFO = {
@ -27,7 +28,7 @@ EXPLOITER_INFO = {
"vulnerable_ports": [], "vulnerable_ports": [],
"executed_cmds": [], "executed_cmds": [],
} }
EXPLOITER_ATTEMPTS = [] EXPLOITER_ATTEMPTS: Iterable = []
RESULT = False RESULT = False
OS_LINUX = "linux" OS_LINUX = "linux"
ERROR_MSG = "failed because yolo" ERROR_MSG = "failed because yolo"

View File

@ -1,4 +1,5 @@
import json import json
from typing import Any, Dict
import pytest import pytest
@ -14,9 +15,9 @@ HOST_AS_DICT = {
"os": {}, "os": {},
"services": {}, "services": {},
"icmp": False, "icmp": False,
"default_server": None, "default_server": "",
} }
HOST_SERVICES = {} HOST_SERVICES: Dict[str, Any] = {}
@pytest.fixture @pytest.fixture

View File

@ -18,7 +18,7 @@ SCAN_DATA_MOCK = [
}, },
}, },
"monkey_exe": None, "monkey_exe": None,
"default_server": None, "default_server": "",
}, },
} }
] ]