#2350: tests + application

tests/integration_tests/game_layer/observations/test_software_observations.py

@@ -0,0 +1,66 @@
+import pytest
+
+from primaite.game.agent.observations.software_observation import ApplicationObservation, ServiceObservation
+from primaite.simulator.network.hardware.nodes.host.computer import Computer
+from primaite.simulator.sim_container import Simulation
+from primaite.simulator.system.applications.database_client import DatabaseClient
+from primaite.simulator.system.applications.web_browser import WebBrowser
+from primaite.simulator.system.services.ntp.ntp_server import NTPServer
+
+
+@pytest.fixture(scope="function")
+def simulation(example_network) -> Simulation:
+    sim = Simulation()
+
+    # set simulation network as example network
+    sim.network = example_network
+
+    return sim
+
+
+def test_service_observation(simulation):
+    """Test the service observation."""
+    pc: Computer = simulation.network.get_node_by_hostname("client_1")
+    # install software on the computer
+    pc.software_manager.install(NTPServer)
+
+    ntp_server = pc.software_manager.software.get("NTPServer")
+    assert ntp_server
+
+    service_obs = ServiceObservation(where=["network", "nodes", pc.hostname, "services", "NTPServer"])
+
+    observation_state = service_obs.observe(simulation.describe_state())
+
+    assert observation_state.get("health_status") == 0
+    assert observation_state.get("operating_status") == 1  # running
+
+    ntp_server.restart()
+    observation_state = service_obs.observe(simulation.describe_state())
+    assert observation_state.get("health_status") == 0
+    assert observation_state.get("operating_status") == 6  # resetting
+
+
+def test_application_observation(simulation):
+    """Test the application observation."""
+    pc: Computer = simulation.network.get_node_by_hostname("client_1")
+    # install software on the computer
+    pc.software_manager.install(DatabaseClient)
+
+    web_browser: WebBrowser = pc.software_manager.software.get("WebBrowser")
+    assert web_browser
+
+    app_obs = ApplicationObservation(where=["network", "nodes", pc.hostname, "applications", "WebBrowser"])
+
+    web_browser.close()
+    observation_state = app_obs.observe(simulation.describe_state())
+    assert observation_state.get("health_status") == 0
+    assert observation_state.get("operating_status") == 2  # stopped
+    assert observation_state.get("num_executions") == 0
+
+    web_browser.run()
+    web_browser.scan()  # scan to update health status
+    web_browser.get_webpage("test")
+    observation_state = app_obs.observe(simulation.describe_state())
+    assert observation_state.get("health_status") == 1
+    assert observation_state.get("operating_status") == 1  # running
+    assert observation_state.get("num_executions") == 1