RemoandGitHub
8b41b71db7
Remove db setup from the Haskell test suite and include it in the test bash scripts.
72 lines
2.7 KiB
Bash
Executable File
72 lines
2.7 KiB
Bash
Executable File
#! /usr/bin/env bash
|
|
|
|
usage() {
|
|
echo "$0 <db-uri> <test-database>"
|
|
exit -1
|
|
}
|
|
|
|
if [ -z "$1" ]; then
|
|
echo "Please supply the connection uri for the user with create database privileges"
|
|
usage
|
|
fi
|
|
|
|
if [ -z "$2" ]; then
|
|
echo "Please supply the test database name"
|
|
usage
|
|
fi
|
|
|
|
if [[ $1 != postgres://* ]]; then
|
|
echo "Please use a valid connection URI (https://www.postgresql.org/docs/current/static/libpq-connect.html#AEN45347)"
|
|
usage
|
|
fi
|
|
|
|
BASEPATH=$( cd $(dirname $0) ; pwd -P )
|
|
URI="$1"
|
|
#Extract host and port--we need this to form the new connection string
|
|
HOST_PORT=$(echo $URI | cut -d'/' -f3 | cut -d'@' -f2 )
|
|
DB=$2
|
|
# Specify the username of choice, or let the script create a random unique user by appending the database name
|
|
TEST_USER_NAME=postgrest_test_authenticator
|
|
# New password will get assigned only if the user does not already exist
|
|
# Otherwise make sure to provide the correct password for the existing user
|
|
TEST_USER_PASS=$(cat /dev/urandom | env LC_CTYPE=C tr -dc 'a-zA-Z0-9' | fold -w 16 | head -n 1)
|
|
|
|
PGOPTIONS='-c client_min_messages=WARNING' psql "$URI" -Xq >/dev/null -c 'select rolcreatedb from pg_authid where rolname = current_user;' 2>/dev/null
|
|
if [ $? -ne 0 ]; then
|
|
echo "ERROR: Please specify the user with 'Create DB' permissions, and ensure that the default database for the username exists."
|
|
exit 1
|
|
fi
|
|
|
|
# plpgsql does not like psql variables, easier to pull this part off with bash variables
|
|
PGOPTIONS='-c client_min_messages=WARNING' psql "$URI" -Xq >/dev/null <<EOF
|
|
SELECT pg_terminate_backend(pg_stat_activity.pid)
|
|
FROM pg_stat_activity
|
|
WHERE pg_stat_activity.datname = '$DB'
|
|
AND pid <> pg_backend_pid();
|
|
|
|
DROP DATABASE IF EXISTS $DB;
|
|
DROP ROLE IF EXISTS $TEST_USER_NAME;
|
|
CREATE USER $TEST_USER_NAME WITH SUPERUSER LOGIN NOINHERIT PASSWORD '$TEST_USER_PASS' CREATEROLE;
|
|
ALTER ROLE postgrest_test_authenticator SET default_text_search_config TO english;
|
|
CREATE DATABASE $DB OWNER $TEST_USER_NAME LC_COLLATE 'POSIX' LC_CTYPE 'POSIX' TEMPLATE template0;
|
|
\\connect $DB
|
|
ALTER SCHEMA public OWNER TO $TEST_USER_NAME;
|
|
ALTER DATABASE $DB SET LC_MESSAGES = 'POSIX';
|
|
ALTER DATABASE $DB SET LC_MONETARY = 'POSIX';
|
|
ALTER DATABASE $DB SET LC_NUMERIC = 'POSIX';
|
|
ALTER DATABASE $DB SET LC_TIME = 'POSIX';
|
|
|
|
\i $BASEPATH/fixtures/load.sql
|
|
EOF
|
|
|
|
#Remove database path from the connection uri--prevents setting up the new database name with PGDATABASE
|
|
URI=$(echo $URI | cut -d'/' -f1-3)
|
|
|
|
PGDATABASE=$DB PGOPTIONS='-c client_min_messages=WARNING' psql "$URI" --set=db=$DB -Xq <<EOF
|
|
CREATE EXTENSION IF NOT EXISTS pgcrypto;
|
|
ALTER DATABASE ${DB} SET request.jwt.claim.id = '-1';
|
|
EOF
|
|
|
|
# Create a new connection string to use with the test runner
|
|
echo 'postgres://'${TEST_USER_NAME}':'$TEST_USER_PASS'@'$HOST_PORT'/'$DB
|