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 |
11 #include <memory> | 11 #include <memory> |
12 | 12 |
13 #include "webrtc/modules/video_coding/include/mock/mock_vcm_callbacks.h" | 13 #include "webrtc/modules/video_coding/include/mock/mock_vcm_callbacks.h" |
14 #include "webrtc/modules/video_coding/include/mock/mock_video_codec_interface.h" | 14 #include "webrtc/modules/video_coding/include/mock/mock_video_codec_interface.h" |
15 #include "webrtc/modules/video_coding/include/video_coding.h" | 15 #include "webrtc/modules/video_coding/include/video_coding.h" |
16 #include "webrtc/modules/video_coding/test/test_util.h" | 16 #include "webrtc/modules/video_coding/test/test_util.h" |
17 #include "webrtc/system_wrappers/include/clock.h" | 17 #include "webrtc/system_wrappers/include/clock.h" |
18 #include "webrtc/test/gmock.h" | 18 #include "webrtc/test/gmock.h" |
19 #include "webrtc/test/gtest.h" | 19 #include "webrtc/test/gtest.h" |
20 | 20 |
21 namespace webrtc { | 21 namespace webrtc { |
22 | 22 |
| 23 using ::testing::_; |
| 24 using ::testing::AllOf; |
| 25 using ::testing::AnyNumber; |
| 26 using ::testing::Args; |
| 27 using ::testing::ElementsAre; |
| 28 using ::testing::Field; |
| 29 using ::testing::NiceMock; |
| 30 using ::testing::Pointee; |
23 using ::testing::Return; | 31 using ::testing::Return; |
24 using ::testing::_; | |
25 using ::testing::ElementsAre; | |
26 using ::testing::AllOf; | |
27 using ::testing::Args; | |
28 using ::testing::Field; | |
29 using ::testing::Pointee; | |
30 using ::testing::NiceMock; | |
31 using ::testing::Sequence; | 32 using ::testing::Sequence; |
32 | 33 |
33 class VCMRobustnessTest : public ::testing::Test { | 34 class VCMRobustnessTest : public ::testing::Test { |
34 protected: | 35 protected: |
35 static const size_t kPayloadLen = 10; | 36 static const size_t kPayloadLen = 10; |
36 | 37 |
37 virtual void SetUp() { | 38 virtual void SetUp() { |
38 clock_.reset(new SimulatedClock(0)); | 39 clock_.reset(new SimulatedClock(0)); |
39 ASSERT_TRUE(clock_.get() != NULL); | 40 ASSERT_TRUE(clock_.get() != NULL); |
40 vcm_.reset(VideoCodingModule::Create(clock_.get(), &event_factory_)); | 41 vcm_.reset(VideoCodingModule::Create(clock_.get(), &event_factory_)); |
41 ASSERT_TRUE(vcm_ != NULL); | 42 ASSERT_TRUE(vcm_ != NULL); |
42 const size_t kMaxNackListSize = 250; | 43 const size_t kMaxNackListSize = 250; |
43 const int kMaxPacketAgeToNack = 450; | 44 const int kMaxPacketAgeToNack = 450; |
44 vcm_->SetNackSettings(kMaxNackListSize, kMaxPacketAgeToNack, 0); | 45 vcm_->SetNackSettings(kMaxNackListSize, kMaxPacketAgeToNack, 0); |
45 ASSERT_EQ(0, vcm_->RegisterFrameTypeCallback(&frame_type_callback_)); | 46 ASSERT_EQ(0, vcm_->RegisterFrameTypeCallback(&frame_type_callback_)); |
46 ASSERT_EQ(0, vcm_->RegisterPacketRequestCallback(&request_callback_)); | 47 ASSERT_EQ(0, vcm_->RegisterPacketRequestCallback(&request_callback_)); |
47 VideoCodingModule::Codec(kVideoCodecVP8, &video_codec_); | 48 VideoCodingModule::Codec(kVideoCodecVP8, &video_codec_); |
48 ASSERT_EQ(VCM_OK, vcm_->RegisterReceiveCodec(&video_codec_, 1)); | 49 ASSERT_EQ(VCM_OK, vcm_->RegisterReceiveCodec(&video_codec_, 1)); |
49 vcm_->RegisterExternalDecoder(&decoder_, video_codec_.plType); | 50 vcm_->RegisterExternalDecoder(&decoder_, video_codec_.plType); |
| 51 |
| 52 // Since we call Decode, we need to provide a valid receive callback. |
| 53 // However, for the purposes of these tests, we ignore the callbacks. |
| 54 EXPECT_CALL(receive_callback_, OnIncomingPayloadType(_)).Times(AnyNumber()); |
| 55 EXPECT_CALL(receive_callback_, OnDecoderImplementationName(_)) |
| 56 .Times(AnyNumber()); |
| 57 vcm_->RegisterReceiveCallback(&receive_callback_); |
50 } | 58 } |
51 | 59 |
52 virtual void TearDown() { vcm_.reset(); } | 60 virtual void TearDown() { vcm_.reset(); } |
53 | 61 |
54 void InsertPacket(uint32_t timestamp, | 62 void InsertPacket(uint32_t timestamp, |
55 uint16_t seq_no, | 63 uint16_t seq_no, |
56 bool first, | 64 bool first, |
57 bool marker_bit, | 65 bool marker_bit, |
58 FrameType frame_type) { | 66 FrameType frame_type) { |
59 const uint8_t payload[kPayloadLen] = {0}; | 67 const uint8_t payload[kPayloadLen] = {0}; |
60 WebRtcRTPHeader rtp_info; | 68 WebRtcRTPHeader rtp_info; |
61 memset(&rtp_info, 0, sizeof(rtp_info)); | 69 memset(&rtp_info, 0, sizeof(rtp_info)); |
62 rtp_info.frameType = frame_type; | 70 rtp_info.frameType = frame_type; |
63 rtp_info.header.timestamp = timestamp; | 71 rtp_info.header.timestamp = timestamp; |
64 rtp_info.header.sequenceNumber = seq_no; | 72 rtp_info.header.sequenceNumber = seq_no; |
65 rtp_info.header.markerBit = marker_bit; | 73 rtp_info.header.markerBit = marker_bit; |
66 rtp_info.header.payloadType = video_codec_.plType; | 74 rtp_info.header.payloadType = video_codec_.plType; |
67 rtp_info.type.Video.codec = kRtpVideoVp8; | 75 rtp_info.type.Video.codec = kRtpVideoVp8; |
68 rtp_info.type.Video.codecHeader.VP8.InitRTPVideoHeaderVP8(); | 76 rtp_info.type.Video.codecHeader.VP8.InitRTPVideoHeaderVP8(); |
69 rtp_info.type.Video.is_first_packet_in_frame = first; | 77 rtp_info.type.Video.is_first_packet_in_frame = first; |
70 | 78 |
71 ASSERT_EQ(VCM_OK, vcm_->IncomingPacket(payload, kPayloadLen, rtp_info)); | 79 ASSERT_EQ(VCM_OK, vcm_->IncomingPacket(payload, kPayloadLen, rtp_info)); |
72 } | 80 } |
73 | 81 |
74 std::unique_ptr<VideoCodingModule> vcm_; | 82 std::unique_ptr<VideoCodingModule> vcm_; |
| 83 MockVCMReceiveCallback receive_callback_; |
75 VideoCodec video_codec_; | 84 VideoCodec video_codec_; |
76 MockVCMFrameTypeCallback frame_type_callback_; | 85 MockVCMFrameTypeCallback frame_type_callback_; |
77 MockPacketRequestCallback request_callback_; | 86 MockPacketRequestCallback request_callback_; |
78 NiceMock<MockVideoDecoder> decoder_; | 87 NiceMock<MockVideoDecoder> decoder_; |
79 NiceMock<MockVideoDecoder> decoderCopy_; | 88 NiceMock<MockVideoDecoder> decoderCopy_; |
80 std::unique_ptr<SimulatedClock> clock_; | 89 std::unique_ptr<SimulatedClock> clock_; |
81 NullEventFactory event_factory_; | 90 NullEventFactory event_factory_; |
82 }; | 91 }; |
83 | 92 |
84 TEST_F(VCMRobustnessTest, TestHardNack) { | 93 TEST_F(VCMRobustnessTest, TestHardNack) { |
(...skipping 134 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
219 | 228 |
220 clock_->AdvanceTimeMilliseconds(23); | 229 clock_->AdvanceTimeMilliseconds(23); |
221 InsertPacket(3000, 4, false, false, kVideoFrameDelta); | 230 InsertPacket(3000, 4, false, false, kVideoFrameDelta); |
222 | 231 |
223 InsertPacket(9000, 9, true, false, kVideoFrameDelta); | 232 InsertPacket(9000, 9, true, false, kVideoFrameDelta); |
224 InsertPacket(9000, 10, false, false, kVideoFrameDelta); | 233 InsertPacket(9000, 10, false, false, kVideoFrameDelta); |
225 InsertPacket(9000, 11, false, true, kVideoFrameDelta); | 234 InsertPacket(9000, 11, false, true, kVideoFrameDelta); |
226 EXPECT_EQ(VCM_OK, vcm_->Decode(33)); // Decode timestamp 9000 complete. | 235 EXPECT_EQ(VCM_OK, vcm_->Decode(33)); // Decode timestamp 9000 complete. |
227 } | 236 } |
228 } // namespace webrtc | 237 } // namespace webrtc |
OLD | NEW |