Run black on moto & test directories.

tests/test_swf/responses/test_activity_tasks.py

@@ -12,18 +12,19 @@ from ..utils import setup_workflow, SCHEDULE_ACTIVITY_TASK_DECISION
 @mock_swf_deprecated
 def test_poll_for_activity_task_when_one():
     conn = setup_workflow()
-    decision_token = conn.poll_for_decision_task(
-        "test-domain", "queue")["taskToken"]
-    conn.respond_decision_task_completed(decision_token, decisions=[
-        SCHEDULE_ACTIVITY_TASK_DECISION
-    ])
+    decision_token = conn.poll_for_decision_task("test-domain", "queue")["taskToken"]
+    conn.respond_decision_task_completed(
+        decision_token, decisions=[SCHEDULE_ACTIVITY_TASK_DECISION]
+    )
     resp = conn.poll_for_activity_task(
-        "test-domain", "activity-task-list", identity="surprise")
+        "test-domain", "activity-task-list", identity="surprise"
+    )
     resp["activityId"].should.equal("my-activity-001")
     resp["taskToken"].should_not.be.none
 
     resp = conn.get_workflow_execution_history(
-        "test-domain", conn.run_id, "uid-abcd1234")
+        "test-domain", conn.run_id, "uid-abcd1234"
+    )
     resp["events"][-1]["eventType"].should.equal("ActivityTaskStarted")
     resp["events"][-1]["activityTaskStartedEventAttributes"].should.equal(
         {"identity": "surprise", "scheduledEventId": 5}
@@ -48,14 +49,12 @@ def test_poll_for_activity_task_on_non_existent_queue():
 @mock_swf_deprecated
 def test_count_pending_activity_tasks():
     conn = setup_workflow()
-    decision_token = conn.poll_for_decision_task(
-        "test-domain", "queue")["taskToken"]
-    conn.respond_decision_task_completed(decision_token, decisions=[
-        SCHEDULE_ACTIVITY_TASK_DECISION
-    ])
+    decision_token = conn.poll_for_decision_task("test-domain", "queue")["taskToken"]
+    conn.respond_decision_task_completed(
+        decision_token, decisions=[SCHEDULE_ACTIVITY_TASK_DECISION]
+    )
 
-    resp = conn.count_pending_activity_tasks(
-        "test-domain", "activity-task-list")
+    resp = conn.count_pending_activity_tasks("test-domain", "activity-task-list")
     resp.should.equal({"count": 1, "truncated": False})
 
 
@@ -70,20 +69,22 @@ def test_count_pending_decision_tasks_on_non_existent_task_list():
 @mock_swf_deprecated
 def test_respond_activity_task_completed():
     conn = setup_workflow()
-    decision_token = conn.poll_for_decision_task(
-        "test-domain", "queue")["taskToken"]
-    conn.respond_decision_task_completed(decision_token, decisions=[
-        SCHEDULE_ACTIVITY_TASK_DECISION
-    ])
-    activity_token = conn.poll_for_activity_task(
-        "test-domain", "activity-task-list")["taskToken"]
+    decision_token = conn.poll_for_decision_task("test-domain", "queue")["taskToken"]
+    conn.respond_decision_task_completed(
+        decision_token, decisions=[SCHEDULE_ACTIVITY_TASK_DECISION]
+    )
+    activity_token = conn.poll_for_activity_task("test-domain", "activity-task-list")[
+        "taskToken"
+    ]
 
     resp = conn.respond_activity_task_completed(
-        activity_token, result="result of the task")
+        activity_token, result="result of the task"
+    )
     resp.should.be.none
 
     resp = conn.get_workflow_execution_history(
-        "test-domain", conn.run_id, "uid-abcd1234")
+        "test-domain", conn.run_id, "uid-abcd1234"
+    )
     resp["events"][-2]["eventType"].should.equal("ActivityTaskCompleted")
     resp["events"][-2]["activityTaskCompletedEventAttributes"].should.equal(
         {"result": "result of the task", "scheduledEventId": 5, "startedEventId": 6}
@@ -93,13 +94,13 @@ def test_respond_activity_task_completed():
 @mock_swf_deprecated
 def test_respond_activity_task_completed_on_closed_workflow_execution():
     conn = setup_workflow()
-    decision_token = conn.poll_for_decision_task(
-        "test-domain", "queue")["taskToken"]
-    conn.respond_decision_task_completed(decision_token, decisions=[
-        SCHEDULE_ACTIVITY_TASK_DECISION
-    ])
-    activity_token = conn.poll_for_activity_task(
-        "test-domain", "activity-task-list")["taskToken"]
+    decision_token = conn.poll_for_decision_task("test-domain", "queue")["taskToken"]
+    conn.respond_decision_task_completed(
+        decision_token, decisions=[SCHEDULE_ACTIVITY_TASK_DECISION]
+    )
+    activity_token = conn.poll_for_activity_task("test-domain", "activity-task-list")[
+        "taskToken"
+    ]
 
     # bad: we're closing workflow execution manually, but endpoints are not
     # coded for now..
@@ -107,52 +108,57 @@ def test_respond_activity_task_completed_on_closed_workflow_execution():
     wfe.execution_status = "CLOSED"
     # /bad
 
-    conn.respond_activity_task_completed.when.called_with(
-        activity_token
-    ).should.throw(SWFResponseError, "WorkflowExecution=")
+    conn.respond_activity_task_completed.when.called_with(activity_token).should.throw(
+        SWFResponseError, "WorkflowExecution="
+    )
 
 
 @mock_swf_deprecated
 def test_respond_activity_task_completed_with_task_already_completed():
     conn = setup_workflow()
-    decision_token = conn.poll_for_decision_task(
-        "test-domain", "queue")["taskToken"]
-    conn.respond_decision_task_completed(decision_token, decisions=[
-        SCHEDULE_ACTIVITY_TASK_DECISION
-    ])
-    activity_token = conn.poll_for_activity_task(
-        "test-domain", "activity-task-list")["taskToken"]
+    decision_token = conn.poll_for_decision_task("test-domain", "queue")["taskToken"]
+    conn.respond_decision_task_completed(
+        decision_token, decisions=[SCHEDULE_ACTIVITY_TASK_DECISION]
+    )
+    activity_token = conn.poll_for_activity_task("test-domain", "activity-task-list")[
+        "taskToken"
+    ]
 
     conn.respond_activity_task_completed(activity_token)
 
-    conn.respond_activity_task_completed.when.called_with(
-        activity_token
-    ).should.throw(SWFResponseError, "Unknown activity, scheduledEventId = 5")
+    conn.respond_activity_task_completed.when.called_with(activity_token).should.throw(
+        SWFResponseError, "Unknown activity, scheduledEventId = 5"
+    )
 
 
 # RespondActivityTaskFailed endpoint
 @mock_swf_deprecated
 def test_respond_activity_task_failed():
     conn = setup_workflow()
-    decision_token = conn.poll_for_decision_task(
-        "test-domain", "queue")["taskToken"]
-    conn.respond_decision_task_completed(decision_token, decisions=[
-        SCHEDULE_ACTIVITY_TASK_DECISION
-    ])
-    activity_token = conn.poll_for_activity_task(
-        "test-domain", "activity-task-list")["taskToken"]
+    decision_token = conn.poll_for_decision_task("test-domain", "queue")["taskToken"]
+    conn.respond_decision_task_completed(
+        decision_token, decisions=[SCHEDULE_ACTIVITY_TASK_DECISION]
+    )
+    activity_token = conn.poll_for_activity_task("test-domain", "activity-task-list")[
+        "taskToken"
+    ]
 
-    resp = conn.respond_activity_task_failed(activity_token,
-                                             reason="short reason",
-                                             details="long details")
+    resp = conn.respond_activity_task_failed(
+        activity_token, reason="short reason", details="long details"
+    )
     resp.should.be.none
 
     resp = conn.get_workflow_execution_history(
-        "test-domain", conn.run_id, "uid-abcd1234")
+        "test-domain", conn.run_id, "uid-abcd1234"
+    )
     resp["events"][-2]["eventType"].should.equal("ActivityTaskFailed")
     resp["events"][-2]["activityTaskFailedEventAttributes"].should.equal(
-        {"reason": "short reason", "details": "long details",
-         "scheduledEventId": 5, "startedEventId": 6}
+        {
+            "reason": "short reason",
+            "details": "long details",
+            "scheduledEventId": 5,
+            "startedEventId": 6,
+        }
     )
 
 
