35 lines
1.4 KiB
Python
35 lines
1.4 KiB
Python
|
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")
|