|
| 1 | +#!/bin/bash |
| 2 | +# Shared test helpers for functional tests |
| 3 | + |
| 4 | +PASS_COUNT=0 |
| 5 | +FAIL_COUNT=0 |
| 6 | +SKIP_COUNT=0 |
| 7 | +ORC_URL="http://localhost:3099" |
| 8 | + |
| 9 | +pass() { |
| 10 | + echo " ✅ PASS: $1" |
| 11 | + ((PASS_COUNT++)) |
| 12 | +} |
| 13 | + |
| 14 | +fail() { |
| 15 | + echo " ❌ FAIL: $1" |
| 16 | + [ -n "$2" ] && echo " $2" |
| 17 | + ((FAIL_COUNT++)) |
| 18 | +} |
| 19 | + |
| 20 | +skip() { |
| 21 | + echo " ⚠️ SKIP: $1" |
| 22 | + ((SKIP_COUNT++)) |
| 23 | +} |
| 24 | + |
| 25 | +summary() { |
| 26 | + echo "" |
| 27 | + echo "=== RESULTS: $PASS_COUNT passed, $FAIL_COUNT failed, $SKIP_COUNT skipped ===" |
| 28 | + [ "$FAIL_COUNT" -gt 0 ] && exit 1 |
| 29 | + exit 0 |
| 30 | +} |
| 31 | + |
| 32 | +# Test that an HTTP endpoint returns expected status code |
| 33 | +test_endpoint() { |
| 34 | + local NAME="$1" URL="$2" EXPECT="$3" |
| 35 | + local CODE |
| 36 | + CODE=$(curl -s -o /dev/null -w "%{http_code}" "$URL" 2>&1) |
| 37 | + if [ "$CODE" = "$EXPECT" ]; then |
| 38 | + pass "$NAME (HTTP $CODE)" |
| 39 | + else |
| 40 | + fail "$NAME (HTTP $CODE, expected $EXPECT)" |
| 41 | + fi |
| 42 | +} |
| 43 | + |
| 44 | +# Test that response body contains a string |
| 45 | +test_body_contains() { |
| 46 | + local NAME="$1" URL="$2" EXPECT="$3" |
| 47 | + local BODY |
| 48 | + BODY=$(curl -s "$URL" 2>&1) |
| 49 | + if echo "$BODY" | grep -q "$EXPECT"; then |
| 50 | + pass "$NAME" |
| 51 | + else |
| 52 | + fail "$NAME" "Response does not contain '$EXPECT'" |
| 53 | + fi |
| 54 | +} |
| 55 | + |
| 56 | +# Wait for orchestrator to be ready |
| 57 | +wait_for_orchestrator() { |
| 58 | + echo "Waiting for orchestrator to be ready..." |
| 59 | + for i in $(seq 1 30); do |
| 60 | + if curl -s -o /dev/null "$ORC_URL/api/clusters" 2>/dev/null; then |
| 61 | + echo "Orchestrator ready after ${i}s" |
| 62 | + return 0 |
| 63 | + fi |
| 64 | + sleep 1 |
| 65 | + done |
| 66 | + echo "Orchestrator not ready after 30s" |
| 67 | + return 1 |
| 68 | +} |
| 69 | + |
| 70 | +# Seed discovery and wait for all instances |
| 71 | +discover_topology() { |
| 72 | + local MASTER_HOST="$1" |
| 73 | + echo "Seeding discovery with $MASTER_HOST..." |
| 74 | + curl -s "$ORC_URL/api/discover/$MASTER_HOST/3306" > /dev/null |
| 75 | + |
| 76 | + echo "Waiting for topology discovery..." |
| 77 | + for i in $(seq 1 60); do |
| 78 | + local COUNT |
| 79 | + COUNT=$(curl -s "$ORC_URL/api/cluster/$MASTER_HOST:3306" 2>/dev/null | python3 -c "import json,sys; print(len(json.load(sys.stdin)))" 2>/dev/null) |
| 80 | + if [ "$COUNT" = "3" ]; then |
| 81 | + echo "Full topology discovered (3 instances) after ${i}s" |
| 82 | + return 0 |
| 83 | + fi |
| 84 | + # Also try to discover replicas directly |
| 85 | + if [ "$i" = "10" ] || [ "$i" = "20" ]; then |
| 86 | + curl -s "$ORC_URL/api/discover/mysql2/3306" > /dev/null 2>&1 |
| 87 | + curl -s "$ORC_URL/api/discover/mysql3/3306" > /dev/null 2>&1 |
| 88 | + fi |
| 89 | + sleep 1 |
| 90 | + done |
| 91 | + echo "WARNING: Only discovered $COUNT instances after 60s" |
| 92 | + return 1 |
| 93 | +} |
| 94 | + |
| 95 | +# Get ProxySQL servers for a hostgroup |
| 96 | +proxysql_servers() { |
| 97 | + local HG="$1" |
| 98 | + docker compose -f tests/functional/docker-compose.yml exec -T proxysql \ |
| 99 | + mysql -h127.0.0.1 -P6032 -uradmin -pradmin -Nse \ |
| 100 | + "SELECT hostname, port, status FROM runtime_mysql_servers WHERE hostgroup_id=$HG" 2>/dev/null |
| 101 | +} |
| 102 | + |
| 103 | +# Get MySQL read_only status |
| 104 | +mysql_read_only() { |
| 105 | + local CONTAINER="$1" |
| 106 | + docker compose -f tests/functional/docker-compose.yml exec -T "$CONTAINER" \ |
| 107 | + mysql -uroot -ptestpass -Nse "SELECT @@read_only" 2>/dev/null |
| 108 | +} |
| 109 | + |
| 110 | +# Get MySQL replication source |
| 111 | +mysql_source_host() { |
| 112 | + local CONTAINER="$1" |
| 113 | + docker compose -f tests/functional/docker-compose.yml exec -T "$CONTAINER" \ |
| 114 | + mysql -uroot -ptestpass -Nse "SHOW REPLICA STATUS\G" 2>/dev/null | grep "Source_Host" | awk '{print $2}' |
| 115 | +} |
0 commit comments