test-local.sh

#!/bin/bash
set -e

# Configuration
ECR_REPOSITORY_NAME="pandoc-lambda"
IMAGE_TAG="local-test"

# Enable debug mode
set -x

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

# Build the Docker image using the same configuration as AWS
echo "Building Docker image..."
docker buildx build \
    --platform linux/amd64 \
    --provenance=false \
    --load \
    -t ${ECR_REPOSITORY_NAME}:${IMAGE_TAG} \
    .

# 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
- PDF support via LaTeX

## Code Example

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

## Math Example

$E = mc^2$
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": "pdf",
  "content": "${CONTENT_BASE64}",
  "options": ["--pdf-engine=pdflatex"]
}
EOL

# Create test script
echo "Creating test script..."
cat > test.py << 'EOL'
import json
import base64
import os
import subprocess

# Print system information
print("\nSystem information:")
subprocess.run(["ls", "-la", "/usr/bin/pandoc"])
subprocess.run(["which", "pdflatex"])

# Print pandoc version
try:
    version = subprocess.check_output(["pandoc", "--version"]).decode()
    print("\nPandoc version information:")
    print(version)
except Exception as e:
    print(f"Error checking pandoc version: {e}")

# Import the handler from app.py
from app import handler

# Load test event
with open("/function/test-event.json", "r") as f:
    event = json.load(f)

# Call the handler
result = handler(event, {})

# Save the output
if result["statusCode"] == 200:
    output_content = base64.b64decode(result["body"])
    with open("/tmp/output/output.pdf", "wb") as f:
        f.write(output_content)
    print("Conversion successful! Output saved to output.pdf")
else:
    print("Conversion failed:", result["body"])
EOL

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

# Run the Docker container with the test event
echo "Running Lambda function locally..."
docker run --rm \
  --platform linux/amd64 \
  -v "$(pwd)/test-event.json:/function/test-event.json" \
  -v "$(pwd)/test.py:/function/test.py" \
  -v "$(pwd)/output:/tmp/output" \
  --entrypoint python3 \
  ${ECR_REPOSITORY_NAME}:${IMAGE_TAG} \
  /function/test.py

echo "Test completed!"