mirror of
https://github.com/neondatabase/neon.git
synced 2026-01-05 20:42:54 +00:00
cf9d817a21
## Problem Some extensions do not contain tests, which can be easily run on top of docker-compose or staging. ## Summary of changes Added the pg_regress based tests for `pg_tiktoken`, `pgx_ulid`, `pg_rag` Now they will be run on top of docker-compose, but I intend to adopt them to be run on top staging in the next PRs
90 lines
2.6 KiB
Bash
Executable File
90 lines
2.6 KiB
Bash
Executable File
#!/bin/bash
|
|
set -eux
|
|
|
|
# Generate a random tenant or timeline ID
|
|
#
|
|
# Takes a variable name as argument. The result is stored in that variable.
|
|
generate_id() {
|
|
local -n resvar=$1
|
|
printf -v resvar '%08x%08x%08x%08x' $SRANDOM $SRANDOM $SRANDOM $SRANDOM
|
|
}
|
|
|
|
PG_VERSION=${PG_VERSION:-14}
|
|
|
|
SPEC_FILE_ORG=/var/db/postgres/specs/spec.json
|
|
SPEC_FILE=/tmp/spec.json
|
|
|
|
echo "Waiting pageserver become ready."
|
|
while ! nc -z pageserver 6400; do
|
|
sleep 1;
|
|
done
|
|
echo "Page server is ready."
|
|
|
|
cp ${SPEC_FILE_ORG} ${SPEC_FILE}
|
|
|
|
if [ -n "${TENANT_ID:-}" ] && [ -n "${TIMELINE_ID:-}" ]; then
|
|
tenant_id=${TENANT_ID}
|
|
timeline_id=${TIMELINE_ID}
|
|
else
|
|
echo "Check if a tenant present"
|
|
PARAMS=(
|
|
-X GET
|
|
-H "Content-Type: application/json"
|
|
"http://pageserver:9898/v1/tenant"
|
|
)
|
|
tenant_id=$(curl "${PARAMS[@]}" | jq -r .[0].id)
|
|
if [ -z "${tenant_id}" ] || [ "${tenant_id}" = null ]; then
|
|
echo "Create a tenant"
|
|
generate_id tenant_id
|
|
PARAMS=(
|
|
-X PUT
|
|
-H "Content-Type: application/json"
|
|
-d "{\"mode\": \"AttachedSingle\", \"generation\": 1, \"tenant_conf\": {}}"
|
|
"http://pageserver:9898/v1/tenant/${tenant_id}/location_config"
|
|
)
|
|
result=$(curl "${PARAMS[@]}")
|
|
echo $result | jq .
|
|
fi
|
|
|
|
echo "Check if a timeline present"
|
|
PARAMS=(
|
|
-X GET
|
|
-H "Content-Type: application/json"
|
|
"http://pageserver:9898/v1/tenant/${tenant_id}/timeline"
|
|
)
|
|
timeline_id=$(curl "${PARAMS[@]}" | jq -r .[0].timeline_id)
|
|
if [ -z "${timeline_id}" ] || [ "${timeline_id}" = null ]; then
|
|
generate_id timeline_id
|
|
PARAMS=(
|
|
-sbf
|
|
-X POST
|
|
-H "Content-Type: application/json"
|
|
-d "{\"new_timeline_id\": \"${timeline_id}\", \"pg_version\": ${PG_VERSION}}"
|
|
"http://pageserver:9898/v1/tenant/${tenant_id}/timeline/"
|
|
)
|
|
result=$(curl "${PARAMS[@]}")
|
|
echo $result | jq .
|
|
fi
|
|
fi
|
|
|
|
if [[ ${PG_VERSION} -ge 17 ]]; then
|
|
ulid_extension=pgx_ulid
|
|
else
|
|
ulid_extension=ulid
|
|
fi
|
|
echo "Adding pgx_ulid"
|
|
shared_libraries=$(jq -r '.cluster.settings[] | select(.name=="shared_preload_libraries").value' ${SPEC_FILE})
|
|
sed -i "s/${shared_libraries}/${shared_libraries},${ulid_extension}/" ${SPEC_FILE}
|
|
echo "Overwrite tenant id and timeline id in spec file"
|
|
sed -i "s/TENANT_ID/${tenant_id}/" ${SPEC_FILE}
|
|
sed -i "s/TIMELINE_ID/${timeline_id}/" ${SPEC_FILE}
|
|
|
|
cat ${SPEC_FILE}
|
|
|
|
echo "Start compute node"
|
|
/usr/local/bin/compute_ctl --pgdata /var/db/postgres/compute \
|
|
-C "postgresql://cloud_admin@localhost:55433/postgres" \
|
|
-b /usr/local/bin/postgres \
|
|
--compute-id "compute-$RANDOM" \
|
|
-S ${SPEC_FILE}
|