Use correct python X-Ray env

mhart · mhart · commit 6c0739be2ae2 · 2018-01-23T09:32:30.000-05:00
diff --git a/python2.7/run/runtime-mock.py b/python2.7/run/runtime-mock.py
@@ -49,10 +49,10 @@ def _arn(region, account_id, fct_name):
     'session': _GLOBAL_SESSION_TOKEN
 }
 _GLOBAL_INVOKED_FUNCTION_ARN = _arn(_GLOBAL_REGION, _GLOBAL_ACCOUNT_ID, _GLOBAL_FCT_NAME)
-_GLOBAL_XRAY_TRACE_ID = None
+_GLOBAL_XRAY_TRACE_ID = os.environ.get('_X_AMZN_TRACE_ID', None)
 _GLOBAL_XRAY_PARENT_ID = None
 _GLOBAL_XRAY_SAMPLED = None
-_GLOBAL_X_AMZN_TRACE_ID = os.environ.get('_X_AMZN_TRACE_ID', None)
+_GLOBAL_X_AMZN_TRACE_ID
 _GLOBAL_INVOKED = False
 _GLOBAL_ERRORED = False
 _GLOBAL_START_TIME = None
diff --git a/python3.6/run/runtime-mock.py b/python3.6/run/runtime-mock.py
@@ -49,10 +49,10 @@ def _arn(region, account_id, fct_name):
     'session': _GLOBAL_SESSION_TOKEN
 }
 _GLOBAL_INVOKED_FUNCTION_ARN = _arn(_GLOBAL_REGION, _GLOBAL_ACCOUNT_ID, _GLOBAL_FCT_NAME)
-_GLOBAL_XRAY_TRACE_ID = None
+_GLOBAL_XRAY_TRACE_ID = os.environ.get('_X_AMZN_TRACE_ID', None)
 _GLOBAL_XRAY_PARENT_ID = None
 _GLOBAL_XRAY_SAMPLED = None
-_GLOBAL_X_AMZN_TRACE_ID = os.environ.get('_X_AMZN_TRACE_ID', None)
+_GLOBAL_X_AMZN_TRACE_ID
 _GLOBAL_INVOKED = False
 _GLOBAL_ERRORED = False
 _GLOBAL_START_TIME = None
