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")
