Merge 'origin/dev-game-layer' into feature/1924-Agent-Interface

tests/conftest.py

@@ -4,7 +4,7 @@ import shutil
 import tempfile
 from datetime import datetime
 from pathlib import Path
-from typing import Union
+from typing import Any, Union
 from unittest.mock import patch
 
 import pytest
@@ -14,6 +14,9 @@ from primaite.environment.primaite_env import Primaite
 from primaite.primaite_session import PrimaiteSession
 from primaite.simulator.network.container import Network
 from primaite.simulator.network.networks import arcd_uc2_network
+from primaite.simulator.network.transmission.transport_layer import Port
+from primaite.simulator.system.core.sys_log import SysLog
+from primaite.simulator.system.services.service import Service
 from tests.mock_and_patch.get_session_path_mock import get_temp_session_path
 
 ACTION_SPACE_NODE_VALUES = 1
@@ -26,11 +29,25 @@ from primaite.simulator.file_system.file_system import FileSystem
 from primaite.simulator.network.hardware.base import Node
 
 
+class TestService(Service):
+    """Test Service class"""
+
+    def receive(self, payload: Any, session_id: str, **kwargs) -> bool:
+        pass
+
+
 @pytest.fixture(scope="function")
 def uc2_network() -> Network:
     return arcd_uc2_network()
 
 
+@pytest.fixture(scope="function")
+def service(file_system) -> TestService:
+    return TestService(
+        name="TestService", port=Port.ARP, file_system=file_system, sys_log=SysLog(hostname="test_service")
+    )
+
+
 @pytest.fixture(scope="function")
 def file_system() -> FileSystem:
     return Node(hostname="fs_node").file_system

tests/integration_tests/network/test_frame_transmission.py

@@ -1,17 +1,15 @@
-from primaite.simulator.network.hardware.base import Link, NIC, Node
+from primaite.simulator.network.hardware.base import Link, NIC, Node, NodeOperatingState
 
 
 def test_node_to_node_ping():
     """Tests two Nodes are able to ping each other."""
-    node_a = Node(hostname="node_a")
-    nic_a = NIC(ip_address="192.168.0.10", subnet_mask="255.255.255.0")
+    node_a = Node(hostname="node_a", operating_state=NodeOperatingState.ON)
+    nic_a = NIC(ip_address="192.168.0.10", subnet_mask="255.255.255.0", operating_state=NodeOperatingState.ON)
     node_a.connect_nic(nic_a)
-    node_a.power_on()
 
-    node_b = Node(hostname="node_b")
+    node_b = Node(hostname="node_b", operating_state=NodeOperatingState.ON)
     nic_b = NIC(ip_address="192.168.0.11", subnet_mask="255.255.255.0")
     node_b.connect_nic(nic_b)
-    node_b.power_on()
 
     Link(endpoint_a=nic_a, endpoint_b=nic_b)
 
@@ -20,22 +18,19 @@ def test_node_to_node_ping():
 
 def test_multi_nic():
     """Tests that Nodes with multiple NICs can ping each other and the data go across the correct links."""
-    node_a = Node(hostname="node_a")
+    node_a = Node(hostname="node_a", operating_state=NodeOperatingState.ON)
     nic_a = NIC(ip_address="192.168.0.10", subnet_mask="255.255.255.0")
     node_a.connect_nic(nic_a)
-    node_a.power_on()
 
-    node_b = Node(hostname="node_b")
+    node_b = Node(hostname="node_b", operating_state=NodeOperatingState.ON)
     nic_b1 = NIC(ip_address="192.168.0.11", subnet_mask="255.255.255.0")
     nic_b2 = NIC(ip_address="10.0.0.12", subnet_mask="255.0.0.0")
     node_b.connect_nic(nic_b1)
     node_b.connect_nic(nic_b2)
-    node_b.power_on()
 
-    node_c = Node(hostname="node_c")
+    node_c = Node(hostname="node_c", operating_state=NodeOperatingState.ON)
     nic_c = NIC(ip_address="10.0.0.13", subnet_mask="255.0.0.0")
     node_c.connect_nic(nic_c)
-    node_c.power_on()
 
     Link(endpoint_a=nic_a, endpoint_b=nic_b1)
 

tests/integration_tests/network/test_link_connection.py

