diff options
Diffstat (limited to 'test/testsupport/frame_reader_unittest.cc')
-rw-r--r-- | test/testsupport/frame_reader_unittest.cc | 72 |
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 |