From 3f775d2df194554f686d1e5ce29957032bda8844 Mon Sep 17 00:00:00 2001
From: Alexey Masterov <alexeymasterov@neon.tech>
Date: Tue, 1 Apr 2025 12:06:04 +0200
Subject: [PATCH] Change the script

---
 docker-compose/run-tests.sh | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/docker-compose/run-tests.sh b/docker-compose/run-tests.sh
index 3117950cc0..cb2de9770b 100644
--- a/docker-compose/run-tests.sh
+++ b/docker-compose/run-tests.sh
@@ -1,6 +1,14 @@
 #!/bin/bash
 set -x
 
+if [[ -v BENCHMARK_CONNSTR ]] && [[ ${BENCHMARK_CONNSTR} =~ ^postgres(ql)?:\/\/([^:]+):([^@]+)@([^:/]+):?([0-9]*)\/(.+)$ ]]; then
+  export PGUSER="${BASH_REMATCH[2]}"
+  export PGPASSWORD="${BASH_REMATCH[3]}"
+  export PGHOST="${BASH_REMATCH[4]}"
+  export PGPORT="${BASH_REMATCH[5]:-5432}"
+  export PGDATABASE="${BASH_REMATCH[6]}"
+fi
+
 extdir=${1}
 
 cd "${extdir}" || exit 2