testing/smoketest/function/main.py

import json

coldstart = "true"
def handler(event, context):
    global coldstart
    print("Example function log", context.aws_request_id)
    resp = {
        "statusCode": 200,
        "body": json.dumps("Hello from Lambda!"+context.aws_request_id),
        "headers": {
            "coldstart": coldstart,
        }
    }
    coldstart = "false"
    return resp