| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright 2016 The WebRTC project authors. All Rights Reserved. | 2 * Copyright 2016 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 |
| 11 #include <set> | 11 #include <set> |
| 12 #include <vector> | 12 #include <vector> |
| 13 | 13 |
| 14 #include "webrtc/api/audio_codecs/builtin_audio_decoder_factory.h" | 14 #include "webrtc/api/audio_codecs/builtin_audio_decoder_factory.h" |
| 15 #include "webrtc/api/audio_codecs/builtin_audio_encoder_factory.h" | 15 #include "webrtc/api/audio_codecs/builtin_audio_encoder_factory.h" |
| 16 #include "webrtc/api/datachannelinterface.h" | 16 #include "webrtc/api/datachannelinterface.h" |
| 17 #include "webrtc/api/peerconnectioninterface.h" | 17 #include "webrtc/api/peerconnectioninterface.h" |
| 18 #include "webrtc/api/stats/rtcstats_objects.h" | 18 #include "webrtc/api/stats/rtcstats_objects.h" |
| 19 #include "webrtc/api/stats/rtcstatsreport.h" | 19 #include "webrtc/api/stats/rtcstatsreport.h" |
| 20 #include "webrtc/pc/test/peerconnectiontestwrapper.h" | 20 #include "webrtc/pc/test/peerconnectiontestwrapper.h" |
| 21 #include "webrtc/pc/test/rtcstatsobtainer.h" | 21 #include "webrtc/pc/test/rtcstatsobtainer.h" |
| 22 #include "webrtc/rtc_base/checks.h" | 22 #include "webrtc/rtc_base/checks.h" |
| 23 #include "webrtc/rtc_base/event_tracer.h" | 23 #include "webrtc/rtc_base/event_tracer.h" |
| 24 #include "webrtc/rtc_base/gunit.h" | 24 #include "webrtc/rtc_base/gunit.h" |
| 25 #include "webrtc/rtc_base/refcountedobject.h" | 25 #include "webrtc/rtc_base/refcountedobject.h" |
| 26 #include "webrtc/rtc_base/scoped_ref_ptr.h" | 26 #include "webrtc/rtc_base/scoped_ref_ptr.h" |
| 27 #include "webrtc/rtc_base/trace_event.h" | |
| 28 #include "webrtc/rtc_base/virtualsocketserver.h" | 27 #include "webrtc/rtc_base/virtualsocketserver.h" |
| 29 | 28 |
| 30 namespace webrtc { | 29 namespace webrtc { |
| 31 | 30 |
| 32 namespace { | 31 namespace { |
| 33 | 32 |
| 34 const int64_t kGetStatsTimeoutMs = 10000; | 33 const int64_t kGetStatsTimeoutMs = 10000; |
| 35 | 34 |
| 36 const unsigned char* GetCategoryEnabledHandler(const char* name) { | 35 const unsigned char* GetCategoryEnabledHandler(const char* name) { |
| 37 if (strcmp("webrtc_stats", name) != 0) { | 36 return reinterpret_cast<const unsigned char*>("webrtc_stats"); |
| 38 return reinterpret_cast<const unsigned char*>(""); | |
| 39 } | |
| 40 return reinterpret_cast<const unsigned char*>(name); | |
| 41 } | 37 } |
| 42 | 38 |
| 43 class RTCStatsReportTraceListener { | 39 void AddTraceEventHandler(char phase, |
| 44 public: | 40 const unsigned char* category_enabled, |
| 45 static void SetUp() { | 41 const char* name, |
| 46 if (!traced_report_) | 42 unsigned long long id, |
| 47 traced_report_ = new RTCStatsReportTraceListener(); | 43 int num_args, |
| 48 traced_report_->last_trace_ = ""; | 44 const char** arg_names, |
| 49 SetupEventTracer(&GetCategoryEnabledHandler, | 45 const unsigned char* arg_types, |
| 50 &RTCStatsReportTraceListener::AddTraceEventHandler); | 46 const unsigned long long* arg_values, |
| 51 } | 47 unsigned char flags) { |
| 52 | 48 // Do nothing |
| 53 static const std::string& last_trace() { | 49 } |
| 54 RTC_DCHECK(traced_report_); | |
| 55 return traced_report_->last_trace_; | |
| 56 } | |
| 57 | |
| 58 private: | |
| 59 static void AddTraceEventHandler(char phase, | |
| 60 const unsigned char* category_enabled, | |
| 61 const char* name, | |
| 62 unsigned long long id, | |
| 63 int num_args, | |
| 64 const char** arg_names, | |
| 65 const unsigned char* arg_types, | |
| 66 const unsigned long long* arg_values, | |
| 67 unsigned char flags) { | |
| 68 RTC_DCHECK(traced_report_); | |
| 69 EXPECT_EQ(traced_report_->last_trace_, ""); | |
| 70 EXPECT_STREQ("webrtc_stats", | |
| 71 reinterpret_cast<const char*>(category_enabled)); | |
| 72 EXPECT_STREQ("webrtc_stats", name); | |
| 73 EXPECT_EQ(1, num_args); | |
| 74 EXPECT_STREQ("report", arg_names[0]); | |
| 75 EXPECT_EQ(TRACE_VALUE_TYPE_COPY_STRING, arg_types[0]); | |
| 76 | |
| 77 traced_report_->last_trace_ = reinterpret_cast<const char*>(arg_values[0]); | |
| 78 } | |
| 79 | |
| 80 static RTCStatsReportTraceListener* traced_report_; | |
| 81 std::string last_trace_; | |
| 82 }; | |
| 83 | |
| 84 RTCStatsReportTraceListener* RTCStatsReportTraceListener::traced_report_ = | |
| 85 nullptr; | |
| 86 | 50 |
| 87 class RTCStatsIntegrationTest : public testing::Test { | 51 class RTCStatsIntegrationTest : public testing::Test { |
| 88 public: | 52 public: |
| 89 RTCStatsIntegrationTest() | 53 RTCStatsIntegrationTest() |
| 90 : network_thread_(new rtc::Thread(&virtual_socket_server_)), | 54 : network_thread_(new rtc::Thread(&virtual_socket_server_)), |
| 91 worker_thread_(rtc::Thread::Create()) { | 55 worker_thread_(rtc::Thread::Create()) { |
| 92 RTCStatsReportTraceListener::SetUp(); | 56 SetupEventTracer(&GetCategoryEnabledHandler, &AddTraceEventHandler); |
| 93 | 57 |
| 94 RTC_CHECK(network_thread_->Start()); | 58 RTC_CHECK(network_thread_->Start()); |
| 95 RTC_CHECK(worker_thread_->Start()); | 59 RTC_CHECK(worker_thread_->Start()); |
| 96 | 60 |
| 97 caller_ = new rtc::RefCountedObject<PeerConnectionTestWrapper>( | 61 caller_ = new rtc::RefCountedObject<PeerConnectionTestWrapper>( |
| 98 "caller", network_thread_.get(), worker_thread_.get()); | 62 "caller", network_thread_.get(), worker_thread_.get()); |
| 99 callee_ = new rtc::RefCountedObject<PeerConnectionTestWrapper>( | 63 callee_ = new rtc::RefCountedObject<PeerConnectionTestWrapper>( |
| 100 "callee", network_thread_.get(), worker_thread_.get()); | 64 "callee", network_thread_.get(), worker_thread_.get()); |
| 101 } | 65 } |
| 102 | 66 |
| (...skipping 576 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 679 private: | 643 private: |
| 680 rtc::scoped_refptr<const RTCStatsReport> report_; | 644 rtc::scoped_refptr<const RTCStatsReport> report_; |
| 681 }; | 645 }; |
| 682 | 646 |
| 683 #ifdef HAVE_SCTP | 647 #ifdef HAVE_SCTP |
| 684 TEST_F(RTCStatsIntegrationTest, GetStatsFromCaller) { | 648 TEST_F(RTCStatsIntegrationTest, GetStatsFromCaller) { |
| 685 StartCall(); | 649 StartCall(); |
| 686 | 650 |
| 687 rtc::scoped_refptr<const RTCStatsReport> report = GetStatsFromCaller(); | 651 rtc::scoped_refptr<const RTCStatsReport> report = GetStatsFromCaller(); |
| 688 RTCStatsReportVerifier(report.get()).VerifyReport(); | 652 RTCStatsReportVerifier(report.get()).VerifyReport(); |
| 689 EXPECT_EQ(report->ToJson(), RTCStatsReportTraceListener::last_trace()); | |
| 690 } | 653 } |
| 691 | 654 |
| 692 TEST_F(RTCStatsIntegrationTest, GetStatsFromCallee) { | 655 TEST_F(RTCStatsIntegrationTest, GetStatsFromCallee) { |
| 693 StartCall(); | 656 StartCall(); |
| 694 | 657 |
| 695 rtc::scoped_refptr<const RTCStatsReport> report = GetStatsFromCallee(); | 658 rtc::scoped_refptr<const RTCStatsReport> report = GetStatsFromCallee(); |
| 696 RTCStatsReportVerifier(report.get()).VerifyReport(); | 659 RTCStatsReportVerifier(report.get()).VerifyReport(); |
| 697 EXPECT_EQ(report->ToJson(), RTCStatsReportTraceListener::last_trace()); | |
| 698 } | 660 } |
| 699 | 661 |
| 700 TEST_F(RTCStatsIntegrationTest, GetsStatsWhileDestroyingPeerConnections) { | 662 TEST_F(RTCStatsIntegrationTest, GetsStatsWhileDestroyingPeerConnections) { |
| 701 StartCall(); | 663 StartCall(); |
| 702 | 664 |
| 703 rtc::scoped_refptr<RTCStatsObtainer> stats_obtainer = | 665 rtc::scoped_refptr<RTCStatsObtainer> stats_obtainer = |
| 704 RTCStatsObtainer::Create(); | 666 RTCStatsObtainer::Create(); |
| 705 caller_->pc()->GetStats(stats_obtainer); | 667 caller_->pc()->GetStats(stats_obtainer); |
| 706 // This will destroy the peer connection. | 668 // This will destroy the peer connection. |
| 707 caller_ = nullptr; | 669 caller_ = nullptr; |
| 708 // Any pending stats requests should have completed in the act of destroying | 670 // Any pending stats requests should have completed in the act of destroying |
| 709 // the peer connection. | 671 // the peer connection. |
| 710 EXPECT_TRUE(stats_obtainer->report()); | 672 EXPECT_TRUE(stats_obtainer->report()); |
| 711 EXPECT_EQ(stats_obtainer->report()->ToJson(), | |
| 712 RTCStatsReportTraceListener::last_trace()); | |
| 713 } | 673 } |
| 714 #endif // HAVE_SCTP | 674 #endif // HAVE_SCTP |
| 715 | 675 |
| 716 } // namespace | 676 } // namespace |
| 717 | 677 |
| 718 } // namespace webrtc | 678 } // namespace webrtc |
| OLD | NEW |