run_test.sh

#!/bin/bash

# Simple Load Testing Script
set -e

K6_CONTAINER="demo_k6"

check_container() {
    if ! docker ps --format '{{.Names}}' | grep -q "^${K6_CONTAINER}$"; then
        echo "Error: k6 container not running. Start with: docker-compose up -d"
        exit 1
    fi
}

smoke() {
    echo "Running Smoke Test..."
    docker exec -it ${K6_CONTAINER} k6 run /scripts/smoke.js
}

baseline() {
    echo "Running Baseline Test..."
    docker exec -it ${K6_CONTAINER} k6 run /scripts/capacity.js
}

stepped() {
    echo "Running Stepped Baseline Test (EXTREME)..."
    docker exec -it ${K6_CONTAINER} k6 run /scripts/baseline_stepped.js
}

# Main
check_container

case "${1:-help}" in
    smoke)
        smoke
        ;;
    baseline)
        baseline
        ;;
    stepped)
        stepped
        ;;
    *)
        echo "Usage: ./run_test.sh [smoke|baseline|stepped]"
        echo ""
        echo "  smoke     - Quick sanity check (20 seconds)"
        echo "  baseline  - Smooth ramp: 100→2000 RPS, 50ms work (7 min)"
        echo "  stepped   - Find capacity: 100→3000 RPS by steps, 50ms work (~16 min)"
        exit 1
        ;;
esac

echo ""
echo "Done! View metrics at http://localhost:3000"