Merge pull request #4 from NarrativeScience/account-id-sts

QPT-28740 Source default aws account id value from sts

Author: jdrake

src/sfn_workflow_client/__init__.py

@@ -1,3 +1,3 @@
 """Enhanced, asyncio-compatible client for AWS Step Functions."""
 
-__version__ = "0.3.0"
+__version__ = "0.4.0"

src/sfn_workflow_client/config.py

@@ -1,7 +1,11 @@
 """Constants and configuration for the workflow client"""
 import os
 
-AWS_ACCOUNT_ID = os.environ.get("AWS_ACCOUNT_ID")
+import boto3
+
+AWS_ACCOUNT_ID = os.environ.get(
+    "AWS_ACCOUNT_ID", boto3.client("sts").get_caller_identity()["Account"]
+)
 
 # URL for making requests to the Step Functions API. You would most likely only set
 # this in order to hit the local py2sfn mock server.

tests/unit/test_workflow.py

@@ -3,6 +3,7 @@
 import os
 import unittest
 
+from sfn_workflow_client.config import AWS_ACCOUNT_ID
 from sfn_workflow_client.workflow import Workflow
 
 
@@ -15,7 +16,7 @@ def test_init(self):
         self.assertEqual(workflow.name, "test")
         self.assertEqual(
             workflow.state_machine_arn,
-            f"arn:aws:states:{os.environ.get('AWS_DEFAULT_REGION')}:{os.environ.get('AWS_ACCOUNT_ID')}:stateMachine:test",
+            f"arn:aws:states:{os.environ.get('AWS_DEFAULT_REGION')}:{AWS_ACCOUNT_ID}:stateMachine:test",
         )
         self.assertEqual(len(workflow.executions), 0)