Function TEST

bolt/common/base/tests/SpillConfigTest.cpp:40–93

using namespace bytedance::bolt::exec;

TEST(SpillConfig, spillLevel) {
  const uint8_t kInitialBitOffset = 16;
  const uint8_t kNumPartitionsBits = 3;
  static const std::string emptyPath = "";
  GetSpillDirectoryPathCB getPathCb = [&]() -> const std::string& {
    return emptyPath;
  };
  UpdateAndCheckSpillLimitCB checkLimitCb = [&](uint64_t) {};
  const SpillConfig config(
      getPathCb,
      checkLimitCb,
      "fakeSpillPath",
      0,
      false,
      0,
      nullptr,
      0,
      0,
      kInitialBitOffset,
      kNumPartitionsBits,
      0,
      0,
      0,
      0,
      "none");
  struct {
    uint8_t bitOffset;
    // Indicates an invalid if 'expectedLevel' is negative.
    int32_t expectedLevel;

    std::string debugString() const {
      return fmt::format(
          "bitOffset:{}, expectedLevel:{}", bitOffset, expectedLevel);
    }
  } testSettings[] = {
      {0, -1},
      {kInitialBitOffset - 1, -1},
      {kInitialBitOffset - kNumPartitionsBits, -1},
      {kInitialBitOffset, 0},
      {kInitialBitOffset + 1, -1},
      {kInitialBitOffset + kNumPartitionsBits, 1},
      {kInitialBitOffset + 3 * kNumPartitionsBits, 3},
      {kInitialBitOffset + 15 * kNumPartitionsBits, 15},
      {kInitialBitOffset + 16 * kNumPartitionsBits, -1}};
  for (const auto& testData : testSettings) {
    SCOPED_TRACE(testData.debugString());
    if (testData.expectedLevel == -1) {
      ASSERT_ANY_THROW(config.joinSpillLevel(testData.bitOffset));
    } else {
      ASSERT_EQ(
          config.joinSpillLevel(testData.bitOffset), testData.expectedLevel);
    }
  }
}

TEST(SpillConfig, spillLevelLimit) {
  struct {