| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. |
| 3 * | 3 * |
| 4 * Use of this source code is governed by a BSD-style license | 4 * Use of this source code is governed by a BSD-style license |
| 5 * that can be found in the LICENSE file in the root of the source | 5 * that can be found in the LICENSE file in the root of the source |
| 6 * tree. An additional intellectual property rights grant can be found | 6 * tree. An additional intellectual property rights grant can be found |
| 7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
| 8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
| 9 */ | 9 */ |
| 10 | 10 |
| (...skipping 10 matching lines...) Expand all Loading... |
| 21 #include "webrtc/test/testsupport/unittest_utils.h" | 21 #include "webrtc/test/testsupport/unittest_utils.h" |
| 22 #include "webrtc/typedefs.h" | 22 #include "webrtc/typedefs.h" |
| 23 | 23 |
| 24 using ::testing::_; | 24 using ::testing::_; |
| 25 using ::testing::AtLeast; | 25 using ::testing::AtLeast; |
| 26 using ::testing::Return; | 26 using ::testing::Return; |
| 27 | 27 |
| 28 namespace webrtc { | 28 namespace webrtc { |
| 29 namespace test { | 29 namespace test { |
| 30 | 30 |
| 31 // Very basic testing for VideoProcessor. It's mostly tested by running the | 31 namespace { |
| 32 // video_quality_measurement program. | 32 |
| 33 const VideoCodecType kCodecType = kVideoCodecVP8; |
| 34 |
| 35 const int kWidth = 352; |
| 36 const int kHeight = 288; |
| 37 const int kFrameLength = |
| 38 kWidth * kHeight + 2 * ((kWidth + 1) / 2) * ((kHeight + 1) / 2); |
| 39 const int kNumFrames = 2; |
| 40 |
| 41 const int kStartBitrate = 100; |
| 42 const int kStartFramerate = 15; |
| 43 |
| 44 } // namespace |
| 45 |
| 33 class VideoProcessorTest : public testing::Test { | 46 class VideoProcessorTest : public testing::Test { |
| 34 protected: | 47 protected: |
| 35 MockVideoEncoder encoder_mock_; | 48 VideoProcessorTest() { |
| 36 MockVideoDecoder decoder_mock_; | 49 VideoCodingModule::Codec(kCodecType, &codec_settings_); |
| 37 MockFrameReader frame_reader_mock_; | 50 config_.codec_settings = &codec_settings_; |
| 38 MockFrameWriter frame_writer_mock_; | 51 config_.codec_settings->startBitrate = kStartBitrate; |
| 39 MockPacketManipulator packet_manipulator_mock_; | 52 config_.codec_settings->width = kWidth; |
| 40 Stats stats_; | 53 config_.codec_settings->height = kHeight; |
| 54 |
| 55 EXPECT_CALL(frame_reader_mock_, NumberOfFrames()) |
| 56 .WillOnce(Return(kNumFrames)); |
| 57 EXPECT_CALL(frame_reader_mock_, FrameLength()) |
| 58 .WillRepeatedly(Return(kFrameLength)); |
| 59 processor_.reset(new VideoProcessorImpl( |
| 60 &encoder_mock_, &decoder_mock_, &frame_reader_mock_, |
| 61 &frame_writer_mock_, &packet_manipulator_mock_, config_, &stats_, |
| 62 nullptr /* source_frame_writer */, nullptr /* encoded_frame_writer */, |
| 63 nullptr /* decoded_frame_writer */)); |
| 64 |
| 65 Init(); |
| 66 } |
| 67 |
| 68 ~VideoProcessorTest() override { Release(); } |
| 69 |
| 70 void Init() { |
| 71 EXPECT_CALL(encoder_mock_, InitEncode(_, _, _)).Times(1); |
| 72 EXPECT_CALL(encoder_mock_, RegisterEncodeCompleteCallback(_)).Times(1); |
| 73 EXPECT_CALL(decoder_mock_, InitDecode(_, _)).Times(1); |
| 74 EXPECT_CALL(decoder_mock_, RegisterDecodeCompleteCallback(_)).Times(1); |
| 75 |
| 76 processor_->Init(); |
| 77 } |
| 78 |
| 79 void ExpectProcessOneFrame() { |
| 80 EXPECT_CALL(frame_reader_mock_, ReadFrame()) |
| 81 .WillOnce(Return(I420Buffer::Create(50, 50))); |
| 82 EXPECT_CALL(encoder_mock_, Encode(_, _, _)).Times(1); |
| 83 // Cannot EXPECT callbacks, since our MockVideo{En,De}coders do not |
| 84 // generate callbacks. |
| 85 } |
| 86 |
| 87 void Release() { |
| 88 EXPECT_CALL(encoder_mock_, RegisterEncodeCompleteCallback(_)).Times(1); |
| 89 EXPECT_CALL(decoder_mock_, RegisterDecodeCompleteCallback(_)).Times(1); |
| 90 |
| 91 EXPECT_CALL(encoder_mock_, Release()).Times(1); |
| 92 EXPECT_CALL(decoder_mock_, Release()).Times(1); |
| 93 |
| 94 processor_->Release(); |
| 95 } |
| 96 |
| 97 testing::StrictMock<MockVideoEncoder> encoder_mock_; |
| 98 testing::StrictMock<MockVideoDecoder> decoder_mock_; |
| 99 testing::StrictMock<MockFrameReader> frame_reader_mock_; |
| 100 testing::StrictMock<MockFrameWriter> frame_writer_mock_; |
| 101 testing::StrictMock<MockPacketManipulator> packet_manipulator_mock_; |
| 41 TestConfig config_; | 102 TestConfig config_; |
| 42 VideoCodec codec_settings_; | 103 VideoCodec codec_settings_; |
| 43 | 104 Stats stats_; |
| 44 VideoProcessorTest() {} | 105 std::unique_ptr<VideoProcessor> processor_; |
| 45 virtual ~VideoProcessorTest() {} | |
| 46 void SetUp() { | |
| 47 // Get a codec configuration struct and configure it. | |
| 48 VideoCodingModule::Codec(kVideoCodecVP8, &codec_settings_); | |
| 49 config_.codec_settings = &codec_settings_; | |
| 50 config_.codec_settings->startBitrate = 100; | |
| 51 config_.codec_settings->width = 352; | |
| 52 config_.codec_settings->height = 288; | |
| 53 } | |
| 54 void TearDown() {} | |
| 55 | |
| 56 void ExpectInit() { | |
| 57 EXPECT_CALL(encoder_mock_, InitEncode(_, _, _)).Times(1); | |
| 58 EXPECT_CALL(encoder_mock_, RegisterEncodeCompleteCallback(_)) | |
| 59 .Times(AtLeast(1)); | |
| 60 EXPECT_CALL(decoder_mock_, InitDecode(_, _)).Times(1); | |
| 61 EXPECT_CALL(decoder_mock_, RegisterDecodeCompleteCallback(_)) | |
| 62 .Times(AtLeast(1)); | |
| 63 EXPECT_CALL(frame_reader_mock_, NumberOfFrames()).WillRepeatedly(Return(1)); | |
| 64 EXPECT_CALL(frame_reader_mock_, FrameLength()).WillOnce(Return(152064)); | |
| 65 } | |
| 66 }; | 106 }; |
| 67 | 107 |
| 68 TEST_F(VideoProcessorTest, Init) { | 108 TEST_F(VideoProcessorTest, InitRelease) {} |
| 69 ExpectInit(); | 109 |
| 70 VideoProcessorImpl video_processor( | 110 TEST_F(VideoProcessorTest, SetRates) { |
| 71 &encoder_mock_, &decoder_mock_, &frame_reader_mock_, &frame_writer_mock_, | 111 EXPECT_CALL(encoder_mock_, SetRateAllocation(_, kStartFramerate)).Times(1); |
| 72 &packet_manipulator_mock_, config_, &stats_, | 112 processor_->SetRates(kStartBitrate, kStartFramerate); |
| 73 nullptr /* source_frame_writer */, nullptr /* encoded_frame_writer */, | 113 |
| 74 nullptr /* decoded_frame_writer */); | 114 // Frame statistics should be reset after new rates have been set. |
| 75 video_processor.Init(); | 115 EXPECT_EQ(0, processor_->NumberDroppedFrames()); |
| 116 EXPECT_EQ(0, processor_->NumberSpatialResizes()); |
| 76 } | 117 } |
| 77 | 118 |
| 78 TEST_F(VideoProcessorTest, ProcessFrame) { | 119 TEST_F(VideoProcessorTest, ProcessFrames) { |
| 79 ExpectInit(); | 120 ExpectProcessOneFrame(); |
| 80 EXPECT_CALL(encoder_mock_, Encode(_, _, _)).Times(1); | 121 EXPECT_TRUE(processor_->ProcessFrame(0)); |
| 81 EXPECT_CALL(frame_reader_mock_, ReadFrame()) | 122 |
| 82 .WillOnce(Return(I420Buffer::Create(50, 50))); | 123 ExpectProcessOneFrame(); |
| 83 // Since we don't return any callback from the mock, the decoder will not | 124 EXPECT_TRUE(processor_->ProcessFrame(1)); |
| 84 // be more than initialized... | 125 |
| 85 VideoProcessorImpl video_processor( | 126 // End of file. |
| 86 &encoder_mock_, &decoder_mock_, &frame_reader_mock_, &frame_writer_mock_, | 127 EXPECT_CALL(frame_reader_mock_, ReadFrame()).WillOnce(Return(nullptr)); |
| 87 &packet_manipulator_mock_, config_, &stats_, | 128 EXPECT_FALSE(processor_->ProcessFrame(2)); |
| 88 nullptr /* source_frame_writer */, nullptr /* encoded_frame_writer */, | |
| 89 nullptr /* decoded_frame_writer */); | |
| 90 video_processor.Init(); | |
| 91 video_processor.ProcessFrame(0); | |
| 92 } | 129 } |
| 93 | 130 |
| 94 } // namespace test | 131 } // namespace test |
| 95 } // namespace webrtc | 132 } // namespace webrtc |
| OLD | NEW |