OLD | NEW |
| (Empty) |
1 /* | |
2 * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. | |
3 * | |
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 | |
6 * tree. An additional intellectual property rights grant can be found | |
7 * in the file PATENTS. All contributing project authors may | |
8 * be found in the AUTHORS file in the root of the source tree. | |
9 */ | |
10 | |
11 #include "webrtc/call/rtc_event_log_unittest_helper.h" | |
12 | |
13 #include <string.h> | |
14 | |
15 #include <string> | |
16 | |
17 #include "webrtc/test/gtest.h" | |
18 #include "webrtc/base/checks.h" | |
19 #include "webrtc/test/test_suite.h" | |
20 #include "webrtc/test/testsupport/fileutils.h" | |
21 | |
22 // Files generated at build-time by the protobuf compiler. | |
23 #ifdef WEBRTC_ANDROID_PLATFORM_BUILD | |
24 #include "external/webrtc/webrtc/call/rtc_event_log.pb.h" | |
25 #else | |
26 #include "webrtc/call/rtc_event_log.pb.h" | |
27 #endif | |
28 | |
29 namespace webrtc { | |
30 | |
31 namespace { | |
32 MediaType GetRuntimeMediaType(rtclog::MediaType media_type) { | |
33 switch (media_type) { | |
34 case rtclog::MediaType::ANY: | |
35 return MediaType::ANY; | |
36 case rtclog::MediaType::AUDIO: | |
37 return MediaType::AUDIO; | |
38 case rtclog::MediaType::VIDEO: | |
39 return MediaType::VIDEO; | |
40 case rtclog::MediaType::DATA: | |
41 return MediaType::DATA; | |
42 } | |
43 RTC_NOTREACHED(); | |
44 return MediaType::ANY; | |
45 } | |
46 } // namespace | |
47 | |
48 // Checks that the event has a timestamp, a type and exactly the data field | |
49 // corresponding to the type. | |
50 ::testing::AssertionResult IsValidBasicEvent(const rtclog::Event& event) { | |
51 if (!event.has_timestamp_us()) { | |
52 return ::testing::AssertionFailure() << "Event has no timestamp"; | |
53 } | |
54 if (!event.has_type()) { | |
55 return ::testing::AssertionFailure() << "Event has no event type"; | |
56 } | |
57 rtclog::Event_EventType type = event.type(); | |
58 if ((type == rtclog::Event::RTP_EVENT) != event.has_rtp_packet()) { | |
59 return ::testing::AssertionFailure() | |
60 << "Event of type " << type << " has " | |
61 << (event.has_rtp_packet() ? "" : "no ") << "RTP packet"; | |
62 } | |
63 if ((type == rtclog::Event::RTCP_EVENT) != event.has_rtcp_packet()) { | |
64 return ::testing::AssertionFailure() | |
65 << "Event of type " << type << " has " | |
66 << (event.has_rtcp_packet() ? "" : "no ") << "RTCP packet"; | |
67 } | |
68 if ((type == rtclog::Event::AUDIO_PLAYOUT_EVENT) != | |
69 event.has_audio_playout_event()) { | |
70 return ::testing::AssertionFailure() | |
71 << "Event of type " << type << " has " | |
72 << (event.has_audio_playout_event() ? "" : "no ") | |
73 << "audio_playout event"; | |
74 } | |
75 if ((type == rtclog::Event::VIDEO_RECEIVER_CONFIG_EVENT) != | |
76 event.has_video_receiver_config()) { | |
77 return ::testing::AssertionFailure() | |
78 << "Event of type " << type << " has " | |
79 << (event.has_video_receiver_config() ? "" : "no ") | |
80 << "receiver config"; | |
81 } | |
82 if ((type == rtclog::Event::VIDEO_SENDER_CONFIG_EVENT) != | |
83 event.has_video_sender_config()) { | |
84 return ::testing::AssertionFailure() | |
85 << "Event of type " << type << " has " | |
86 << (event.has_video_sender_config() ? "" : "no ") << "sender config"; | |
87 } | |
88 if ((type == rtclog::Event::AUDIO_RECEIVER_CONFIG_EVENT) != | |
89 event.has_audio_receiver_config()) { | |
90 return ::testing::AssertionFailure() | |
91 << "Event of type " << type << " has " | |
92 << (event.has_audio_receiver_config() ? "" : "no ") | |
93 << "audio receiver config"; | |
94 } | |
95 if ((type == rtclog::Event::AUDIO_SENDER_CONFIG_EVENT) != | |
96 event.has_audio_sender_config()) { | |
97 return ::testing::AssertionFailure() | |
98 << "Event of type " << type << " has " | |
99 << (event.has_audio_sender_config() ? "" : "no ") | |
100 << "audio sender config"; | |
101 } | |
102 return ::testing::AssertionSuccess(); | |
103 } | |
104 | |
105 void RtcEventLogTestHelper::VerifyReceiveStreamConfig( | |
106 const ParsedRtcEventLog& parsed_log, | |
107 size_t index, | |
108 const VideoReceiveStream::Config& config) { | |
109 const rtclog::Event& event = parsed_log.events_[index]; | |
110 ASSERT_TRUE(IsValidBasicEvent(event)); | |
111 ASSERT_EQ(rtclog::Event::VIDEO_RECEIVER_CONFIG_EVENT, event.type()); | |
112 const rtclog::VideoReceiveConfig& receiver_config = | |
113 event.video_receiver_config(); | |
114 // Check SSRCs. | |
115 ASSERT_TRUE(receiver_config.has_remote_ssrc()); | |
116 EXPECT_EQ(config.rtp.remote_ssrc, receiver_config.remote_ssrc()); | |
117 ASSERT_TRUE(receiver_config.has_local_ssrc()); | |
118 EXPECT_EQ(config.rtp.local_ssrc, receiver_config.local_ssrc()); | |
119 // Check RTCP settings. | |
120 ASSERT_TRUE(receiver_config.has_rtcp_mode()); | |
121 if (config.rtp.rtcp_mode == RtcpMode::kCompound) { | |
122 EXPECT_EQ(rtclog::VideoReceiveConfig::RTCP_COMPOUND, | |
123 receiver_config.rtcp_mode()); | |
124 } else { | |
125 EXPECT_EQ(rtclog::VideoReceiveConfig::RTCP_REDUCEDSIZE, | |
126 receiver_config.rtcp_mode()); | |
127 } | |
128 ASSERT_TRUE(receiver_config.has_remb()); | |
129 EXPECT_EQ(config.rtp.remb, receiver_config.remb()); | |
130 // Check RTX map. | |
131 ASSERT_EQ(static_cast<int>(config.rtp.rtx.size()), | |
132 receiver_config.rtx_map_size()); | |
133 for (const rtclog::RtxMap& rtx_map : receiver_config.rtx_map()) { | |
134 ASSERT_TRUE(rtx_map.has_payload_type()); | |
135 ASSERT_TRUE(rtx_map.has_config()); | |
136 EXPECT_EQ(1u, config.rtp.rtx.count(rtx_map.payload_type())); | |
137 const rtclog::RtxConfig& rtx_config = rtx_map.config(); | |
138 const VideoReceiveStream::Config::Rtp::Rtx& rtx = | |
139 config.rtp.rtx.at(rtx_map.payload_type()); | |
140 ASSERT_TRUE(rtx_config.has_rtx_ssrc()); | |
141 ASSERT_TRUE(rtx_config.has_rtx_payload_type()); | |
142 EXPECT_EQ(rtx.ssrc, rtx_config.rtx_ssrc()); | |
143 EXPECT_EQ(rtx.payload_type, rtx_config.rtx_payload_type()); | |
144 } | |
145 // Check header extensions. | |
146 ASSERT_EQ(static_cast<int>(config.rtp.extensions.size()), | |
147 receiver_config.header_extensions_size()); | |
148 for (int i = 0; i < receiver_config.header_extensions_size(); i++) { | |
149 ASSERT_TRUE(receiver_config.header_extensions(i).has_name()); | |
150 ASSERT_TRUE(receiver_config.header_extensions(i).has_id()); | |
151 const std::string& name = receiver_config.header_extensions(i).name(); | |
152 int id = receiver_config.header_extensions(i).id(); | |
153 EXPECT_EQ(config.rtp.extensions[i].id, id); | |
154 EXPECT_EQ(config.rtp.extensions[i].uri, name); | |
155 } | |
156 // Check decoders. | |
157 ASSERT_EQ(static_cast<int>(config.decoders.size()), | |
158 receiver_config.decoders_size()); | |
159 for (int i = 0; i < receiver_config.decoders_size(); i++) { | |
160 ASSERT_TRUE(receiver_config.decoders(i).has_name()); | |
161 ASSERT_TRUE(receiver_config.decoders(i).has_payload_type()); | |
162 const std::string& decoder_name = receiver_config.decoders(i).name(); | |
163 int decoder_type = receiver_config.decoders(i).payload_type(); | |
164 EXPECT_EQ(config.decoders[i].payload_name, decoder_name); | |
165 EXPECT_EQ(config.decoders[i].payload_type, decoder_type); | |
166 } | |
167 | |
168 // Check consistency of the parser. | |
169 VideoReceiveStream::Config parsed_config(nullptr); | |
170 parsed_log.GetVideoReceiveConfig(index, &parsed_config); | |
171 EXPECT_EQ(config.rtp.remote_ssrc, parsed_config.rtp.remote_ssrc); | |
172 EXPECT_EQ(config.rtp.local_ssrc, parsed_config.rtp.local_ssrc); | |
173 // Check RTCP settings. | |
174 EXPECT_EQ(config.rtp.rtcp_mode, parsed_config.rtp.rtcp_mode); | |
175 EXPECT_EQ(config.rtp.remb, parsed_config.rtp.remb); | |
176 // Check RTX map. | |
177 EXPECT_EQ(config.rtp.rtx.size(), parsed_config.rtp.rtx.size()); | |
178 for (const auto& kv : config.rtp.rtx) { | |
179 auto parsed_kv = parsed_config.rtp.rtx.find(kv.first); | |
180 EXPECT_EQ(kv.first, parsed_kv->first); | |
181 EXPECT_EQ(kv.second.ssrc, parsed_kv->second.ssrc); | |
182 EXPECT_EQ(kv.second.payload_type, parsed_kv->second.payload_type); | |
183 } | |
184 // Check header extensions. | |
185 EXPECT_EQ(config.rtp.extensions.size(), parsed_config.rtp.extensions.size()); | |
186 for (size_t i = 0; i < parsed_config.rtp.extensions.size(); i++) { | |
187 EXPECT_EQ(config.rtp.extensions[i].uri, | |
188 parsed_config.rtp.extensions[i].uri); | |
189 EXPECT_EQ(config.rtp.extensions[i].id, parsed_config.rtp.extensions[i].id); | |
190 } | |
191 // Check decoders. | |
192 EXPECT_EQ(config.decoders.size(), parsed_config.decoders.size()); | |
193 for (size_t i = 0; i < parsed_config.decoders.size(); i++) { | |
194 EXPECT_EQ(config.decoders[i].payload_name, | |
195 parsed_config.decoders[i].payload_name); | |
196 EXPECT_EQ(config.decoders[i].payload_type, | |
197 parsed_config.decoders[i].payload_type); | |
198 } | |
199 } | |
200 | |
201 void RtcEventLogTestHelper::VerifySendStreamConfig( | |
202 const ParsedRtcEventLog& parsed_log, | |
203 size_t index, | |
204 const VideoSendStream::Config& config) { | |
205 const rtclog::Event& event = parsed_log.events_[index]; | |
206 ASSERT_TRUE(IsValidBasicEvent(event)); | |
207 ASSERT_EQ(rtclog::Event::VIDEO_SENDER_CONFIG_EVENT, event.type()); | |
208 const rtclog::VideoSendConfig& sender_config = event.video_sender_config(); | |
209 // Check SSRCs. | |
210 ASSERT_EQ(static_cast<int>(config.rtp.ssrcs.size()), | |
211 sender_config.ssrcs_size()); | |
212 for (int i = 0; i < sender_config.ssrcs_size(); i++) { | |
213 EXPECT_EQ(config.rtp.ssrcs[i], sender_config.ssrcs(i)); | |
214 } | |
215 // Check header extensions. | |
216 ASSERT_EQ(static_cast<int>(config.rtp.extensions.size()), | |
217 sender_config.header_extensions_size()); | |
218 for (int i = 0; i < sender_config.header_extensions_size(); i++) { | |
219 ASSERT_TRUE(sender_config.header_extensions(i).has_name()); | |
220 ASSERT_TRUE(sender_config.header_extensions(i).has_id()); | |
221 const std::string& name = sender_config.header_extensions(i).name(); | |
222 int id = sender_config.header_extensions(i).id(); | |
223 EXPECT_EQ(config.rtp.extensions[i].id, id); | |
224 EXPECT_EQ(config.rtp.extensions[i].uri, name); | |
225 } | |
226 // Check RTX settings. | |
227 ASSERT_EQ(static_cast<int>(config.rtp.rtx.ssrcs.size()), | |
228 sender_config.rtx_ssrcs_size()); | |
229 for (int i = 0; i < sender_config.rtx_ssrcs_size(); i++) { | |
230 EXPECT_EQ(config.rtp.rtx.ssrcs[i], sender_config.rtx_ssrcs(i)); | |
231 } | |
232 if (sender_config.rtx_ssrcs_size() > 0) { | |
233 ASSERT_TRUE(sender_config.has_rtx_payload_type()); | |
234 EXPECT_EQ(config.rtp.rtx.payload_type, sender_config.rtx_payload_type()); | |
235 } | |
236 // Check encoder. | |
237 ASSERT_TRUE(sender_config.has_encoder()); | |
238 ASSERT_TRUE(sender_config.encoder().has_name()); | |
239 ASSERT_TRUE(sender_config.encoder().has_payload_type()); | |
240 EXPECT_EQ(config.encoder_settings.payload_name, | |
241 sender_config.encoder().name()); | |
242 EXPECT_EQ(config.encoder_settings.payload_type, | |
243 sender_config.encoder().payload_type()); | |
244 | |
245 // Check consistency of the parser. | |
246 VideoSendStream::Config parsed_config(nullptr); | |
247 parsed_log.GetVideoSendConfig(index, &parsed_config); | |
248 // Check SSRCs | |
249 EXPECT_EQ(config.rtp.ssrcs.size(), parsed_config.rtp.ssrcs.size()); | |
250 for (size_t i = 0; i < config.rtp.ssrcs.size(); i++) { | |
251 EXPECT_EQ(config.rtp.ssrcs[i], parsed_config.rtp.ssrcs[i]); | |
252 } | |
253 // Check header extensions. | |
254 EXPECT_EQ(config.rtp.extensions.size(), parsed_config.rtp.extensions.size()); | |
255 for (size_t i = 0; i < parsed_config.rtp.extensions.size(); i++) { | |
256 EXPECT_EQ(config.rtp.extensions[i].uri, | |
257 parsed_config.rtp.extensions[i].uri); | |
258 EXPECT_EQ(config.rtp.extensions[i].id, parsed_config.rtp.extensions[i].id); | |
259 } | |
260 // Check RTX settings. | |
261 EXPECT_EQ(config.rtp.rtx.ssrcs.size(), parsed_config.rtp.rtx.ssrcs.size()); | |
262 for (size_t i = 0; i < config.rtp.rtx.ssrcs.size(); i++) { | |
263 EXPECT_EQ(config.rtp.rtx.ssrcs[i], parsed_config.rtp.rtx.ssrcs[i]); | |
264 } | |
265 EXPECT_EQ(config.rtp.rtx.payload_type, parsed_config.rtp.rtx.payload_type); | |
266 // Check encoder. | |
267 EXPECT_EQ(config.encoder_settings.payload_name, | |
268 parsed_config.encoder_settings.payload_name); | |
269 EXPECT_EQ(config.encoder_settings.payload_type, | |
270 parsed_config.encoder_settings.payload_type); | |
271 } | |
272 | |
273 void RtcEventLogTestHelper::VerifyRtpEvent(const ParsedRtcEventLog& parsed_log, | |
274 size_t index, | |
275 PacketDirection direction, | |
276 MediaType media_type, | |
277 const uint8_t* header, | |
278 size_t header_size, | |
279 size_t total_size) { | |
280 const rtclog::Event& event = parsed_log.events_[index]; | |
281 ASSERT_TRUE(IsValidBasicEvent(event)); | |
282 ASSERT_EQ(rtclog::Event::RTP_EVENT, event.type()); | |
283 const rtclog::RtpPacket& rtp_packet = event.rtp_packet(); | |
284 ASSERT_TRUE(rtp_packet.has_incoming()); | |
285 EXPECT_EQ(direction == kIncomingPacket, rtp_packet.incoming()); | |
286 ASSERT_TRUE(rtp_packet.has_type()); | |
287 EXPECT_EQ(media_type, GetRuntimeMediaType(rtp_packet.type())); | |
288 ASSERT_TRUE(rtp_packet.has_packet_length()); | |
289 EXPECT_EQ(total_size, rtp_packet.packet_length()); | |
290 ASSERT_TRUE(rtp_packet.has_header()); | |
291 ASSERT_EQ(header_size, rtp_packet.header().size()); | |
292 for (size_t i = 0; i < header_size; i++) { | |
293 EXPECT_EQ(header[i], static_cast<uint8_t>(rtp_packet.header()[i])); | |
294 } | |
295 | |
296 // Check consistency of the parser. | |
297 PacketDirection parsed_direction; | |
298 MediaType parsed_media_type; | |
299 uint8_t parsed_header[1500]; | |
300 size_t parsed_header_size, parsed_total_size; | |
301 parsed_log.GetRtpHeader(index, &parsed_direction, &parsed_media_type, | |
302 parsed_header, &parsed_header_size, | |
303 &parsed_total_size); | |
304 EXPECT_EQ(direction, parsed_direction); | |
305 EXPECT_EQ(media_type, parsed_media_type); | |
306 ASSERT_EQ(header_size, parsed_header_size); | |
307 EXPECT_EQ(0, std::memcmp(header, parsed_header, header_size)); | |
308 EXPECT_EQ(total_size, parsed_total_size); | |
309 } | |
310 | |
311 void RtcEventLogTestHelper::VerifyRtcpEvent(const ParsedRtcEventLog& parsed_log, | |
312 size_t index, | |
313 PacketDirection direction, | |
314 MediaType media_type, | |
315 const uint8_t* packet, | |
316 size_t total_size) { | |
317 const rtclog::Event& event = parsed_log.events_[index]; | |
318 ASSERT_TRUE(IsValidBasicEvent(event)); | |
319 ASSERT_EQ(rtclog::Event::RTCP_EVENT, event.type()); | |
320 const rtclog::RtcpPacket& rtcp_packet = event.rtcp_packet(); | |
321 ASSERT_TRUE(rtcp_packet.has_incoming()); | |
322 EXPECT_EQ(direction == kIncomingPacket, rtcp_packet.incoming()); | |
323 ASSERT_TRUE(rtcp_packet.has_type()); | |
324 EXPECT_EQ(media_type, GetRuntimeMediaType(rtcp_packet.type())); | |
325 ASSERT_TRUE(rtcp_packet.has_packet_data()); | |
326 ASSERT_EQ(total_size, rtcp_packet.packet_data().size()); | |
327 for (size_t i = 0; i < total_size; i++) { | |
328 EXPECT_EQ(packet[i], static_cast<uint8_t>(rtcp_packet.packet_data()[i])); | |
329 } | |
330 | |
331 // Check consistency of the parser. | |
332 PacketDirection parsed_direction; | |
333 MediaType parsed_media_type; | |
334 uint8_t parsed_packet[1500]; | |
335 size_t parsed_total_size; | |
336 parsed_log.GetRtcpPacket(index, &parsed_direction, &parsed_media_type, | |
337 parsed_packet, &parsed_total_size); | |
338 EXPECT_EQ(direction, parsed_direction); | |
339 EXPECT_EQ(media_type, parsed_media_type); | |
340 ASSERT_EQ(total_size, parsed_total_size); | |
341 EXPECT_EQ(0, std::memcmp(packet, parsed_packet, total_size)); | |
342 } | |
343 | |
344 void RtcEventLogTestHelper::VerifyPlayoutEvent( | |
345 const ParsedRtcEventLog& parsed_log, | |
346 size_t index, | |
347 uint32_t ssrc) { | |
348 const rtclog::Event& event = parsed_log.events_[index]; | |
349 ASSERT_TRUE(IsValidBasicEvent(event)); | |
350 ASSERT_EQ(rtclog::Event::AUDIO_PLAYOUT_EVENT, event.type()); | |
351 const rtclog::AudioPlayoutEvent& playout_event = event.audio_playout_event(); | |
352 ASSERT_TRUE(playout_event.has_local_ssrc()); | |
353 EXPECT_EQ(ssrc, playout_event.local_ssrc()); | |
354 | |
355 // Check consistency of the parser. | |
356 uint32_t parsed_ssrc; | |
357 parsed_log.GetAudioPlayout(index, &parsed_ssrc); | |
358 EXPECT_EQ(ssrc, parsed_ssrc); | |
359 } | |
360 | |
361 void RtcEventLogTestHelper::VerifyBweLossEvent( | |
362 const ParsedRtcEventLog& parsed_log, | |
363 size_t index, | |
364 int32_t bitrate, | |
365 uint8_t fraction_loss, | |
366 int32_t total_packets) { | |
367 const rtclog::Event& event = parsed_log.events_[index]; | |
368 ASSERT_TRUE(IsValidBasicEvent(event)); | |
369 ASSERT_EQ(rtclog::Event::BWE_PACKET_LOSS_EVENT, event.type()); | |
370 const rtclog::BwePacketLossEvent& bwe_event = event.bwe_packet_loss_event(); | |
371 ASSERT_TRUE(bwe_event.has_bitrate()); | |
372 EXPECT_EQ(bitrate, bwe_event.bitrate()); | |
373 ASSERT_TRUE(bwe_event.has_fraction_loss()); | |
374 EXPECT_EQ(fraction_loss, bwe_event.fraction_loss()); | |
375 ASSERT_TRUE(bwe_event.has_total_packets()); | |
376 EXPECT_EQ(total_packets, bwe_event.total_packets()); | |
377 | |
378 // Check consistency of the parser. | |
379 int32_t parsed_bitrate; | |
380 uint8_t parsed_fraction_loss; | |
381 int32_t parsed_total_packets; | |
382 parsed_log.GetBwePacketLossEvent( | |
383 index, &parsed_bitrate, &parsed_fraction_loss, &parsed_total_packets); | |
384 EXPECT_EQ(bitrate, parsed_bitrate); | |
385 EXPECT_EQ(fraction_loss, parsed_fraction_loss); | |
386 EXPECT_EQ(total_packets, parsed_total_packets); | |
387 } | |
388 | |
389 void RtcEventLogTestHelper::VerifyLogStartEvent( | |
390 const ParsedRtcEventLog& parsed_log, | |
391 size_t index) { | |
392 const rtclog::Event& event = parsed_log.events_[index]; | |
393 ASSERT_TRUE(IsValidBasicEvent(event)); | |
394 EXPECT_EQ(rtclog::Event::LOG_START, event.type()); | |
395 } | |
396 | |
397 void RtcEventLogTestHelper::VerifyLogEndEvent( | |
398 const ParsedRtcEventLog& parsed_log, | |
399 size_t index) { | |
400 const rtclog::Event& event = parsed_log.events_[index]; | |
401 ASSERT_TRUE(IsValidBasicEvent(event)); | |
402 EXPECT_EQ(rtclog::Event::LOG_END, event.type()); | |
403 } | |
404 | |
405 } // namespace webrtc | |
OLD | NEW |