Run black on moto & test directories.
This commit is contained in:
Asher Foa
parent
c820395dbf
commit
96e5b1993d
507 changed files with 52541 additions and 47814 deletions
|
|
@ -3,6 +3,7 @@ from boto.swf.exceptions import SWFResponseError
|
|||
from datetime import datetime, timedelta
|
||||
|
||||
import sure # noqa
|
||||
|
||||
# Ensure 'assert_raises' context manager support for Python 2.6
|
||||
import tests.backport_assert_raises # noqa
|
||||
|
||||
|
|
@ -16,8 +17,11 @@ def setup_swf_environment():
|
|||
conn = boto.connect_swf("the_key", "the_secret")
|
||||
conn.register_domain("test-domain", "60", description="A test domain")
|
||||
conn.register_workflow_type(
|
||||
"test-domain", "test-workflow", "v1.0",
|
||||
task_list="queue", default_child_policy="TERMINATE",
|
||||
"test-domain",
|
||||
"test-workflow",
|
||||
"v1.0",
|
||||
task_list="queue",
|
||||
default_child_policy="TERMINATE",
|
||||
default_execution_start_to_close_timeout="300",
|
||||
default_task_start_to_close_timeout="300",
|
||||
)
|
||||
|
|
@ -31,29 +35,34 @@ def test_start_workflow_execution():
|
|||
conn = setup_swf_environment()
|
||||
|
||||
wf = conn.start_workflow_execution(
|
||||
"test-domain", "uid-abcd1234", "test-workflow", "v1.0")
|
||||
"test-domain", "uid-abcd1234", "test-workflow", "v1.0"
|
||||
)
|
||||
wf.should.contain("runId")
|
||||
|
||||
|
||||
@mock_swf_deprecated
|
||||
def test_signal_workflow_execution():
|
||||
conn = setup_swf_environment()
|
||||
hsh = conn.start_workflow_execution(
|
||||
"test-domain", "uid-abcd1234", "test-workflow", "v1.0")
|
||||
"test-domain", "uid-abcd1234", "test-workflow", "v1.0"
|
||||
)
|
||||
run_id = hsh["runId"]
|
||||
|
||||
wfe = conn.signal_workflow_execution(
|
||||
"test-domain", "my_signal", "uid-abcd1234", "my_input", run_id)
|
||||
"test-domain", "my_signal", "uid-abcd1234", "my_input", run_id
|
||||
)
|
||||
|
||||
wfe = conn.describe_workflow_execution(
|
||||
"test-domain", run_id, "uid-abcd1234")
|
||||
wfe = conn.describe_workflow_execution("test-domain", run_id, "uid-abcd1234")
|
||||
|
||||
wfe["openCounts"]["openDecisionTasks"].should.equal(2)
|
||||
|
||||
|
||||
@mock_swf_deprecated
|
||||
def test_start_already_started_workflow_execution():
|
||||
conn = setup_swf_environment()
|
||||
conn.start_workflow_execution(
|
||||
"test-domain", "uid-abcd1234", "test-workflow", "v1.0")
|
||||
"test-domain", "uid-abcd1234", "test-workflow", "v1.0"
|
||||
)
|
||||
|
||||
conn.start_workflow_execution.when.called_with(
|
||||
"test-domain", "uid-abcd1234", "test-workflow", "v1.0"
|
||||
|
|
@ -75,13 +84,12 @@ def test_start_workflow_execution_on_deprecated_type():
|
|||
def test_describe_workflow_execution():
|
||||
conn = setup_swf_environment()
|
||||
hsh = conn.start_workflow_execution(
|
||||
"test-domain", "uid-abcd1234", "test-workflow", "v1.0")
|
||||
"test-domain", "uid-abcd1234", "test-workflow", "v1.0"
|
||||
)
|
||||
run_id = hsh["runId"]
|
||||
|
||||
wfe = conn.describe_workflow_execution(
|
||||
"test-domain", run_id, "uid-abcd1234")
|
||||
wfe["executionInfo"]["execution"][
|
||||
"workflowId"].should.equal("uid-abcd1234")
|
||||
wfe = conn.describe_workflow_execution("test-domain", run_id, "uid-abcd1234")
|
||||
wfe["executionInfo"]["execution"]["workflowId"].should.equal("uid-abcd1234")
|
||||
wfe["executionInfo"]["executionStatus"].should.equal("OPEN")
|
||||
|
||||
|
||||
|
|
@ -99,11 +107,11 @@ def test_describe_non_existent_workflow_execution():
|
|||
def test_get_workflow_execution_history():
|
||||
conn = setup_swf_environment()
|
||||
hsh = conn.start_workflow_execution(
|
||||
"test-domain", "uid-abcd1234", "test-workflow", "v1.0")
|
||||
"test-domain", "uid-abcd1234", "test-workflow", "v1.0"
|
||||
)
|
||||
run_id = hsh["runId"]
|
||||
|
||||
resp = conn.get_workflow_execution_history(
|
||||
"test-domain", run_id, "uid-abcd1234")
|
||||
resp = conn.get_workflow_execution_history("test-domain", run_id, "uid-abcd1234")
|
||||
types = [evt["eventType"] for evt in resp["events"]]
|
||||
types.should.equal(["WorkflowExecutionStarted", "DecisionTaskScheduled"])
|
||||
|
||||
|
|
@ -112,11 +120,13 @@ def test_get_workflow_execution_history():
|
|||
def test_get_workflow_execution_history_with_reverse_order():
|
||||
conn = setup_swf_environment()
|
||||
hsh = conn.start_workflow_execution(
|
||||
"test-domain", "uid-abcd1234", "test-workflow", "v1.0")
|
||||
"test-domain", "uid-abcd1234", "test-workflow", "v1.0"
|
||||
)
|
||||
run_id = hsh["runId"]
|
||||
|
||||
resp = conn.get_workflow_execution_history("test-domain", run_id, "uid-abcd1234",
|
||||
reverse_order=True)
|
||||
resp = conn.get_workflow_execution_history(
|
||||
"test-domain", run_id, "uid-abcd1234", reverse_order=True
|
||||
)
|
||||
types = [evt["eventType"] for evt in resp["events"]]
|
||||
types.should.equal(["DecisionTaskScheduled", "WorkflowExecutionStarted"])
|
||||
|
||||
|
|
@ -136,32 +146,36 @@ def test_list_open_workflow_executions():
|
|||
conn = setup_swf_environment()
|
||||
# One open workflow execution
|
||||
conn.start_workflow_execution(
|
||||
'test-domain', 'uid-abcd1234', 'test-workflow', 'v1.0'
|
||||
"test-domain", "uid-abcd1234", "test-workflow", "v1.0"
|
||||
)
|
||||
# One closed workflow execution to make sure it isn't displayed
|
||||
run_id = conn.start_workflow_execution(
|
||||
'test-domain', 'uid-abcd12345', 'test-workflow', 'v1.0'
|
||||
)['runId']
|
||||
conn.terminate_workflow_execution('test-domain', 'uid-abcd12345',
|
||||
details='some details',
|
||||
reason='a more complete reason',
|
||||
run_id=run_id)
|
||||
"test-domain", "uid-abcd12345", "test-workflow", "v1.0"
|
||||
)["runId"]
|
||||
conn.terminate_workflow_execution(
|
||||
"test-domain",
|
||||
"uid-abcd12345",
|
||||
details="some details",
|
||||
reason="a more complete reason",
|
||||
run_id=run_id,
|
||||
)
|
||||
|
||||
yesterday = datetime.utcnow() - timedelta(days=1)
|
||||
oldest_date = unix_time(yesterday)
|
||||
response = conn.list_open_workflow_executions('test-domain',
|
||||
oldest_date,
|
||||
workflow_id='test-workflow')
|
||||
execution_infos = response['executionInfos']
|
||||
response = conn.list_open_workflow_executions(
|
||||
"test-domain", oldest_date, workflow_id="test-workflow"
|
||||
)
|
||||
execution_infos = response["executionInfos"]
|
||||
len(execution_infos).should.equal(1)
|
||||
open_workflow = execution_infos[0]
|
||||
open_workflow['workflowType'].should.equal({'version': 'v1.0',
|
||||
'name': 'test-workflow'})
|
||||
open_workflow.should.contain('startTimestamp')
|
||||
open_workflow['execution']['workflowId'].should.equal('uid-abcd1234')
|
||||
open_workflow['execution'].should.contain('runId')
|
||||
open_workflow['cancelRequested'].should.be(False)
|
||||
open_workflow['executionStatus'].should.equal('OPEN')
|
||||
open_workflow["workflowType"].should.equal(
|
||||
{"version": "v1.0", "name": "test-workflow"}
|
||||
)
|
||||
open_workflow.should.contain("startTimestamp")
|
||||
open_workflow["execution"]["workflowId"].should.equal("uid-abcd1234")
|
||||
open_workflow["execution"].should.contain("runId")
|
||||
open_workflow["cancelRequested"].should.be(False)
|
||||
open_workflow["executionStatus"].should.equal("OPEN")
|
||||
|
||||
|
||||
# ListClosedWorkflowExecutions endpoint
|
||||
|
|
@ -170,33 +184,36 @@ def test_list_closed_workflow_executions():
|
|||
conn = setup_swf_environment()
|
||||
# Leave one workflow execution open to make sure it isn't displayed
|
||||
conn.start_workflow_execution(
|
||||
'test-domain', 'uid-abcd1234', 'test-workflow', 'v1.0'
|
||||
"test-domain", "uid-abcd1234", "test-workflow", "v1.0"
|
||||
)
|
||||
# One closed workflow execution
|
||||
run_id = conn.start_workflow_execution(
|
||||
'test-domain', 'uid-abcd12345', 'test-workflow', 'v1.0'
|
||||
)['runId']
|
||||
conn.terminate_workflow_execution('test-domain', 'uid-abcd12345',
|
||||
details='some details',
|
||||
reason='a more complete reason',
|
||||
run_id=run_id)
|
||||
"test-domain", "uid-abcd12345", "test-workflow", "v1.0"
|
||||
)["runId"]
|
||||
conn.terminate_workflow_execution(
|
||||
"test-domain",
|
||||
"uid-abcd12345",
|
||||
details="some details",
|
||||
reason="a more complete reason",
|
||||
run_id=run_id,
|
||||
)
|
||||
|
||||
yesterday = datetime.utcnow() - timedelta(days=1)
|
||||
oldest_date = unix_time(yesterday)
|
||||
response = conn.list_closed_workflow_executions(
|
||||
'test-domain',
|
||||
start_oldest_date=oldest_date,
|
||||
workflow_id='test-workflow')
|
||||
execution_infos = response['executionInfos']
|
||||
"test-domain", start_oldest_date=oldest_date, workflow_id="test-workflow"
|
||||
)
|
||||
execution_infos = response["executionInfos"]
|
||||
len(execution_infos).should.equal(1)
|
||||
open_workflow = execution_infos[0]
|
||||
open_workflow['workflowType'].should.equal({'version': 'v1.0',
|
||||
'name': 'test-workflow'})
|
||||
open_workflow.should.contain('startTimestamp')
|
||||
open_workflow['execution']['workflowId'].should.equal('uid-abcd12345')
|
||||
open_workflow['execution'].should.contain('runId')
|
||||
open_workflow['cancelRequested'].should.be(False)
|
||||
open_workflow['executionStatus'].should.equal('CLOSED')
|
||||
open_workflow["workflowType"].should.equal(
|
||||
{"version": "v1.0", "name": "test-workflow"}
|
||||
)
|
||||
open_workflow.should.contain("startTimestamp")
|
||||
open_workflow["execution"]["workflowId"].should.equal("uid-abcd12345")
|
||||
open_workflow["execution"].should.contain("runId")
|
||||
open_workflow["cancelRequested"].should.be(False)
|
||||
open_workflow["executionStatus"].should.equal("CLOSED")
|
||||
|
||||
|
||||
# TerminateWorkflowExecution endpoint
|
||||
|
|
@ -207,14 +224,16 @@ def test_terminate_workflow_execution():
|
|||
"test-domain", "uid-abcd1234", "test-workflow", "v1.0"
|
||||
)["runId"]
|
||||
|
||||
resp = conn.terminate_workflow_execution("test-domain", "uid-abcd1234",
|
||||
details="some details",
|
||||
reason="a more complete reason",
|
||||
run_id=run_id)
|
||||
resp = conn.terminate_workflow_execution(
|
||||
"test-domain",
|
||||
"uid-abcd1234",
|
||||
details="some details",
|
||||
reason="a more complete reason",
|
||||
run_id=run_id,
|
||||
)
|
||||
resp.should.be.none
|
||||
|
||||
resp = conn.get_workflow_execution_history(
|
||||
"test-domain", run_id, "uid-abcd1234")
|
||||
resp = conn.get_workflow_execution_history("test-domain", run_id, "uid-abcd1234")
|
||||
evt = resp["events"][-1]
|
||||
evt["eventType"].should.equal("WorkflowExecutionTerminated")
|
||||
attrs = evt["workflowExecutionTerminatedEventAttributes"]
|
||||
|
|
@ -243,16 +262,12 @@ def test_terminate_workflow_execution_with_wrong_workflow_or_run_id():
|
|||
# already closed, without run_id
|
||||
conn.terminate_workflow_execution.when.called_with(
|
||||
"test-domain", "uid-abcd1234"
|
||||
).should.throw(
|
||||
SWFResponseError, "Unknown execution, workflowId = uid-abcd1234"
|
||||
)
|
||||
).should.throw(SWFResponseError, "Unknown execution, workflowId = uid-abcd1234")
|
||||
|
||||
# wrong workflow id
|
||||
conn.terminate_workflow_execution.when.called_with(
|
||||
"test-domain", "uid-non-existent"
|
||||
).should.throw(
|
||||
SWFResponseError, "Unknown execution, workflowId = uid-non-existent"
|
||||
)
|
||||
).should.throw(SWFResponseError, "Unknown execution, workflowId = uid-non-existent")
|
||||
|
||||
# wrong run_id
|
||||
conn.terminate_workflow_execution.when.called_with(
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue