Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix(swf_tests): compare tz-aware timestamps (#5039)

Browse Source
This commit is contained in:
Iain Samuel McLean Elder 2022-04-20 00:22:49 +02:00 committed by GitHub
parent ba7c852b64
commit fc170df796
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 19 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/test_swf/responses/test_decision_tasks.py
View File

@ -1,5 +1,6 @@
from botocore.exceptions import ClientError from botocore.exceptions import ClientError
from datetime import datetime from datetime import datetime
from dateutil.parser import parse as dtparse
from freezegun import freeze_time from freezegun import freeze_time
from time import sleep from time import sleep
import sure # noqa # pylint: disable=unused-import import sure # noqa # pylint: disable=unused-import
@ -380,7 +381,7 @@ def test_respond_decision_task_completed_with_fail_workflow_execution_boto3():
@mock_swf @mock_swf
@freeze_time("2015-01-01 12:00:00") @freeze_time("2015-01-01 12:00:00 UTC")
def test_respond_decision_task_completed_with_schedule_activity_task_boto3(): def test_respond_decision_task_completed_with_schedule_activity_task_boto3():
client = setup_workflow_boto3() client = setup_workflow_boto3()
resp = client.poll_for_decision_task( resp = client.poll_for_decision_task(
@ -433,8 +434,8 @@ def test_respond_decision_task_completed_with_schedule_activity_task_boto3():
) )
resp["latestActivityTaskTimestamp"].should.be.a(datetime) resp["latestActivityTaskTimestamp"].should.be.a(datetime)
if not settings.TEST_SERVER_MODE: if not settings.TEST_SERVER_MODE:
ts = resp["latestActivityTaskTimestamp"].strftime("%Y-%m-%d %H:%M:%S") ts = resp["latestActivityTaskTimestamp"]
ts.should.equal("2015-01-01 12:00:00") ts.should.equal(dtparse("2015-01-01 12:00:00 UTC"))
@mock_swf @mock_swf

29
tests/test_swf/responses/test_timeouts.py
View File

@ -1,4 +1,5 @@
from datetime import datetime from datetime import datetime
from dateutil.parser import parse as dtparse
from freezegun import freeze_time from freezegun import freeze_time
import sure # noqa # pylint: disable=unused-import import sure # noqa # pylint: disable=unused-import
from unittest import SkipTest from unittest import SkipTest
@ -29,14 +30,14 @@ def test_activity_task_heartbeat_timeout_boto3():
identity="surprise", identity="surprise",
) )
with freeze_time("2015-01-01 12:04:30"): with freeze_time("2015-01-01 12:04:30 UTC"):
resp = client.get_workflow_execution_history( resp = client.get_workflow_execution_history(
domain="test-domain", domain="test-domain",
execution={"runId": client.run_id, "workflowId": "uid-abcd1234"}, execution={"runId": client.run_id, "workflowId": "uid-abcd1234"},
) )
resp["events"][-1]["eventType"].should.equal("ActivityTaskStarted") resp["events"][-1]["eventType"].should.equal("ActivityTaskStarted")
with freeze_time("2015-01-01 12:05:30"): with freeze_time("2015-01-01 12:05:30 UTC"):
# => Activity Task Heartbeat timeout reached!! # => Activity Task Heartbeat timeout reached!!
resp = client.get_workflow_execution_history( resp = client.get_workflow_execution_history(
domain="test-domain", domain="test-domain",
@ -48,8 +49,8 @@ def test_activity_task_heartbeat_timeout_boto3():
attrs["timeoutType"].should.equal("HEARTBEAT") attrs["timeoutType"].should.equal("HEARTBEAT")
# checks that event has been emitted at 12:05:00, not 12:05:30 # checks that event has been emitted at 12:05:00, not 12:05:30
resp["events"][-2]["eventTimestamp"].should.be.a(datetime) resp["events"][-2]["eventTimestamp"].should.be.a(datetime)
ts = resp["events"][-2]["eventTimestamp"].strftime("%Y-%m-%d %H:%M:%S") ts = resp["events"][-2]["eventTimestamp"]
ts.should.equal("2015-01-01 12:05:00") ts.should.equal(dtparse("2015-01-01 12:05:00 UTC"))
# Decision Task Start to Close timeout # Decision Task Start to Close timeout
@ -59,11 +60,11 @@ def test_decision_task_start_to_close_timeout_boto3():
if settings.TEST_SERVER_MODE: if settings.TEST_SERVER_MODE:
raise SkipTest("Unable to manipulate time in ServerMode") raise SkipTest("Unable to manipulate time in ServerMode")
with freeze_time("2015-01-01 12:00:00"): with freeze_time("2015-01-01 12:00:00 UTC"):
client = setup_workflow_boto3() client = setup_workflow_boto3()
client.poll_for_decision_task(domain="test-domain", taskList={"name": "queue"}) client.poll_for_decision_task(domain="test-domain", taskList={"name": "queue"})
with freeze_time("2015-01-01 12:04:30"): with freeze_time("2015-01-01 12:04:30 UTC"):
resp = client.get_workflow_execution_history( resp = client.get_workflow_execution_history(
domain="test-domain", domain="test-domain",
execution={"runId": client.run_id, "workflowId": "uid-abcd1234"}, execution={"runId": client.run_id, "workflowId": "uid-abcd1234"},
@ -74,7 +75,7 @@ def test_decision_task_start_to_close_timeout_boto3():
["WorkflowExecutionStarted", "DecisionTaskScheduled", "DecisionTaskStarted"] ["WorkflowExecutionStarted", "DecisionTaskScheduled", "DecisionTaskStarted"]
) )
with freeze_time("2015-01-01 12:05:30"): with freeze_time("2015-01-01 12:05:30 UTC"):
# => Decision Task Start to Close timeout reached!! # => Decision Task Start to Close timeout reached!!
resp = client.get_workflow_execution_history( resp = client.get_workflow_execution_history(
domain="test-domain", domain="test-domain",
@ -101,8 +102,8 @@ def test_decision_task_start_to_close_timeout_boto3():
) )
# checks that event has been emitted at 12:05:00, not 12:05:30 # checks that event has been emitted at 12:05:00, not 12:05:30
resp["events"][-2]["eventTimestamp"].should.be.a(datetime) resp["events"][-2]["eventTimestamp"].should.be.a(datetime)
ts = resp["events"][-2]["eventTimestamp"].strftime("%Y-%m-%d %H:%M:%S") ts = resp["events"][-2]["eventTimestamp"]
ts.should.equal("2015-01-01 12:05:00") ts.should.equal(dtparse("2015-01-01 12:05:00 UTC"))
# Workflow Execution Start to Close timeout # Workflow Execution Start to Close timeout
@ -111,10 +112,10 @@ def test_decision_task_start_to_close_timeout_boto3():
def test_workflow_execution_start_to_close_timeout_boto3(): def test_workflow_execution_start_to_close_timeout_boto3():
if settings.TEST_SERVER_MODE: if settings.TEST_SERVER_MODE:
raise SkipTest("Unable to manipulate time in ServerMode") raise SkipTest("Unable to manipulate time in ServerMode")
with freeze_time("2015-01-01 12:00:00"): with freeze_time("2015-01-01 12:00:00 UTC"):
client = setup_workflow_boto3() client = setup_workflow_boto3()
with freeze_time("2015-01-01 13:59:30"): with freeze_time("2015-01-01 13:59:30 UTC"):
resp = client.get_workflow_execution_history( resp = client.get_workflow_execution_history(
domain="test-domain", domain="test-domain",
execution={"runId": client.run_id, "workflowId": "uid-abcd1234"}, execution={"runId": client.run_id, "workflowId": "uid-abcd1234"},
@ -123,7 +124,7 @@ def test_workflow_execution_start_to_close_timeout_boto3():
event_types = [evt["eventType"] for evt in resp["events"]] event_types = [evt["eventType"] for evt in resp["events"]]
event_types.should.equal(["WorkflowExecutionStarted", "DecisionTaskScheduled"]) event_types.should.equal(["WorkflowExecutionStarted", "DecisionTaskScheduled"])
with freeze_time("2015-01-01 14:00:30"): with freeze_time("2015-01-01 14:00:30 UTC"):
# => Workflow Execution Start to Close timeout reached!! # => Workflow Execution Start to Close timeout reached!!
resp = client.get_workflow_execution_history( resp = client.get_workflow_execution_history(
domain="test-domain", domain="test-domain",
@ -142,5 +143,5 @@ def test_workflow_execution_start_to_close_timeout_boto3():
attrs.should.equal({"childPolicy": "ABANDON", "timeoutType": "START_TO_CLOSE"}) attrs.should.equal({"childPolicy": "ABANDON", "timeoutType": "START_TO_CLOSE"})
# checks that event has been emitted at 14:00:00, not 14:00:30 # checks that event has been emitted at 14:00:00, not 14:00:30
resp["events"][-1]["eventTimestamp"].should.be.a(datetime) resp["events"][-1]["eventTimestamp"].should.be.a(datetime)
ts = resp["events"][-1]["eventTimestamp"].strftime("%Y-%m-%d %H:%M:%S") ts = resp["events"][-1]["eventTimestamp"]
ts.should.equal("2015-01-01 14:00:00") ts.should.equal(dtparse("2015-01-01 14:00:00 UTC"))