1 files changed, 72 insertions, 0 deletions
diff --git a/test/testsupport/frame_reader_unittest.cc b/test/testsupport/frame_reader_unittest.cc
new file mode 100644
index 0000000000..f1da5ce8cb
--- /dev/null
+++ b/test/testsupport/frame_reader_unittest.cc
@@ -0,0 +1,72 @@
+/*
+ *  Copyright (c) 2011 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 "testsupport/frame_reader.h"
+
+#include "gtest/gtest.h"
+#include "testsupport/fileutils.h"
+
+namespace webrtc {
+namespace test {
+
+const std::string kInputFilename = "temp_inputfile.tmp";
+const std::string kInputFileContents = "baz";
+// Setting the kFrameLength value to a value much larger than the
+// file to test causes the ReadFrame test to fail on Windows.
+const int kFrameLength = 1000;
+
+class FrameReaderTest: public testing::Test {
+ protected:
+  FrameReaderTest() {}
+  virtual ~FrameReaderTest() {}
+  void SetUp() {
+    // Cleanup any previous dummy input file.
+    std::remove(kInputFilename.c_str());
+
+    // Create a dummy input file.
+    FILE* dummy = fopen(kInputFilename.c_str(), "wb");
+    fprintf(dummy, "%s", kInputFileContents.c_str());
+    fclose(dummy);
+
+    frame_reader_ = new FrameReaderImpl(kInputFilename, kFrameLength);
+    ASSERT_TRUE(frame_reader_->Init());
+  }
+  void TearDown() {
+    delete frame_reader_;
+    // Cleanup the dummy input file.
+    std::remove(kInputFilename.c_str());
+  }
+  FrameReader* frame_reader_;
+};
+
+TEST_F(FrameReaderTest, InitSuccess) {
+  FrameReaderImpl frame_reader(kInputFilename, kFrameLength);
+  ASSERT_TRUE(frame_reader.Init());
+  ASSERT_EQ(kFrameLength, frame_reader.FrameLength());
+  ASSERT_EQ(0, frame_reader.NumberOfFrames());
+}
+
+TEST_F(FrameReaderTest, ReadFrame) {
+  WebRtc_UWord8 buffer[3];
+  bool result = frame_reader_->ReadFrame(buffer);
+  ASSERT_FALSE(result);  // No more files to read.
+  ASSERT_EQ(kInputFileContents[0], buffer[0]);
+  ASSERT_EQ(kInputFileContents[1], buffer[1]);
+  ASSERT_EQ(kInputFileContents[2], buffer[2]);
+}
+
+TEST_F(FrameReaderTest, ReadFrameUninitialized) {
+  WebRtc_UWord8 buffer[3];
+  FrameReaderImpl file_reader(kInputFilename, kFrameLength);
+  ASSERT_FALSE(file_reader.ReadFrame(buffer));
+}
+
+}  // namespace test
+}  // namespace webrtc