test.sh

#!/bin/sh
# automated smoke test for the mysql (mariadb) container
# builds the image, starts it standalone, waits for mariadb to accept
# connections and then does a REAL db round-trip (create/insert/select)
set -eu

IMAGE=mysql-test
NAME=mysql-test-run

ADMIN_USER=admin
ADMIN_PASSWORD=adminsecret
DB_NAME=testdb
DB_USER=testuser
DB_PASSWORD=testsecret

FAILED=0
fail() {
  echo "FAIL: $*" >&2
  FAILED=1
}

cleanup() {
  echo ">> cleanup: removing container $NAME"
  docker rm -f "$NAME" >/dev/null 2>&1 || true
}
trap cleanup EXIT INT TERM

# run a query as the given user, printing the result (no column headers)
# usage: q <user> <password> <sql> [db]
q() {
  _u="$1"; _p="$2"; _sql="$3"; _db="${4:-}"
  docker exec "$NAME" mysql -u"$_u" -p"$_p" -N -B ${_db:+"$_db"} -e "$_sql" 2>/dev/null
}

echo ">> building image $IMAGE"
docker build -t "$IMAGE" .

echo ">> (re)starting container $NAME"
docker rm -f "$NAME" >/dev/null 2>&1 || true
docker run -d --name "$NAME" \
  -e ADMIN_USER="$ADMIN_USER" \
  -e ADMIN_PASSWORD="$ADMIN_PASSWORD" \
  -e DB_NAME="$DB_NAME" \
  -e DB_USER="$DB_USER" \
  -e DB_PASSWORD="$DB_PASSWORD" \
  "$IMAGE"

# wait until the admin user can actually run a query (mariadb init takes a while)
echo ">> waiting for mariadb to accept connections (up to ~120s)"
READY=0
i=0
while [ "$i" -lt 60 ]; do
  if ! docker ps --format '{{.Names}}' | grep -q "^${NAME}$"; then
    echo "!! container is not running anymore, dumping logs:" >&2
    docker logs "$NAME" >&2 2>&1 || true
    fail "container exited during startup"
    break
  fi
  if [ "$(q "$ADMIN_USER" "$ADMIN_PASSWORD" 'SELECT 1')" = "1" ]; then
    READY=1
    break
  fi
  i=$((i + 1))
  sleep 2
done

if [ "$READY" -ne 1 ] && [ "$FAILED" -eq 0 ]; then
  echo "!! mariadb did not accept connections in time, dumping logs:" >&2
  docker logs "$NAME" >&2 2>&1 || true
  fail "timed out waiting for mariadb"
fi

# only run the deeper assertions if the db is up
if [ "$READY" -eq 1 ]; then

  echo ">> assert: admin user works"
  if [ "$(q "$ADMIN_USER" "$ADMIN_PASSWORD" 'SELECT 42')" = "42" ]; then
    echo "ok - admin user can query"
  else
    fail "admin user could not run a query"
  fi

  # the auto-created db/user is provisioned asynchronously by the entrypoint,
  # so wait until the app user can connect to its database
  echo ">> waiting for auto-created db '$DB_NAME' + user '$DB_USER'"
  DBREADY=0
  j=0
  while [ "$j" -lt 30 ]; do
    if [ "$(q "$DB_USER" "$DB_PASSWORD" 'SELECT 1' "$DB_NAME")" = "1" ]; then
      DBREADY=1
      break
    fi
    j=$((j + 1))
    sleep 2
  done
  if [ "$DBREADY" -eq 1 ]; then
    echo "ok - db user can connect to '$DB_NAME'"
  else
    fail "db user '$DB_USER' could not connect to '$DB_NAME'"
  fi

  if [ "$DBREADY" -eq 1 ]; then
    echo ">> assert: real round-trip (CREATE TABLE / INSERT / SELECT) as db user"
    q "$DB_USER" "$DB_PASSWORD" \
      'CREATE TABLE roundtrip (id INT PRIMARY KEY, name VARCHAR(32));' "$DB_NAME"
    q "$DB_USER" "$DB_PASSWORD" \
      "INSERT INTO roundtrip (id, name) VALUES (1, 'servercontainers');" "$DB_NAME"
    GOT=$(q "$DB_USER" "$DB_PASSWORD" \
      'SELECT name FROM roundtrip WHERE id = 1;' "$DB_NAME")
    if [ "$GOT" = "servercontainers" ]; then
      echo "ok - round-trip returned expected row ('$GOT')"
    else
      fail "round-trip did not return expected row (got: '$GOT')"
    fi

    echo ">> assert: admin can see the app database"
    if q "$ADMIN_USER" "$ADMIN_PASSWORD" 'SHOW DATABASES;' | grep -q "^${DB_NAME}$"; then
      echo "ok - admin sees database '$DB_NAME'"
    else
      fail "admin does not see database '$DB_NAME'"
    fi
  fi

fi

echo
if [ "$FAILED" -eq 0 ]; then
  echo "ALL TESTS PASSED"
  exit 0
else
  echo "SOME TESTS FAILED"
  exit 1
fi