def test_move_active_to_completed(self):
"""FSM can update after completing a step"""
f = FSM(state_id)
active_step = {"plugin": "not real"}
f.active = active_step.copy()
f.completed = []
# For .called_once_with()
_update_state = {
'$set': {
'active_step': None,
'completed_steps': [active_step]
}
}
with mock.patch.object(f, 'update_state') as (us):
f.move_active_to_completed()
us.assert_called_once_with(_update_state)
self.assertEqual(f.active, None)
self.assertEqual(f.completed, [active_step])
def test_move_active_to_completed(self):
"""FSM can update after completing a step"""
f = FSM(state_id)
f.active_step = _active_step_string
f.active_sequence = new_active_sequence()
f.active_sequence['completed_steps'] = []
f.executed = []
f.execution = new_playbook()['execution']
f.completed_steps = []
# For .called_once_with()
_update_state = {
'$set': {
'active_step': None,
'active_sequence': f.active_sequence
}
}
with mock.patch.object(f, 'update_state') as (us):
f.move_active_to_completed()
us.assert_called_once_with(_update_state)