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())