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)