wehub-resource-sync
115 lines
3.4 KiB
Bash
115 lines
3.4 KiB
Bash
# Common helpers for doltgresql compatibility BATS tests.
|
||
# Loaded by all test files in this directory tree via:
|
||
# load $BATS_TEST_DIRNAME/../helper/common.bash (from subdirs)
|
||
# or
|
||
# load $BATS_TEST_DIRNAME/helper/common.bash (from top level)
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# Server lifecycle
|
||
# ---------------------------------------------------------------------------
|
||
|
||
COMPAT_SERVER_PID=""
|
||
COMPAT_SERVER_PORT=""
|
||
|
||
# pick_port — find an unused TCP port in the range 2048–6144.
|
||
pick_port() {
|
||
for i in {0..99}; do
|
||
port=$((RANDOM % 4096 + 2048))
|
||
if ! nc -z localhost "$port" 2>/dev/null; then
|
||
echo "$port"
|
||
return 0
|
||
fi
|
||
done
|
||
echo "ERROR: could not find a free port" >&2
|
||
return 1
|
||
}
|
||
|
||
# write_config <dir> <port> — write a minimal doltgres config.yaml into <dir>.
|
||
write_config() {
|
||
local dir="$1"
|
||
local port="$2"
|
||
cat > "$dir/compat-config.yaml" <<EOF
|
||
log_level: warning
|
||
behavior:
|
||
read_only: false
|
||
disable_client_multi_statements: false
|
||
listener:
|
||
host: localhost
|
||
port: $port
|
||
EOF
|
||
}
|
||
|
||
# start_doltgres <binary> <datadir> [logfile]
|
||
# Starts a doltgres server and waits for it to accept connections.
|
||
# Sets COMPAT_SERVER_PID and COMPAT_SERVER_PORT.
|
||
start_doltgres() {
|
||
local binary="${1:?start_doltgres: binary required}"
|
||
local datadir="${2:?start_doltgres: datadir required}"
|
||
local logfile="${3:-/dev/null}"
|
||
|
||
COMPAT_SERVER_PORT=$(pick_port)
|
||
write_config "$datadir" "$COMPAT_SERVER_PORT"
|
||
|
||
PGPASSWORD=password "$binary" -data-dir="$datadir" --config="$datadir/compat-config.yaml" \
|
||
> "$logfile" 2>&1 &
|
||
COMPAT_SERVER_PID=$!
|
||
|
||
# Wait up to 15 s
|
||
local end=$((SECONDS + 15))
|
||
while [ $SECONDS -lt $end ]; do
|
||
if PGPASSWORD=password psql -U postgres -h localhost -p "$COMPAT_SERVER_PORT" \
|
||
-c "SELECT 1;" postgres >/dev/null 2>&1; then
|
||
return 0
|
||
fi
|
||
sleep 0.5
|
||
done
|
||
|
||
echo "ERROR: doltgres server failed to start on port $COMPAT_SERVER_PORT" >&2
|
||
if [ -f "$logfile" ]; then cat "$logfile" >&2; fi
|
||
kill "$COMPAT_SERVER_PID" 2>/dev/null
|
||
COMPAT_SERVER_PID=""
|
||
return 1
|
||
}
|
||
|
||
# stop_doltgres — kill the server started by start_doltgres.
|
||
stop_doltgres() {
|
||
if [ -n "$COMPAT_SERVER_PID" ]; then
|
||
kill "$COMPAT_SERVER_PID" 2>/dev/null || true
|
||
wait "$COMPAT_SERVER_PID" 2>/dev/null || true
|
||
COMPAT_SERVER_PID=""
|
||
COMPAT_SERVER_PORT=""
|
||
fi
|
||
}
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# Query helpers
|
||
# ---------------------------------------------------------------------------
|
||
|
||
# sql [psql-args...] — run psql against the current server's postgres database.
|
||
# Use with heredoc for multi-statement sessions:
|
||
# sql <<SQL
|
||
# SELECT dolt_checkout('other');
|
||
# SELECT * FROM abc;
|
||
# SQL
|
||
# Or with -c for a single statement:
|
||
# run sql -c "SELECT count(*) FROM abc;"
|
||
sql() {
|
||
PGPASSWORD=password psql -U postgres -h localhost -p "$COMPAT_SERVER_PORT" \
|
||
-v ON_ERROR_STOP=1 "$@" postgres
|
||
}
|
||
|
||
# sql_csv [psql-args...] — same as sql but outputs CSV (--csv flag).
|
||
sql_csv() {
|
||
PGPASSWORD=password psql --csv -U postgres -h localhost -p "$COMPAT_SERVER_PORT" \
|
||
-v ON_ERROR_STOP=1 "$@" postgres
|
||
}
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# Repo isolation
|
||
# ---------------------------------------------------------------------------
|
||
|
||
# copy_repo <src> <dst> — copy a data directory for test isolation.
|
||
copy_repo() {
|
||
cp -Rpf "$1" "$2"
|
||
}
|