Method CreateTmpFileMgr

be/src/runtime/tmp-file-mgr-test.cc:115–129

Helper to create a TmpFileMgr and initialise it with InitCustom(). Adds the mgr to 'obj_pool_' for automatic cleanup at the end of each test. Fails the test if InitCustom() fails.

  /// 'obj_pool_' for automatic cleanup at the end of each test. Fails the test if
  /// InitCustom() fails.
  TmpFileMgr* CreateTmpFileMgr(const string& tmp_dirs_spec, bool expect_ok = true,
      HdfsFsCache::HdfsFsMap* hdfs_conn_map = nullptr) {
    // Allocate a new metrics group for each TmpFileMgr so they don't get confused by
    // the pre-existing metrics (TmpFileMgr assumes it's a singleton in product code).
    MetricGroup* metrics = obj_pool_.Add(new MetricGroup(""));
    TmpFileMgr* mgr = obj_pool_.Add(new TmpFileMgr());
    if (hdfs_conn_map != nullptr) mgr->hdfs_conns_ = *hdfs_conn_map;
    Status status = mgr->InitCustom(tmp_dirs_spec, false, "", false, metrics);
    if (expect_ok) {
      EXPECT_TRUE(status.ok());
    } else {
      EXPECT_FALSE(status.ok());
    }
    return mgr;
  }

  /// Check that metric values are consistent with TmpFileMgr state.
  void CheckMetrics(TmpFileMgr* tmp_file_mgr) {