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

- Added unit tests
2023-05-25 13:02:15 +01:00
parent e35360edb8
commit e1bcbd1a34
4 changed files with 320 additions and 64 deletions
--- a/tests/test_active_node.py
+++ b/tests/test_active_node.py
@@ -0,0 +1,121 @@
+"""Used to test Active Node functions."""
+import pytest
+
+from primaite.common.enums import FILE_SYSTEM_STATE, HARDWARE_STATE, SOFTWARE_STATE
+from primaite.nodes.active_node import ActiveNode
+
+
+@pytest.mark.parametrize(
+    "operating_state, expected_state",
+    [
+        (HARDWARE_STATE.OFF, SOFTWARE_STATE.GOOD),
+        (HARDWARE_STATE.ON, SOFTWARE_STATE.OVERWHELMED),
+    ],
+)
+def test_os_state_change(operating_state, expected_state):
+    """
+    Test that a node cannot change its operating system state.
+
+    When its operating state is OFF.
+    """
+    active_node = ActiveNode(
+        0,
+        "node",
+        "COMPUTER",
+        "1",
+        operating_state,
+        "192.168.0.1",
+        SOFTWARE_STATE.GOOD,
+        "GOOD",
+        1,
+    )
+
+    active_node.set_os_state(SOFTWARE_STATE.OVERWHELMED)
+
+    assert active_node.get_os_state() == expected_state
+
+
+@pytest.mark.parametrize(
+    "operating_state, expected_state",
+    [
+        (HARDWARE_STATE.OFF, SOFTWARE_STATE.GOOD),
+        (HARDWARE_STATE.ON, SOFTWARE_STATE.OVERWHELMED),
+    ],
+)
+def test_os_state_change_if_not_compromised(operating_state, expected_state):
+    """
+    Test that a node cannot change its operating system state.
+
+    If not compromised) when its operating state is OFF.
+    """
+    active_node = ActiveNode(
+        0,
+        "node",
+        "COMPUTER",
+        "1",
+        operating_state,
+        "192.168.0.1",
+        SOFTWARE_STATE.GOOD,
+        "GOOD",
+        1,
+    )
+
+    active_node.set_os_state_if_not_compromised(SOFTWARE_STATE.OVERWHELMED)
+
+    assert active_node.get_os_state() == expected_state
+
+
+@pytest.mark.parametrize(
+    "operating_state, expected_state",
+    [
+        (HARDWARE_STATE.OFF, FILE_SYSTEM_STATE.GOOD),
+        (HARDWARE_STATE.ON, FILE_SYSTEM_STATE.CORRUPT),
+    ],
+)
+def test_file_system_change(operating_state, expected_state):
+    """Test that a node cannot change its file system state when its operating state is ON."""
+    active_node = ActiveNode(
+        0,
+        "node",
+        "COMPUTER",
+        "1",
+        operating_state,
+        "192.168.0.1",
+        "COMPROMISED",
+        FILE_SYSTEM_STATE.GOOD,
+        1,
+    )
+
+    active_node.set_file_system_state(FILE_SYSTEM_STATE.CORRUPT)
+
+    assert active_node.get_file_system_state_actual() == expected_state
+
+
+@pytest.mark.parametrize(
+    "operating_state, expected_state",
+    [
+        (HARDWARE_STATE.OFF, FILE_SYSTEM_STATE.GOOD),
+        (HARDWARE_STATE.ON, FILE_SYSTEM_STATE.CORRUPT),
+    ],
+)
+def test_file_system_change_if_not_compromised(operating_state, expected_state):
+    """
+    Test that a node cannot change its file system state.
+
+    If not compromised) when its operating state is OFF.
+    """
+    active_node = ActiveNode(
+        0,
+        "node",
+        "COMPUTER",
+        "1",
+        operating_state,
+        "192.168.0.1",
+        "GOOD",
+        FILE_SYSTEM_STATE.GOOD,
+        1,
+    )
+
+    active_node.set_file_system_state_if_not_compromised(FILE_SYSTEM_STATE.CORRUPT)
+
+    assert active_node.get_file_system_state_actual() == expected_state
--- a/tests/test_service_node.py
+++ b/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