@@ -1,17 +1,15 @@
-from primaite.simulator.network.hardware.base import Link, NIC, Node
+from primaite.simulator.network.hardware.base import Link, NIC, Node, NodeOperatingState
 
 
 def test_link_up():
     """Tests Nodes, NICs, and Links can all be connected and be in an enabled/up state."""
-    node_a = Node(hostname="node_a")
+    node_a = Node(hostname="node_a", operating_state=NodeOperatingState.ON)
     nic_a = NIC(ip_address="192.168.0.10", subnet_mask="255.255.255.0")
     node_a.connect_nic(nic_a)
-    node_a.power_on()
 
-    node_b = Node(hostname="node_b")
+    node_b = Node(hostname="node_b", operating_state=NodeOperatingState.ON)
     nic_b = NIC(ip_address="192.168.0.11", subnet_mask="255.255.255.0")
     node_b.connect_nic(nic_b)
-    node_b.power_on()
 
     link = Link(endpoint_a=nic_a, endpoint_b=nic_b)
 

tests/integration_tests/network/test_routing.py

@@ -2,7 +2,7 @@ from typing import Tuple
 
 import pytest
 
-from primaite.simulator.network.hardware.base import Link, NIC, Node
+from primaite.simulator.network.hardware.base import Link, NIC, Node, NodeOperatingState
 from primaite.simulator.network.hardware.nodes.router import ACLAction, Router
 from primaite.simulator.network.transmission.network_layer import IPProtocol
 from primaite.simulator.network.transmission.transport_layer import Port
@@ -10,18 +10,15 @@ from primaite.simulator.network.transmission.transport_layer import Port
 
 @pytest.fixture(scope="function")
 def pc_a_pc_b_router_1() -> Tuple[Node, Node, Router]:
-    pc_a = Node(hostname="pc_a", default_gateway="192.168.0.1")
+    pc_a = Node(hostname="pc_a", default_gateway="192.168.0.1", operating_state=NodeOperatingState.ON)
     nic_a = NIC(ip_address="192.168.0.10", subnet_mask="255.255.255.0")
     pc_a.connect_nic(nic_a)
-    pc_a.power_on()
 
-    pc_b = Node(hostname="pc_b", default_gateway="192.168.1.1")
+    pc_b = Node(hostname="pc_b", default_gateway="192.168.1.1", operating_state=NodeOperatingState.ON)
     nic_b = NIC(ip_address="192.168.1.10", subnet_mask="255.255.255.0")
     pc_b.connect_nic(nic_b)
-    pc_b.power_on()
 
-    router_1 = Router(hostname="router_1")
-    router_1.power_on()
+    router_1 = Router(hostname="router_1", operating_state=NodeOperatingState.ON)
 
     router_1.configure_port(1, "192.168.0.1", "255.255.255.0")
     router_1.configure_port(2, "192.168.1.1", "255.255.255.0")

tests/integration_tests/network/test_switched_network.py

@@ -1,4 +1,4 @@
-from primaite.simulator.network.hardware.base import Link
+from primaite.simulator.network.hardware.base import Link, NodeOperatingState
 from primaite.simulator.network.hardware.nodes.computer import Computer
 from primaite.simulator.network.hardware.nodes.server import Server
 from primaite.simulator.network.hardware.nodes.switch import Switch
@@ -7,17 +7,22 @@ from primaite.simulator.network.hardware.nodes.switch import Switch
 def test_switched_network():
     """Tests a node can ping another node via the switch."""
     client_1 = Computer(
-        hostname="client_1", ip_address="192.168.1.10", subnet_mask="255.255.255.0", default_gateway="192.168.1.0"
+        hostname="client_1",
+        ip_address="192.168.1.10",
+        subnet_mask="255.255.255.0",
+        default_gateway="192.168.1.0",
+        operating_state=NodeOperatingState.ON,
     )
-    client_1.power_on()
 
     server_1 = Server(
-        hostname=" server_1", ip_address="192.168.1.11", subnet_mask="255.255.255.0", default_gateway="192.168.1.11"
+        hostname=" server_1",
+        ip_address="192.168.1.11",
+        subnet_mask="255.255.255.0",
+        default_gateway="192.168.1.11",
+        operating_state=NodeOperatingState.ON,
     )
-    server_1.power_on()
 
