Add SWF endpoint PollForActivityTask

tests/test_swf/models/test_activity_task.py

@@ -1,6 +1,9 @@
 from sure import expect
 
-from moto.swf.models import ActivityTask
+from moto.swf.models import (
+    ActivityTask,
+    ActivityType,
+)
 
 from ..utils import make_workflow_execution
 
@@ -11,6 +14,7 @@ def test_activity_task_creation():
         activity_id="my-activity-123",
         activity_type="foo",
         input="optional",
+        scheduled_event_id=117,
         workflow_execution=wfe,
     )
     task.workflow_execution.should.equal(wfe)
@@ -24,3 +28,27 @@ def test_activity_task_creation():
 
     task.complete()
     task.state.should.equal("COMPLETED")
+
+def test_activity_task_full_dict_representation():
+    wfe = make_workflow_execution()
+    wft = wfe.workflow_type
+    at = ActivityTask(
+        activity_id="my-activity-123",
+        activity_type=ActivityType("foo", "v1.0"),
+        input="optional",
+        scheduled_event_id=117,
+        workflow_execution=wfe,
+    )
+    at.start(1234)
+
+    fd = at.to_full_dict()
+    fd["activityId"].should.equal("my-activity-123")
+    fd["activityType"]["version"].should.equal("v1.0")
+    fd["input"].should.equal("optional")
+    fd["startedEventId"].should.equal(1234)
+    fd.should.contain("taskToken")
+    fd["workflowExecution"].should.equal(wfe.to_short_dict())
+
+    at.start(1234)
+    fd = at.to_full_dict()
+    fd["startedEventId"].should.equal(1234)