#!/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 -vf $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 \
		"$@"
}

: ${POSTGRES_TEST_TRIES:=10}
: ${POSTGRES_TEST_SLEEP:=2}

tries="$POSTGRES_TEST_TRIES"
while ! echo 'SELECT 1' | psql &> /dev/null; do
	(( tries-- ))
	if [ $tries -le 0 ]; then
		echo >&2 'postgres failed to accept connections in a reasonable amount of time!'
		( set -x && docker logs "$cid" ) >&2 || true
		echo 'SELECT 1' | psql # to hopefully get a useful error message
		false
	fi
	echo >&2 -n .
	sleep "$POSTGRES_TEST_SLEEP"
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