-    switch_1 = Switch(hostname="switch_1", num_ports=6)
-    switch_1.power_on()
+    switch_1 = Switch(hostname="switch_1", num_ports=6, operating_state=NodeOperatingState.ON)
 
     Link(endpoint_a=client_1.ethernet_port[1], endpoint_b=switch_1.switch_ports[1])
     Link(endpoint_a=server_1.ethernet_port[1], endpoint_b=switch_1.switch_ports[2])

tests/unit_tests/_primaite/_simulator/_file_system/test_file_system.py

@@ -1,6 +1,6 @@
 import pytest
 
-from primaite.simulator.file_system.file_system import FileSystem
+from primaite.simulator.file_system.file_system import File, FileSystem, FileSystemItemHealthStatus, Folder
 from primaite.simulator.file_system.file_type import FileType
 
 
@@ -122,6 +122,7 @@ def test_copy_file(file_system):
     assert file_system.get_file("dst_folder", "test_file.txt").uuid != original_uuid
 
 
+@pytest.mark.skip(reason="Implementation for quarantine not needed yet")
 def test_folder_quarantine_state(file_system):
     """Tests the changing of folder quarantine status."""
     folder = file_system.get_folder("root")
@@ -135,6 +136,158 @@ def test_folder_quarantine_state(file_system):
     assert folder.quarantine_status() is False
 
 
