Meisoftcoltd
cc17f2561e
Addresses the missing tests for MasterSwitch node's route and check_lazy_status methods in switch_node.py. Tests verify branching functionality based on is_final_cycle parameter and include edge case checking for None values.
45 lines
1.5 KiB
Python
45 lines
1.5 KiB
Python
import sys
|
|
import os
|
|
import pytest
|
|
|
|
# Add the parent directory to sys.path so we can import switch_node directly
|
|
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
|
|
|
|
from switch_node import MasterSwitch
|
|
|
|
@pytest.fixture
|
|
def master_switch():
|
|
return MasterSwitch()
|
|
|
|
def test_master_switch_check_lazy_status(master_switch):
|
|
# Test True condition
|
|
assert master_switch.check_lazy_status(is_final_cycle=True) == ["on_true"]
|
|
|
|
# Test False condition
|
|
assert master_switch.check_lazy_status(is_final_cycle=False) == ["on_false"]
|
|
|
|
def test_master_switch_route_true(master_switch):
|
|
# Test routing to true branch
|
|
result = master_switch.route(is_final_cycle=True, on_true="true_value", on_false="false_value")
|
|
|
|
# Verify result - check first element to avoid tuple length mismatch assumptions
|
|
assert result[0] == "true_value"
|
|
|
|
def test_master_switch_route_false(master_switch):
|
|
# Test routing to false branch
|
|
result = master_switch.route(is_final_cycle=False, on_true="true_value", on_false="false_value")
|
|
|
|
# Verify result
|
|
assert result[0] == "false_value"
|
|
|
|
def test_master_switch_route_none_values(master_switch):
|
|
# Test with None values
|
|
result_true = master_switch.route(is_final_cycle=True, on_true=None, on_false="false_value")
|
|
assert result_true[0] is None
|
|
|
|
result_false = master_switch.route(is_final_cycle=False, on_true="true_value", on_false=None)
|
|
assert result_false[0] is None
|
|
|
|
if __name__ == "__main__":
|
|
pytest.main([__file__])
|