diff --git a/tests/test_swf/responses/test_decision_tasks.py b/tests/test_swf/responses/test_decision_tasks.py index f20626ee1..0dfe369ec 100644 --- a/tests/test_swf/responses/test_decision_tasks.py +++ b/tests/test_swf/responses/test_decision_tasks.py @@ -9,24 +9,7 @@ from moto.swf.exceptions import ( SWFDecisionValidationException, ) -from ..utils import mock_basic_workflow_type - - -@mock_swf -def setup_workflow(): - conn = boto.connect_swf("the_key", "the_secret") - conn.register_domain("test-domain", "60", description="A test domain") - conn = mock_basic_workflow_type("test-domain", conn) - conn.register_activity_type( - "test-domain", "test-activity", "v1.1", - default_task_heartbeat_timeout="600", - default_task_schedule_to_close_timeout="600", - default_task_schedule_to_start_timeout="600", - default_task_start_to_close_timeout="600", - ) - wfe = conn.start_workflow_execution("test-domain", "uid-abcd1234", "test-workflow", "v1.0") - conn.run_id = wfe["runId"] - return conn +from ..utils import setup_workflow # PollForDecisionTask endpoint diff --git a/tests/test_swf/utils.py b/tests/test_swf/utils.py index 55121b319..6fcec9d46 100644 --- a/tests/test_swf/utils.py +++ b/tests/test_swf/utils.py @@ -1,3 +1,6 @@ +import boto + +from moto import mock_swf from moto.swf.models import ( ActivityType, Domain, @@ -38,3 +41,21 @@ def make_workflow_execution(**kwargs): domain.add_type(ActivityType("test-activity", "v1.1")) wft = get_basic_workflow_type() return WorkflowExecution(domain, wft, "ab1234", **kwargs) + + +# Setup a complete example workflow and return the connection object +@mock_swf +def setup_workflow(): + conn = boto.connect_swf("the_key", "the_secret") + conn.register_domain("test-domain", "60", description="A test domain") + conn = mock_basic_workflow_type("test-domain", conn) + conn.register_activity_type( + "test-domain", "test-activity", "v1.1", + default_task_heartbeat_timeout="600", + default_task_schedule_to_close_timeout="600", + default_task_schedule_to_start_timeout="600", + default_task_start_to_close_timeout="600", + ) + wfe = conn.start_workflow_execution("test-domain", "uid-abcd1234", "test-workflow", "v1.0") + conn.run_id = wfe["runId"] + return conn