92 lines
2.4 KiB
Python
92 lines
2.4 KiB
Python
|
"""Used to test Active Node functions."""
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from primaite.common.enums import FileSystemState, HardwareState, SoftwareState
|
||
|
|
from primaite.nodes.active_node import ActiveNode
|
||
|
|
from primaite.nodes.service_node import ServiceNode
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
"starting_operating_state, expected_operating_state",
|
||
|
|
[
|
||
|
|
(HardwareState.RESETTING, HardwareState.ON)
|
||
|
|
],
|
||
|
|
)
|
||
|
|
def test_node_resets_correctly(starting_operating_state, expected_operating_state):
|
||
|
|
"""
|
||
|
|
Tests that a node resets correctly.
|
||
|
|
"""
|
||
|
|
active_node = ActiveNode(
|
||
|
|
node_id = 0,
|
||
|
|
name = "node",
|
||
|
|
node_type = "COMPUTER",
|
||
|
|
priority = "1",
|
||
|
|
hardware_state = starting_operating_state,
|
||
|
|
ip_address = "192.168.0.1",
|
||
|
|
software_state = SoftwareState.GOOD,
|
||
|
|
file_system_state = "GOOD",
|
||
|
|
config_values = 1,
|
||
|
|
)
|
||
|
|
|
||
|
|
for x in range(5):
|
||
|
|
active_node.update_resetting_status()
|
||
|
|
|
||
|
|
assert active_node.hardware_state == expected_operating_state
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
"operating_state, expected_operating_state",
|
||
|
|
[
|
||
|
|
(HardwareState.BOOTING, HardwareState.ON)
|
||
|
|
],
|
||
|
|
)
|
||
|
|
def test_node_boots_correctly(operating_state, expected_operating_state):
|
||
|
|
"""
|
||
|
|
Tests that a node boots correctly.
|
||
|
|
"""
|
||
|
|
active_node = ActiveNode(
|
||
|
|
node_id = 0,
|
||
|
|
name = "node",
|
||
|
|
node_type = "COMPUTER",
|
||
|
|
priority = "1",
|
||
|
|
hardware_state = operating_state,
|
||
|
|
ip_address = "192.168.0.1",
|
||
|
|
software_state = SoftwareState.GOOD,
|
||
|
|
file_system_state = "GOOD",
|
||
|
|
config_values = 1,
|
||
|
|
)
|
||
|
|
|
||
|
|
for x in range(5):
|
||
|
|
active_node.update_booting_status()
|
||
|
|
|
||
|
|
assert active_node.hardware_state == expected_operating_state
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
"operating_state, expected_operating_state",
|
||
|
|
[
|
||
|
|
(HardwareState.SHUTTING_DOWN, HardwareState.OFF)
|
||
|
|
],
|
||
|
|
)
|
||
|
|
def test_node_shutdown_correctly(operating_state, expected_operating_state):
|
||
|
|
"""
|
||
|
|
Tests that a node shutdown correctly.
|
||
|
|
"""
|
||
|
|
active_node = ActiveNode(
|
||
|
|
node_id = 0,
|
||
|
|
name = "node",
|
||
|
|
node_type = "COMPUTER",
|
||
|
|
priority = "1",
|
||
|
|
hardware_state = operating_state,
|
||
|
|
ip_address = "192.168.0.1",
|
||
|
|
software_state = SoftwareState.GOOD,
|
||
|
|
file_system_state = "GOOD",
|
||
|
|
config_values = 1,
|
||
|
|
)
|
||
|
|
|
||
|
|
for x in range(5):
|
||
|
|
active_node.update_shutdown_status()
|
||
|
|
|
||
|
|
assert active_node.hardware_state == expected_operating_state
|
||
|
|
|
||
|
|
|