@@ -162,11 +168,10 @@ def test_respond_activity_task_completed_with_wrong_token():
     # because the safeguards are shared with RespondActivityTaskCompleted, so
     # no need to retest everything end-to-end.
     conn = setup_workflow()
-    decision_token = conn.poll_for_decision_task(
-        "test-domain", "queue")["taskToken"]
-    conn.respond_decision_task_completed(decision_token, decisions=[
-        SCHEDULE_ACTIVITY_TASK_DECISION
-    ])
+    decision_token = conn.poll_for_decision_task("test-domain", "queue")["taskToken"]
+    conn.respond_decision_task_completed(
+        decision_token, decisions=[SCHEDULE_ACTIVITY_TASK_DECISION]
+    )
     conn.poll_for_activity_task("test-domain", "activity-task-list")
     conn.respond_activity_task_failed.when.called_with(
         "not-a-correct-token"
@@ -177,13 +182,13 @@ def test_respond_activity_task_completed_with_wrong_token():
 @mock_swf_deprecated
 def test_record_activity_task_heartbeat():
     conn = setup_workflow()
-    decision_token = conn.poll_for_decision_task(
-        "test-domain", "queue")["taskToken"]
-    conn.respond_decision_task_completed(decision_token, decisions=[
-        SCHEDULE_ACTIVITY_TASK_DECISION
-    ])
-    activity_token = conn.poll_for_activity_task(
-        "test-domain", "activity-task-list")["taskToken"]
+    decision_token = conn.poll_for_decision_task("test-domain", "queue")["taskToken"]
+    conn.respond_decision_task_completed(
+        decision_token, decisions=[SCHEDULE_ACTIVITY_TASK_DECISION]
+    )
+    activity_token = conn.poll_for_activity_task("test-domain", "activity-task-list")[
+        "taskToken"
+    ]
 
     resp = conn.record_activity_task_heartbeat(activity_token)
     resp.should.equal({"cancelRequested": False})
@@ -192,13 +197,11 @@ def test_record_activity_task_heartbeat():
 @mock_swf_deprecated
 def test_record_activity_task_heartbeat_with_wrong_token():
     conn = setup_workflow()
-    decision_token = conn.poll_for_decision_task(
-        "test-domain", "queue")["taskToken"]
-    conn.respond_decision_task_completed(decision_token, decisions=[
-        SCHEDULE_ACTIVITY_TASK_DECISION
-    ])
-    conn.poll_for_activity_task(
-        "test-domain", "activity-task-list")["taskToken"]
+    decision_token = conn.poll_for_decision_task("test-domain", "queue")["taskToken"]
+    conn.respond_decision_task_completed(
+        decision_token, decisions=[SCHEDULE_ACTIVITY_TASK_DECISION]
+    )
+    conn.poll_for_activity_task("test-domain", "activity-task-list")["taskToken"]
 
     conn.record_activity_task_heartbeat.when.called_with(
         "bad-token", details="some progress details"
@@ -208,21 +211,23 @@ def test_record_activity_task_heartbeat_with_wrong_token():
 @mock_swf_deprecated
 def test_record_activity_task_heartbeat_sets_details_in_case_of_timeout():
     conn = setup_workflow()
-    decision_token = conn.poll_for_decision_task(
-        "test-domain", "queue")["taskToken"]
-    conn.respond_decision_task_completed(decision_token, decisions=[
-        SCHEDULE_ACTIVITY_TASK_DECISION
-    ])
+    decision_token = conn.poll_for_decision_task("test-domain", "queue")["taskToken"]
+    conn.respond_decision_task_completed(
+        decision_token, decisions=[SCHEDULE_ACTIVITY_TASK_DECISION]
+    )
     with freeze_time("2015-01-01 12:00:00"):
         activity_token = conn.poll_for_activity_task(
-            "test-domain", "activity-task-list")["taskToken"]
+            "test-domain", "activity-task-list"
+        )["taskToken"]
         conn.record_activity_task_heartbeat(
-            activity_token, details="some progress details")
+            activity_token, details="some progress details"
+        )
 
     with freeze_time("2015-01-01 12:05:30"):
         # => Activity Task Heartbeat timeout reached!!
         resp = conn.get_workflow_execution_history(
-            "test-domain", conn.run_id, "uid-abcd1234")
+            "test-domain", conn.run_id, "uid-abcd1234"
+        )
         resp["events"][-2]["eventType"].should.equal("ActivityTaskTimedOut")
         attrs = resp["events"][-2]["activityTaskTimedOutEventAttributes"]
         attrs["details"].should.equal("some progress details")