diff --git a/webrtc/modules/modules.gyp b/webrtc/modules/modules.gyp index c1de3b479..939e623fd 100644 --- a/webrtc/modules/modules.gyp +++ b/webrtc/modules/modules.gyp @@ -206,7 +206,8 @@ 'video_coding/main/source/session_info_unittest.cc', 'video_coding/main/source/timing_unittest.cc', 'video_coding/main/source/video_coding_robustness_unittest.cc', - 'video_coding/main/source/video_coding_impl_unittest.cc', + 'video_coding/main/source/video_receiver_unittest.cc', + 'video_coding/main/source/video_sender_unittest.cc', 'video_coding/main/source/qm_select_unittest.cc', 'video_coding/main/source/test/stream_generator.cc', 'video_coding/main/source/test/stream_generator.h', diff --git a/webrtc/modules/video_coding/main/source/video_coding_impl_unittest.cc b/webrtc/modules/video_coding/main/source/video_coding_impl_unittest.cc deleted file mode 100644 index 0a3858296..000000000 --- a/webrtc/modules/video_coding/main/source/video_coding_impl_unittest.cc +++ /dev/null @@ -1,309 +0,0 @@ -/* - * Copyright (c) 2012 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 - -#include "testing/gtest/include/gtest/gtest.h" -#include "webrtc/modules/video_coding/codecs/interface/mock/mock_video_codec_interface.h" -#include "webrtc/modules/video_coding/main/interface/mock/mock_vcm_callbacks.h" -#include "webrtc/modules/video_coding/main/interface/video_coding.h" -#include "webrtc/modules/video_coding/main/test/test_util.h" -#include "webrtc/system_wrappers/interface/clock.h" -#include "webrtc/system_wrappers/interface/scoped_ptr.h" - -using ::testing::_; -using ::testing::AllOf; -using ::testing::ElementsAre; -using ::testing::ElementsAreArray; -using ::testing::Field; -using ::testing::NiceMock; -using ::testing::Pointee; -using ::testing::Return; - -namespace webrtc { - -class TestVideoCodingModule : public ::testing::Test { - protected: - static const int kDefaultWidth = 1280; - static const int kDefaultHeight = 720; - static const int kNumberOfStreams = 3; - static const int kNumberOfLayers = 3; - static const int kUnusedPayloadType = 10; - - virtual void SetUp() { - clock_.reset(new SimulatedClock(0)); - vcm_ = VideoCodingModule::Create(0, clock_.get(), &event_factory_); - EXPECT_EQ(0, vcm_->InitializeReceiver()); - EXPECT_EQ(0, vcm_->InitializeSender()); - EXPECT_EQ(0, vcm_->RegisterExternalEncoder(&encoder_, kUnusedPayloadType, - false)); - EXPECT_EQ(0, vcm_->RegisterExternalDecoder(&decoder_, kUnusedPayloadType, - true)); - const size_t kMaxNackListSize = 250; - const int kMaxPacketAgeToNack = 450; - vcm_->SetNackSettings(kMaxNackListSize, kMaxPacketAgeToNack, 0); - memset(&settings_, 0, sizeof(settings_)); - EXPECT_EQ(0, vcm_->Codec(kVideoCodecVP8, &settings_)); - settings_.numberOfSimulcastStreams = kNumberOfStreams; - ConfigureStream(kDefaultWidth / 4, kDefaultHeight / 4, 100, - &settings_.simulcastStream[0]); - ConfigureStream(kDefaultWidth / 2, kDefaultHeight / 2, 500, - &settings_.simulcastStream[1]); - ConfigureStream(kDefaultWidth, kDefaultHeight, 1200, - &settings_.simulcastStream[2]); - settings_.plType = kUnusedPayloadType; // Use the mocked encoder. - EXPECT_EQ(0, vcm_->RegisterSendCodec(&settings_, 1, 1200)); - EXPECT_EQ(0, vcm_->RegisterReceiveCodec(&settings_, 1, true)); - } - - virtual void TearDown() { - VideoCodingModule::Destroy(vcm_); - } - - void ExpectIntraRequest(int stream) { - if (stream == -1) { - // No intra request expected. - EXPECT_CALL(encoder_, Encode( - _, _, Pointee(ElementsAre(kDeltaFrame, kDeltaFrame, kDeltaFrame)))) - .Times(1) - .WillRepeatedly(Return(0)); - return; - } - assert(stream >= 0); - assert(stream < kNumberOfStreams); - std::vector frame_types(kNumberOfStreams, kDeltaFrame); - frame_types[stream] = kKeyFrame; - EXPECT_CALL(encoder_, Encode( - _, _, Pointee(ElementsAreArray(&frame_types[0], frame_types.size())))) - .Times(1) - .WillRepeatedly(Return(0)); - } - - static void ConfigureStream(int width, int height, int max_bitrate, - SimulcastStream* stream) { - assert(stream); - stream->width = width; - stream->height = height; - stream->maxBitrate = max_bitrate; - stream->numberOfTemporalLayers = kNumberOfLayers; - stream->qpMax = 45; - } - - void InsertAndVerifyPaddingFrame(const uint8_t* payload, int length, - WebRtcRTPHeader* header) { - ASSERT_TRUE(header != NULL); - for (int j = 0; j < 5; ++j) { - // Padding only packets are passed to the VCM with payload size 0. - EXPECT_EQ(0, vcm_->IncomingPacket(payload, 0, *header)); - ++header->header.sequenceNumber; - } - EXPECT_EQ(0, vcm_->Process()); - EXPECT_CALL(decoder_, Decode(_, _, _, _, _)) - .Times(0); - EXPECT_EQ(VCM_FRAME_NOT_READY, vcm_->Decode(0)); - } - - void InsertAndVerifyDecodableFrame(const uint8_t* payload, int length, - WebRtcRTPHeader* header) { - ASSERT_TRUE(header != NULL); - EXPECT_EQ(0, vcm_->IncomingPacket(payload, length, *header)); - ++header->header.sequenceNumber; - EXPECT_CALL(packet_request_callback_, ResendPackets(_, _)) - .Times(0); - EXPECT_EQ(0, vcm_->Process()); - EXPECT_CALL(decoder_, Decode(_, _, _, _, _)) - .Times(1); - EXPECT_EQ(0, vcm_->Decode(0)); - } - - VideoCodingModule* vcm_; - scoped_ptr clock_; - NullEventFactory event_factory_; - NiceMock decoder_; - NiceMock encoder_; - I420VideoFrame input_frame_; - VideoCodec settings_; - NiceMock packet_request_callback_; -}; - -TEST_F(TestVideoCodingModule, TestIntraRequests) { - EXPECT_EQ(0, vcm_->IntraFrameRequest(0)); - ExpectIntraRequest(0); - EXPECT_EQ(0, vcm_->AddVideoFrame(input_frame_, NULL, NULL)); - ExpectIntraRequest(-1); - EXPECT_EQ(0, vcm_->AddVideoFrame(input_frame_, NULL, NULL)); - - EXPECT_EQ(0, vcm_->IntraFrameRequest(1)); - ExpectIntraRequest(1); - EXPECT_EQ(0, vcm_->AddVideoFrame(input_frame_, NULL, NULL)); - ExpectIntraRequest(-1); - EXPECT_EQ(0, vcm_->AddVideoFrame(input_frame_, NULL, NULL)); - - EXPECT_EQ(0, vcm_->IntraFrameRequest(2)); - ExpectIntraRequest(2); - EXPECT_EQ(0, vcm_->AddVideoFrame(input_frame_, NULL, NULL)); - ExpectIntraRequest(-1); - EXPECT_EQ(0, vcm_->AddVideoFrame(input_frame_, NULL, NULL)); - - EXPECT_EQ(-1, vcm_->IntraFrameRequest(3)); - ExpectIntraRequest(-1); - EXPECT_EQ(0, vcm_->AddVideoFrame(input_frame_, NULL, NULL)); - - EXPECT_EQ(-1, vcm_->IntraFrameRequest(-1)); - ExpectIntraRequest(-1); - EXPECT_EQ(0, vcm_->AddVideoFrame(input_frame_, NULL, NULL)); -} - -TEST_F(TestVideoCodingModule, TestIntraRequestsInternalCapture) { - // De-register current external encoder. - EXPECT_EQ(0, vcm_->RegisterExternalEncoder(NULL, kUnusedPayloadType, false)); - // Register encoder with internal capture. - EXPECT_EQ(0, vcm_->RegisterExternalEncoder(&encoder_, kUnusedPayloadType, - true)); - EXPECT_EQ(0, vcm_->RegisterSendCodec(&settings_, 1, 1200)); - ExpectIntraRequest(0); - EXPECT_EQ(0, vcm_->IntraFrameRequest(0)); - ExpectIntraRequest(1); - EXPECT_EQ(0, vcm_->IntraFrameRequest(1)); - ExpectIntraRequest(2); - EXPECT_EQ(0, vcm_->IntraFrameRequest(2)); - // No requests expected since these indices are out of bounds. - EXPECT_EQ(-1, vcm_->IntraFrameRequest(3)); - EXPECT_EQ(-1, vcm_->IntraFrameRequest(-1)); -} - -TEST_F(TestVideoCodingModule, PaddingOnlyFrames) { - EXPECT_EQ(0, vcm_->SetVideoProtection(kProtectionNack, true)); - EXPECT_EQ(0, vcm_->RegisterPacketRequestCallback(&packet_request_callback_)); - const unsigned int kPaddingSize = 220; - const uint8_t payload[kPaddingSize] = {0}; - WebRtcRTPHeader header; - memset(&header, 0, sizeof(header)); - header.frameType = kFrameEmpty; - header.header.markerBit = false; - header.header.paddingLength = kPaddingSize; - header.header.payloadType = kUnusedPayloadType; - header.header.ssrc = 1; - header.header.headerLength = 12; - header.type.Video.codec = kRtpVideoVp8; - for (int i = 0; i < 10; ++i) { - EXPECT_CALL(packet_request_callback_, ResendPackets(_, _)) - .Times(0); - InsertAndVerifyPaddingFrame(payload, 0, &header); - clock_->AdvanceTimeMilliseconds(33); - header.header.timestamp += 3000; - } -} - -TEST_F(TestVideoCodingModule, PaddingOnlyFramesWithLosses) { - EXPECT_EQ(0, vcm_->SetVideoProtection(kProtectionNack, true)); - EXPECT_EQ(0, vcm_->RegisterPacketRequestCallback(&packet_request_callback_)); - const unsigned int kFrameSize = 1200; - const unsigned int kPaddingSize = 220; - const uint8_t payload[kFrameSize] = {0}; - WebRtcRTPHeader header; - memset(&header, 0, sizeof(header)); - header.frameType = kFrameEmpty; - header.header.markerBit = false; - header.header.paddingLength = kPaddingSize; - header.header.payloadType = kUnusedPayloadType; - header.header.ssrc = 1; - header.header.headerLength = 12; - header.type.Video.codec = kRtpVideoVp8; - // Insert one video frame to get one frame decoded. - header.frameType = kVideoFrameKey; - header.type.Video.isFirstPacket = true; - header.header.markerBit = true; - InsertAndVerifyDecodableFrame(payload, kFrameSize, &header); - clock_->AdvanceTimeMilliseconds(33); - header.header.timestamp += 3000; - - header.frameType = kFrameEmpty; - header.type.Video.isFirstPacket = false; - header.header.markerBit = false; - // Insert padding frames. - for (int i = 0; i < 10; ++i) { - // Lose one packet from the 6th frame. - if (i == 5) { - ++header.header.sequenceNumber; - } - // Lose the 4th frame. - if (i == 3) { - header.header.sequenceNumber += 5; - } else { - if (i > 3 && i < 5) { - EXPECT_CALL(packet_request_callback_, ResendPackets(_, 5)) - .Times(1); - } else if (i >= 5) { - EXPECT_CALL(packet_request_callback_, ResendPackets(_, 6)) - .Times(1); - } else { - EXPECT_CALL(packet_request_callback_, ResendPackets(_, _)) - .Times(0); - } - InsertAndVerifyPaddingFrame(payload, 0, &header); - } - clock_->AdvanceTimeMilliseconds(33); - header.header.timestamp += 3000; - } -} - -TEST_F(TestVideoCodingModule, PaddingOnlyAndVideo) { - EXPECT_EQ(0, vcm_->SetVideoProtection(kProtectionNack, true)); - EXPECT_EQ(0, vcm_->RegisterPacketRequestCallback(&packet_request_callback_)); - const unsigned int kFrameSize = 1200; - const unsigned int kPaddingSize = 220; - const uint8_t payload[kFrameSize] = {0}; - WebRtcRTPHeader header; - memset(&header, 0, sizeof(header)); - header.frameType = kFrameEmpty; - header.type.Video.isFirstPacket = false; - header.header.markerBit = false; - header.header.paddingLength = kPaddingSize; - header.header.payloadType = kUnusedPayloadType; - header.header.ssrc = 1; - header.header.headerLength = 12; - header.type.Video.codec = kRtpVideoVp8; - header.type.Video.codecHeader.VP8.pictureId = -1; - header.type.Video.codecHeader.VP8.tl0PicIdx = -1; - for (int i = 0; i < 3; ++i) { - // Insert 2 video frames. - for (int j = 0; j < 2; ++j) { - if (i == 0 && j == 0) // First frame should be a key frame. - header.frameType = kVideoFrameKey; - else - header.frameType = kVideoFrameDelta; - header.type.Video.isFirstPacket = true; - header.header.markerBit = true; - InsertAndVerifyDecodableFrame(payload, kFrameSize, &header); - clock_->AdvanceTimeMilliseconds(33); - header.header.timestamp += 3000; - } - - // Insert 2 padding only frames. - header.frameType = kFrameEmpty; - header.type.Video.isFirstPacket = false; - header.header.markerBit = false; - for (int j = 0; j < 2; ++j) { - InsertAndVerifyPaddingFrame(payload, 0, &header); - clock_->AdvanceTimeMilliseconds(33); - header.header.timestamp += 3000; - } - } -} - -TEST_F(TestVideoCodingModule, ReceiverDelay) { - EXPECT_EQ(0, vcm_->SetMinReceiverDelay(0)); - EXPECT_EQ(0, vcm_->SetMinReceiverDelay(5000)); - EXPECT_EQ(-1, vcm_->SetMinReceiverDelay(-100)); - EXPECT_EQ(-1, vcm_->SetMinReceiverDelay(10010)); -} - -} // namespace webrtc diff --git a/webrtc/modules/video_coding/main/source/video_receiver_unittest.cc b/webrtc/modules/video_coding/main/source/video_receiver_unittest.cc new file mode 100644 index 000000000..4fd524d4b --- /dev/null +++ b/webrtc/modules/video_coding/main/source/video_receiver_unittest.cc @@ -0,0 +1,214 @@ +/* + * 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 + +#include "testing/gtest/include/gtest/gtest.h" +#include "webrtc/modules/video_coding/codecs/interface/mock/mock_video_codec_interface.h" +#include "webrtc/modules/video_coding/main/interface/mock/mock_vcm_callbacks.h" +#include "webrtc/modules/video_coding/main/interface/video_coding.h" +#include "webrtc/modules/video_coding/main/source/video_coding_impl.h" +#include "webrtc/modules/video_coding/main/test/test_util.h" +#include "webrtc/system_wrappers/interface/clock.h" +#include "webrtc/system_wrappers/interface/scoped_ptr.h" + +using ::testing::_; +using ::testing::NiceMock; + +namespace webrtc { +namespace vcm { +namespace { + +class TestVideoReceiver : public ::testing::Test { + protected: + static const int kUnusedPayloadType = 10; + + TestVideoReceiver() : clock_(0) {} + + virtual void SetUp() { + receiver_.reset(new VideoReceiver(0, &clock_, &event_factory_)); + EXPECT_EQ(0, receiver_->InitializeReceiver()); + EXPECT_EQ(0, + receiver_->RegisterExternalDecoder( + &decoder_, kUnusedPayloadType, true)); + const size_t kMaxNackListSize = 250; + const int kMaxPacketAgeToNack = 450; + receiver_->SetNackSettings(kMaxNackListSize, kMaxPacketAgeToNack, 0); + + memset(&settings_, 0, sizeof(settings_)); + EXPECT_EQ(0, VideoCodingModule::Codec(kVideoCodecVP8, &settings_)); + settings_.plType = kUnusedPayloadType; // Use the mocked encoder. + EXPECT_EQ(0, receiver_->RegisterReceiveCodec(&settings_, 1, true)); + } + + void InsertAndVerifyPaddingFrame(const uint8_t* payload, + int length, + WebRtcRTPHeader* header) { + ASSERT_TRUE(header != NULL); + for (int j = 0; j < 5; ++j) { + // Padding only packets are passed to the VCM with payload size 0. + EXPECT_EQ(0, receiver_->IncomingPacket(payload, 0, *header)); + ++header->header.sequenceNumber; + } + EXPECT_EQ(0, receiver_->Process()); + EXPECT_CALL(decoder_, Decode(_, _, _, _, _)).Times(0); + EXPECT_EQ(VCM_FRAME_NOT_READY, receiver_->Decode(0)); + } + + void InsertAndVerifyDecodableFrame(const uint8_t* payload, + int length, + WebRtcRTPHeader* header) { + ASSERT_TRUE(header != NULL); + EXPECT_EQ(0, receiver_->IncomingPacket(payload, length, *header)); + ++header->header.sequenceNumber; + EXPECT_CALL(packet_request_callback_, ResendPackets(_, _)).Times(0); + EXPECT_EQ(0, receiver_->Process()); + EXPECT_CALL(decoder_, Decode(_, _, _, _, _)).Times(1); + EXPECT_EQ(0, receiver_->Decode(0)); + } + + SimulatedClock clock_; + NullEventFactory event_factory_; + VideoCodec settings_; + NiceMock decoder_; + NiceMock packet_request_callback_; + + scoped_ptr receiver_; +}; + +TEST_F(TestVideoReceiver, PaddingOnlyFrames) { + EXPECT_EQ(0, receiver_->SetVideoProtection(kProtectionNack, true)); + EXPECT_EQ( + 0, receiver_->RegisterPacketRequestCallback(&packet_request_callback_)); + const unsigned int kPaddingSize = 220; + const uint8_t payload[kPaddingSize] = {0}; + WebRtcRTPHeader header; + memset(&header, 0, sizeof(header)); + header.frameType = kFrameEmpty; + header.header.markerBit = false; + header.header.paddingLength = kPaddingSize; + header.header.payloadType = kUnusedPayloadType; + header.header.ssrc = 1; + header.header.headerLength = 12; + header.type.Video.codec = kRtpVideoVp8; + for (int i = 0; i < 10; ++i) { + EXPECT_CALL(packet_request_callback_, ResendPackets(_, _)).Times(0); + InsertAndVerifyPaddingFrame(payload, 0, &header); + clock_.AdvanceTimeMilliseconds(33); + header.header.timestamp += 3000; + } +} + +TEST_F(TestVideoReceiver, PaddingOnlyFramesWithLosses) { + EXPECT_EQ(0, receiver_->SetVideoProtection(kProtectionNack, true)); + EXPECT_EQ( + 0, receiver_->RegisterPacketRequestCallback(&packet_request_callback_)); + const unsigned int kFrameSize = 1200; + const unsigned int kPaddingSize = 220; + const uint8_t payload[kFrameSize] = {0}; + WebRtcRTPHeader header; + memset(&header, 0, sizeof(header)); + header.frameType = kFrameEmpty; + header.header.markerBit = false; + header.header.paddingLength = kPaddingSize; + header.header.payloadType = kUnusedPayloadType; + header.header.ssrc = 1; + header.header.headerLength = 12; + header.type.Video.codec = kRtpVideoVp8; + // Insert one video frame to get one frame decoded. + header.frameType = kVideoFrameKey; + header.type.Video.isFirstPacket = true; + header.header.markerBit = true; + InsertAndVerifyDecodableFrame(payload, kFrameSize, &header); + clock_.AdvanceTimeMilliseconds(33); + header.header.timestamp += 3000; + + header.frameType = kFrameEmpty; + header.type.Video.isFirstPacket = false; + header.header.markerBit = false; + // Insert padding frames. + for (int i = 0; i < 10; ++i) { + // Lose one packet from the 6th frame. + if (i == 5) { + ++header.header.sequenceNumber; + } + // Lose the 4th frame. + if (i == 3) { + header.header.sequenceNumber += 5; + } else { + if (i > 3 && i < 5) { + EXPECT_CALL(packet_request_callback_, ResendPackets(_, 5)).Times(1); + } else if (i >= 5) { + EXPECT_CALL(packet_request_callback_, ResendPackets(_, 6)).Times(1); + } else { + EXPECT_CALL(packet_request_callback_, ResendPackets(_, _)).Times(0); + } + InsertAndVerifyPaddingFrame(payload, 0, &header); + } + clock_.AdvanceTimeMilliseconds(33); + header.header.timestamp += 3000; + } +} + +TEST_F(TestVideoReceiver, PaddingOnlyAndVideo) { + EXPECT_EQ(0, receiver_->SetVideoProtection(kProtectionNack, true)); + EXPECT_EQ( + 0, receiver_->RegisterPacketRequestCallback(&packet_request_callback_)); + const unsigned int kFrameSize = 1200; + const unsigned int kPaddingSize = 220; + const uint8_t payload[kFrameSize] = {0}; + WebRtcRTPHeader header; + memset(&header, 0, sizeof(header)); + header.frameType = kFrameEmpty; + header.type.Video.isFirstPacket = false; + header.header.markerBit = false; + header.header.paddingLength = kPaddingSize; + header.header.payloadType = kUnusedPayloadType; + header.header.ssrc = 1; + header.header.headerLength = 12; + header.type.Video.codec = kRtpVideoVp8; + header.type.Video.codecHeader.VP8.pictureId = -1; + header.type.Video.codecHeader.VP8.tl0PicIdx = -1; + for (int i = 0; i < 3; ++i) { + // Insert 2 video frames. + for (int j = 0; j < 2; ++j) { + if (i == 0 && j == 0) // First frame should be a key frame. + header.frameType = kVideoFrameKey; + else + header.frameType = kVideoFrameDelta; + header.type.Video.isFirstPacket = true; + header.header.markerBit = true; + InsertAndVerifyDecodableFrame(payload, kFrameSize, &header); + clock_.AdvanceTimeMilliseconds(33); + header.header.timestamp += 3000; + } + + // Insert 2 padding only frames. + header.frameType = kFrameEmpty; + header.type.Video.isFirstPacket = false; + header.header.markerBit = false; + for (int j = 0; j < 2; ++j) { + // InsertAndVerifyPaddingFrame(payload, 0, &header); + clock_.AdvanceTimeMilliseconds(33); + header.header.timestamp += 3000; + } + } +} + +TEST_F(TestVideoReceiver, ReceiverDelay) { + EXPECT_EQ(0, receiver_->SetMinReceiverDelay(0)); + EXPECT_EQ(0, receiver_->SetMinReceiverDelay(5000)); + EXPECT_EQ(-1, receiver_->SetMinReceiverDelay(-100)); + EXPECT_EQ(-1, receiver_->SetMinReceiverDelay(10010)); +} + +} // namespace +} // namespace vcm +} // namespace webrtc diff --git a/webrtc/modules/video_coding/main/source/video_sender_unittest.cc b/webrtc/modules/video_coding/main/source/video_sender_unittest.cc new file mode 100644 index 000000000..9d6ae2bea --- /dev/null +++ b/webrtc/modules/video_coding/main/source/video_sender_unittest.cc @@ -0,0 +1,168 @@ +/* + * 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 + +#include "testing/gtest/include/gtest/gtest.h" +#include "webrtc/modules/video_coding/codecs/interface/mock/mock_video_codec_interface.h" +#include "webrtc/modules/video_coding/main/interface/mock/mock_vcm_callbacks.h" +#include "webrtc/modules/video_coding/main/interface/video_coding.h" +#include "webrtc/modules/video_coding/main/source/video_coding_impl.h" +#include "webrtc/modules/video_coding/main/test/test_util.h" +#include "webrtc/system_wrappers/interface/clock.h" +#include "webrtc/system_wrappers/interface/scoped_ptr.h" + +using ::testing::_; +using ::testing::AllOf; +using ::testing::ElementsAre; +using ::testing::ElementsAreArray; +using ::testing::Field; +using ::testing::NiceMock; +using ::testing::Pointee; +using ::testing::Return; + +namespace webrtc { +namespace vcm { +namespace { + +class TestVideoSender : public ::testing::Test { + protected: + TestVideoSender() : clock_(0) {} + + virtual void SetUp() { + sender_.reset(new VideoSender(0, &clock_)); + EXPECT_EQ(0, sender_->InitializeSender()); + } + + virtual void TearDown() { sender_.reset(); } + + SimulatedClock clock_; + I420VideoFrame input_frame_; + scoped_ptr sender_; +}; + +class TestVideoSenderWithMockEncoder : public TestVideoSender { + protected: + static const int kDefaultWidth = 1280; + static const int kDefaultHeight = 720; + static const int kNumberOfStreams = 3; + static const int kNumberOfLayers = 3; + static const int kUnusedPayloadType = 10; + + virtual void SetUp() { + TestVideoSender::SetUp(); + EXPECT_EQ( + 0, + sender_->RegisterExternalEncoder(&encoder_, kUnusedPayloadType, false)); + memset(&settings_, 0, sizeof(settings_)); + EXPECT_EQ(0, VideoCodingModule::Codec(kVideoCodecVP8, &settings_)); + settings_.numberOfSimulcastStreams = kNumberOfStreams; + ConfigureStream(kDefaultWidth / 4, + kDefaultHeight / 4, + 100, + &settings_.simulcastStream[0]); + ConfigureStream(kDefaultWidth / 2, + kDefaultHeight / 2, + 500, + &settings_.simulcastStream[1]); + ConfigureStream( + kDefaultWidth, kDefaultHeight, 1200, &settings_.simulcastStream[2]); + settings_.plType = kUnusedPayloadType; // Use the mocked encoder. + EXPECT_EQ(0, sender_->RegisterSendCodec(&settings_, 1, 1200)); + } + + void ExpectIntraRequest(int stream) { + if (stream == -1) { + // No intra request expected. + EXPECT_CALL( + encoder_, + Encode(_, + _, + Pointee(ElementsAre(kDeltaFrame, kDeltaFrame, kDeltaFrame)))) + .Times(1).WillRepeatedly(Return(0)); + return; + } + assert(stream >= 0); + assert(stream < kNumberOfStreams); + std::vector frame_types(kNumberOfStreams, kDeltaFrame); + frame_types[stream] = kKeyFrame; + EXPECT_CALL( + encoder_, + Encode(_, + _, + Pointee(ElementsAreArray(&frame_types[0], frame_types.size())))) + .Times(1).WillRepeatedly(Return(0)); + } + + static void ConfigureStream(int width, + int height, + int max_bitrate, + SimulcastStream* stream) { + assert(stream); + stream->width = width; + stream->height = height; + stream->maxBitrate = max_bitrate; + stream->numberOfTemporalLayers = kNumberOfLayers; + stream->qpMax = 45; + } + + VideoCodec settings_; + NiceMock encoder_; +}; + +TEST_F(TestVideoSenderWithMockEncoder, TestIntraRequests) { + EXPECT_EQ(0, sender_->IntraFrameRequest(0)); + ExpectIntraRequest(0); + EXPECT_EQ(0, sender_->AddVideoFrame(input_frame_, NULL, NULL)); + ExpectIntraRequest(-1); + EXPECT_EQ(0, sender_->AddVideoFrame(input_frame_, NULL, NULL)); + + EXPECT_EQ(0, sender_->IntraFrameRequest(1)); + ExpectIntraRequest(1); + EXPECT_EQ(0, sender_->AddVideoFrame(input_frame_, NULL, NULL)); + ExpectIntraRequest(-1); + EXPECT_EQ(0, sender_->AddVideoFrame(input_frame_, NULL, NULL)); + + EXPECT_EQ(0, sender_->IntraFrameRequest(2)); + ExpectIntraRequest(2); + EXPECT_EQ(0, sender_->AddVideoFrame(input_frame_, NULL, NULL)); + ExpectIntraRequest(-1); + EXPECT_EQ(0, sender_->AddVideoFrame(input_frame_, NULL, NULL)); + + EXPECT_EQ(-1, sender_->IntraFrameRequest(3)); + ExpectIntraRequest(-1); + EXPECT_EQ(0, sender_->AddVideoFrame(input_frame_, NULL, NULL)); + + EXPECT_EQ(-1, sender_->IntraFrameRequest(-1)); + ExpectIntraRequest(-1); + EXPECT_EQ(0, sender_->AddVideoFrame(input_frame_, NULL, NULL)); +} + +TEST_F(TestVideoSenderWithMockEncoder, TestIntraRequestsInternalCapture) { + // De-register current external encoder. + EXPECT_EQ(0, + sender_->RegisterExternalEncoder(NULL, kUnusedPayloadType, false)); + // Register encoder with internal capture. + EXPECT_EQ( + 0, sender_->RegisterExternalEncoder(&encoder_, kUnusedPayloadType, true)); + EXPECT_EQ(0, sender_->RegisterSendCodec(&settings_, 1, 1200)); + ExpectIntraRequest(0); + EXPECT_EQ(0, sender_->IntraFrameRequest(0)); + ExpectIntraRequest(1); + EXPECT_EQ(0, sender_->IntraFrameRequest(1)); + ExpectIntraRequest(2); + EXPECT_EQ(0, sender_->IntraFrameRequest(2)); + // No requests expected since these indices are out of bounds. + EXPECT_EQ(-1, sender_->IntraFrameRequest(3)); + EXPECT_EQ(-1, sender_->IntraFrameRequest(-1)); +} +} // namespace +} // namespace vcm +} // namespace webrtc