122 lines
3.0 KiB
Python
122 lines
3.0 KiB
Python
|
"""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
|