catalin/moto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

stepfunctions: start_execution accepts name arg

Browse source
This commit is contained in:
James Thorniley 2019-10-03 14:47:24 +01:00
commit 31310760ea
3 changed files with 20 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
moto/stepfunctions/models.py
View file

@ -96,12 +96,12 @@ class StepFunctionBackend(BaseBackend):
if sm:
self.state_machines.remove(sm)
def start_execution(self, state_machine_arn):
def start_execution(self, state_machine_arn, name=None):
state_machine_name = self.describe_state_machine(state_machine_arn).name
execution = Execution(region_name=self.region_name,
account_id=self._get_account_id(),
state_machine_name=state_machine_name,
execution_name=str(uuid4()),
execution_name=name or str(uuid4()),
state_machine_arn=state_machine_arn)
self.executions.append(execution)
return execution

3
moto/stepfunctions/responses.py
View file

@ -86,7 +86,8 @@ class StepFunctionResponse(BaseResponse):
@amzn_request_id
def start_execution(self):
arn = self._get_param('stateMachineArn')
execution = self.stepfunction_backend.start_execution(arn)
name = self._get_param('name')
execution = self.stepfunction_backend.start_execution(arn, name)
response = {'executionArn': execution.execution_arn,
'startDate': execution.start_date}
return 200, {}, json.dumps(response)