+def test_file_corrupt_repair(file_system):
+    """Test the ability to corrupt and repair files."""
+    folder: Folder = file_system.create_folder(folder_name="test_folder")
+    file: File = file_system.create_file(file_name="test_file.txt", folder_name="test_folder")
+
+    file.corrupt()
+
+    assert folder.health_status == FileSystemItemHealthStatus.GOOD
+    assert file.health_status == FileSystemItemHealthStatus.CORRUPT
+
+    file.repair()
+
+    assert folder.health_status == FileSystemItemHealthStatus.GOOD
+    assert file.health_status == FileSystemItemHealthStatus.GOOD
+
+
+def test_folder_corrupt_repair(file_system):
+    """Test the ability to corrupt and repair folders."""
+    folder: Folder = file_system.create_folder(folder_name="test_folder")
+    file_system.create_file(file_name="test_file.txt", folder_name="test_folder")
+
+    folder.corrupt()
+
+    file = folder.get_file(file_name="test_file.txt")
+    assert folder.health_status == FileSystemItemHealthStatus.CORRUPT
+    assert file.health_status == FileSystemItemHealthStatus.CORRUPT
+
+    folder.repair()
+
+    file = folder.get_file(file_name="test_file.txt")
+    assert folder.health_status == FileSystemItemHealthStatus.GOOD
+    assert file.health_status == FileSystemItemHealthStatus.GOOD
+
+
+def test_file_scan(file_system):
+    """Test the ability to update visible status."""
+    folder: Folder = file_system.create_folder(folder_name="test_folder")
+    file: File = file_system.create_file(file_name="test_file.txt", folder_name="test_folder")
+
+    assert file.health_status == FileSystemItemHealthStatus.GOOD
+    assert file.visible_health_status == FileSystemItemHealthStatus.GOOD
+
+    file.corrupt()
+
+    assert file.health_status == FileSystemItemHealthStatus.CORRUPT
+    assert file.visible_health_status == FileSystemItemHealthStatus.GOOD
+
+    file.scan()
+
+    assert file.health_status == FileSystemItemHealthStatus.CORRUPT
+    assert file.visible_health_status == FileSystemItemHealthStatus.CORRUPT
+
+
+def test_folder_scan(file_system):
+    """Test the ability to update visible status."""
+    folder: Folder = file_system.create_folder(folder_name="test_folder")
+    file_system.create_file(file_name="test_file.txt", folder_name="test_folder")
+    file_system.create_file(file_name="test_file2.txt", folder_name="test_folder")
+
+    file1: File = folder.get_file_by_id(file_uuid=list(folder.files)[1])
+    file2: File = folder.get_file_by_id(file_uuid=list(folder.files)[0])
+
+    assert folder.health_status == FileSystemItemHealthStatus.GOOD
+    assert folder.visible_health_status == FileSystemItemHealthStatus.GOOD
+    assert file1.visible_health_status == FileSystemItemHealthStatus.GOOD
+    assert file2.visible_health_status == FileSystemItemHealthStatus.GOOD
+
+    folder.corrupt()
+
+    assert folder.health_status == FileSystemItemHealthStatus.CORRUPT
+    assert folder.visible_health_status == FileSystemItemHealthStatus.GOOD
+    assert file1.visible_health_status == FileSystemItemHealthStatus.GOOD
+    assert file2.visible_health_status == FileSystemItemHealthStatus.GOOD
+
+    folder.scan()
+
+    folder.apply_timestep(timestep=0)
+
+    assert folder.health_status == FileSystemItemHealthStatus.CORRUPT
+    assert folder.visible_health_status == FileSystemItemHealthStatus.CORRUPT
+    assert file1.visible_health_status == FileSystemItemHealthStatus.CORRUPT
+    assert file2.visible_health_status == FileSystemItemHealthStatus.GOOD
+
+    folder.apply_timestep(timestep=1)
+
+    assert folder.health_status == FileSystemItemHealthStatus.CORRUPT
+    assert folder.visible_health_status == FileSystemItemHealthStatus.CORRUPT
+    assert file1.visible_health_status == FileSystemItemHealthStatus.CORRUPT
+    assert file2.visible_health_status == FileSystemItemHealthStatus.CORRUPT
+
+    folder.apply_timestep(timestep=2)
+
+    assert folder.health_status == FileSystemItemHealthStatus.CORRUPT
+    assert folder.visible_health_status == FileSystemItemHealthStatus.CORRUPT
+    assert file1.visible_health_status == FileSystemItemHealthStatus.CORRUPT
+    assert file2.visible_health_status == FileSystemItemHealthStatus.CORRUPT
+
+
+def test_simulated_file_check_hash(file_system):
+    file: File = file_system.create_file(file_name="test_file.txt", folder_name="test_folder")
+
+    assert file.check_hash() is True
+
+    # change simulated file size
+    file.sim_size = 0
+    assert file.check_hash() is False
+    assert file.health_status == FileSystemItemHealthStatus.CORRUPT
+
+
+def test_real_file_check_hash(file_system):
+    file: File = file_system.create_file(file_name="test_file.txt", folder_name="test_folder", real=True)
+
+    assert file.check_hash() is True
+
+    # change file content
+    with open(file.sim_path, "a") as f:
+        f.write("get hacked scrub lol xD\n")
+
+    assert file.check_hash() is False
+    assert file.health_status == FileSystemItemHealthStatus.CORRUPT
+
+
+def test_simulated_folder_check_hash(file_system):
+    folder: Folder = file_system.create_folder(folder_name="test_folder")
+    file_system.create_file(file_name="test_file.txt", folder_name="test_folder")
+
+    assert folder.check_hash() is True
+
+    # change simulated file size
+    file = folder.get_file(file_name="test_file.txt")
+    file.sim_size = 0
+    assert folder.check_hash() is False
+    assert folder.health_status == FileSystemItemHealthStatus.CORRUPT
+
+
+def test_real_folder_check_hash(file_system):
+    folder: Folder = file_system.create_folder(folder_name="test_folder")
+    file_system.create_file(file_name="test_file.txt", folder_name="test_folder", real=True)
+
+    assert folder.check_hash() is True
+
+    # change simulated file size
+    file = folder.get_file(file_name="test_file.txt")
+
+    # change file content
+    with open(file.sim_path, "a") as f:
+        f.write("get hacked scrub lol xD\n")
+
+    assert folder.check_hash() is False
+    assert folder.health_status == FileSystemItemHealthStatus.CORRUPT
+
+
 @pytest.mark.skip(reason="Skipping until we tackle serialisation")
 def test_serialisation(file_system):
     """Test to check that the object serialisation works correctly."""

tests/unit_tests/_primaite/_simulator/_file_system/test_file_system_actions.py

