def test_dequeue_next_active_step(self): """The FSM can remove the next step and update Mongo with it""" f = FSM(TEST_PBID, state_id) f.active_step = _active_step_string f.active_sequence = new_active_sequence() f.executed = [] f.execution = new_playbook()['execution'] _update_state = { '$set': { 'active_step': _active_step_string, 'active_sequence': f.active_sequence, 'executed': [], 'execution': new_playbook()['execution'], } } with mock.patch.object(f, 'update_state') as ( us): set_field = mock.MagicMock() filter = mock.MagicMock(return_value=set_field) f.filter = filter f.dequeue_next_active_step() us.assert_called_once_with(_update_state) self.assertEqual(f.active_step, _active_step_string)
def test_dequeue_next_active_step(self): """The FSM can remove the next step and update Mongo with it""" f = FSM(state_id) f.remaining = ["Step 1", "Step 2"] _update_state = { '$set': { 'active_step': "Step 1", 'remaining_steps': ["Step 2"] } } with mock.patch.object(f, 'update_state') as ( us): f.dequeue_next_active_step() us.assert_called_once_with(_update_state) self.assertEqual(f.active, "Step 1")