Skip to content

Commit 8d614b9

Browse files
committed
chore: update lint for all files
1 parent 6d60f5a commit 8d614b9

8 files changed

Lines changed: 52 additions & 34 deletions

tests/system/test_asyncpg_connection.py

Lines changed: 12 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -167,9 +167,11 @@ async def test_sqlalchemy_connection_with_asyncpg() -> None:
167167
user = os.environ["POSTGRES_USER"]
168168
password = os.environ["POSTGRES_PASS"]
169169
db = os.environ["POSTGRES_DB"]
170-
ip_type = os.environ.get("IP_TYPE", "public") # can be "public", "private" or "psc"
170+
ip_type = os.environ.get("IP_TYPE", "public") # can be "public", "private" or "psc"
171171

172-
pool, connector = await create_sqlalchemy_engine(inst_conn_name, user, password, db, ip_type)
172+
pool, connector = await create_sqlalchemy_engine(
173+
inst_conn_name, user, password, db, ip_type
174+
)
173175

174176
async with pool.connect() as conn:
175177
res = (await conn.execute(sqlalchemy.text("SELECT 1"))).fetchone()
@@ -184,7 +186,7 @@ async def test_lazy_sqlalchemy_connection_with_asyncpg() -> None:
184186
user = os.environ["POSTGRES_USER"]
185187
password = os.environ["POSTGRES_PASS"]
186188
db = os.environ["POSTGRES_DB"]
187-
ip_type = os.environ.get("IP_TYPE", "public") # can be "public", "private" or "psc"
189+
ip_type = os.environ.get("IP_TYPE", "public") # can be "public", "private" or "psc"
188190

