Method DatabaseTest

be/src/service/internal-server-test.cc:126–193

class DatabaseTest {
  public:
    DatabaseTest(shared_ptr<ImpalaServer> impala_server, const string& name_prefix,
        const bool create_table = false, const int record_count = 5000) {
      // See the warning on the category_count_ class member definition.
      EXPECT_LE(category_count_, 11);

      impala_server_ = move(impala_server);
      database_name_ = StrCat(name_prefix, "_", GetCurrentTimeMicros());
      TUniqueId query_id;
      EXPECT_OK(impala_server_->ExecuteIgnoreResults("impala", StrCat("create database ",
          database_name_, " comment 'Temporary database created and managed by "
          "internal-server-test'"), {}, false, &query_id));
      assertQueryState(query_id, QUERY_STATE_SUCCESS);

      if (create_table) {
        table_name_ = StrCat(database_name_, ".", "products");
        EXPECT_OK(impala_server_->ExecuteIgnoreResults("impala", StrCat("create table ",
            table_name_, "(id INT, name STRING, first_sold TIMESTAMP, "
            "last_sold TIMESTAMP, price DECIMAL(30, 2)) partitioned by (category INT)"),
            {}, false, &query_id));
        assertQueryState(query_id, QUERY_STATE_SUCCESS);

        // Insert some products that have a last_sold time.
        string sql1 = StrCat("insert into ", table_name_ , "(id,category,name,"
            "first_sold,last_sold,price) values ");

        const int secs_per_year = 31536000; // Seconds in 1 year.
        for (int i=0; i<record_count; i+=2) {
          int cat = (i % category_count_) + 1;
          double price = cat * .01;
          // Calculate a first sold offset.
          uint32_t first_sold = secs_per_year * (cat * cat);
          // Calculate a last sold offset that is a minimum 1 year after first_sold.
          uint32_t last_sold = first_sold - (secs_per_year * cat);
          sql1 += StrCat("(", i, ",", cat, ",'prod_", i,
              "',seconds_sub(now(),", first_sold, "),seconds_sub(now(),", last_sold,
              "),cast(", price, " as DECIMAL(30, 2)))");

          if (i < record_count - 2) {
            sql1 += ",";
          }
        }

        EXPECT_OK(impala_server_->ExecuteIgnoreResults("impala", sql1, {},
            false, &query_id));
        assertQueryState(query_id, QUERY_STATE_SUCCESS);

        // Insert some products that do not have a last_sold time.
        string sql2 = StrCat("insert into ", table_name_, "(id, category,name,first_sold,"
            "price) values ");

        for (int i=1; i<record_count; i+=2) {
          int cat = (i % category_count_) + 1;
          double price = cat * .01;
          // Calculate a sold offset.
          uint32_t sold = secs_per_year * (cat * cat);
          sql2 += StrCat("(", i, ",", cat, ",'prod_", i,
              "',seconds_sub(now(),", sold, "),cast(", price, " as DECIMAL(30, 2)))");