Blame: python/example_code/lambda/test/test_lambda_basics.py - awsdocs/aws-doc-sdk-examples

awsdocs / aws-doc-sdk-examples UNCLAIMED

Welcome to the AWS Code Examples Repository. This repo contains code examples used in the AWS documentation, AWS SDK Developer Guides, and more. For more information, see the Readme.md file below.

0 0 31 Java

Normal View History Raw

Cleaned up Lambda basics and API Gateway REST API examples. Cleaned up scheduled Lambda example. Unit tests, comments, README. 2020-07-09 17:27:30 -07:00			`# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.`
			`# SPDX-License-Identifier: Apache-2.0`

			`"""`
			`Unit tests for lambda_basics.py functions.`
			`"""`

			`import json`
			`import unittest.mock`
			`import zipfile`
			`import boto3`
			`from botocore.exceptions import ClientError`
			`import pytest`

Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00			`from lambda_basics import LambdaWrapper`
Cleaned up Lambda basics and API Gateway REST API examples. Cleaned up scheduled Lambda example. Unit tests, comments, README. 2020-07-09 17:27:30 -07:00

			`def test_create_lambda_deployment_package(monkeypatch):`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`monkeypatch.setattr(zipfile.ZipFile, "write", lambda x, y, z: None)`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00			`wrapper = LambdaWrapper(None, None)`
Cleaned up Lambda basics and API Gateway REST API examples. Cleaned up scheduled Lambda example. Unit tests, comments, README. 2020-07-09 17:27:30 -07:00
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`got_package = wrapper.create_deployment_package("test-file", "other-file")`
Cleaned up Lambda basics and API Gateway REST API examples. Cleaned up scheduled Lambda example. Unit tests, comments, README. 2020-07-09 17:27:30 -07:00			`assert got_package is not None`


			`@pytest.mark.parametrize(`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`"error_code,stop_on_method",`
			`[`
Cleaned up Lambda basics and API Gateway REST API examples. Cleaned up scheduled Lambda example. Unit tests, comments, README. 2020-07-09 17:27:30 -07:00			`(None, None),`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`("TestException", "stub_create_role"),`
			`("TestException", "stub_attach_role_policy"),`
			`],`
			`)`
Cleaned up Lambda basics and API Gateway REST API examples. Cleaned up scheduled Lambda example. Unit tests, comments, README. 2020-07-09 17:27:30 -07:00			`def test_create_iam_role_for_lambda(`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`make_stubber, make_unique_name, stub_runner, error_code, stop_on_method`
			`):`
			`iam_resource = boto3.resource("iam")`
Cleaned up Lambda basics and API Gateway REST API examples. Cleaned up scheduled Lambda example. Unit tests, comments, README. 2020-07-09 17:27:30 -07:00			`iam_stubber = make_stubber(iam_resource.meta.client)`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00			`wrapper = LambdaWrapper(None, iam_resource)`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`role_name = make_unique_name("role-")`
Cleaned up Lambda basics and API Gateway REST API examples. Cleaned up scheduled Lambda example. Unit tests, comments, README. 2020-07-09 17:27:30 -07:00
Refactored all tests to use stub_runner, removed unused stub_controller code. 2020-09-15 18:20:16 -07:00			`with stub_runner(error_code, stop_on_method) as runner:`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`runner.add(iam_stubber.stub_get_role, role_name, error_code="NoSuchEntity")`
Refactored all tests to use stub_runner, removed unused stub_controller code. 2020-09-15 18:20:16 -07:00			`runner.add(iam_stubber.stub_create_role, role_name)`
			`runner.add(`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`iam_stubber.stub_attach_role_policy,`
			`role_name,`
			`"arn:aws:iam::aws:policy/service-role/AWSLambdaBasicExecutionRole",`
			`)`
Cleaned up Lambda basics and API Gateway REST API examples. Cleaned up scheduled Lambda example. Unit tests, comments, README. 2020-07-09 17:27:30 -07:00
			`if error_code is None:`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00			`got_role, got_created = wrapper.create_iam_role_for_lambda(role_name)`
Cleaned up Lambda basics and API Gateway REST API examples. Cleaned up scheduled Lambda example. Unit tests, comments, README. 2020-07-09 17:27:30 -07:00			`assert got_role.name == role_name`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00			`assert got_created`
Cleaned up Lambda basics and API Gateway REST API examples. Cleaned up scheduled Lambda example. Unit tests, comments, README. 2020-07-09 17:27:30 -07:00			`else:`
			`with pytest.raises(ClientError) as exc_info:`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00			`wrapper.create_iam_role_for_lambda(role_name)`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`assert exc_info.value.response["Error"]["Code"] == error_code`
