def test_run_to_first_checkpoint_after_method(self):
machine = ThreePhaseMachine()
player = ProcessPlayer(machine)
cp = player.add_checkpoint_after(machine.second_phase)
conductor = ProcessConductor([player], [cp])
next(conductor)
self.assertEqual(machine.steps, [1, 2])
def test_first_thread_finishes_then_second_starts(self):
first_machine = ThreePhaseMachine()
second_machine = ThreePhaseMachine()
first_starter = ProcessPlayer(first_machine)
cp1 = first_starter.add_checkpoint_after(first_machine.third_phase)
second_starter = ProcessPlayer(second_machine)
cp2 = second_starter.add_checkpoint_before(second_machine.first_phase)
conductor = ProcessConductor([first_starter, second_starter], [cp1, cp2])
self.assertIs(next(conductor), cp1)
self.assertEqual(first_machine.steps, [1, 2, 3])
self.assertEqual(second_machine.steps, [])
self.assertIs(next(conductor), cp2)
next(conductor)
self.assertEqual(second_machine.steps, [1, 2, 3])
def test_thread_state_changes_after_each_checkpoint(self):
machine = ThreePhaseMachine()
player = ProcessPlayer(machine)
cp1 = player.add_checkpoint_before(machine.first_phase)
cp2 = player.add_checkpoint_before(machine.second_phase)
cp3 = player.add_checkpoint_before(machine.third_phase)
cp4 = player.add_checkpoint_after(machine.third_phase)
conductor = ProcessConductor([player], [cp1, cp2, cp3, cp4])
self.assertEqual(machine.steps, [])
next(conductor)
self.assertEqual(machine.steps, [])
next(conductor)
self.assertEqual(machine.steps, [1])
next(conductor)
self.assertEqual(machine.steps, [1, 2])
next(conductor)
self.assertEqual(machine.steps, [1, 2, 3])
def test_simple_order_for_2_processes(self):
# Time -> ...
# Process 1: CP1 CP2 CP3
# Process 2: CP1 CP2 CP3
machine1 = ThreePhaseMachine()
machine2 = ThreePhaseMachine()
starter1 = ProcessPlayer(machine1)
starter2 = ProcessPlayer(machine2)
cp1_1 = starter1.add_checkpoint_before(machine1.first_phase)
cp1_2 = starter1.add_checkpoint_before(machine1.second_phase)
cp1_3 = starter1.add_checkpoint_after(machine1.third_phase)
cp2_1 = starter2.add_checkpoint_before(machine2.first_phase)
cp2_2 = starter2.add_checkpoint_before(machine2.second_phase)
cp2_3 = starter2.add_checkpoint_before(machine2.third_phase)
conductor = ProcessConductor(
[starter1, starter2],
[cp1_1, cp1_2, cp1_3, cp2_1, cp2_2, cp2_3]
)
next(conductor)
self.assertEqual(machine1.steps, [])
next(conductor)
self.assertEqual(machine1.steps, [1])
next(conductor)
self.assertEqual(machine1.steps, [1, 2, 3])
next(conductor)
self.assertEqual(machine2.steps, [])
self.assertEqual(machine1.steps, [1, 2, 3])
next(conductor)
self.assertEqual(machine2.steps, [1])
next(conductor)
self.assertEqual(machine2.steps, [1, 2])
next(conductor)
self.assertEqual(machine2.steps, [1, 2, 3])