@@ -0,0 +1,160 @@
+from typing import Tuple
+
+import pytest
+
+from primaite.simulator.file_system.file_system import File, FileSystem, FileSystemItemHealthStatus, Folder
+
+
+@pytest.fixture(scope="function")
+def populated_file_system(file_system) -> Tuple[FileSystem, Folder, File]:
+    """Test that an agent can request a file scan."""
+    folder = file_system.create_folder(folder_name="test_folder")
+    file = file_system.create_file(folder_name="test_folder", file_name="test_file.txt")
+
+    return file_system, folder, file
+
+
+def test_file_scan_request(populated_file_system):
+    """Test that an agent can request a file scan."""
+    fs, folder, file = populated_file_system
+
+    file.corrupt()
+    assert file.health_status == FileSystemItemHealthStatus.CORRUPT
+    assert file.visible_health_status == FileSystemItemHealthStatus.GOOD
+
+    fs.apply_request(request=["file", file.uuid, "scan"])
+
+    assert file.health_status == FileSystemItemHealthStatus.CORRUPT
+    assert file.visible_health_status == FileSystemItemHealthStatus.CORRUPT
+
+
+def test_folder_scan_request(populated_file_system):
+    """Test that an agent can request a folder scan."""
+    fs, folder, file = populated_file_system
+    fs.create_file(file_name="test_file2.txt", folder_name="test_folder")
+
+    file1: File = folder.get_file_by_id(file_uuid=list(folder.files)[1])
+    file2: File = folder.get_file_by_id(file_uuid=list(folder.files)[0])
+
+    folder.corrupt()
+    assert folder.health_status == FileSystemItemHealthStatus.CORRUPT
+    assert folder.visible_health_status == FileSystemItemHealthStatus.GOOD
+    assert file1.visible_health_status == FileSystemItemHealthStatus.GOOD
+    assert file2.visible_health_status == FileSystemItemHealthStatus.GOOD
+
+    fs.apply_request(request=["folder", folder.uuid, "scan"])
+
+    folder.apply_timestep(timestep=0)
+
+    assert folder.health_status == FileSystemItemHealthStatus.CORRUPT
+    assert folder.visible_health_status == FileSystemItemHealthStatus.CORRUPT
+    assert file1.visible_health_status == FileSystemItemHealthStatus.CORRUPT
+    assert file2.visible_health_status == FileSystemItemHealthStatus.GOOD
+
+    folder.apply_timestep(timestep=1)
+
+    assert folder.health_status == FileSystemItemHealthStatus.CORRUPT
+    assert folder.visible_health_status == FileSystemItemHealthStatus.CORRUPT
+    assert file1.visible_health_status == FileSystemItemHealthStatus.CORRUPT
+    assert file2.visible_health_status == FileSystemItemHealthStatus.CORRUPT
+
+    folder.apply_timestep(timestep=2)
+
+    assert folder.health_status == FileSystemItemHealthStatus.CORRUPT
+    assert folder.visible_health_status == FileSystemItemHealthStatus.CORRUPT
+    assert file1.visible_health_status == FileSystemItemHealthStatus.CORRUPT
+    assert file2.visible_health_status == FileSystemItemHealthStatus.CORRUPT
+
+
+def test_file_checkhash_request(populated_file_system):
+    """Test that an agent can request a file hash check."""
+    fs, folder, file = populated_file_system
+
+    fs.apply_request(request=["file", file.uuid, "checkhash"])
+
+    assert file.health_status == FileSystemItemHealthStatus.GOOD
+    file.sim_size = 0
+
+    fs.apply_request(request=["file", file.uuid, "checkhash"])
+
+    assert file.health_status == FileSystemItemHealthStatus.CORRUPT
+
+
+def test_folder_checkhash_request(populated_file_system):
+    """Test that an agent can request a folder hash check."""
+    fs, folder, file = populated_file_system
+
+    fs.apply_request(request=["folder", folder.uuid, "checkhash"])
+
+    assert folder.health_status == FileSystemItemHealthStatus.GOOD
+    file.sim_size = 0
+
+    fs.apply_request(request=["folder", folder.uuid, "checkhash"])
+    assert folder.health_status == FileSystemItemHealthStatus.CORRUPT
+
+
+def test_file_repair_request(populated_file_system):
+    """Test that an agent can request a file repair."""
+    fs, folder, file = populated_file_system
+
+    file.corrupt()
+    assert file.health_status == FileSystemItemHealthStatus.CORRUPT
+
+    fs.apply_request(request=["file", file.uuid, "repair"])
+    assert file.health_status == FileSystemItemHealthStatus.GOOD
+
+
+def test_folder_repair_request(populated_file_system):
+    """Test that an agent can request a folder repair."""
+    fs, folder, file = populated_file_system
+
+    folder.corrupt()
+    assert file.health_status == FileSystemItemHealthStatus.CORRUPT
+    assert folder.health_status == FileSystemItemHealthStatus.CORRUPT
+
+    fs.apply_request(request=["folder", folder.uuid, "repair"])
+    assert file.health_status == FileSystemItemHealthStatus.GOOD
+    assert folder.health_status == FileSystemItemHealthStatus.GOOD
+
+
+def test_file_restore_request(populated_file_system):
+    pass
+
+
+def test_folder_restore_request(populated_file_system):
+    pass
+
+
+def test_file_corrupt_request(populated_file_system):
+    """Test that an agent can request a file corruption."""
+    fs, folder, file = populated_file_system
+    fs.apply_request(request=["file", file.uuid, "corrupt"])
+    assert file.health_status == FileSystemItemHealthStatus.CORRUPT
+
+
+def test_folder_corrupt_request(populated_file_system):
+    """Test that an agent can request a folder corruption."""
+    fs, folder, file = populated_file_system
+    fs.apply_request(request=["folder", folder.uuid, "corrupt"])
+    assert file.health_status == FileSystemItemHealthStatus.CORRUPT
+    assert folder.health_status == FileSystemItemHealthStatus.CORRUPT
+
+
+def test_file_delete_request(populated_file_system):
+    """Test that an agent can request a file deletion."""
+    fs, folder, file = populated_file_system
+    assert folder.get_file_by_id(file_uuid=file.uuid) is not None
+
+    fs.apply_request(request=["folder", folder.uuid, "delete", file.uuid])
+    assert folder.get_file_by_id(file_uuid=file.uuid) is None
+
+
+def test_folder_delete_request(populated_file_system):
+    """Test that an agent can request a folder deletion."""
+    fs, folder, file = populated_file_system
+    assert folder.get_file_by_id(file_uuid=file.uuid) is not None
+    assert fs.get_folder_by_id(folder_uuid=folder.uuid) is not None
+
+    fs.apply_request(request=["delete", folder.uuid])
+    assert fs.get_folder_by_id(folder_uuid=folder.uuid) is None
+    assert folder.get_file_by_id(file_uuid=file.uuid) is None

