33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
from typing import Any, List, Optional, Union
|
|
|
|
from moto.awslambda.models import LambdaBackend
|
|
from moto.core.base_backend import BackendDict
|
|
|
|
|
|
class LambdaSimpleBackend(LambdaBackend):
|
|
"""
|
|
Implements a Lambda-Backend that does not use Docker containers, will always succeed.
|
|
Annotate your tests with `@mock_aws(config={"lambda": {"use_docker": False}}) to use this Lambda-implementation.
|
|
"""
|
|
|
|
def __init__(self, region_name: str, account_id: str):
|
|
super().__init__(region_name, account_id)
|
|
self.lambda_simple_results_queue: List[str] = []
|
|
|
|
# pylint: disable=unused-argument
|
|
def invoke(
|
|
self,
|
|
function_name: str,
|
|
qualifier: Optional[str],
|
|
body: Any,
|
|
headers: Any,
|
|
response_headers: Any,
|
|
) -> Optional[Union[str, bytes]]:
|
|
default_result = body or "Simple Lambda happy path OK"
|
|
if self.lambda_simple_results_queue:
|
|
default_result = self.lambda_simple_results_queue.pop(0)
|
|
return str.encode(default_result)
|
|
|
|
|
|
lambda_simple_backends = BackendDict(LambdaSimpleBackend, "lambda")
|