def test_tear_down_repeat(self, mocker, decider):
info = {
'executionInfo': {
'workflowType': {
'name': 'wf_name',
'version': 'v1'
}
}
}
mocker.patch('floto.decider.Base.get_workflow_execution_description',
return_value=info)
mocker.patch('floto.api.Swf.start_workflow_execution')
mocker.patch('floto.History.get_workflow_input',
return_value='wf_input')
decider.repeat_workflow = True
decider.domain = 'd'
decider.task_list = 'tl'
decider.tear_down()
expected_args = {
'domain': 'd',
'workflow_type_name': 'wf_name',
'workflow_type_version': 'v1',
'task_list': 'tl',
'input': 'wf_input'
}
decider.swf.start_workflow_execution.assert_called_once_with(
**expected_args)
def test_tear_down_repeat(self, mocker, decider):
info = {'executionInfo':{'workflowType':{'name':'wf_name', 'version':'v1'}}}
mocker.patch('floto.decider.Base.get_workflow_execution_description', return_value=info)
mocker.patch('floto.api.Swf.start_workflow_execution')
mocker.patch('floto.History.get_workflow_input', return_value='wf_input')
decider.repeat_workflow = True
decider.domain = 'd'
decider.task_list = 'tl'
decider.tear_down()
expected_args = {'domain':'d',
'workflow_type_name':'wf_name',
'workflow_type_version':'v1',
'task_list':'tl',
'input':'wf_input'}
decider.swf.start_workflow_execution.assert_called_once_with(**expected_args)
def test_tear_down_do_not_repeat(self, decider):
decider.swf.start_workflow_execution = Mock()
decider.terminate_decider = False
decider.tear_down()
assert decider.terminate_decider == True
assert not decider.swf.start_workflow_execution.called
def test_tear_down_do_not_repeat(self, decider):
decider.swf.start_workflow_execution = Mock()
decider.terminate_decider = False
decider.tear_down()
assert decider.terminate_decider == True
assert not decider.swf.start_workflow_execution.called
