def test_must_work_with_partial_aws_creds(self):
creds = {
"region": "some other region",
"sessiontoken": "my awesome token"
}
expected = {
"AWS_SAM_LOCAL": "true",
"AWS_LAMBDA_FUNCTION_MEMORY_SIZE": "1024",
"AWS_LAMBDA_FUNCTION_TIMEOUT": "123",
"AWS_LAMBDA_FUNCTION_HANDLER": "handler",
# Values from the input creds
"AWS_REGION": "some other region",
"AWS_DEFAULT_REGION": "some other region",
"AWS_SESSION_TOKEN": "my awesome token",
# These variables still get the default value
"AWS_ACCESS_KEY_ID": "defaultkey",
"AWS_SECRET_ACCESS_KEY": "defaultsecret",
}
environ = EnvironmentVariables(self.memory,
self.timeout,
self.handler,
aws_creds=creds)
self.assertEqual(expected, environ._get_aws_variables())
def test_must_work_without_any_aws_creds_extensions_preview(
self, PreviewEnabledMock):
PreviewEnabledMock.return_value = True
expected = {
"AWS_SAM_LOCAL": "true",
"AWS_LAMBDA_FUNCTION_MEMORY_SIZE": "1024",
"AWS_LAMBDA_FUNCTION_TIMEOUT": "123",
"AWS_LAMBDA_FUNCTION_HANDLER": "handler",
# Default values assigned to these variables
"AWS_REGION": "us-east-1",
"AWS_DEFAULT_REGION": "us-east-1",
"AWS_ACCESS_KEY_ID": "defaultkey",
"AWS_SECRET_ACCESS_KEY": "defaultsecret",
"AWS_LAMBDA_FUNCTION_NAME": "name",
"AWS_LAMBDA_FUNCTION_VERSION": "$LATEST",
"AWS_LAMBDA_LOG_GROUP_NAME": f"aws/lambda/{self.name}",
"AWS_LAMBDA_LOG_STREAM_NAME": "$LATEST",
"AWS_ACCOUNT_ID": "123456789012",
}
environ = EnvironmentVariables(self.name, self.memory, self.timeout,
self.handler)
self.assertEqual(expected, environ._get_aws_variables())
def test_must_work_with_partial_aws_creds(self):
creds = {
"region": "some other region",
"sessiontoken": "my awesome token"
}
expected = {
"AWS_SAM_LOCAL": "true",
"AWS_LAMBDA_FUNCTION_MEMORY_SIZE": "1024",
"AWS_LAMBDA_FUNCTION_TIMEOUT": "123",
"AWS_LAMBDA_FUNCTION_HANDLER": "handler",
# Values from the input creds
"AWS_REGION": "some other region",
"AWS_DEFAULT_REGION": "some other region",
"AWS_SESSION_TOKEN": "my awesome token",
# These variables still get the default value
"AWS_ACCESS_KEY_ID": "defaultkey",
"AWS_SECRET_ACCESS_KEY": "defaultsecret",
}
environ = EnvironmentVariables(self.memory, self.timeout, self.handler, aws_creds=creds)
self.assertEquals(expected, environ._get_aws_variables())
def test_must_work_with_overridden_aws_creds(self):
expected = {
"AWS_SAM_LOCAL": "true",
"AWS_LAMBDA_FUNCTION_MEMORY_SIZE": "1024",
"AWS_LAMBDA_FUNCTION_TIMEOUT": "123",
"AWS_LAMBDA_FUNCTION_HANDLER": "handler",
"AWS_LAMBDA_FUNCTION_NAME": self.name,
"AWS_LAMBDA_FUNCTION_VERSION": "$LATEST",
"AWS_REGION": "some region",
"AWS_DEFAULT_REGION": "some region",
"AWS_ACCESS_KEY_ID": "some key",
"AWS_SECRET_ACCESS_KEY": "some other secret",
"AWS_LAMBDA_LOG_GROUP_NAME": f"aws/lambda/{self.name}",
"AWS_LAMBDA_LOG_STREAM_NAME": "$LATEST",
"AWS_ACCOUNT_ID": "123456789012",
"AWS_SESSION_TOKEN": "some other token",
}
environ = EnvironmentVariables(self.name,
self.memory,
self.timeout,
self.handler,
aws_creds=self.aws_creds)
self.assertEqual(expected, environ._get_aws_variables())
def test_must_work_without_any_aws_creds(self):
expected = {
"AWS_SAM_LOCAL": "true",
"AWS_LAMBDA_FUNCTION_MEMORY_SIZE": "1024",
"AWS_LAMBDA_FUNCTION_TIMEOUT": "123",
"AWS_LAMBDA_FUNCTION_HANDLER": "handler",
# Default values assigned to these variables
"AWS_REGION": "us-east-1",
"AWS_DEFAULT_REGION": "us-east-1",
"AWS_ACCESS_KEY_ID": "defaultkey",
"AWS_SECRET_ACCESS_KEY": "defaultsecret",
}
environ = EnvironmentVariables(self.memory, self.timeout, self.handler)
self.assertEqual(expected, environ._get_aws_variables())
def test_must_work_without_any_aws_creds(self):
expected = {
"AWS_SAM_LOCAL": "true",
"AWS_LAMBDA_FUNCTION_MEMORY_SIZE": "1024",
"AWS_LAMBDA_FUNCTION_TIMEOUT": "123",
"AWS_LAMBDA_FUNCTION_HANDLER": "handler",
# Default values assigned to these variables
"AWS_REGION": "us-east-1",
"AWS_DEFAULT_REGION": "us-east-1",
"AWS_ACCESS_KEY_ID": "defaultkey",
"AWS_SECRET_ACCESS_KEY": "defaultsecret",
}
environ = EnvironmentVariables(self.memory, self.timeout, self.handler)
self.assertEquals(expected, environ._get_aws_variables())
def test_must_work_with_overridden_aws_creds(self):
expected = {
"AWS_SAM_LOCAL": "true",
"AWS_LAMBDA_FUNCTION_MEMORY_SIZE": "1024",
"AWS_LAMBDA_FUNCTION_TIMEOUT": "123",
"AWS_LAMBDA_FUNCTION_HANDLER": "handler",
"AWS_REGION": "some region",
"AWS_DEFAULT_REGION": "some region",
"AWS_ACCESS_KEY_ID": "some key",
"AWS_SECRET_ACCESS_KEY": "some other secret",
"AWS_SESSION_TOKEN": "some other token"
}
environ = EnvironmentVariables(self.memory, self.timeout, self.handler, aws_creds=self.aws_creds)
self.assertEquals(expected, environ._get_aws_variables())
def test_must_work_with_overridden_aws_creds(self):
expected = {
"AWS_SAM_LOCAL": "true",
"AWS_LAMBDA_FUNCTION_MEMORY_SIZE": "1024",
"AWS_LAMBDA_FUNCTION_TIMEOUT": "123",
"AWS_LAMBDA_FUNCTION_HANDLER": "handler",
"AWS_REGION": "some region",
"AWS_DEFAULT_REGION": "some region",
"AWS_ACCESS_KEY_ID": "some key",
"AWS_SECRET_ACCESS_KEY": "some other secret",
"AWS_SESSION_TOKEN": "some other token",
}
environ = EnvironmentVariables(self.memory,
self.timeout,
self.handler,
aws_creds=self.aws_creds)
self.assertEqual(expected, environ._get_aws_variables())