Function TEST

be/src/service/internal-server-test.cc:223–248

Asserts that a timeout error is returned if the query times out during execution. This test must select from a table because selecting directly from the sleep() function does not cause a timeout.

// This test must select from a table because selecting directly from the sleep() function
// does not cause a timeout.
TEST(InternalServerTest, QueryTimeout) {
  DatabaseTest db_test = DatabaseTest(impala_server_, "query_timeout", true, 5);
  InternalServer* fixture = impala_server_.get();

  TUniqueId session_id;
  TUniqueId query_id;

  ASSERT_OK(fixture->OpenSession("impala", session_id,
      {{TImpalaQueryOptions::FETCH_ROWS_TIMEOUT_MS, "1"}}));

  // Run a query that will execute for longer than the configured exec timeout.
  ASSERT_OK(fixture->SubmitQuery(StrCat("select * from ", db_test.GetTableName(),
      " where id=sleep(2000000)"), session_id, query_id));

  Status stat = fixture->WaitForResults(query_id);

  // Assert the expected timeout error was returned.
  EXPECT_FALSE(stat.ok());
  EXPECT_EQ(TErrorCode::GENERAL, stat.code());
  EXPECT_EQ("query timed out waiting for results", stat.msg().msg());

  fixture->CloseQuery(query_id);
  fixture->CloseSession(session_id);

  assertQueryState(query_id, QUERY_STATE_FAILED);
} // TEST QueryTimeout

// Asserts the expected error is returned when a query option is set to an invalid value.
TEST(InternalServerTest, InvalidQueryOption) {