use pure-sasl

dropbox · Jun 20, 2023 · 78691ce · 78691ce
1 parent e38eb23
commit 78691ce
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyhive/hive.py b/pyhive/hive.py
@@ -60,7 +60,7 @@ def get_sasl_client(host, sasl_auth, service=None, username=None, password=None)
         sasl_client.setAttr('username', username)
         sasl_client.setAttr('password', password)
     else:
-        raise AssertionError
+        raise ValueError("sasl_auth only supports GSSAPI and PLAIN")
 
     sasl_client.init()
     return sasl_client
@@ -74,7 +74,7 @@ def get_pure_sasl_client(host, sasl_auth, service=None, username=None, password=
     elif sasl_auth == 'PLAIN':
         sasl_kwargs = {'username': username, 'password': password}
     else:
-        raise AssertionError
+        raise ValueError("sasl_auth only supports GSSAPI and PLAIN")
 
     return PureSASLClient(host=host, **sasl_kwargs)