Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(468)

Side by Side Diff: webrtc/modules/rtp_rtcp/source/rtp_sender_unittest.cc

Issue 1623543002: Refactor RtpSender and SSRCDatabase a bit. (Closed) Base URL: https://chromium.googlesource.com/external/webrtc.git@master
Patch Set: Add locking back for voe_auto_test Created 4 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
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 117 matching lines...) Expand 10 before | Expand all | Expand 10 after
128 transport_(), 128 transport_(),
129 kMarkerBit(true) { 129 kMarkerBit(true) {
130 } 130 }
131 131
132 void SetUp() override { SetUpRtpSender(true); } 132 void SetUp() override { SetUpRtpSender(true); }
133 133
134 void SetUpRtpSender(bool pacer) { 134 void SetUpRtpSender(bool pacer) {
135 rtp_sender_.reset(new RTPSender(false, &fake_clock_, &transport_, nullptr, 135 rtp_sender_.reset(new RTPSender(false, &fake_clock_, &transport_, nullptr,
136 pacer ? &mock_paced_sender_ : nullptr, 136 pacer ? &mock_paced_sender_ : nullptr,
137 nullptr, nullptr, nullptr, nullptr, nullptr, 137 nullptr, nullptr, nullptr, nullptr, nullptr,
138 &mock_rtc_event_log_)); 138 &mock_rtc_event_log_, &ssrc_database_));
139 rtp_sender_->SetSequenceNumber(kSeqNum); 139 rtp_sender_->SetSequenceNumber(kSeqNum);
140 } 140 }
141 141
142 // TODO: These variables should be below the methods.
stefan-webrtc 2016/01/23 17:38:40 Add an owner to the todo, or could you simply move
143 SSRCDatabase ssrc_database_;
142 SimulatedClock fake_clock_; 144 SimulatedClock fake_clock_;
143 MockRtcEventLog mock_rtc_event_log_; 145 MockRtcEventLog mock_rtc_event_log_;
144 MockRtpPacketSender mock_paced_sender_; 146 MockRtpPacketSender mock_paced_sender_;
145 rtc::scoped_ptr<RTPSender> rtp_sender_; 147 rtc::scoped_ptr<RTPSender> rtp_sender_;
146 int payload_; 148 int payload_;
147 LoopbackTransportTest transport_; 149 LoopbackTransportTest transport_;
148 const bool kMarkerBit; 150 const bool kMarkerBit;
149 uint8_t packet_[kMaxPacketLength]; 151 uint8_t packet_[kMaxPacketLength];
150 152
151 void VerifyRTPHeaderCommon(const RTPHeader& rtp_header) { 153 void VerifyRTPHeaderCommon(const RTPHeader& rtp_header) {
(...skipping 747 matching lines...) Expand 10 before | Expand all | Expand 10 after
899 EXPECT_EQ(0, rtp_header.extension.transmissionTimeOffset); 901 EXPECT_EQ(0, rtp_header.extension.transmissionTimeOffset);
900 uint64_t expected_send_time = 902 uint64_t expected_send_time =
901 ConvertMsToAbsSendTime(fake_clock_.TimeInMilliseconds()); 903 ConvertMsToAbsSendTime(fake_clock_.TimeInMilliseconds());
902 EXPECT_EQ(expected_send_time, rtp_header.extension.absoluteSendTime); 904 EXPECT_EQ(expected_send_time, rtp_header.extension.absoluteSendTime);
903 } 905 }
904 906
905 TEST_F(RtpSenderTest, SendRedundantPayloads) { 907 TEST_F(RtpSenderTest, SendRedundantPayloads) {
906 MockTransport transport; 908 MockTransport transport;
907 rtp_sender_.reset(new RTPSender( 909 rtp_sender_.reset(new RTPSender(
908 false, &fake_clock_, &transport, nullptr, &mock_paced_sender_, nullptr, 910 false, &fake_clock_, &transport, nullptr, &mock_paced_sender_, nullptr,
909 nullptr, nullptr, nullptr, nullptr, &mock_rtc_event_log_)); 911 nullptr, nullptr, nullptr, nullptr, &mock_rtc_event_log_,
912 &ssrc_database_));
910 rtp_sender_->SetSequenceNumber(kSeqNum); 913 rtp_sender_->SetSequenceNumber(kSeqNum);
911 rtp_sender_->SetRtxPayloadType(kRtxPayload, kPayload); 914 rtp_sender_->SetRtxPayloadType(kRtxPayload, kPayload);
912 915
913 uint16_t seq_num = kSeqNum; 916 uint16_t seq_num = kSeqNum;
914 rtp_sender_->SetStorePacketsStatus(true, 10); 917 rtp_sender_->SetStorePacketsStatus(true, 10);
915 int32_t rtp_header_len = kRtpHeaderSize; 918 int32_t rtp_header_len = kRtpHeaderSize;
916 EXPECT_EQ( 919 EXPECT_EQ(
917 0, rtp_sender_->RegisterRtpHeaderExtension(kRtpExtensionAbsoluteSendTime, 920 0, rtp_sender_->RegisterRtpHeaderExtension(kRtpExtensionAbsoluteSendTime,
918 kAbsoluteSendTimeExtensionId)); 921 kAbsoluteSendTimeExtensionId));
919 rtp_header_len += 4; // 4 bytes extension. 922 rtp_header_len += 4; // 4 bytes extension.
(...skipping 122 matching lines...) Expand 10 before | Expand all | Expand 10 after
1042 frame_counts_ = frame_counts; 1045 frame_counts_ = frame_counts;
1043 } 1046 }
1044 1047
1045 uint32_t num_calls_; 1048 uint32_t num_calls_;
1046 uint32_t ssrc_; 1049 uint32_t ssrc_;
1047 FrameCounts frame_counts_; 1050 FrameCounts frame_counts_;
1048 } callback; 1051 } callback;
1049 1052
1050 rtp_sender_.reset(new RTPSender(false, &fake_clock_, &transport_, nullptr, 1053 rtp_sender_.reset(new RTPSender(false, &fake_clock_, &transport_, nullptr,
1051 &mock_paced_sender_, nullptr, nullptr, 1054 &mock_paced_sender_, nullptr, nullptr,
1052 nullptr, &callback, nullptr, nullptr)); 1055 nullptr, &callback, nullptr, nullptr,
1056 &ssrc_database_));
1053 1057
1054 char payload_name[RTP_PAYLOAD_NAME_SIZE] = "GENERIC"; 1058 char payload_name[RTP_PAYLOAD_NAME_SIZE] = "GENERIC";
1055 const uint8_t payload_type = 127; 1059 const uint8_t payload_type = 127;
1056 ASSERT_EQ(0, rtp_sender_->RegisterPayload(payload_name, payload_type, 90000, 1060 ASSERT_EQ(0, rtp_sender_->RegisterPayload(payload_name, payload_type, 90000,
1057 0, 1500)); 1061 0, 1500));
1058 uint8_t payload[] = {47, 11, 32, 93, 89}; 1062 uint8_t payload[] = {47, 11, 32, 93, 89};
1059 rtp_sender_->SetStorePacketsStatus(true, 1); 1063 rtp_sender_->SetStorePacketsStatus(true, 1);
1060 uint32_t ssrc = rtp_sender_->SSRC(); 1064 uint32_t ssrc = rtp_sender_->SSRC();
1061 1065
1062 EXPECT_CALL(mock_paced_sender_, InsertPacket(_, _, _, _, _, _)) 1066 EXPECT_CALL(mock_paced_sender_, InsertPacket(_, _, _, _, _, _))
(...skipping 35 matching lines...) Expand 10 before | Expand all | Expand 10 after
1098 retransmit_stats_ = retransmit_stats; 1102 retransmit_stats_ = retransmit_stats;
1099 } 1103 }
1100 1104
1101 uint32_t num_calls_; 1105 uint32_t num_calls_;
1102 uint32_t ssrc_; 1106 uint32_t ssrc_;
1103 BitrateStatistics total_stats_; 1107 BitrateStatistics total_stats_;
1104 BitrateStatistics retransmit_stats_; 1108 BitrateStatistics retransmit_stats_;
1105 } callback; 1109 } callback;
1106 rtp_sender_.reset(new RTPSender(false, &fake_clock_, &transport_, nullptr, 1110 rtp_sender_.reset(new RTPSender(false, &fake_clock_, &transport_, nullptr,
1107 nullptr, nullptr, nullptr, &callback, nullptr, 1111 nullptr, nullptr, nullptr, &callback, nullptr,
1108 nullptr, nullptr)); 1112 nullptr, nullptr, &ssrc_database_));
1109 1113
1110 // Simulate kNumPackets sent with kPacketInterval ms intervals. 1114 // Simulate kNumPackets sent with kPacketInterval ms intervals.
1111 const uint32_t kNumPackets = 15; 1115 const uint32_t kNumPackets = 15;
1112 const uint32_t kPacketInterval = 20; 1116 const uint32_t kPacketInterval = 20;
1113 // Overhead = 12 bytes RTP header + 1 byte generic header. 1117 // Overhead = 12 bytes RTP header + 1 byte generic header.
1114 const uint32_t kPacketOverhead = 13; 1118 const uint32_t kPacketOverhead = 13;
1115 1119
1116 char payload_name[RTP_PAYLOAD_NAME_SIZE] = "GENERIC"; 1120 char payload_name[RTP_PAYLOAD_NAME_SIZE] = "GENERIC";
1117 const uint8_t payload_type = 127; 1121 const uint8_t payload_type = 127;
1118 ASSERT_EQ(0, rtp_sender_->RegisterPayload(payload_name, payload_type, 90000, 1122 ASSERT_EQ(0, rtp_sender_->RegisterPayload(payload_name, payload_type, 90000,
(...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after
1151 } 1155 }
1152 1156
1153 class RtpSenderAudioTest : public RtpSenderTest { 1157 class RtpSenderAudioTest : public RtpSenderTest {
1154 protected: 1158 protected:
1155 RtpSenderAudioTest() {} 1159 RtpSenderAudioTest() {}
1156 1160
1157 void SetUp() override { 1161 void SetUp() override {
1158 payload_ = kAudioPayload; 1162 payload_ = kAudioPayload;
1159 rtp_sender_.reset(new RTPSender(true, &fake_clock_, &transport_, nullptr, 1163 rtp_sender_.reset(new RTPSender(true, &fake_clock_, &transport_, nullptr,
1160 nullptr, nullptr, nullptr, nullptr, nullptr, 1164 nullptr, nullptr, nullptr, nullptr, nullptr,
1161 nullptr, nullptr)); 1165 nullptr, nullptr, &ssrc_database_));
1162 rtp_sender_->SetSequenceNumber(kSeqNum); 1166 rtp_sender_->SetSequenceNumber(kSeqNum);
1163 } 1167 }
1164 }; 1168 };
1165 1169
1166 TEST_F(RtpSenderTestWithoutPacer, StreamDataCountersCallbacks) { 1170 TEST_F(RtpSenderTestWithoutPacer, StreamDataCountersCallbacks) {
1167 class TestCallback : public StreamDataCountersCallback { 1171 class TestCallback : public StreamDataCountersCallback {
1168 public: 1172 public:
1169 TestCallback() : StreamDataCountersCallback(), ssrc_(0), counters_() {} 1173 TestCallback() : StreamDataCountersCallback(), ssrc_(0), counters_() {}
1170 virtual ~TestCallback() {} 1174 virtual ~TestCallback() {}
1171 1175
(...skipping 305 matching lines...) Expand 10 before | Expand all | Expand 10 after
1477 reinterpret_cast<uint8_t*>(transport_.sent_packets_[0]->data()), 1481 reinterpret_cast<uint8_t*>(transport_.sent_packets_[0]->data()),
1478 transport_.sent_packets_[0]->size(), true, &map, kSeqNum, hdr.rotation); 1482 transport_.sent_packets_[0]->size(), true, &map, kSeqNum, hdr.rotation);
1479 1483
1480 // Verify that this packet does have CVO byte. 1484 // Verify that this packet does have CVO byte.
1481 VerifyCVOPacket( 1485 VerifyCVOPacket(
1482 reinterpret_cast<uint8_t*>(transport_.sent_packets_[1]->data()), 1486 reinterpret_cast<uint8_t*>(transport_.sent_packets_[1]->data()),
1483 transport_.sent_packets_[1]->size(), true, &map, kSeqNum + 1, 1487 transport_.sent_packets_[1]->size(), true, &map, kSeqNum + 1,
1484 hdr.rotation); 1488 hdr.rotation);
1485 } 1489 }
1486 } // namespace webrtc 1490 } // namespace webrtc
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698