Function TEST

tensorflow/contrib/ffmpeg/default/ffmpeg_lib_utility_test.cc:33–77

namespace {

TEST(FfmpegLibTest, TestTempDirectoryThreading) {
  // Testing a fix for a bug that allowed different threads to create
  // conflicting temp files.
  // See github.com/tensorflow/tensorflow/issues/5804 for details.
  const int32 kNumThreads = 10;
  const int32 kNumWorkItems = 10000;
  static constexpr size_t kStringsPerItem = 100;
  Env* environment = Env::Default();
  thread::ThreadPool pool(environment, "test", kNumThreads);

  mutex mu;
  std::vector<string> temp_filenames;
  temp_filenames.reserve(kNumWorkItems * kStringsPerItem);

  // Queue a large number of work items for the threads to process. Each work
  // item creates a temp file and then deletes it.
  for (int i = 0; i < kNumWorkItems; ++i) {
    pool.Schedule([&mu, &temp_filenames, environment]() {
      std::array<string, kStringsPerItem> buffer;
      for (int32 j = 0; j < kStringsPerItem; ++j) {
        buffer[j] = io::GetTempFilename("mp3");
        TF_QCHECK_OK(environment->DeleteFile(buffer[j]));
      }
      mutex_lock l(mu);
      for (const auto& fn : buffer) {
        temp_filenames.push_back(fn);
      }
    });
  }

  // Wait until all work items are complete.
  while (true) {
    mutex_lock l(mu);
    if (temp_filenames.size() == kNumWorkItems * kStringsPerItem) {
      break;
    }
  }

  // Check that no duplicates are created.
  std::set<string> unique_filenames;
  mutex_lock l(mu);
  for (const auto& fn : temp_filenames) {
    ASSERT_TRUE(unique_filenames.insert(fn).second);
  }
}

}  // namespace
}  // namespace ffmpeg