189191
pool, connector = await create_sqlalchemy_engine(
190192
inst_conn_name, user, password, db, ip_type, "lazy"
@@ -203,10 +205,10 @@ async def test_custom_SAN_with_dns_sqlalchemy_connection_with_asyncpg() -> None:
203205
user = os.environ["POSTGRES_USER"]
204206
password = os.environ["POSTGRES_CUSTOMER_CAS_PASS"]
205207
db = os.environ["POSTGRES_DB"]
206-
ip_type = os.environ.get("IP_TYPE", "public") # can be "public", "private" or "psc"
208+
ip_type = os.environ.get("IP_TYPE", "public") # can be "public", "private" or "psc"
207209

208210
pool, connector = await create_sqlalchemy_engine(
209-
inst_conn_name, user, password, db, ip_type,resolver=DnsResolver
211+
inst_conn_name, user, password, db, ip_type, resolver=DnsResolver
210212
)
211213

212214
async with pool.connect() as conn:
@@ -222,9 +224,11 @@ async def test_connection_with_asyncpg() -> None:
222224
user = os.environ["POSTGRES_USER"]
223225
password = os.environ["POSTGRES_PASS"]
224226
db = os.environ["POSTGRES_DB"]
225-
ip_type = os.environ.get("IP_TYPE", "public") # can be "public", "private" or "psc"
227+
ip_type = os.environ.get("IP_TYPE", "public") # can be "public", "private" or "psc"
226228

227-
pool, connector = await create_asyncpg_pool(inst_conn_name, user, password, db, ip_type)
229+
pool, connector = await create_asyncpg_pool(
230+
inst_conn_name, user, password, db, ip_type
231+
)
228232

229233
async with pool.acquire() as conn:
230234
res = await conn.fetch("SELECT 1")
@@ -239,7 +243,7 @@ async def test_lazy_connection_with_asyncpg() -> None:
239243
user = os.environ["POSTGRES_USER"]
240244
password = os.environ["POSTGRES_PASS"]
241245
db = os.environ["POSTGRES_DB"]
242-
ip_type = os.environ.get("IP_TYPE", "public") # can be "public", "private" or "psc"
246+
ip_type = os.environ.get("IP_TYPE", "public") # can be "public", "private" or "psc"
243247

244248
pool, connector = await create_asyncpg_pool(
245249
inst_conn_name, user, password, db, ip_type, "lazy"

tests/system/test_asyncpg_iam_auth.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ async def test_iam_authn_connection_with_asyncpg() -> None:
8787
inst_conn_name = os.environ["POSTGRES_CONNECTION_NAME"]
8888
user = os.environ["POSTGRES_IAM_USER"]
8989
db = os.environ["POSTGRES_DB"]
90-
ip_type = os.environ.get("IP_TYPE", "public") # can be "public", "private" or "psc"
90+
ip_type = os.environ.get("IP_TYPE", "public") # can be "public", "private" or "psc"
9191

9292
pool, connector = await create_sqlalchemy_engine(inst_conn_name, user, db, ip_type)
9393

@@ -103,9 +103,11 @@ async def test_lazy_iam_authn_connection_with_asyncpg() -> None:
103103
inst_conn_name = os.environ["POSTGRES_CONNECTION_NAME"]
104104
user = os.environ["POSTGRES_IAM_USER"]
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-
pool, connector = await create_sqlalchemy_engine(inst_conn_name, user, db, ip_type, "lazy")
108+
pool, connector = await create_sqlalchemy_engine(
109+
inst_conn_name, user, db, ip_type, "lazy"
110+
)
109111

110112
async with pool.connect() as conn:
111113
res = (await conn.execute(sqlalchemy.text("SELECT 1"))).fetchone()

tests/system/test_connector_object.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -39,9 +39,7 @@ def getconn() -> pymysql.connections.Connection:
3939
user=os.environ["MYSQL_USER"],
4040
password=os.environ["MYSQL_PASS"],
4141
db=os.environ["MYSQL_DB"],
42-
ip_type=os.environ.get(
43-
"IP_TYPE", "public"
44-
),
42+
ip_type=os.environ.get("IP_TYPE", "public"),
4543
)
4644
return conn
4745

tests/system/test_pg8000_connection.py

Lines changed: 14 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -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

tests/system/test_pg8000_iam_auth.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ def test_pg8000_iam_authn_connection() -> None:
8585
inst_conn_name = os.environ["POSTGRES_CONNECTION_NAME"]
8686
user = os.environ["POSTGRES_IAM_USER"]
8787
db = os.environ["POSTGRES_DB"]
88-
ip_type = os.environ.get("IP_TYPE", "public") # can be "public", "private" or "psc"
88+
ip_type = os.environ.get("IP_TYPE", "public") # can be "public", "private" or "psc"
8989

9090
engine, connector = create_sqlalchemy_engine(inst_conn_name, user, db, ip_type)
9191
with engine.connect() as conn:
@@ -101,9 +101,11 @@ def test_lazy_pg8000_iam_authn_connection() -> None:
101101
inst_conn_name = os.environ["POSTGRES_CONNECTION_NAME"]
102102
user = os.environ["POSTGRES_IAM_USER"]
103103
db = os.environ["POSTGRES_DB"]
104-
ip_type = os.environ.get("IP_TYPE", "public") # can be "public", "private" or "psc"
104+
ip_type = os.environ.get("IP_TYPE", "public") # can be "public", "private" or "psc"
105105

106-
engine, connector = create_sqlalchemy_engine(inst_conn_name, user, db, ip_type, "lazy")
106+
engine, connector = create_sqlalchemy_engine(
107+
inst_conn_name, user, db, ip_type, "lazy"
108+
)
107109
with engine.connect() as conn:
108110
time = conn.execute(sqlalchemy.text("SELECT NOW()")).fetchone()
109111
conn.commit()

tests/system/test_pymysql_connection.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -93,9 +93,11 @@ def test_pymysql_connection() -> None:
9393
user = os.environ["MYSQL_USER"]
9494
password = os.environ["MYSQL_PASS"]
9595
db = os.environ["MYSQL_DB"]
96-
ip_type = os.environ.get("IP_TYPE", "public") # can be "public", "private" or "psc"
96+
ip_type = os.environ.get("IP_TYPE", "public") # can be "public", "private" or "psc"
9797

98-
engine, connector = create_sqlalchemy_engine(inst_conn_name, user, password, db, ip_type)
98+
engine, connector = create_sqlalchemy_engine(
99+
inst_conn_name, user, password, db, ip_type
100+
)
99101
with engine.connect() as conn:
100102
time = conn.execute(sqlalchemy.text("SELECT NOW()")).fetchone()
101103
conn.commit()
@@ -110,7 +112,7 @@ def test_lazy_pymysql_connection() -> None:
110112
user = os.environ["MYSQL_USER"]
111113
password = os.environ["MYSQL_PASS"]
112114
db = os.environ["MYSQL_DB"]
113-
ip_type = os.environ.get("IP_TYPE", "public") # can be "public", "private" or "psc"
115+
ip_type = os.environ.get("IP_TYPE", "public") # can be "public", "private" or "psc"
114116

115117
engine, connector = create_sqlalchemy_engine(
116118
inst_conn_name, user, password, db, ip_type, "lazy"

tests/system/test_pymysql_iam_auth.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ def test_pymysql_iam_authn_connection() -> None:
8585
inst_conn_name = os.environ["MYSQL_CONNECTION_NAME"]
8686
user = os.environ["MYSQL_IAM_USER"]
8787
db = os.environ["MYSQL_DB"]
88-
ip_type = os.environ.get("IP_TYPE", "public") # can be "public", "private" or "psc"
88+
ip_type = os.environ.get("IP_TYPE", "public") # can be "public", "private" or "psc"
8989

9090
engine, connector = create_sqlalchemy_engine(inst_conn_name, user, db, ip_type)
9191
with engine.connect() as conn:
@@ -101,9 +101,11 @@ def test_lazy_pymysql_iam_authn_connection() -> None:
101101
inst_conn_name = os.environ["MYSQL_CONNECTION_NAME"]
102102
user = os.environ["MYSQL_IAM_USER"]
103103
db = os.environ["MYSQL_DB"]
104-
ip_type = os.environ.get("IP_TYPE", "public") # can be "public", "private" or "psc"
104+
ip_type = os.environ.get("IP_TYPE", "public") # can be "public", "private" or "psc"
105105

106-
engine, connector = create_sqlalchemy_engine(inst_conn_name, user, db, ip_type, "lazy")
106+
engine, connector = create_sqlalchemy_engine(
107+
inst_conn_name, user, db, ip_type, "lazy"
108+
)
107109
with engine.connect() as conn:
108110
time = conn.execute(sqlalchemy.text("SELECT NOW()")).fetchone()
109111
conn.commit()

tests/system/test_pytds_connection.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -91,9 +91,11 @@ def test_pytds_connection() -> None:
9191
user = os.environ["SQLSERVER_USER"]
9292
password = os.environ["SQLSERVER_PASS"]
9393
db = os.environ["SQLSERVER_DB"]
94-
ip_type = os.environ.get("IP_TYPE", "public") # can be "public", "private" or "psc"
94+
ip_type = os.environ.get("IP_TYPE", "public") # can be "public", "private" or "psc"
9595

96-
engine, connector = create_sqlalchemy_engine(inst_conn_name, user, password, db, ip_type)
96+
engine, connector = create_sqlalchemy_engine(
97+
inst_conn_name, user, password, db, ip_type
98+
)
9799
with engine.connect() as conn:
98100
res = conn.execute(sqlalchemy.text("SELECT 1")).fetchone()
99101
conn.commit()
@@ -107,7 +109,7 @@ def test_lazy_pytds_connection() -> None:
107109
user = os.environ["SQLSERVER_USER"]
108110
password = os.environ["SQLSERVER_PASS"]
109111
db = os.environ["SQLSERVER_DB"]
110-
ip_type = os.environ.get("IP_TYPE", "public") # can be "public", "private" or "psc"
112+
ip_type = os.environ.get("IP_TYPE", "public") # can be "public", "private" or "psc"
111113

112114
engine, connector = create_sqlalchemy_engine(
113115
inst_conn_name, user, password, db, ip_type, "lazy"

0 commit comments

Comments
 (0)