run.sh 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  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 -vf $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. : ${POSTGRES_TEST_TRIES:=10}
  32. : ${POSTGRES_TEST_SLEEP:=2}
  33. tries="$POSTGRES_TEST_TRIES"
  34. while ! echo 'SELECT 1' | psql &> /dev/null; do
  35. (( tries-- ))
  36. if [ $tries -le 0 ]; then
  37. echo >&2 'postgres failed to accept connections in a reasonable amount of time!'
  38. echo 'SELECT 1' | psql # to hopefully get a useful error message
  39. false
  40. fi
  41. echo >&2 -n .
  42. sleep "${POSTGRES_TEST_SLEEP}"
  43. done
  44. [ "$(echo 'SELECT COUNT(*) FROM test' | psql)" = 1 ]
  45. [ "$(echo 'SELECT c FROM test' | psql)" = 'goodbye!' ]