Update setup_db.sh to use CONNECTION_URI

.github/actions/setup-postgres-macos/action.yml

@@ -5,7 +5,6 @@ runs:
   steps:
     - shell: bash
       run: |
-        brew services start postgresql
         echo "Check PostgreSQL service is running"
         i=10
         COMMAND='pg_isready'

.github/actions/setup-postgres-windows/action.yml

@@ -2,6 +2,8 @@ name: "Set up postgres (windows)"
 description: "Set up postgres service on windows vm for dbt integration tests"
 runs:
   using: "composite"
+  env:
+    PQ_LIB_DIR: 'C:\Program Files\PostgreSQL\16\lib'
   steps:
     - shell: pwsh
       run: |

.github/workflows/main.yml

@@ -186,17 +186,29 @@ jobs:
         with:
           python-version: ${{ matrix.python-version }}
 
+      - name: Install postgres 16
+        uses: ikalnytskyi/action-setup-postgres@v6
+        with:
+          postgres-version: "16"
+        id: postgres
+
       - name: Set up postgres (linux)
         if: runner.os == 'Linux'
         uses: ./.github/actions/setup-postgres-linux
+        env:
+          CONNECTION_STR: ${{ steps.postgres.outputs.connection-uri }}
 
       - name: Set up postgres (macos)
         if: runner.os == 'macOS'
         uses: ./.github/actions/setup-postgres-macos
+        env:
+          CONNECTION_STR: ${{ steps.postgres.outputs.connection-uri }}
 
       - name: Set up postgres (windows)
         if: runner.os == 'Windows'
         uses: ./.github/actions/setup-postgres-windows
+        env:
+          CONNECTION_STR: ${{ steps.postgres.outputs.connection-uri }}
 
       - name: Install python tools
         run: |
@@ -213,6 +225,7 @@ jobs:
           command: tox -- --ddtrace
         env:
           PYTEST_ADDOPTS: ${{ format('--splits {0} --group {1}', env.PYTHON_INTEGRATION_TEST_WORKERS, matrix.split-group) }}
+          CONNECTION_STR: ${{ steps.postgres.outputs.connection-uri }}
 
       - name: Get current date
         if: always()

test/setup_db.sh

@@ -5,6 +5,8 @@ env | grep '^PG'
 # If you want to run this script for your own postgresql (run with
 # docker-compose) it will look like this:
 # PGHOST=127.0.0.1 PGUSER=root PGPASSWORD=password PGDATABASE=postgres \
+PG_CONNECTION_URI="${CONNECTION_URI}"
+export PG_CONNECTION_URI
 PGUSER="${PGUSER:-postgres}"
 export PGUSER
 PGPORT="${PGPORT:-5432}"
@@ -15,11 +17,11 @@ function connect_circle() {
 	# try to handle circleci/docker oddness
 	let rc=1
 	while [[ $rc -eq 1 ]]; do
-		nc -z ${PGHOST} ${PGPORT}
+		nc -z ${PG_CONNECTION_URI}
 		let rc=$?
 	done
 	if [[ $rc -ne 0 ]]; then
-		echo "Fatal: Could not connect to $PGHOST"
+		echo "Fatal: Could not connect to $PG_CONNECTION_URI"
 		exit 1
 	fi
 }
@@ -30,7 +32,7 @@ if [[ -n $CIRCLECI ]]; then
 fi
 
 for i in {1..10}; do
-	if pg_isready -h "${PGHOST}" -p "${PGPORT}" -U "${PGUSER}" ; then
+	if pg_isready --d "${PG_CONNECTION_URI}"; then
 		break
 	fi