@@ -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
359360def 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