71 lines
1.8 KiB
Python
71 lines
1.8 KiB
Python
"""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
|