@@ -84,13 +84,13 @@ def read_json(config_path):
84
84
# We expect these to fail
85
85
if conf ["useAuthentication" ]:
86
86
conn = get_authenticated_https_connection (coordinator_host , "admin" , "admin" , untrusted_ca )
87
- test_query_failure (conn , query , requests .exceptions .SSLError , "Could connect with wrong certificate" )
87
+ test_query_failure (conn , query , trino .exceptions .TrinoConnectionError , "Could connect with wrong certificate" )
88
88
conn = get_authenticated_https_connection (coordinator_host , "admin" , "wrong_password" , trusted_ca )
89
89
test_query_failure (conn , query , trino .exceptions .HttpError , "Could connect with wrong password" )
90
90
conn = get_authenticated_https_connection (coordinator_host , "wrong_user" , "wrong_password" , trusted_ca )
91
91
test_query_failure (conn , query , trino .exceptions .HttpError , "Could connect with wrong user and password" )
92
92
elif conf ["useTls" ]:
93
93
conn = get_https_connection (coordinator_host , "admin" , untrusted_ca )
94
- test_query_failure (conn , query , requests .exceptions .SSLError , "Could connect with wrong certificate" )
94
+ test_query_failure (conn , query , trino .exceptions .TrinoConnectionError , "Could connect with wrong certificate" )
95
95
96
96
print ("All TLS tests finished successfully!" )
0 commit comments