@@ -103,9 +103,11 @@ def test_pg8000_connection() -> None:
103103 user = os .environ ["POSTGRES_USER" ]
104104 password = os .environ ["POSTGRES_PASS" ]
105105 db = os .environ ["POSTGRES_DB" ]
106- ip_type = os .environ .get ("IP_TYPE" , "public" ) # can be "public", "private" or "psc"
106+ ip_type = os .environ .get ("IP_TYPE" , "public" ) # can be "public", "private" or "psc"
107107
108- engine , connector = create_sqlalchemy_engine (inst_conn_name , user , password , db , ip_type )
108+ engine , connector = create_sqlalchemy_engine (
109+ inst_conn_name , user , password , db , ip_type
110+ )
109111 with engine .connect () as conn :
110112 time = conn .execute (sqlalchemy .text ("SELECT NOW()" )).fetchone ()
111113 conn .commit ()
@@ -120,7 +122,7 @@ def test_lazy_pg8000_connection() -> None:
120122 user = os .environ ["POSTGRES_USER" ]
121123 password = os .environ ["POSTGRES_PASS" ]
122124 db = os .environ ["POSTGRES_DB" ]
123- ip_type = os .environ .get ("IP_TYPE" , "public" ) # can be "public", "private" or "psc"
125+ ip_type = os .environ .get ("IP_TYPE" , "public" ) # can be "public", "private" or "psc"
124126
125127 engine , connector = create_sqlalchemy_engine (
126128 inst_conn_name , user , password , db , ip_type , "lazy"
@@ -139,9 +141,11 @@ def test_CAS_pg8000_connection() -> None:
139141 user = os .environ ["POSTGRES_USER" ]
140142 password = os .environ ["POSTGRES_CAS_PASS" ]
141143 db = os .environ ["POSTGRES_DB" ]
142- ip_type = os .environ .get ("IP_TYPE" , "public" ) # can be "public", "private" or "psc"
144+ ip_type = os .environ .get ("IP_TYPE" , "public" ) # can be "public", "private" or "psc"
143145
144- engine , connector = create_sqlalchemy_engine (inst_conn_name , user , password , db , ip_type )
146+ engine , connector = create_sqlalchemy_engine (
147+ inst_conn_name , user , password , db , ip_type
148+ )
145149 with engine .connect () as conn :
146150 time = conn .execute (sqlalchemy .text ("SELECT NOW()" )).fetchone ()
147151 conn .commit ()
@@ -156,9 +160,11 @@ def test_customer_managed_CAS_pg8000_connection() -> None:
156160 user = os .environ ["POSTGRES_USER" ]
157161 password = os .environ ["POSTGRES_CUSTOMER_CAS_PASS" ]
158162 db = os .environ ["POSTGRES_DB" ]
159- ip_type = os .environ .get ("IP_TYPE" , "public" ) # can be "public", "private" or "psc"
163+ ip_type = os .environ .get ("IP_TYPE" , "public" ) # can be "public", "private" or "psc"
160164
161- engine , connector = create_sqlalchemy_engine (inst_conn_name , user , password , db , ip_type )
165+ engine , connector = create_sqlalchemy_engine (
166+ inst_conn_name , user , password , db , ip_type
167+ )
162168 with engine .connect () as conn :
163169 time = conn .execute (sqlalchemy .text ("SELECT NOW()" )).fetchone ()
164170 conn .commit ()
@@ -173,7 +179,7 @@ def test_custom_SAN_with_dns_pg8000_connection() -> None:
173179 user = os .environ ["POSTGRES_USER" ]
174180 password = os .environ ["POSTGRES_CUSTOMER_CAS_PASS" ]
175181 db = os .environ ["POSTGRES_DB" ]
176- ip_type = os .environ .get ("IP_TYPE" , "public" ) # can be "public", "private" or "psc"
182+ ip_type = os .environ .get ("IP_TYPE" , "public" ) # can be "public", "private" or "psc"
177183
178184 engine , connector = create_sqlalchemy_engine (
179185 inst_conn_name , user , password , db , ip_type , resolver = DnsResolver
0 commit comments