Chromium Code Reviews| Index: webrtc/video/end_to_end_tests.cc |
| diff --git a/webrtc/video/end_to_end_tests.cc b/webrtc/video/end_to_end_tests.cc |
| index f59044cf01ad61b561d12cf46227a7c223afc4de..525a0299e4d216bfe0208b2e8745a65a2e6badb1 100644 |
| --- a/webrtc/video/end_to_end_tests.cc |
| +++ b/webrtc/video/end_to_end_tests.cc |
| @@ -26,6 +26,7 @@ |
| #include "webrtc/modules/include/module_common_types.h" |
| #include "webrtc/modules/rtp_rtcp/include/rtp_rtcp.h" |
| #include "webrtc/modules/rtp_rtcp/source/byte_io.h" |
| +#include "webrtc/modules/rtp_rtcp/source/rtcp_packet/rapid_resync_request.h" |
| #include "webrtc/modules/rtp_rtcp/source/rtcp_utility.h" |
| #include "webrtc/modules/video_coding/codecs/h264/include/h264.h" |
| #include "webrtc/modules/video_coding/codecs/vp8/include/vp8.h" |
| @@ -111,7 +112,7 @@ class EndToEndTest : public test::CallTest { |
| void RespectsRtcpMode(RtcpMode rtcp_mode); |
| void TestXrReceiverReferenceTimeReport(bool enable_rrtr); |
| void TestSendsSetSsrcs(size_t num_ssrcs, bool send_single_ssrc_first); |
| - void TestRtpStatePreservation(bool use_rtx); |
| + void TestRtpStatePreservation(bool use_rtx, bool wait_rtcp); |
| void VerifyHistogramStats(bool use_rtx, bool use_red, bool screenshare); |
| void VerifyNewVideoSendStreamsRespectNetworkState( |
| MediaType network_to_bring_down, |
| @@ -2871,7 +2872,7 @@ TEST_F(EndToEndTest, DISABLED_RedundantPayloadsTransmittedOnAllSsrcs) { |
| RunBaseTest(&test); |
| } |
| -void EndToEndTest::TestRtpStatePreservation(bool use_rtx) { |
| +void EndToEndTest::TestRtpStatePreservation(bool use_rtx, bool wait_rtcp) { |
| class RtpSequenceObserver : public test::RtpRtcpObserver { |
| public: |
| explicit RtpSequenceObserver(bool use_rtx) |
| @@ -2891,6 +2892,23 @@ void EndToEndTest::TestRtpStatePreservation(bool use_rtx) { |
| } |
| private: |
| + void ValidateTimestampGap(uint32_t ssrc, uint32_t timestamp) |
| + EXCLUSIVE_LOCKS_REQUIRED(crit_) { |
| + static const int32_t kMaxTimestampGap = kDefaultTimeoutMs * 90; |
| + auto timestamp_it = last_observed_timestamp_.find(ssrc); |
| + if (timestamp_it == last_observed_timestamp_.end()) { |
| + last_observed_timestamp_[ssrc] = timestamp; |
| + } else { |
| + // Verify timestamps are reasonably close. |
| + uint32_t latest_observed = timestamp_it->second; |
| + int32_t timestamp_gap = rtc::TimeDiff(timestamp, latest_observed); |
| + EXPECT_LE(std::abs(timestamp_gap), kMaxTimestampGap) |
| + << "Gap in timestamps (" << latest_observed << " -> " << timestamp |
| + << ") too large for SSRC: " << ssrc << "."; |
| + timestamp_it->second = timestamp; |
| + } |
| + } |
| + |
| Action OnSendRtp(const uint8_t* packet, size_t length) override { |
| RTPHeader header; |
| EXPECT_TRUE(parser_->Parse(packet, length, &header)); |
| @@ -2927,21 +2945,9 @@ void EndToEndTest::TestRtpStatePreservation(bool use_rtx) { |
| } |
| } |
| - static const int32_t kMaxTimestampGap = kDefaultTimeoutMs * 90; |
| - auto timestamp_it = last_observed_timestamp_.find(ssrc); |
| - if (timestamp_it == last_observed_timestamp_.end()) { |
| - last_observed_timestamp_[ssrc] = timestamp; |
| - } else { |
| - // Verify timestamps are reasonably close. |
| - uint32_t latest_observed = timestamp_it->second; |
| - int32_t timestamp_gap = rtc::TimeDiff(timestamp, latest_observed); |
| - EXPECT_LE(std::abs(timestamp_gap), kMaxTimestampGap) |
| - << "Gap in timestamps (" << latest_observed << " -> " |
| - << timestamp << ") too large for SSRC: " << ssrc << "."; |
| - timestamp_it->second = timestamp; |
| - } |
| - |
| rtc::CritScope lock(&crit_); |
| + ValidateTimestampGap(ssrc, timestamp); |
| + |
| // Wait for media packets on all ssrcs. |
| if (!ssrc_observed_[ssrc] && !only_padding) { |
| ssrc_observed_[ssrc] = true; |
| @@ -2952,6 +2958,19 @@ void EndToEndTest::TestRtpStatePreservation(bool use_rtx) { |
| return SEND_PACKET; |
| } |
| + Action OnSendRtcp(const uint8_t* packet, size_t length) override { |
| + test::RtcpPacketParser rtcp_parser; |
| + rtcp_parser.Parse(packet, length); |
| + if (rtcp_parser.sender_report()->num_packets() > 0) { |
| + uint32_t ssrc = rtcp_parser.sender_report()->Ssrc(); |
| + uint32_t rtcp_timestamp = rtcp_parser.sender_report()->RtpTimestamp(); |
| + |
| + rtc::CritScope lock(&crit_); |
| + ValidateTimestampGap(ssrc, rtcp_timestamp); |
| + } |
| + return SEND_PACKET; |
| + } |
| + |
| SequenceNumberUnwrapper seq_numbers_unwrapper_; |
| std::map<uint32_t, std::list<int64_t>> last_observed_seq_numbers_; |
| std::map<uint32_t, uint32_t> last_observed_timestamp_; |
| @@ -3021,6 +3040,17 @@ void EndToEndTest::TestRtpStatePreservation(bool use_rtx) { |
| video_send_stream_ = |
| sender_call_->CreateVideoSendStream(video_send_config_, one_stream); |
| video_send_stream_->Start(); |
| + if (wait_rtcp) { |
| + // Wait for SR rtcp packet before generating rtp packets. |
| + // There should be no rtcp packet. |
| + |
| + // Rapid Resync Request forces sending RTCP Sender Report back. |
| + // Alternative approach is to wait several seconds for SR to be generated. |
| + rtcp::RapidResyncRequest rrr; |
|
pbos-webrtc
2016/06/16 11:39:43
call rrr something understandable
danilchap
2016/06/16 12:05:02
Done.
|
| + rtc::Buffer packet = rrr.Build(); |
| + static_cast<webrtc::test::DirectTransport&>(receive_transport) |
| + .SendRtcp(packet.data(), packet.size()); |
| + } |
| CreateFrameGeneratorCapturer(); |
| frame_generator_capturer_->Start(); |
| @@ -3053,11 +3083,15 @@ void EndToEndTest::TestRtpStatePreservation(bool use_rtx) { |
| } |
| TEST_F(EndToEndTest, RestartingSendStreamPreservesRtpState) { |
| - TestRtpStatePreservation(false); |
| + TestRtpStatePreservation(false, false); |
| } |
| TEST_F(EndToEndTest, RestartingSendStreamPreservesRtpStatesWithRtx) { |
| - TestRtpStatePreservation(true); |
| + TestRtpStatePreservation(true, false); |
| +} |
| + |
| +TEST_F(EndToEndTest, RestartingSendStreamKeepsRtpAndRtcpTimestampsSynced) { |
| + TestRtpStatePreservation(true, true); |
| } |
| TEST_F(EndToEndTest, RespectsNetworkState) { |