Mock Lambda handler event context dict in Python

import json


def lambda_handler(event, context):
    """
    A mock AWS Lambda handler that processes an event dict and context object.
    Demonstrates the typical Lambda function signature and basic event/context usage.
    """
    print("Received event:", json.dumps(event, indent=2))
    print("Function name:", context.function_name)
    print("AWS Request ID:", context.aws_request_id)
    print("Memory limit (MB):", context.memory_limit_in_mb)
    
    # Simple response echoing the event data
    return {
        "statusCode": 200,
        "body": json.dumps({
            "message": "Hello from mock Lambda!",
            "received_key": event.get("key", "default"),
            "function_name": context.function_name
        }),
        "headers": {
            "Content-Type": "application/json"
        }
    }


class MockContext:
    """Simple mock of the Lambda context object."""
    function_name = "my-mock-function"
    aws_request_id = "mock-request-id-123456"
    memory_limit_in_mb = 128


if __name__ == "__main__":
    # Simulate an AWS Lambda invocation
    mock_event = {
        "key": "value123",
        "extra": "data"
    }
    mock_context = MockContext()
    
    result = lambda_handler(mock_event, mock_context)
    print("Response:", result)