test-lambda.sh

#!/bin/bash
set -e

# Configuration
LAMBDA_FUNCTION_NAME="pandoc-lambda-function"
AWS_REGION=${AWS_REGION:-"us-east-1"}

# Check if AWS CLI is installed
if ! command -v aws &> /dev/null; then
    echo "AWS CLI not found. Please install it first."
    exit 1
fi

# Create output directory
if [ ! -d "output" ]; then
    mkdir -p output
fi

# Create a sample markdown file for testing
echo "Creating sample markdown file..."
cat > sample.md << 'EOL'
# Sample Markdown

This is a **sample** markdown file to test the Pandoc Lambda function.

## Features

- Document conversion
- Multiple formats
- AWS Lambda integration

## Code Example

```python
def hello_world():
    print("Hello, World!")
```
EOL

# Encode the content as base64
CONTENT_BASE64=$(cat sample.md | base64)

# Create a test event file
echo "Creating test event file..."
cat > test-event.json << EOL
{
  "input_format": "markdown",
  "output_format": "html",
  "content": "${CONTENT_BASE64}",
  "options": ["--standalone", "--toc"]
}
EOL

# Invoke the Lambda function
echo "Invoking Lambda function ${LAMBDA_FUNCTION_NAME}..."
aws lambda invoke \
    --function-name ${LAMBDA_FUNCTION_NAME} \
    --region ${AWS_REGION} \
    --payload file://test-event.json \
    --cli-binary-format raw-in-base64-out \
    output/lambda-response.json

# Check if the Lambda function was invoked successfully
if [ $? -ne 0 ]; then
    echo "Failed to invoke Lambda function."
    exit 1
fi

# Check the response
STATUS_CODE=$(cat output/lambda-response.json | grep -o '"statusCode":[0-9]*' | cut -d':' -f2)

if [ "$STATUS_CODE" == "200" ]; then
    echo "Lambda function executed successfully!"
    
    # Extract and decode the base64 content
    RESPONSE_BODY=$(cat output/lambda-response.json | grep -o '"body":"[^"]*"' | cut -d':' -f2 | sed 's/"//g')
    echo "$RESPONSE_BODY" | base64 --decode > output/lambda-output.html
    
    echo "Output saved to output/lambda-output.html"
    
    # Open the file if on macOS
    if [[ "$OSTYPE" == "darwin"* ]]; then
        open output/lambda-output.html
    fi
else
    echo "Lambda function execution failed. Status code: $STATUS_CODE"
    cat output/lambda-response.json
fi