|
|
@@ -3010,7 +3010,7 @@ def test_hiveserver2_get_security():
|
|
|
default_query_server = {'server_host': 'my_host', 'server_port': 12345}
|
|
|
|
|
|
# Beeswax
|
|
|
- beeswax_query_server = {'server_name': 'beeswax', 'principal': 'hive', 'auth_username': 'hue', 'auth_password': None}
|
|
|
+ beeswax_query_server = {'server_name': 'beeswax', 'principal': 'hive', 'auth_username': 'hue', 'auth_password': None, 'use_sasl': True}
|
|
|
beeswax_query_server.update(default_query_server)
|
|
|
assert_equal((True, 'PLAIN', 'hive', True, 'hue', None), HiveServerClient(beeswax_query_server, user).get_security())
|
|
|
|
|
|
@@ -3019,22 +3019,25 @@ def test_hiveserver2_get_security():
|
|
|
assert_equal((True, 'PLAIN', 'hive', True, 'hueabcd', 'abcd'), HiveServerClient(beeswax_query_server, user).get_security())
|
|
|
beeswax_query_server.update({'auth_username': 'hue', 'auth_password': None})
|
|
|
|
|
|
- hive_site._HIVE_SITE_DICT[hive_site._CNF_HIVESERVER2_AUTHENTICATION] = 'NOSASL'
|
|
|
- hive_site._HIVE_SITE_DICT[hive_site._CNF_HIVESERVER2_IMPERSONATION] = 'false'
|
|
|
- assert_equal((False, 'NOSASL', 'hive', False, 'hue', None), HiveServerClient(beeswax_query_server, user).get_security())
|
|
|
hive_site._HIVE_SITE_DICT[hive_site._CNF_HIVESERVER2_AUTHENTICATION] = 'KERBEROS'
|
|
|
+ hive_site._HIVE_SITE_DICT[hive_site._CNF_HIVESERVER2_IMPERSONATION] = 'false'
|
|
|
assert_equal((True, 'GSSAPI', 'hive', False, 'hue', None), HiveServerClient(beeswax_query_server, user).get_security())
|
|
|
|
|
|
+ hive_site._HIVE_SITE_DICT[hive_site._CNF_HIVESERVER2_AUTHENTICATION] = 'NOSASL'
|
|
|
+ beeswax_query_server.update({'use_sasl': False})
|
|
|
+ assert_equal((False, 'NOSASL', 'hive', False, 'hue', None), HiveServerClient(beeswax_query_server, user).get_security())
|
|
|
+
|
|
|
+
|
|
|
# Impala
|
|
|
cluster_conf = hadoop.cluster.get_cluster_conf_for_job_submission()
|
|
|
|
|
|
finish = cluster_conf.SECURITY_ENABLED.set_for_testing(False)
|
|
|
try:
|
|
|
- impala_query_server = {'server_name': 'impala', 'principal': 'impala', 'impersonation_enabled': False, 'auth_username': 'hue', 'auth_password': None}
|
|
|
+ impala_query_server = {'server_name': 'impala', 'principal': 'impala', 'impersonation_enabled': False, 'auth_username': 'hue', 'auth_password': None, 'use_sasl': False}
|
|
|
impala_query_server.update(default_query_server)
|
|
|
assert_equal((False, 'GSSAPI', 'impala', False, 'hue', None), HiveServerClient(impala_query_server, user).get_security())
|
|
|
|
|
|
- impala_query_server = {'server_name': 'impala', 'principal': 'impala', 'impersonation_enabled': True, 'auth_username': 'hue', 'auth_password': None}
|
|
|
+ impala_query_server = {'server_name': 'impala', 'principal': 'impala', 'impersonation_enabled': True, 'auth_username': 'hue', 'auth_password': None, 'use_sasl': False}
|
|
|
impala_query_server.update(default_query_server)
|
|
|
assert_equal((False, 'GSSAPI', 'impala', True, 'hue', None), HiveServerClient(impala_query_server, user).get_security())
|
|
|
finally:
|
|
|
@@ -3042,6 +3045,7 @@ def test_hiveserver2_get_security():
|
|
|
|
|
|
finish = cluster_conf.SECURITY_ENABLED.set_for_testing(True)
|
|
|
try:
|
|
|
+ impala_query_server.update({'use_sasl': True})
|
|
|
assert_equal((True, 'GSSAPI', 'impala', True, 'hue', None), HiveServerClient(impala_query_server, user).get_security())
|
|
|
finally:
|
|
|
finish()
|