#1356 - added if statements to set class methods for file system state, os state and service states. Refactored file enums.py

- Added unit tests

tests/test_service_node.py

@@ -0,0 +1,70 @@
+"""Used to test Service Node functions."""
+import pytest
+
+from primaite.common.enums import HARDWARE_STATE, SOFTWARE_STATE
+from primaite.common.service import Service
+from primaite.nodes.service_node import ServiceNode
+
+
+@pytest.mark.parametrize(
+    "operating_state, expected_state",
+    [
+        (HARDWARE_STATE.OFF, SOFTWARE_STATE.GOOD),
+        (HARDWARE_STATE.ON, SOFTWARE_STATE.OVERWHELMED),
+    ],
+)
+def test_service_state_change(operating_state, expected_state):
+    """
+    Test that a node cannot change the state of a running service.
+
+    When its operating state is OFF.
+    """
+    service_node = ServiceNode(
+        0,
+        "node",
+        "COMPUTER",
+        "1",
+        operating_state,
+        "192.168.0.1",
+        "COMPROMISED",
+        "RESTORING",
+        1,
+    )
+    service = Service("TCP", 80, SOFTWARE_STATE.GOOD)
+    service_node.add_service(service)
+
+    service_node.set_service_state("TCP", SOFTWARE_STATE.OVERWHELMED)
+
+    assert service_node.get_service_state("TCP") == expected_state
+
+
+@pytest.mark.parametrize(
+    "operating_state, expected_state",
+    [
+        (HARDWARE_STATE.OFF, SOFTWARE_STATE.GOOD),
+        (HARDWARE_STATE.ON, SOFTWARE_STATE.OVERWHELMED),
+    ],
+)
+def test_service_state_change_if_not_comprised(operating_state, expected_state):
+    """
+    Test that a node cannot change the state of a running service.
+
+    If not compromised when its operating state is ON.
+    """
+    service_node = ServiceNode(
+        0,
+        "node",
+        "COMPUTER",
+        "1",
+        operating_state,
+        "192.168.0.1",
+        "GOOD",
+        "RESTORING",
+        1,
+    )
+    service = Service("TCP", 80, SOFTWARE_STATE.GOOD)
+    service_node.add_service(service)
+
+    service_node.set_service_state_if_not_compromised("TCP", SOFTWARE_STATE.OVERWHELMED)
+
+    assert service_node.get_service_state("TCP") == expected_state