From accec497ebde840925dae69e7a7fd65d376ada1a Mon Sep 17 00:00:00 2001
From: xxchan <xxchan22f@gmail.com>
Date: Fri, 19 Apr 2024 13:59:30 +0800
Subject: [PATCH] ci: fix poetry in integration test (#16394)

Signed-off-by: xxchan <xxchan22f@gmail.com>
---
 ci/scripts/integration-tests.sh | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

diff --git a/ci/scripts/integration-tests.sh b/ci/scripts/integration-tests.sh
index 90d23b2d80355..9502c297f87ca 100755
--- a/ci/scripts/integration-tests.sh
+++ b/ci/scripts/integration-tests.sh
@@ -24,17 +24,17 @@ shift $((OPTIND -1))
 
 echo "export INTEGRATION_TEST_CASE=${case}" > env_vars.sh
 
-echo "--- clean up docker"
+echo "~~~ clean up docker"
 if [ $(docker ps -aq |wc -l) -gt 0 ]; then
   docker rm -f $(docker ps -aq)
 fi
 docker network prune -f
 docker volume prune -f
 
-echo "--- ghcr login"
+echo "~~~ ghcr login"
 echo "$GHCR_TOKEN" | docker login ghcr.io -u "$GHCR_USERNAME" --password-stdin
 
-echo "--- case: ${case}, format: ${format}"
+echo "+++ set RW_IMAGE"
 
 if [[ -n "${RW_IMAGE_TAG+x}" ]]; then
   export RW_IMAGE="ghcr.io/risingwavelabs/risingwave:${RW_IMAGE_TAG}"
@@ -47,18 +47,24 @@ if [ "${BUILDKITE_SOURCE}" == "schedule" ]; then
   echo Docker image: "$RW_IMAGE"
 fi
 
+if [ -z "${RW_IMAGE+x}" ]; then
+  echo "RW_IMAGE is not set. The image defined in docker-compose.yml will be used."
+fi
+
+echo "--- case: ${case}, format: ${format}"
+
 if [ "${case}" == "client-library" ]; then
   cd integration_tests/client-library
   python3 client_test.py
   exit 0
 fi
 
-echo "--- install postgresql"
+echo "~~~ install postgresql"
 sudo yum install -y postgresql15
 
-echo "--- install poetry"
+echo "~~~ install poetry"
 curl -sSL https://install.python-poetry.org | POETRY_VERSION=1.8.0 python3 -
-
+export PATH=$PATH:$HOME/.local/bin
 
 
 echo "--- download rwctest-key"