tests/unit_tests/_primaite/_simulator/_network/_hardware/test_node_actions.py

@@ -0,0 +1,41 @@
+import pytest
+
+from primaite.simulator.network.hardware.base import Node, NodeOperatingState
+
+
+@pytest.fixture
+def node() -> Node:
+    return Node(hostname="test")
+
+
+def test_node_startup(node):
+    assert node.operating_state == NodeOperatingState.OFF
+    node.apply_request(["startup"])
+    assert node.operating_state == NodeOperatingState.BOOTING
+
+    idx = 0
+    while node.operating_state == NodeOperatingState.BOOTING:
+        node.apply_timestep(timestep=idx)
+        idx += 1
+
+    assert node.operating_state == NodeOperatingState.ON
+
+
+def test_node_shutdown(node):
+    assert node.operating_state == NodeOperatingState.OFF
+    node.apply_request(["startup"])
+    idx = 0
+    while node.operating_state == NodeOperatingState.BOOTING:
+        node.apply_timestep(timestep=idx)
+        idx += 1
+
+    assert node.operating_state == NodeOperatingState.ON
+
+    node.apply_request(["shutdown"])
+
+    idx = 0
+    while node.operating_state == NodeOperatingState.SHUTTING_DOWN:
+        node.apply_timestep(timestep=idx)
+        idx += 1
+
+    assert node.operating_state == NodeOperatingState.OFF

tests/unit_tests/_primaite/_simulator/_system/_services/test_service_actions.py

