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)