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