test: update ip type for connectors

tests/system/test_asyncpg_connection.py

@@ -87,7 +87,9 @@ async def create_sqlalchemy_engine(
             user=user,
             password=password,
             db=db,
-            ip_type="public",  # can also be "private" or "psc"
+            ip_type=os.environ.get(
+                "IP_TYPE", "public"
+            ),  # can also be "private" or "psc"
         ),
         execution_options={"isolation_level": "AUTOCOMMIT"},
     )
@@ -145,7 +147,9 @@ async def getconn(
             user=user,
             password=password,
             db=db,
-            ip_type="public",  # can also be "private" or "psc",
+            ip_type=os.environ.get(
+                "IP_TYPE", "public"
+            ),  # can also be "private" or "psc",
             **kwargs,
         )
         return conn

tests/system/test_asyncpg_iam_auth.py

@@ -71,7 +71,9 @@ async def create_sqlalchemy_engine(
             "asyncpg",
             user=user,
             db=db,
-            ip_type="public",  # can also be "private" or "psc"
+            ip_type=os.environ.get(
+                "IP_TYPE", "public"
+            ),  # can also be "private" or "psc"
             enable_iam_auth=True,
         ),
         execution_options={"isolation_level": "AUTOCOMMIT"},

tests/system/test_pg8000_connection.py

@@ -85,7 +85,9 @@ def create_sqlalchemy_engine(
             user=user,
             password=password,
             db=db,
-            ip_type="public",  # can also be "private" or "psc"
+            ip_type=os.environ.get(
+                "IP_TYPE", "public"
+            ),  # can also be "private" or "psc"
         ),
     )
     return engine, connector

tests/system/test_pg8000_iam_auth.py

@@ -70,7 +70,9 @@ def create_sqlalchemy_engine(
             "pg8000",
             user=user,
             db=db,
-            ip_type="public",  # can also be "private" or "psc"
+            ip_type=os.environ.get(
+                "IP_TYPE", "public"
+            ),  # can also be "private" or "psc"
             enable_iam_auth=True,
         ),
     )

tests/system/test_pymysql_connection.py

@@ -75,7 +75,9 @@ def create_sqlalchemy_engine(
             user=user,
             password=password,
             db=db,
-            ip_type="public",  # can also be "private" or "psc"
+            ip_type=os.environ.get(
+                "IP_TYPE", "public"
+            ),  # can also be "private" or "psc"
         ),
     )
     return engine, connector

tests/system/test_pymysql_iam_auth.py

@@ -70,7 +70,9 @@ def create_sqlalchemy_engine(
             "pymysql",
             user=user,
             db=db,
-            ip_type="public",  # can also be "private" or "psc"
+            ip_type=os.environ.get(
+                "IP_TYPE", "public"
+            ),  # can also be "private" or "psc"
             enable_iam_auth=True,
         ),
     )

tests/system/test_pytds_connection.py

@@ -73,7 +73,9 @@ def create_sqlalchemy_engine(
             user=user,
             password=password,
             db=db,
-            ip_type="public",  # can also be "private" or "psc"
+            ip_type=os.environ.get(
+                "IP_TYPE", "public"
+            ),  # can also be "private" or "psc"
         ),
     )
     return engine, connector