Cleaned up Lambda basics and API Gateway REST API examples. Cleaned up scheduled Lambda example. Unit tests, comments, README. 2020-07-09 17:27:30 -07:00

Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`@pytest.mark.parametrize("error_code", [None, "TestException"])`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00			`def test_create_function(make_stubber, make_unique_name, error_code):`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`lambda_client = boto3.client("lambda")`
Cleaned up Lambda basics and API Gateway REST API examples. Cleaned up scheduled Lambda example. Unit tests, comments, README. 2020-07-09 17:27:30 -07:00			`lambda_stubber = make_stubber(lambda_client)`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00			`wrapper = LambdaWrapper(lambda_client, None)`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`func_name = make_unique_name("func-")`
			`handler_name = make_unique_name("handler-")`
			`iam_role = unittest.mock.MagicMock(arn="arn:aws:iam:::role/test-role")`
			`test_package = "test-package"`
			`func_arn = f"arn:aws:lambda:::function/{func_name}"`
Cleaned up Lambda basics and API Gateway REST API examples. Cleaned up scheduled Lambda example. Unit tests, comments, README. 2020-07-09 17:27:30 -07:00
			`lambda_stubber.stub_create_function(`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`func_name,`
			`func_arn,`
			`iam_role.arn,`
			`handler_name,`
			`test_package,`
			`error_code=error_code,`
			`)`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00			`if error_code is None:`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`lambda_stubber.stub_get_function(func_name, "Active")`
Cleaned up Lambda basics and API Gateway REST API examples. Cleaned up scheduled Lambda example. Unit tests, comments, README. 2020-07-09 17:27:30 -07:00
			`if error_code is None:`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`got_arn = wrapper.create_function(`
			`func_name, handler_name, iam_role, test_package`
			`)`
Cleaned up Lambda basics and API Gateway REST API examples. Cleaned up scheduled Lambda example. Unit tests, comments, README. 2020-07-09 17:27:30 -07:00			`assert got_arn == func_arn`
			`else:`
			`with pytest.raises(ClientError) as exc_info:`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`wrapper.create_function(func_name, handler_name, iam_role, test_package)`
			`assert exc_info.value.response["Error"]["Code"] == error_code`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00

Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`@pytest.mark.parametrize(`
			`"error_code", [None, "TestException", "ResourceNotFoundException"]`
			`)`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00			`def test_get_function(make_stubber, error_code):`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`lambda_client = boto3.client("lambda")`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00			`lambda_stubber = make_stubber(lambda_client)`
			`wrapper = LambdaWrapper(lambda_client, None)`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`func_name = "test-func_name"`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00
			`lambda_stubber.stub_get_function(func_name, error_code=error_code)`

Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`if error_code in (None, "ResourceNotFoundException"):`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00			`wrapper.get_function(func_name)`
			`else:`
			`with pytest.raises(ClientError) as exc_info:`
			`wrapper.get_function(func_name)`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`assert exc_info.value.response["Error"]["Code"] == error_code`
Cleaned up Lambda basics and API Gateway REST API examples. Cleaned up scheduled Lambda example. Unit tests, comments, README. 2020-07-09 17:27:30 -07:00

Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`@pytest.mark.parametrize("error_code", [None, "TestException"])`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00			`def test_delete_function(make_stubber, make_unique_name, error_code):`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`lambda_client = boto3.client("lambda")`
Cleaned up Lambda basics and API Gateway REST API examples. Cleaned up scheduled Lambda example. Unit tests, comments, README. 2020-07-09 17:27:30 -07:00			`lambda_stubber = make_stubber(lambda_client)`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00			`wrapper = LambdaWrapper(lambda_client, None)`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`func_name = make_unique_name("func-")`
Cleaned up Lambda basics and API Gateway REST API examples. Cleaned up scheduled Lambda example. Unit tests, comments, README. 2020-07-09 17:27:30 -07:00
			`lambda_stubber.stub_delete_function(func_name, error_code=error_code)`

			`if error_code is None:`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00			`wrapper.delete_function(func_name)`
Cleaned up Lambda basics and API Gateway REST API examples. Cleaned up scheduled Lambda example. Unit tests, comments, README. 2020-07-09 17:27:30 -07:00			`else:`
			`with pytest.raises(ClientError) as exc_info:`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00			`wrapper.delete_function(func_name)`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`assert exc_info.value.response["Error"]["Code"] == error_code`
