Function TEST_P

bolt/exec/tests/HashJoinBridgeTest.cpp:182–266

};

TEST_P(HashJoinBridgeTest, withoutSpill) {
  // Test both when morsel-driven is disabled and enabled.
  for (const bool isMorselDriven : {false, true}) {
    for (const bool hasNullKeys : {false, true}) {
      SCOPED_TRACE(fmt::format("hasNullKeys: {}", hasNullKeys));
      auto futures = createEmptyFutures(numProbers_);

      auto joinBridge = createJoinBridge(isMorselDriven);
      // Can't call any other APIs except addBuilder() before start a join
      // bridge first.
      ASSERT_ANY_THROW(
          joinBridge->setHashTable(createFakeHashTable(), {}, false));
      ASSERT_ANY_THROW(joinBridge->setAntiJoinHasNullKeys());
      ASSERT_ANY_THROW(joinBridge->probeFinished());
      ASSERT_ANY_THROW(joinBridge->tableOrFuture(&futures[0]));
      ASSERT_ANY_THROW(joinBridge->spillInputOrFuture(&futures[0]));

      // Can't start a bridge without any builders if morsel-driven is disabled.
      // When morsel-driven is enabled, joinBridge should not throw any error
      // when there is no builder.
      if (!isMorselDriven) {
        ASSERT_ANY_THROW(joinBridge->start());
      }

      joinBridge = createJoinBridge(isMorselDriven);

      for (int32_t i = 0; i < numBuilders_; ++i) {
        joinBridge->addBuilder();
      }
      joinBridge->start();

      for (int32_t i = 0; i < numProbers_; ++i) {
        auto tableOr = joinBridge->tableOrFuture(&futures[i]);
        ASSERT_FALSE(tableOr.has_value());
        ASSERT_TRUE(futures[i].valid());
      }

      BaseHashTable* rawTable = nullptr;
      if (hasNullKeys) {
        joinBridge->setAntiJoinHasNullKeys();
        ASSERT_ANY_THROW(joinBridge->setAntiJoinHasNullKeys());
      } else {
        auto table = createFakeHashTable();
        rawTable = table.get();
        joinBridge->setHashTable(std::move(table), {}, false);
        ASSERT_ANY_THROW(
            joinBridge->setHashTable(createFakeHashTable(), {}, false));
      }

      for (int32_t i = 0; i < numProbers_; ++i) {
        futures[i].wait();
      }

      // Check build results.
      futures = createEmptyFutures(numProbers_ * 2);
      for (int32_t i = 0; i < numProbers_ * 2; ++i) {
        auto tableOr = joinBridge->tableOrFuture(&futures[i]);
        ASSERT_TRUE(tableOr.has_value());