Method TableWriteTest

bolt/exec/tests/TableWriteTest.cpp:236–280

class TableWriteTest : public HiveConnectorTestBase {
 protected:
  explicit TableWriteTest(uint64_t testValue)
      : testParam_(static_cast<TestParam>(testValue)),
        fileFormat_(testParam_.fileFormat()),
        testMode_(testParam_.testMode()),
        numTableWriterCount_(
            testParam_.multiDrivers() ? kNumTableWriterCount : 1),
        numPartitionedTableWriterCount_(
            testParam_.multiDrivers() ? kNumPartitionedTableWriterCount : 1),
        commitStrategy_(testParam_.commitStrategy()),
        compressionKind_(testParam_.compressionKind()) {
    LOG(INFO) << testParam_.toString();

    auto rowType =
        ROW({"c0", "c1", "c2", "c3", "c4", "c5"},
            {BIGINT(), INTEGER(), SMALLINT(), REAL(), DOUBLE(), VARCHAR()});
    setDataTypes(rowType);
    if (testMode_ == TestMode::kPartitioned ||
        testMode_ == TestMode::kBucketed) {
      const std::vector<std::string> partitionBy = {"c0", "c1"};
      setPartitionBy(partitionBy);
      numPartitionKeyValues_ = {4, 4};
    }
    if (testMode_ == TestMode::kBucketed ||
        testMode_ == TestMode::kOnlyBucketed) {
      std::vector<std::string> bucketedBy = {"c3", "c5"};
      std::vector<TypePtr> bucketedTypes = {REAL(), VARCHAR()};
      std::vector<std::shared_ptr<const HiveSortingColumn>> sortedBy;
      if (testParam_.bucketSort()) {
        // The sortedBy key shouldn't contain partitionBy key.
        sortedBy = {std::make_shared<const HiveSortingColumn>(
            "c4", core::SortOrder{true, true})};
        // The sortColumnIndices_ should represent the indices after removing
        // the partition keys.
        if (testMode_ == TestMode::kBucketed) {
          sortColumnIndices_ = {2};
        } else {
          sortColumnIndices_ = {4};
        }

        sortedFlags_ = {{true, true}};
      }
      bucketProperty_ = std::make_shared<HiveBucketProperty>(
          testParam_.bucketKind(), 4, bucketedBy, bucketedTypes, sortedBy);
    }
  }

  void SetUp() override {
    HiveConnectorTestBase::SetUp();