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 "webrtc/api/rtcstatscollector.h" | 11 #include "webrtc/api/rtcstatscollector.h" |
12 | 12 |
13 #include <memory> | 13 #include <memory> |
14 #include <ostream> | 14 #include <ostream> |
15 #include <string> | 15 #include <string> |
16 #include <vector> | 16 #include <vector> |
17 | 17 |
18 #include "webrtc/api/jsepsessiondescription.h" | 18 #include "webrtc/api/jsepsessiondescription.h" |
19 #include "webrtc/api/mediastream.h" | 19 #include "webrtc/api/mediastream.h" |
20 #include "webrtc/api/mediastreamtrack.h" | 20 #include "webrtc/api/mediastreamtrack.h" |
21 #include "webrtc/api/rtpparameters.h" | 21 #include "webrtc/api/rtpparameters.h" |
22 #include "webrtc/api/stats/rtcstats_objects.h" | 22 #include "webrtc/api/stats/rtcstats_objects.h" |
23 #include "webrtc/api/stats/rtcstatsreport.h" | 23 #include "webrtc/api/stats/rtcstatsreport.h" |
24 #include "webrtc/api/test/mock_datachannel.h" | 24 #include "webrtc/api/test/mock_datachannel.h" |
25 #include "webrtc/api/test/mock_peerconnection.h" | 25 #include "webrtc/api/test/mock_peerconnection.h" |
26 #include "webrtc/api/test/mock_webrtcsession.h" | 26 #include "webrtc/api/test/mock_webrtcsession.h" |
| 27 #include "webrtc/api/test/rtcstatsobtainer.h" |
27 #include "webrtc/base/checks.h" | 28 #include "webrtc/base/checks.h" |
28 #include "webrtc/base/fakeclock.h" | 29 #include "webrtc/base/fakeclock.h" |
29 #include "webrtc/base/fakesslidentity.h" | 30 #include "webrtc/base/fakesslidentity.h" |
30 #include "webrtc/base/gunit.h" | 31 #include "webrtc/base/gunit.h" |
31 #include "webrtc/base/logging.h" | 32 #include "webrtc/base/logging.h" |
32 #include "webrtc/base/socketaddress.h" | 33 #include "webrtc/base/socketaddress.h" |
33 #include "webrtc/base/thread_checker.h" | 34 #include "webrtc/base/thread_checker.h" |
34 #include "webrtc/base/timedelta.h" | 35 #include "webrtc/base/timedelta.h" |
35 #include "webrtc/base/timeutils.h" | 36 #include "webrtc/base/timeutils.h" |
36 #include "webrtc/logging/rtc_event_log/rtc_event_log.h" | 37 #include "webrtc/logging/rtc_event_log/rtc_event_log.h" |
(...skipping 428 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
465 rtc::Thread* const worker_thread_; | 466 rtc::Thread* const worker_thread_; |
466 rtc::Thread* const network_thread_; | 467 rtc::Thread* const network_thread_; |
467 | 468 |
468 rtc::CriticalSection lock_; | 469 rtc::CriticalSection lock_; |
469 rtc::scoped_refptr<const RTCStatsReport> delivered_report_; | 470 rtc::scoped_refptr<const RTCStatsReport> delivered_report_; |
470 int produced_on_signaling_thread_ = 0; | 471 int produced_on_signaling_thread_ = 0; |
471 int produced_on_worker_thread_ = 0; | 472 int produced_on_worker_thread_ = 0; |
472 int produced_on_network_thread_ = 0; | 473 int produced_on_network_thread_ = 0; |
473 }; | 474 }; |
474 | 475 |
475 class StatsCallback : public RTCStatsCollectorCallback { | |
476 public: | |
477 static rtc::scoped_refptr<StatsCallback> Create( | |
478 rtc::scoped_refptr<const RTCStatsReport>* report_ptr = nullptr) { | |
479 return rtc::scoped_refptr<StatsCallback>( | |
480 new rtc::RefCountedObject<StatsCallback>(report_ptr)); | |
481 } | |
482 | |
483 void OnStatsDelivered( | |
484 const rtc::scoped_refptr<const RTCStatsReport>& report) override { | |
485 EXPECT_TRUE(thread_checker_.CalledOnValidThread()); | |
486 report_ = report; | |
487 if (report_ptr_) | |
488 *report_ptr_ = report_; | |
489 } | |
490 | |
491 rtc::scoped_refptr<const RTCStatsReport> report() const { | |
492 EXPECT_TRUE(thread_checker_.CalledOnValidThread()); | |
493 return report_; | |
494 } | |
495 | |
496 protected: | |
497 explicit StatsCallback(rtc::scoped_refptr<const RTCStatsReport>* report_ptr) | |
498 : report_ptr_(report_ptr) {} | |
499 | |
500 private: | |
501 rtc::ThreadChecker thread_checker_; | |
502 rtc::scoped_refptr<const RTCStatsReport> report_; | |
503 rtc::scoped_refptr<const RTCStatsReport>* report_ptr_; | |
504 }; | |
505 | |
506 class RTCStatsCollectorTest : public testing::Test { | 476 class RTCStatsCollectorTest : public testing::Test { |
507 public: | 477 public: |
508 RTCStatsCollectorTest() | 478 RTCStatsCollectorTest() |
509 : test_(new rtc::RefCountedObject<RTCStatsCollectorTestHelper>()), | 479 : test_(new rtc::RefCountedObject<RTCStatsCollectorTestHelper>()), |
510 collector_(RTCStatsCollector::Create( | 480 collector_(RTCStatsCollector::Create( |
511 &test_->pc(), 50 * rtc::kNumMicrosecsPerMillisec)) { | 481 &test_->pc(), 50 * rtc::kNumMicrosecsPerMillisec)) { |
512 } | 482 } |
513 | 483 |
514 rtc::scoped_refptr<const RTCStatsReport> GetStatsReport() { | 484 rtc::scoped_refptr<const RTCStatsReport> GetStatsReport() { |
515 rtc::scoped_refptr<StatsCallback> callback = StatsCallback::Create(); | 485 rtc::scoped_refptr<RTCStatsObtainer> callback = RTCStatsObtainer::Create(); |
516 collector_->GetStatsReport(callback); | 486 collector_->GetStatsReport(callback); |
517 EXPECT_TRUE_WAIT(callback->report(), kGetStatsReportTimeoutMs); | 487 EXPECT_TRUE_WAIT(callback->report(), kGetStatsReportTimeoutMs); |
518 int64_t after = rtc::TimeUTCMicros(); | 488 int64_t after = rtc::TimeUTCMicros(); |
519 for (const RTCStats& stats : *callback->report()) { | 489 for (const RTCStats& stats : *callback->report()) { |
520 EXPECT_LE(stats.timestamp_us(), after); | 490 EXPECT_LE(stats.timestamp_us(), after); |
521 } | 491 } |
522 return callback->report(); | 492 return callback->report(); |
523 } | 493 } |
524 | 494 |
525 const RTCIceCandidateStats* ExpectReportContainsCandidate( | 495 const RTCIceCandidateStats* ExpectReportContainsCandidate( |
(...skipping 190 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
716 data_channel.bytes_received()); | 686 data_channel.bytes_received()); |
717 } | 687 } |
718 | 688 |
719 protected: | 689 protected: |
720 rtc::scoped_refptr<RTCStatsCollectorTestHelper> test_; | 690 rtc::scoped_refptr<RTCStatsCollectorTestHelper> test_; |
721 rtc::scoped_refptr<RTCStatsCollector> collector_; | 691 rtc::scoped_refptr<RTCStatsCollector> collector_; |
722 }; | 692 }; |
723 | 693 |
724 TEST_F(RTCStatsCollectorTest, SingleCallback) { | 694 TEST_F(RTCStatsCollectorTest, SingleCallback) { |
725 rtc::scoped_refptr<const RTCStatsReport> result; | 695 rtc::scoped_refptr<const RTCStatsReport> result; |
726 collector_->GetStatsReport(StatsCallback::Create(&result)); | 696 collector_->GetStatsReport(RTCStatsObtainer::Create(&result)); |
727 EXPECT_TRUE_WAIT(result, kGetStatsReportTimeoutMs); | 697 EXPECT_TRUE_WAIT(result, kGetStatsReportTimeoutMs); |
728 } | 698 } |
729 | 699 |
730 TEST_F(RTCStatsCollectorTest, MultipleCallbacks) { | 700 TEST_F(RTCStatsCollectorTest, MultipleCallbacks) { |
731 rtc::scoped_refptr<const RTCStatsReport> a; | 701 rtc::scoped_refptr<const RTCStatsReport> a; |
732 rtc::scoped_refptr<const RTCStatsReport> b; | 702 rtc::scoped_refptr<const RTCStatsReport> b; |
733 rtc::scoped_refptr<const RTCStatsReport> c; | 703 rtc::scoped_refptr<const RTCStatsReport> c; |
734 collector_->GetStatsReport(StatsCallback::Create(&a)); | 704 collector_->GetStatsReport(RTCStatsObtainer::Create(&a)); |
735 collector_->GetStatsReport(StatsCallback::Create(&b)); | 705 collector_->GetStatsReport(RTCStatsObtainer::Create(&b)); |
736 collector_->GetStatsReport(StatsCallback::Create(&c)); | 706 collector_->GetStatsReport(RTCStatsObtainer::Create(&c)); |
737 EXPECT_TRUE_WAIT(a, kGetStatsReportTimeoutMs); | 707 EXPECT_TRUE_WAIT(a, kGetStatsReportTimeoutMs); |
738 EXPECT_TRUE_WAIT(b, kGetStatsReportTimeoutMs); | 708 EXPECT_TRUE_WAIT(b, kGetStatsReportTimeoutMs); |
739 EXPECT_TRUE_WAIT(c, kGetStatsReportTimeoutMs); | 709 EXPECT_TRUE_WAIT(c, kGetStatsReportTimeoutMs); |
740 EXPECT_EQ(a.get(), b.get()); | 710 EXPECT_EQ(a.get(), b.get()); |
741 EXPECT_EQ(b.get(), c.get()); | 711 EXPECT_EQ(b.get(), c.get()); |
742 } | 712 } |
743 | 713 |
744 TEST_F(RTCStatsCollectorTest, CachedStatsReports) { | 714 TEST_F(RTCStatsCollectorTest, CachedStatsReports) { |
745 // Caching should ensure |a| and |b| are the same report. | 715 // Caching should ensure |a| and |b| are the same report. |
746 rtc::scoped_refptr<const RTCStatsReport> a = GetStatsReport(); | 716 rtc::scoped_refptr<const RTCStatsReport> a = GetStatsReport(); |
747 rtc::scoped_refptr<const RTCStatsReport> b = GetStatsReport(); | 717 rtc::scoped_refptr<const RTCStatsReport> b = GetStatsReport(); |
748 EXPECT_EQ(a.get(), b.get()); | 718 EXPECT_EQ(a.get(), b.get()); |
749 // Invalidate cache by clearing it. | 719 // Invalidate cache by clearing it. |
750 collector_->ClearCachedStatsReport(); | 720 collector_->ClearCachedStatsReport(); |
751 rtc::scoped_refptr<const RTCStatsReport> c = GetStatsReport(); | 721 rtc::scoped_refptr<const RTCStatsReport> c = GetStatsReport(); |
752 EXPECT_NE(b.get(), c.get()); | 722 EXPECT_NE(b.get(), c.get()); |
753 // Invalidate cache by advancing time. | 723 // Invalidate cache by advancing time. |
754 test_->fake_clock().AdvanceTime(rtc::TimeDelta::FromMilliseconds(51)); | 724 test_->fake_clock().AdvanceTime(rtc::TimeDelta::FromMilliseconds(51)); |
755 rtc::scoped_refptr<const RTCStatsReport> d = GetStatsReport(); | 725 rtc::scoped_refptr<const RTCStatsReport> d = GetStatsReport(); |
756 EXPECT_TRUE(d); | 726 EXPECT_TRUE(d); |
757 EXPECT_NE(c.get(), d.get()); | 727 EXPECT_NE(c.get(), d.get()); |
758 } | 728 } |
759 | 729 |
760 TEST_F(RTCStatsCollectorTest, MultipleCallbacksWithInvalidatedCacheInBetween) { | 730 TEST_F(RTCStatsCollectorTest, MultipleCallbacksWithInvalidatedCacheInBetween) { |
761 rtc::scoped_refptr<const RTCStatsReport> a; | 731 rtc::scoped_refptr<const RTCStatsReport> a; |
762 rtc::scoped_refptr<const RTCStatsReport> b; | 732 rtc::scoped_refptr<const RTCStatsReport> b; |
763 rtc::scoped_refptr<const RTCStatsReport> c; | 733 rtc::scoped_refptr<const RTCStatsReport> c; |
764 collector_->GetStatsReport(StatsCallback::Create(&a)); | 734 collector_->GetStatsReport(RTCStatsObtainer::Create(&a)); |
765 collector_->GetStatsReport(StatsCallback::Create(&b)); | 735 collector_->GetStatsReport(RTCStatsObtainer::Create(&b)); |
766 // Cache is invalidated after 50 ms. | 736 // Cache is invalidated after 50 ms. |
767 test_->fake_clock().AdvanceTime(rtc::TimeDelta::FromMilliseconds(51)); | 737 test_->fake_clock().AdvanceTime(rtc::TimeDelta::FromMilliseconds(51)); |
768 collector_->GetStatsReport(StatsCallback::Create(&c)); | 738 collector_->GetStatsReport(RTCStatsObtainer::Create(&c)); |
769 EXPECT_TRUE_WAIT(a, kGetStatsReportTimeoutMs); | 739 EXPECT_TRUE_WAIT(a, kGetStatsReportTimeoutMs); |
770 EXPECT_TRUE_WAIT(b, kGetStatsReportTimeoutMs); | 740 EXPECT_TRUE_WAIT(b, kGetStatsReportTimeoutMs); |
771 EXPECT_TRUE_WAIT(c, kGetStatsReportTimeoutMs); | 741 EXPECT_TRUE_WAIT(c, kGetStatsReportTimeoutMs); |
772 EXPECT_EQ(a.get(), b.get()); | 742 EXPECT_EQ(a.get(), b.get()); |
773 // The act of doing |AdvanceTime| processes all messages. If this was not the | 743 // The act of doing |AdvanceTime| processes all messages. If this was not the |
774 // case we might not require |c| to be fresher than |b|. | 744 // case we might not require |c| to be fresher than |b|. |
775 EXPECT_NE(c.get(), b.get()); | 745 EXPECT_NE(c.get(), b.get()); |
776 } | 746 } |
777 | 747 |
778 TEST_F(RTCStatsCollectorTest, CollectRTCCertificateStatsSingle) { | 748 TEST_F(RTCStatsCollectorTest, CollectRTCCertificateStatsSingle) { |
(...skipping 1038 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1817 rtc::scoped_refptr<FakeRTCStatsCollector> collector_; | 1787 rtc::scoped_refptr<FakeRTCStatsCollector> collector_; |
1818 }; | 1788 }; |
1819 | 1789 |
1820 TEST_F(RTCStatsCollectorTestWithFakeCollector, ThreadUsageAndResultsMerging) { | 1790 TEST_F(RTCStatsCollectorTestWithFakeCollector, ThreadUsageAndResultsMerging) { |
1821 collector_->VerifyThreadUsageAndResultsMerging(); | 1791 collector_->VerifyThreadUsageAndResultsMerging(); |
1822 } | 1792 } |
1823 | 1793 |
1824 } // namespace | 1794 } // namespace |
1825 | 1795 |
1826 } // namespace webrtc | 1796 } // namespace webrtc |
OLD | NEW |