Skip to content

Commit 4671516

Browse files
Improve the readability of the conn view tests (#32529)
1 parent f7d4d98 commit 4671516

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

tests/www/views/test_views_connection.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -344,16 +344,17 @@ def test_duplicate_connection(admin_client):
344344

345345
data = {"action": "mulduplicate", "rowid": [conn1.id, conn3.id]}
346346
resp = admin_client.post("/connection/action_post", data=data, follow_redirects=True)
347-
expected_result = {
347+
assert resp.status_code == 200
348+
349+
expected_connections_ids = {
348350
"test_duplicate_gcp_connection",
349351
"test_duplicate_gcp_connection_copy1",
350352
"test_duplicate_mysql_connection",
351353
"test_duplicate_postgres_connection_copy1",
352354
"test_duplicate_postgres_connection_copy2",
353355
}
354-
response = {conn[0] for conn in session.query(Connection.conn_id).all()}
355-
assert resp.status_code == 200
356-
assert expected_result == response
356+
connections_ids = {conn.conn_id for conn in session.query(Connection.conn_id)}
357+
assert expected_connections_ids == connections_ids
357358

358359

359360
def test_duplicate_connection_error(admin_client):
@@ -380,12 +381,11 @@ def test_duplicate_connection_error(admin_client):
380381

381382
data = {"action": "mulduplicate", "rowid": [connections[0].id]}
382383
resp = admin_client.post("/connection/action_post", data=data, follow_redirects=True)
383-
384-
expected_result = {f"test_duplicate_postgres_connection_copy{i}" for i in range(1, 11)}
385-
386384
assert resp.status_code == 200
387-
response = {conn[0] for conn in session.query(Connection.conn_id).all()}
388-
assert expected_result == response
385+
386+
expected_connections_ids = {f"test_duplicate_postgres_connection_copy{i}" for i in range(1, 11)}
387+
connections_ids = {conn.conn_id for conn in session.query(Connection.conn_id)}
388+
assert expected_connections_ids == connections_ids
389389

390390

391391
@pytest.fixture()

0 commit comments

Comments
 (0)