65 lines
1.6 KiB
Python
65 lines
1.6 KiB
Python
import json
|
|
from unittest import SkipTest
|
|
|
|
import boto3
|
|
|
|
from moto import mock_aws, settings
|
|
|
|
from ..test_awslambda.utilities import get_role_name, get_test_zip_file1
|
|
|
|
LAMBDA_REGION = "us-west-2"
|
|
PYTHON_VERSION = "3.11"
|
|
FUNCTION_NAME = "test-function-123"
|
|
|
|
if settings.TEST_SERVER_MODE:
|
|
raise SkipTest("No point in testing batch_simple in ServerMode")
|
|
|
|
|
|
@mock_aws(config={"lambda": {"use_docker": False}})
|
|
def test_run_function():
|
|
# Setup
|
|
client = setup_lambda()
|
|
|
|
# Execute
|
|
result = client.invoke(
|
|
FunctionName=FUNCTION_NAME,
|
|
LogType="Tail",
|
|
)
|
|
|
|
# Verify
|
|
assert result["StatusCode"] == 200
|
|
assert result["Payload"].read().decode("utf-8") == "Simple Lambda happy path OK"
|
|
|
|
|
|
@mock_aws(config={"lambda": {"use_docker": False}})
|
|
def test_run_function_no_log():
|
|
# Setup
|
|
client = setup_lambda()
|
|
payload = {"results": "results"}
|
|
|
|
# Execute
|
|
result = client.invoke(FunctionName=FUNCTION_NAME, Payload=json.dumps(payload))
|
|
|
|
# Verify
|
|
assert result["StatusCode"] == 200
|
|
assert json.loads(result["Payload"].read().decode("utf-8")) == payload
|
|
|
|
|
|
def setup_lambda():
|
|
client = boto3.client("lambda", LAMBDA_REGION)
|
|
zip_content = get_test_zip_file1()
|
|
function_name = FUNCTION_NAME
|
|
role = get_role_name()
|
|
client.create_function(
|
|
FunctionName=function_name,
|
|
Runtime=PYTHON_VERSION,
|
|
Role=role,
|
|
Handler="lambda_function.lambda_handler",
|
|
Code={"ZipFile": zip_content},
|
|
Description="test lambda function",
|
|
Timeout=3,
|
|
MemorySize=128,
|
|
Publish=True,
|
|
)
|
|
return client
|