Update test workflow to match new example

Author: danielfrankcom

.github/workflows/python-psycopg3-integ-tests.yml

@@ -34,6 +34,20 @@ jobs:
     - name: Checkout code
       uses: actions/checkout@v4
 
+    - name: Test IPv4 and IPv6 connectivity
+      env:
+        CLUSTER_ENDPOINT: ${{ secrets.PYTHON_PSYCOPG3_CLUSTER_ENDPOINT }}
+      run: |
+        echo -e "\n=== Testing IPv4 Connection to Cluster ==="
+        nc -zv -4 $CLUSTER_ENDPOINT 5432 || echo "IPv4 connection to cluster failed"
+
+        echo -e "\n=== Testing IPv6 Connection to Cluster ==="
+        nc -zv -6 $CLUSTER_ENDPOINT 5432 || echo "IPv6 connection to cluster failed"
+
+    - name: Configure DNS to prefer IPv4
+      run: |
+        echo "precedence ::ffff:0:0/96  100" | sudo tee -a /etc/gai.conf
+
     - name: Set up Python
       uses: actions/setup-python@v5
       with:
@@ -53,11 +67,13 @@ jobs:
         role-to-assume: ${{ secrets.PYTHON_IAM_ROLE }}
         aws-region: us-east-1
 
-    - name: Configure and run integration for psycopg3
+    - name: Configure and run integration for psycopg3 - admin
       working-directory: ./python/psycopg
       env:
+        CLUSTER_USER: "admin"
         CLUSTER_ENDPOINT: ${{ secrets.PYTHON_PSYCOPG3_CLUSTER_ENDPOINT }}
         REGION: ${{ secrets.PYTHON_PSYCOPG3_CLUSTER_REGION }}
+        FORCE_IPV4: "true"
       run: |
         python3 -m venv psycopg3-integ
         source psycopg3-integ/bin/activate

python/psycopg/requirements.txt

@@ -1,3 +1,3 @@
 boto3>=1.35.74
-psycopg[binary]>=3
+psycopg[binary,pool]>=3
 pytest>=8

python/psycopg/src/example.py

@@ -14,6 +14,7 @@
 import boto3
 import psycopg
 import os
+import socket
 import sys
 
 
@@ -37,6 +38,15 @@ def create_connection(cluster_user, cluster_endpoint, region):
         "password": password_token
     }
 
+
+    if os.environ.get("FORCE_IPV4", False):
+        try:
+            # Get the IPv4 address for the host
+            conn_params["hostaddr"] = socket.gethostbyname(cluster_endpoint)
+        except socket.gaierror:
+            # If DNS resolution fails, continue without hostaddr
+            pass
+
     # Make a connection to the cluster
     conn = psycopg.connect(**conn_params)