|
|
@@ -1198,6 +1198,13 @@ for x in sys.stdin:
|
|
|
assert_equal('localhost', history.server_host)
|
|
|
assert_equal(BEESWAXD_TEST_PORT, history.server_port)
|
|
|
|
|
|
+ query_server = history.get_query_server_config()
|
|
|
+ assert_equal('beeswax', query_server['server_name'])
|
|
|
+ assert_equal('localhost', query_server['server_host'])
|
|
|
+ assert_equal(BEESWAXD_TEST_PORT, query_server['server_port'])
|
|
|
+ assert_equal('beeswax', query_server['server_type'])
|
|
|
+ assert_true(query_server['principal'].startswith('hue/'), query_server['principal'])
|
|
|
+
|
|
|
|
|
|
def test_select_multi_db(self):
|
|
|
response = _make_query(self.client, 'SELECT * FROM test LIMIT 5', local=False, database='default')
|
|
|
@@ -1477,20 +1484,23 @@ def test_search_log_line():
|
|
|
|
|
|
|
|
|
def test_beeswax_get_kerberos_security():
|
|
|
- query_server = {'server_name': 'beeswax'}
|
|
|
- assert_equal((False, 'hue'), BeeswaxClient.get_security(query_server))
|
|
|
+ principal = get_query_server_config('beeswax')['principal']
|
|
|
+ assert_true(principal.startswith('hue/'), principal)
|
|
|
+
|
|
|
+ principal = get_query_server_config('impala')['principal']
|
|
|
+ assert_true(principal.startswith('impala/'), principal)
|
|
|
|
|
|
- query_server = {'server_name': 'impala'}
|
|
|
- assert_equal((False, 'impala'), BeeswaxClient.get_security(query_server))
|
|
|
+ beeswax_query_server = {'server_name': 'beeswax', 'principal': 'hue'}
|
|
|
+ impala_query_server = {'server_name': 'impala', 'principal': 'impala'}
|
|
|
+
|
|
|
+ assert_equal((False, 'hue'), BeeswaxClient.get_security(beeswax_query_server))
|
|
|
+ assert_equal((False, 'impala'), BeeswaxClient.get_security(impala_query_server))
|
|
|
|
|
|
cluster_conf = hadoop.cluster.get_cluster_conf_for_job_submission()
|
|
|
finish = cluster_conf.SECURITY_ENABLED.set_for_testing(True)
|
|
|
try:
|
|
|
- query_server = {'server_name': 'beeswax'}
|
|
|
- assert_equal((True, 'hue'), BeeswaxClient.get_security(query_server))
|
|
|
-
|
|
|
- query_server = {'server_name': 'impala'}
|
|
|
- assert_equal((True, 'impala'), BeeswaxClient.get_security(query_server))
|
|
|
+ assert_equal((True, 'hue'), BeeswaxClient.get_security(beeswax_query_server))
|
|
|
+ assert_equal((True, 'impala'), BeeswaxClient.get_security(impala_query_server))
|
|
|
finally:
|
|
|
finish()
|
|
|
|