Update test workflow to match new example

Author: danielfrankcom

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

@@ -53,19 +53,21 @@ 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
+        python3 -m venv .venv
+        source .venv/bin/activate
         pip install --upgrade pip
         pip install --force-reinstall -r requirements.txt
         python3 -c "import boto3; print(boto3.__version__)"
-        pip install pytest pytest-cov
         pip list
         echo "$GITHUB_WORKSPACE" >> $GITHUB_PATH
-        pytest -v test/
+        wget https://www.amazontrust.com/repository/AmazonRootCA1.pem -O root.pem
+        python src/example.py
         

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)