1 files changed, 2 insertions, 14 deletions
diff --git a/src/traced/probes/common/cpu_freq_info_for_testing.h b/src/traced/probes/common/cpu_freq_info_for_testing.h
index e0e57c2f8..3ba3aa5cd 100644
--- a/src/traced/probes/common/cpu_freq_info_for_testing.h
+++ b/src/traced/probes/common/cpu_freq_info_for_testing.h
@@ -20,31 +20,19 @@
 #include "src/traced/probes/common/cpu_freq_info.h"
 
 #include <memory>
-#include <string>
-#include <vector>
 
-#include "perfetto/ext/base/scoped_file.h"
-#include "perfetto/ext/base/temp_file.h"
+#include "src/base/test/tmp_dir_tree.h"
 
 namespace perfetto {
 
 class CpuFreqInfoForTesting {
  public:
   CpuFreqInfoForTesting();
-  ~CpuFreqInfoForTesting();
 
   std::unique_ptr<CpuFreqInfo> GetInstance();
 
  private:
-  base::TempDir fake_cpu_dir_;
-  std::vector<std::string> dirs_to_remove_;
-  std::vector<std::string> files_to_remove_;
-
-  void AddDir(std::string path);
-  void AddFile(std::string path, std::string content);
-  void RmDir(std::string path);
-  void RmFile(std::string path);
-  std::string AbsolutePath(std::string path);
+  base::TmpDirTree tmpdir_;
 };
 
 }  // namespace perfetto