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)