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

Step Functions - Remove STS-client and refer to hardcoded account-id

Browse Source
This commit is contained in:
Bert Blommers 2019-09-24 14:36:34 +01:00
parent 6a1a8df7cc
commit 38455c8e19
1 changed files with 2 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
moto/stepfunctions/models.py
View File

@ -1,9 +1,9 @@
import boto import boto
import boto3
import re import re
from datetime import datetime from datetime import datetime
from moto.core import BaseBackend from moto.core import BaseBackend
from moto.core.utils import iso_8601_datetime_without_milliseconds from moto.core.utils import iso_8601_datetime_without_milliseconds
from moto.sts.models import ACCOUNT_ID
from uuid import uuid4 from uuid import uuid4
from .exceptions import ExecutionDoesNotExist, InvalidArn, InvalidName, StateMachineDoesNotExist from .exceptions import ExecutionDoesNotExist, InvalidArn, InvalidName, StateMachineDoesNotExist
@ -156,12 +156,7 @@ class StepFunctionBackend(BaseBackend):
raise InvalidArn(invalid_msg) raise InvalidArn(invalid_msg)
def _get_account_id(self): def _get_account_id(self):
if self._account_id: return ACCOUNT_ID
return self._account_id
sts = boto3.client("sts")
identity = sts.get_caller_identity()
self._account_id = identity['Account']
return self._account_id
stepfunction_backends = {_region.name: StepFunctionBackend(_region.name) for _region in boto.awslambda.regions()} stepfunction_backends = {_region.name: StepFunctionBackend(_region.name) for _region in boto.awslambda.regions()}