From c917ac6ecb76fc319869a4e1c575877d25c54344 Mon Sep 17 00:00:00 2001
From: "Kamil Mankowski (kam193)" <kam193@wp.pl>
Date: Wed, 19 Aug 2020 23:18:32 +0200
Subject: [PATCH] Set more lambda runtime variables

---
 moto/awslambda/models.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/moto/awslambda/models.py b/moto/awslambda/models.py
index a234fbe01..2c0d4d9e2 100644
--- a/moto/awslambda/models.py
+++ b/moto/awslambda/models.py
@@ -388,11 +388,16 @@ class LambdaFunction(CloudFormationModel):
             #       also need to hook it up to the other services so it can make kws/s3 etc calls
             #  Should get invoke_id /RequestId from invocation
             env_vars = {
+                "_HANDLER": self.handler,
+                "AWS_EXECUTION_ENV": "AWS_Lambda_{}".format(self.run_time),
                 "AWS_LAMBDA_FUNCTION_TIMEOUT": self.timeout,
                 "AWS_LAMBDA_FUNCTION_NAME": self.function_name,
                 "AWS_LAMBDA_FUNCTION_MEMORY_SIZE": self.memory_size,
                 "AWS_LAMBDA_FUNCTION_VERSION": self.version,
                 "AWS_REGION": self.region,
+                "AWS_ACCESS_KEY_ID": "role-account-id",
+                "AWS_SECRET_ACCESS_KEY": "role-secret-key",
+                "AWS_SESSION_TOKEN": "session-token",
             }
 
             env_vars.update(self.environment_vars)