(impala) 03/05: IMPALA-13045: Wait for impala_query_live to exist

[bo...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

boroknagyz pushed a commit to branch branch-4.4.0
in repository https://gitbox.apache.org/repos/asf/impala.git

commit 39233ba3d134b8c18f6f208a7d85c3fadf8ee371
Author: Michael Smith <mi...@cloudera.com>
AuthorDate: Mon Apr 29 11:56:46 2024 -0700

    IMPALA-13045: Wait for impala_query_live to exist
    
    Waits for creation of 'sys.impala_query_live' in tests to ensure it has
    been registered with HMS.
    
    Change-Id: I5cc3fa3c43be7af9a5f097359a0d4f20d057a207
    Reviewed-on: http://gerrit.cloudera.org:8080/21372
    Reviewed-by: Impala Public Jenkins <im...@cloudera.com>
    Tested-by: Michael Smith <mi...@cloudera.com>
    (cherry picked from commit b35aa819653dce062109e61d8f30171234dce5f9)
---
 tests/custom_cluster/test_query_live.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/tests/custom_cluster/test_query_live.py b/tests/custom_cluster/test_query_live.py
index 9629afb5d..58324b23b 100644
--- a/tests/custom_cluster/test_query_live.py
+++ b/tests/custom_cluster/test_query_live.py
@@ -31,6 +31,13 @@ from time import sleep
 class TestQueryLive(CustomClusterTestSuite):
   """Tests to assert the query live table is correctly populated."""
 
+  def setup_method(self, method):
+    super(TestQueryLive, self).setup_method(method)
+    create_match = self.assert_impalad_log_contains("INFO", r'\]\s+(\w+:\w+)\]\s+'
+        r'Analyzing query: CREATE TABLE IF NOT EXISTS sys.impala_query_live')
+    self.assert_impalad_log_contains("INFO", r'Query successfully unregistered: '
+        r'query_id={}'.format(create_match.group(1)))
+
   def assert_impalads(self, profile, present=[0, 1, 2], absent=[]):
     for port_idx in present:
       assert ":" + str(DEFAULT_KRPC_PORT + port_idx) + ":" in profile