@@ -0,0 +1,80 @@
+from primaite.simulator.system.services.service import ServiceOperatingState
+from primaite.simulator.system.software import SoftwareHealthState
+
+
+def test_service_scan(service):
+    """Test that an agent can request a service scan."""
+    service.start()
+    assert service.operating_state == ServiceOperatingState.RUNNING
+    assert service.health_state_visible == SoftwareHealthState.UNUSED
+
+    service.apply_request(["scan"])
+    assert service.operating_state == ServiceOperatingState.RUNNING
+    assert service.health_state_visible == SoftwareHealthState.GOOD
+
+
+def test_service_stop(service):
+    """Test that an agent can request to stop a service."""
+    service.start()
+    assert service.operating_state == ServiceOperatingState.RUNNING
+
+    service.apply_request(["stop"])
+    assert service.operating_state == ServiceOperatingState.STOPPED
+
+
+def test_service_start(service):
+    """Test that an agent can request to start a service."""
+    assert service.operating_state == ServiceOperatingState.STOPPED
+    service.apply_request(["start"])
+    assert service.operating_state == ServiceOperatingState.RUNNING
+
+
+def test_service_pause(service):
+    """Test that an agent can request to pause a service."""
+    service.start()
+    assert service.operating_state == ServiceOperatingState.RUNNING
+
+    service.apply_request(["pause"])
+    assert service.operating_state == ServiceOperatingState.PAUSED
+
+
+def test_service_resume(service):
+    """Test that an agent can request to resume a service."""
+    service.start()
+    assert service.operating_state == ServiceOperatingState.RUNNING
+
+    service.apply_request(["pause"])
+    assert service.operating_state == ServiceOperatingState.PAUSED
+
+    service.apply_request(["resume"])
+    assert service.operating_state == ServiceOperatingState.RUNNING
+
+
+def test_service_restart(service):
+    """Test that an agent can request to restart a service."""
+    service.start()
+    assert service.operating_state == ServiceOperatingState.RUNNING
+
+    service.apply_request(["restart"])
+    assert service.operating_state == ServiceOperatingState.RESTARTING
+
+
+def test_service_disable(service):
+    """Test that an agent can request to disable a service."""
+    service.start()
+    assert service.operating_state == ServiceOperatingState.RUNNING
+
+    service.apply_request(["disable"])
+    assert service.operating_state == ServiceOperatingState.DISABLED
+
+
+def test_service_enable(service):
+    """Test that an agent can request to enable a service."""
+    service.start()
+    assert service.operating_state == ServiceOperatingState.RUNNING
+
+    service.apply_request(["disable"])
+    assert service.operating_state == ServiceOperatingState.DISABLED
+
+    service.apply_request(["enable"])
+    assert service.operating_state == ServiceOperatingState.STOPPED

tests/unit_tests/_primaite/_simulator/_system/_services/test_services.py

@@ -0,0 +1,68 @@
+from primaite.simulator.system.services.service import ServiceOperatingState
+from primaite.simulator.system.software import SoftwareHealthState
+
+
+def test_scan(service):
+    assert service.operating_state == ServiceOperatingState.STOPPED
+    assert service.health_state_visible == SoftwareHealthState.UNUSED
+
+    service.start()
+    assert service.operating_state == ServiceOperatingState.RUNNING
+    assert service.health_state_visible == SoftwareHealthState.UNUSED
+
+    service.scan()
+    assert service.operating_state == ServiceOperatingState.RUNNING
+    assert service.health_state_visible == SoftwareHealthState.GOOD
+
+
+def test_start_service(service):
+    assert service.operating_state == ServiceOperatingState.STOPPED
+    service.start()
+
+    assert service.operating_state == ServiceOperatingState.RUNNING
+
+
+def test_stop_service(service):
+    service.start()
+    assert service.operating_state == ServiceOperatingState.RUNNING
+
+    service.stop()
+    assert service.operating_state == ServiceOperatingState.STOPPED
+
+
+def test_pause_and_resume_service(service):
+    assert service.operating_state == ServiceOperatingState.STOPPED
+    service.resume()
+    assert service.operating_state == ServiceOperatingState.STOPPED
+
+    service.start()
+    service.pause()
+    assert service.operating_state == ServiceOperatingState.PAUSED
+
+    service.resume()
+    assert service.operating_state == ServiceOperatingState.RUNNING
+
+
+def test_restart(service):
+    assert service.operating_state == ServiceOperatingState.STOPPED
+    service.restart()
+    assert service.operating_state == ServiceOperatingState.STOPPED
+
+    service.start()
+    service.restart()
+    assert service.operating_state == ServiceOperatingState.RESTARTING
+
+    timestep = 0
+    while service.operating_state == ServiceOperatingState.RESTARTING:
+        service.apply_timestep(timestep)
+        timestep += 1
+
+    assert service.operating_state == ServiceOperatingState.RUNNING
+
+
+def test_enable_disable(service):
+    service.disable()
+    assert service.operating_state == ServiceOperatingState.DISABLED
+
+    service.enable()
+    assert service.operating_state == ServiceOperatingState.STOPPED