run.sh 1.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748
  1. #!/bin/bash
  2. set -e
  3. image="$1"
  4. testDir="$(readlink -f "$(dirname "$BASH_SOURCE")")"
  5. export POSTGRES_USER='my cool postgres user'
  6. export POSTGRES_PASSWORD='my cool postgres password'
  7. export POSTGRES_DB='my cool postgres database'
  8. cname="postgres-container-$RANDOM-$RANDOM"
  9. cid="$(
  10. docker run -d \
  11. -e POSTGRES_USER \
  12. -e POSTGRES_PASSWORD \
  13. -e POSTGRES_DB \
  14. --name "$cname" \
  15. -v "$testDir/initdb.sql:/docker-entrypoint-initdb.d/test.sql":ro \
  16. "$image"
  17. )"
  18. trap "docker rm -f $cid > /dev/null" EXIT
  19. psql() {
  20. docker run --rm -i \
  21. --link "$cname":postgres \
  22. --entrypoint psql \
  23. -e PGPASSWORD="$POSTGRES_PASSWORD" \
  24. "$image" \
  25. --host postgres \
  26. --username "$POSTGRES_USER" \
  27. --dbname "$POSTGRES_DB" \
  28. --quiet --no-align --tuples-only \
  29. "$@"
  30. }
  31. tries=10
  32. while ! echo 'SELECT 1' | psql &> /dev/null; do
  33. (( tries-- ))
  34. if [ $tries -le 0 ]; then
  35. echo >&2 'postgres failed to accept connetions in a reasonable amount of time!'
  36. echo 'SELECT 1' | psql # to hopefully get a useful error message
  37. false
  38. fi
  39. sleep 2
  40. done
  41. [ "$(echo 'SELECT COUNT(*) FROM test' | psql)" = 1 ]
  42. [ "$(echo 'SELECT c FROM test' | psql)" = 'goodbye!' ]