Channel Manager State Machine with Tests

[nullcount]

example:

import unittest

class StateMachine:
    def __init__(self, initial_state):
        self.state = initial_state

    def transition(self, event):
        self.state = self.state.transition(event)

class State:
    def transition(self, event):
        raise NotImplementedError

class InitialState(State):
    def transition(self, event):
        if event == "start":
            return StartedState()
        return self

class StartedState(State):
    def transition(self, event):
        if event == "stop":
            return StoppedState()
        return self

class StoppedState(State):
    def transition(self, event):
        if event == "start":
            return StartedState()
        return self

class TestStateMachine(unittest.TestCase):
    def test_state_machine(self):
        machine = StateMachine(InitialState())

        # Test initial state
        self.assertEqual(machine.state.__class__, InitialState)

        # Test transition to StartedState
        machine.transition("start")
        self.assertEqual(machine.state.__class__, StartedState)

        # Test transition to StoppedState
        machine.transition("stop")
        self.assertEqual(machine.state.__class__, StoppedState)

        # Test transition back to StartedState
        machine.transition("start")
        self.assertEqual(machine.state.__class__, StartedState)

if __name__ == "__main__":
    unittest.main()