dolthub--doltgresql
wehub-resource-sync
115 行
3.4 KiB
Bash
115 行
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"
|
|
}
|