Cleanup temporary files created by tests.

webrtc/modules/audio_processing/transient/file_utils_unittest.cc

 /*
  *  Copyright (c) 2013 The WebRTC project authors. All Rights Reserved.
  *
  *  Use of this source code is governed by a BSD-style license
  *  that can be found in the LICENSE file in the root of the source
  *  tree. An additional intellectual property rights grant can be found
  *  in the file PATENTS.  All contributing project authors may
  *  be found in the AUTHORS file in the root of the source tree.
  */
 
 #include "webrtc/modules/audio_processing/transient/file_utils.h"
 
 #include <string.h>
 #include <string>
+#include <vector>
 
 #include "testing/gtest/include/gtest/gtest.h"
 #include "webrtc/base/scoped_ptr.h"
 #include "webrtc/system_wrappers/include/file_wrapper.h"
 #include "webrtc/test/testsupport/fileutils.h"
 #include "webrtc/typedefs.h"
 
 namespace webrtc {
 
 static const uint8_t kPiBytesf[4] = {0xDB, 0x0F, 0x49, 0x40};
@@ skipping 13 matching lines @@
 
 class TransientFileUtilsTest: public ::testing::Test {
  protected:
   TransientFileUtilsTest()
       : kTestFileName(
             test::ResourcePath("audio_processing/transient/double-utils",
                                "dat")),
         kTestFileNamef(
             test::ResourcePath("audio_processing/transient/float-utils",
                                "dat")) {}
+
+  ~TransientFileUtilsTest() override {
+    CleanupTempFiles();
+  }
+
+  std::string CreateTempFilename(const std::string& dir,
+      const std::string& prefix) {
+    std::string filename = test::TempFilename(dir, prefix);
+    temp_filenames_.push_back(filename);
+    return filename;
+  }
+
+  void CleanupTempFiles() {
+    for (const std::string& filename : temp_filenames_) {
+      remove(filename.c_str());
+    }
+    temp_filenames_.clear();
+  }
+
   // This file (used in some tests) contains binary data. The data correspond to
   // the double representation of the constants: Pi, E, and the Avogadro's
   // Number;
   // appended in that order.
   const std::string kTestFileName;
 
   // This file (used in some tests) contains binary data. The data correspond to
   // the float representation of the constants: Pi, E, and the Avogadro's
   // Number;
   // appended in that order.
   const std::string kTestFileNamef;
+
+  // List of temporary filenames created by CreateTempFilename.
+  std::vector<std::string> temp_filenames_;
 };
 
 #if defined(WEBRTC_IOS)
 #define MAYBE_ConvertByteArrayToFloat DISABLED_ConvertByteArrayToFloat
 #else
 #define MAYBE_ConvertByteArrayToFloat ConvertByteArrayToFloat
 #endif
 TEST_F(TransientFileUtilsTest, MAYBE_ConvertByteArrayToFloat) {
   float value = 0.0;
 
@@ skipping 282 matching lines @@
 }
 
 #if defined(WEBRTC_IOS)
 #define MAYBE_WriteInt16BufferToFile DISABLED_WriteInt16BufferToFile
 #else
 #define MAYBE_WriteInt16BufferToFile WriteInt16BufferToFile
 #endif
 TEST_F(TransientFileUtilsTest, MAYBE_WriteInt16BufferToFile) {
   rtc::scoped_ptr<FileWrapper> file(FileWrapper::Create());
 
-  std::string kOutFileName = test::TempFilename(test::OutputPath(),
+  std::string kOutFileName = CreateTempFilename(test::OutputPath(),
                                                 "utils_test");
 
   file->OpenFile(kOutFileName.c_str(),
                  false,   // Write mode.
                  false,   // No loop.
                  false);  // No text.
   ASSERT_TRUE(file->Open()) << "File could not be opened:\n"
       << kOutFileName.c_str();
 
   const size_t kBufferLength = 3;
@@ skipping 26 matching lines @@
 }
 
 #if defined(WEBRTC_IOS)
 #define MAYBE_WriteFloatBufferToFile DISABLED_WriteFloatBufferToFile
 #else
 #define MAYBE_WriteFloatBufferToFile WriteFloatBufferToFile
 #endif
 TEST_F(TransientFileUtilsTest, MAYBE_WriteFloatBufferToFile) {
   rtc::scoped_ptr<FileWrapper> file(FileWrapper::Create());
 
-  std::string kOutFileName = test::TempFilename(test::OutputPath(),
+  std::string kOutFileName = CreateTempFilename(test::OutputPath(),
                                                 "utils_test");
 
   file->OpenFile(kOutFileName.c_str(),
                  false,   // Write mode.
                  false,   // No loop.
                  false);  // No text.
   ASSERT_TRUE(file->Open()) << "File could not be opened:\n"
       << kOutFileName.c_str();
 
   const size_t kBufferLength = 3;
@@ skipping 26 matching lines @@
 }
 
 #if defined(WEBRTC_IOS)
 #define MAYBE_WriteDoubleBufferToFile DISABLED_WriteDoubleBufferToFile
 #else
 #define MAYBE_WriteDoubleBufferToFile WriteDoubleBufferToFile
 #endif
 TEST_F(TransientFileUtilsTest, MAYBE_WriteDoubleBufferToFile) {
   rtc::scoped_ptr<FileWrapper> file(FileWrapper::Create());
 
-  std::string kOutFileName = test::TempFilename(test::OutputPath(),
+  std::string kOutFileName = CreateTempFilename(test::OutputPath(),
                                                 "utils_test");
 
   file->OpenFile(kOutFileName.c_str(),
                  false,   // Write mode.
                  false,   // No loop.
                  false);  // No text.
   ASSERT_TRUE(file->Open()) << "File could not be opened:\n"
       << kOutFileName.c_str();
 
   const size_t kBufferLength = 3;
@@ skipping 77 matching lines @@
   EXPECT_EQ(0u, WriteInt16BufferToFile(file.get(), 1, NULL));
   EXPECT_EQ(0u, WriteInt16BufferToFile(file.get(), 0, int16_buffer.get()));
 
   EXPECT_EQ(0u, WriteDoubleBufferToFile(NULL, 1, double_buffer.get()));
   EXPECT_EQ(0u, WriteDoubleBufferToFile(file.get(), 1, NULL));
   EXPECT_EQ(0u, WriteDoubleBufferToFile(file.get(), 0, double_buffer.get()));
 }
 
 }  // namespace webrtc