Implement heartbeat timeout on SWF activity tasks

tests/test_swf/responses/test_timeouts.py

@@ -0,0 +1,34 @@
+import boto
+from freezegun import freeze_time
+from sure import expect
+
+from moto import mock_swf
+
+from ..utils import setup_workflow, SCHEDULE_ACTIVITY_TASK_DECISION
+
+
+# Activity Task Heartbeat timeout
+# Default value in workflow helpers: 5 mins
+@mock_swf
+def test_activity_task_heartbeat_timeout():
+    with freeze_time("2015-01-01 12:00:00"):
+        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", identity="surprise")
+
+    with freeze_time("2015-01-01 12:04:30"):
+        resp = conn.get_workflow_execution_history("test-domain", conn.run_id, "uid-abcd1234")
+        resp["events"][-1]["eventType"].should.equal("ActivityTaskStarted")
+
+    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")
+
+        resp["events"][-2]["eventType"].should.equal("ActivityTaskTimedOut")
+        attrs = resp["events"][-2]["activityTaskTimedOutEventAttributes"]
+        attrs["timeoutType"].should.equal("HEARTBEAT")
+
+        resp["events"][-1]["eventType"].should.equal("DecisionTaskScheduled")