|
@@ -0,0 +1,51 @@
|
|
|
|
|
+#!/bin/bash
|
|
|
|
|
+set -e
|
|
|
|
|
+
|
|
|
|
|
+image="$1"
|
|
|
|
|
+
|
|
|
|
|
+export POSTGRES_USER='my cool postgres user'
|
|
|
|
|
+export POSTGRES_PASSWORD='my cool postgres password'
|
|
|
|
|
+export POSTGRES_DB='my cool postgres database'
|
|
|
|
|
+
|
|
|
|
|
+cname="postgres-container-$RANDOM-$RANDOM"
|
|
|
|
|
+cid="$(docker run -d -e POSTGRES_USER -e POSTGRES_PASSWORD -e POSTGRES_DB --name "$cname" "$image")"
|
|
|
|
|
+trap "docker rm -f $cid > /dev/null" EXIT
|
|
|
|
|
+
|
|
|
|
|
+psql() {
|
|
|
|
|
+ docker run --rm -i \
|
|
|
|
|
+ --link "$cname":postgres \
|
|
|
|
|
+ --entrypoint psql \
|
|
|
|
|
+ -e PGPASSWORD="$POSTGRES_PASSWORD" \
|
|
|
|
|
+ "$image" \
|
|
|
|
|
+ --host postgres \
|
|
|
|
|
+ --username "$POSTGRES_USER" \
|
|
|
|
|
+ --dbname "$POSTGRES_DB" \
|
|
|
|
|
+ --quiet --no-align --tuples-only \
|
|
|
|
|
+ "$@"
|
|
|
|
|
+}
|
|
|
|
|
+
|
|
|
|
|
+tries=10
|
|
|
|
|
+while ! echo 'SELECT 1' | psql &> /dev/null; do
|
|
|
|
|
+ (( tries-- ))
|
|
|
|
|
+ if [ $tries -le 0 ]; then
|
|
|
|
|
+ echo >&2 'postgres failed to accept connetions in a reasonable amount of time!'
|
|
|
|
|
+ echo 'SELECT 1' | psql # to hopefully get a useful error message
|
|
|
|
|
+ false
|
|
|
|
|
+ fi
|
|
|
|
|
+ sleep 2
|
|
|
|
|
+done
|
|
|
|
|
+
|
|
|
|
|
+echo 'CREATE TABLE test (a INT, b INT, c VARCHAR(255))' | psql
|
|
|
|
|
+[ "$(echo 'SELECT COUNT(*) FROM test' | psql)" = 0 ]
|
|
|
|
|
+psql <<'EOSQL'
|
|
|
|
|
+ INSERT INTO test VALUES (1, 2, 'hello')
|
|
|
|
|
+EOSQL
|
|
|
|
|
+[ "$(echo 'SELECT COUNT(*) FROM test' | psql)" = 1 ]
|
|
|
|
|
+psql <<'EOSQL'
|
|
|
|
|
+ INSERT INTO test VALUES (2, 3, 'goodbye!')
|
|
|
|
|
+EOSQL
|
|
|
|
|
+[ "$(echo 'SELECT COUNT(*) FROM test' | psql)" = 2 ]
|
|
|
|
|
+echo 'DELETE FROM test WHERE a = 1' | psql
|
|
|
|
|
+[ "$(echo 'SELECT COUNT(*) FROM test' | psql)" = 1 ]
|
|
|
|
|
+[ "$(echo 'SELECT c FROM test' | psql)" = "goodbye!" ]
|
|
|
|
|
+echo 'DROP TABLE test' | psql
|