diff --git a/src/primaite/game/game.py b/src/primaite/game/game.py index a8e23d56..f42d6824 100644 --- a/src/primaite/game/game.py +++ b/src/primaite/game/game.py @@ -384,7 +384,7 @@ class PrimaiteGame: if service_class is not None: _LOGGER.debug(f"installing {service_type} on node {new_node.hostname}") new_node.software_manager.install(service_class, software_config=service_cfg.get("options", {})) - new_service = new_node.software_manager.software[service_class.__name__] + new_service = new_node.software_manager.software[service_type] # fixing duration for the service if "fixing_duration" in service_cfg.get("options", {}): diff --git a/src/primaite/simulator/system/services/arp/arp.py b/src/primaite/simulator/system/services/arp/arp.py index 0946f985..b0630d5d 100644 --- a/src/primaite/simulator/system/services/arp/arp.py +++ b/src/primaite/simulator/system/services/arp/arp.py @@ -33,7 +33,7 @@ class ARP(Service, discriminator="arp"): arp: Dict[IPV4Address, ARPEntry] = {} def __init__(self, **kwargs): - kwargs["name"] = "ARP" + kwargs["name"] = "arp" kwargs["port"] = PORT_LOOKUP["ARP"] kwargs["protocol"] = PROTOCOL_LOOKUP["UDP"] super().__init__(**kwargs) diff --git a/tests/unit_tests/_primaite/_simulator/_system/_services/test_terminal.py b/tests/unit_tests/_primaite/_simulator/_system/_services/test_terminal.py index f0906de9..144037f5 100644 --- a/tests/unit_tests/_primaite/_simulator/_system/_services/test_terminal.py +++ b/tests/unit_tests/_primaite/_simulator/_system/_services/test_terminal.py @@ -18,7 +18,7 @@ from primaite.simulator.network.protocols.ssh import ( SSHTransportMessage, SSHUserCredentials, ) -from primaite.simulator.system.applications.red_applications.ransomware_script import ransomware_script +from primaite.simulator.system.applications.red_applications.ransomware_script import RansomwareScript from primaite.simulator.system.services.dns.dns_server import DNSServer from primaite.simulator.system.services.service import ServiceOperatingState from primaite.simulator.system.services.terminal.terminal import RemoteTerminalConnection, Terminal