Class DatabaseTest

be/src/service/internal-server-test.cc:124–218

Helper class to set up a uniquely named database. Not every test will need its own database, thus an instance of this class must be instantiated in every test that needs its own database. Upon construction, instances of this class create a database consisting of the name_prefix parameter with the current time appended. Upon destruction, instances of this class drop cascade the database that was c

//
// DO NOT provide a value for record_count that is less than 5.
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,