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