Cleaned up Lambda basics and API Gateway REST API examples. Cleaned up scheduled Lambda example. Unit tests, comments, README. 2020-07-09 17:27:30 -07:00

Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`@pytest.mark.parametrize("error_code", [None, "TestException"])`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00			`def test_invoke_function(make_stubber, make_unique_name, error_code):`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`lambda_client = boto3.client("lambda")`
Cleaned up Lambda basics and API Gateway REST API examples. Cleaned up scheduled Lambda example. Unit tests, comments, README. 2020-07-09 17:27:30 -07:00			`lambda_stubber = make_stubber(lambda_client)`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00			`wrapper = LambdaWrapper(lambda_client, None)`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`func_name = make_unique_name("func-")`
			`func_params = {"param1": "test", "param2": 35}`
			`response_payload = "ahoy there"`
Cleaned up Lambda basics and API Gateway REST API examples. Cleaned up scheduled Lambda example. Unit tests, comments, README. 2020-07-09 17:27:30 -07:00
			`lambda_stubber.stub_invoke(`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`func_name,`
			`json.dumps(func_params),`
			`response_payload,`
			`log_type="None",`
			`error_code=error_code,`
			`)`
Cleaned up Lambda basics and API Gateway REST API examples. Cleaned up scheduled Lambda example. Unit tests, comments, README. 2020-07-09 17:27:30 -07:00
			`if error_code is None:`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00			`response = wrapper.invoke_function(func_name, func_params)`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`assert response["Payload"] == response_payload`
Cleaned up Lambda basics and API Gateway REST API examples. Cleaned up scheduled Lambda example. Unit tests, comments, README. 2020-07-09 17:27:30 -07:00			`else:`
			`with pytest.raises(ClientError) as exc_info:`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00			`wrapper.invoke_function(func_name, func_params)`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`assert exc_info.value.response["Error"]["Code"] == error_code`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00

Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`@pytest.mark.parametrize("error_code", [None, "TestException"])`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00			`def test_update_function_code(make_stubber, error_code):`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`lambda_client = boto3.client("lambda")`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00			`lambda_stubber = make_stubber(lambda_client)`
			`wrapper = LambdaWrapper(lambda_client, None)`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`func_name = "test-func_name"`
			`package = "test-package"`
			`update_status = "InProgress"`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00
			`lambda_stubber.stub_update_function_code(`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`func_name, update_status, package=package, error_code=error_code`
			`)`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00
			`if error_code is None:`
			`got_response = wrapper.update_function_code(func_name, package)`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`assert got_response["LastUpdateStatus"] == update_status`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00			`else:`
			`with pytest.raises(ClientError) as exc_info:`
			`wrapper.update_function_code(func_name, package)`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`assert exc_info.value.response["Error"]["Code"] == error_code`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00

Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`@pytest.mark.parametrize("error_code", [None, "TestException"])`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00			`def test_update_function_configuration(make_stubber, error_code):`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`lambda_client = boto3.client("lambda")`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00			`lambda_stubber = make_stubber(lambda_client)`
			`wrapper = LambdaWrapper(lambda_client, None)`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`func_name = "test-func_name"`
			`env_vars = {"test-key": "test-val"}`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00
			`lambda_stubber.stub_update_function_configuration(`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`func_name, env_vars, error_code=error_code`
			`)`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00
			`if error_code is None:`
			`got_response = wrapper.update_function_configuration(func_name, env_vars)`
			`assert got_response`
			`else:`
			`with pytest.raises(ClientError) as exc_info:`
			`wrapper.update_function_configuration(func_name, env_vars)`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`assert exc_info.value.response["Error"]["Code"] == error_code`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00

Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`@pytest.mark.parametrize("error_code", [None, "TestException"])`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00			`def test_list_functions(make_stubber, error_code):`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`lambda_client = boto3.client("lambda")`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00			`lambda_stubber = make_stubber(lambda_client)`
			`wrapper = LambdaWrapper(lambda_client, None)`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`funcs = [`
			`{`
			`"FunctionName": f"test-func-{index}",`
			`"Description": f"test description {index}",`
			`"Runtime": f"test-runtime-{index}",`
			`"Handler": f"test-handler-{index}",`
			`}`
			`for index in range(3)`
			`]`
Python Lambda MVP: scenario, actions...and more! Typo. Lint error and minor edits. 2022-04-08 15:42:13 -07:00
			`lambda_stubber.stub_list_functions(funcs, error_code=error_code)`

			`if error_code is None:`
			`wrapper.list_functions()`
			`else:`
			`with pytest.raises(ClientError) as exc_info:`
			`wrapper.list_functions()`
Python: Format with black and switch superlint (#5513) 2023-10-18 10:35:05 -07:00			`assert exc_info.value.response["Error"]["Code"] == error_code`