#2064: turn on everything when node is turned on

tests/conftest.py

@@ -52,6 +52,12 @@ class TestService(Service):
 class TestApplication(Application):
     """Test Application class"""
 
+    def __init__(self, **kwargs):
+        kwargs["name"] = "TestApplication"
+        kwargs["port"] = Port.HTTP
+        kwargs["protocol"] = IPProtocol.TCP
+        super().__init__(**kwargs)
+
     def describe_state(self) -> Dict:
         pass
 

tests/e2e_integration_tests/test_uc2_data_manipulation_scenario.py

@@ -23,6 +23,7 @@ def test_data_manipulation(uc2_network):
 
     # Now we run the DataManipulationBot
     db_manipulation_bot.run()
+    db_manipulation_bot.attack()
 
     # Now check that the DB client on the web_server cannot query the users table on the database
     assert not db_client.query("SELECT")

tests/integration_tests/system/test_app_service_on_node.py

@@ -0,0 +1,95 @@
+from typing import Tuple
+
+import pytest
+from conftest import TestApplication, TestService
+
+from primaite.simulator.network.hardware.node_operating_state import NodeOperatingState
+from primaite.simulator.network.hardware.nodes.server import Server
+from primaite.simulator.system.applications.application import Application, ApplicationOperatingState
+from primaite.simulator.system.services.service import Service, ServiceOperatingState
+
+
+@pytest.fixture(scope="function")
+def populated_node() -> Tuple[Application, Server, Service]:
+    server = Server(
+        hostname="server", ip_address="192.168.0.1", subnet_mask="255.255.255.0", operating_state=NodeOperatingState.ON
+    )
+    server.software_manager.install(TestService)
+    server.software_manager.install(TestApplication)
+
+    app = server.software_manager.software["TestApplication"]
+    app.run()
+    service = server.software_manager.software["TestService"]
+    service.start()
+
+    return app, server, service
+
+
+def test_server_turns_off_service(populated_node):
+    """Check that the service is turned off when the server is turned off"""
+    app, server, service = populated_node
+
+    assert server.operating_state is NodeOperatingState.ON
+    assert service.operating_state is ServiceOperatingState.RUNNING
+    assert app.operating_state is ApplicationOperatingState.RUNNING
+
+    server.power_off()
+
+    for i in range(server.shut_down_duration + 1):
+        server.apply_timestep(timestep=i)
+
+    assert server.operating_state is NodeOperatingState.OFF
+    assert service.operating_state is ServiceOperatingState.STOPPED
+    assert app.operating_state is ApplicationOperatingState.CLOSED
+
+
+def test_service_cannot_be_turned_on_when_server_is_off(populated_node):
+    """Check that the service cannot be started when the server is off."""
+    app, server, service = populated_node
+
+    assert server.operating_state is NodeOperatingState.ON
+    assert service.operating_state is ServiceOperatingState.RUNNING
+    assert app.operating_state is ApplicationOperatingState.RUNNING
+
+    server.power_off()
+
+    for i in range(server.shut_down_duration + 1):
+        server.apply_timestep(timestep=i)
+
+    assert server.operating_state is NodeOperatingState.OFF
+    assert service.operating_state is ServiceOperatingState.STOPPED
+    assert app.operating_state is ApplicationOperatingState.CLOSED
+
+    service.start()
+    app.run()
+
+    assert server.operating_state is NodeOperatingState.OFF
+    assert service.operating_state is ServiceOperatingState.STOPPED
+    assert app.operating_state is ApplicationOperatingState.CLOSED
+
+
+def test_server_turns_on_service(populated_node):
+    """Check that turning on the server turns on service."""
+    app, server, service = populated_node
+
+    assert server.operating_state is NodeOperatingState.ON
+    assert service.operating_state is ServiceOperatingState.RUNNING
+    assert app.operating_state is ApplicationOperatingState.RUNNING
+
+    server.power_off()
+
+    for i in range(server.shut_down_duration + 1):
+        server.apply_timestep(timestep=i)
+
+    assert server.operating_state is NodeOperatingState.OFF
+    assert service.operating_state is ServiceOperatingState.STOPPED
+    assert app.operating_state is ApplicationOperatingState.CLOSED
+
+    server.power_on()
+
+    for i in range(server.start_up_duration + 1):
+        server.apply_timestep(timestep=i)
+
+    assert server.operating_state is NodeOperatingState.ON
+    assert service.operating_state is ServiceOperatingState.RUNNING
+    assert app.operating_state is ApplicationOperatingState.RUNNING

tests/integration_tests/system/test_service_on_node.py

@@ -1,42 +0,0 @@
-from typing import Tuple
-
-import pytest
-from conftest import TestService
-
-from primaite.simulator.network.hardware.node_operating_state import NodeOperatingState
-from primaite.simulator.network.hardware.nodes.server import Server
-from primaite.simulator.system.services.service import Service, ServiceOperatingState
-
-
-@pytest.fixture(scope="function")
-def service_on_node() -> Tuple[Server, Service]:
-    server = Server(
-        hostname="server", ip_address="192.168.0.1", subnet_mask="255.255.255.0", operating_state=NodeOperatingState.ON
-    )
-    server.software_manager.install(TestService)
-
-    service = server.software_manager.software["TestService"]
-    service.start()
-
-    return server, service
-
-
-def test_server_turns_off_service(service_on_node):
-    """Check that the service is turned off when the server is turned off"""
-    server, service = service_on_node
-
-    assert server.operating_state is NodeOperatingState.ON
-    assert service.operating_state is ServiceOperatingState.RUNNING
-
-    server.power_off()
-
-    for i in range(server.shut_down_duration + 1):
-        server.apply_timestep(timestep=i)
-
-    assert server.operating_state is NodeOperatingState.OFF
-    assert service.operating_state is ServiceOperatingState.STOPPED
-
-
-def test_server_turns_on_service(service_on_node):
-    """Check that turning on the server turns on service."""
-    pass