Function error_msg_startswith

tests/common/test_result_verifier.py:805–843

As defined in `ImpalaServer::QUERY_ERROR_FORMAT`, an error message is expected to has the following form: Query failed:\n \n This function returns True if 1. `actual_msg` contains the error prompt "Query failed:\n", and 2. `expected_msg` follows ri

(actual_msg, expected_msg="", query_id=None)

def error_msg_startswith(actual_msg, expected_msg="", query_id=None):
  """
  As defined in `ImpalaServer::QUERY_ERROR_FORMAT`, an error message is expected to
  has the following form:

    Query <query_id> failed:\n<error_detail>\n

  This function returns True if
  1. `actual_msg` contains the error prompt "Query <query_id> failed:\n", and
  2. `expected_msg` follows right after the error prompt.

  For `query_id`,
  - If the query_id argument is not None, the function checks if the actual error
    message contains exactly the query id.
  - Otherwise, it checks if the `query_id` part in the actual error message matches the
    format using the regular expression.

  `expected_msg` may also be an array of strings, in which case the function checks
  whether the actual error message starts with any of the strings in the array.

  NOTE: Messages of errors such as "Invalid session id" do not contain a query id
  since such an error may occur before a query id is generated.
  """
  if query_id is not None:
    ERROR_PROMPT = "Query " + query_id + " failed:\n"
    start = actual_msg.find(ERROR_PROMPT)
    if start == -1:
      return False
    start += len(ERROR_PROMPT)
  else:
    ERROR_PROMPT = "Query " + QUERY_ID_REGEX + " failed:\n"
    m = re.search(ERROR_PROMPT, actual_msg)
    if m is None:
      return False
    start = m.end()
  for msg in expected_msg if isinstance(expected_msg, list) else [expected_msg]:
    if actual_msg.startswith(msg, start):
      return True
  return False


def error_msg_equal(msg1, msg2):