Function TEST_F

bolt/exec/tests/TraceUtilTest.cpp:49–94

};

TEST_F(TraceUtilTest, traceDir) {
  const auto outputDir = TempDirectoryPath::create();
  const auto rootDir = outputDir->getPath();
  const auto fs = filesystems::getFileSystem(rootDir, nullptr);
  auto dir1 = fmt::format("{}/{}", outputDir->getPath(), "t1");
  trace::createTraceDirectory(dir1);
  ASSERT_TRUE(fs->exists(dir1));

  auto dir2 = fmt::format("{}/{}", dir1, "t1_1");
  trace::createTraceDirectory(dir2);
  ASSERT_TRUE(fs->exists(dir2));

  // It will remove the old dir1 along with its subdir when created the dir1
  // again.
  trace::createTraceDirectory(dir1);
  ASSERT_TRUE(fs->exists(dir1));
  ASSERT_FALSE(fs->exists(dir2));

  const auto parentDir = fmt::format("{}/{}", outputDir->getPath(), "p");
  fs->mkdir(parentDir);

  constexpr auto numThreads = 5;
  std::vector<std::thread> traceThreads;
  traceThreads.reserve(numThreads);
  std::mutex mutex;
  std::set<std::string> expectedDirs;
  for (int i = 0; i < numThreads; ++i) {
    traceThreads.emplace_back([&, i]() {
      const auto dir = fmt::format("{}/s{}", parentDir, i);
      trace::createTraceDirectory(dir);
      std::lock_guard<std::mutex> l(mutex);
      expectedDirs.insert(dir);
    });
  }

  for (auto& traceThread : traceThreads) {
    traceThread.join();
  }

  const auto actualDirs = fs->list(parentDir);
  ASSERT_EQ(actualDirs.size(), numThreads);
  ASSERT_EQ(actualDirs.size(), expectedDirs.size());
  for (const auto& dir : actualDirs) {
    ASSERT_EQ(expectedDirs.count(dir), 1);
  }
}

TEST_F(TraceUtilTest, OperatorTraceSummary) {
  exec::trace::OperatorTraceSummary summary;