hbos | 6d183ac | 2016-08-29 07:20:33 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 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 | |
Steve Anton | 10542f2 | 2019-01-11 09:11:00 -0800 | [diff] [blame] | 11 | #include "api/stats/rtc_stats_report.h" |
hbos | 6d183ac | 2016-08-29 07:20:33 -0700 | [diff] [blame] | 12 | |
Steve Anton | 10542f2 | 2019-01-11 09:11:00 -0800 | [diff] [blame] | 13 | #include "api/stats/rtc_stats.h" |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 14 | #include "rtc_base/checks.h" |
Yves Gerey | 3e70781 | 2018-11-28 16:47:49 +0100 | [diff] [blame] | 15 | #include "test/gtest.h" |
hbos | 6d183ac | 2016-08-29 07:20:33 -0700 | [diff] [blame] | 16 | |
| 17 | namespace webrtc { |
| 18 | |
| 19 | class RTCTestStats1 : public RTCStats { |
| 20 | public: |
hbos | fc5e050 | 2016-10-06 02:06:10 -0700 | [diff] [blame] | 21 | WEBRTC_RTCSTATS_DECL(); |
| 22 | |
hbos | 0e6758d | 2016-08-31 07:57:36 -0700 | [diff] [blame] | 23 | RTCTestStats1(const std::string& id, int64_t timestamp_us) |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 24 | : RTCStats(id, timestamp_us), integer("integer") {} |
hbos | 6d183ac | 2016-08-29 07:20:33 -0700 | [diff] [blame] | 25 | |
hbos | 6d183ac | 2016-08-29 07:20:33 -0700 | [diff] [blame] | 26 | RTCStatsMember<int32_t> integer; |
| 27 | }; |
| 28 | |
Mirko Bonadei | c4dd730 | 2019-02-25 09:12:02 +0100 | [diff] [blame] | 29 | WEBRTC_RTCSTATS_IMPL(RTCTestStats1, RTCStats, "test-stats-1", &integer) |
hbos | 6d183ac | 2016-08-29 07:20:33 -0700 | [diff] [blame] | 30 | |
| 31 | class RTCTestStats2 : public RTCStats { |
| 32 | public: |
hbos | fc5e050 | 2016-10-06 02:06:10 -0700 | [diff] [blame] | 33 | WEBRTC_RTCSTATS_DECL(); |
| 34 | |
hbos | 0e6758d | 2016-08-31 07:57:36 -0700 | [diff] [blame] | 35 | RTCTestStats2(const std::string& id, int64_t timestamp_us) |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 36 | : RTCStats(id, timestamp_us), number("number") {} |
hbos | 6d183ac | 2016-08-29 07:20:33 -0700 | [diff] [blame] | 37 | |
hbos | 6d183ac | 2016-08-29 07:20:33 -0700 | [diff] [blame] | 38 | RTCStatsMember<double> number; |
| 39 | }; |
| 40 | |
Mirko Bonadei | c4dd730 | 2019-02-25 09:12:02 +0100 | [diff] [blame] | 41 | WEBRTC_RTCSTATS_IMPL(RTCTestStats2, RTCStats, "test-stats-2", &number) |
hbos | 6d183ac | 2016-08-29 07:20:33 -0700 | [diff] [blame] | 42 | |
| 43 | class RTCTestStats3 : public RTCStats { |
| 44 | public: |
hbos | fc5e050 | 2016-10-06 02:06:10 -0700 | [diff] [blame] | 45 | WEBRTC_RTCSTATS_DECL(); |
| 46 | |
hbos | 0e6758d | 2016-08-31 07:57:36 -0700 | [diff] [blame] | 47 | RTCTestStats3(const std::string& id, int64_t timestamp_us) |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 48 | : RTCStats(id, timestamp_us), string("string") {} |
hbos | 6d183ac | 2016-08-29 07:20:33 -0700 | [diff] [blame] | 49 | |
hbos | 6d183ac | 2016-08-29 07:20:33 -0700 | [diff] [blame] | 50 | RTCStatsMember<std::string> string; |
| 51 | }; |
| 52 | |
Mirko Bonadei | c4dd730 | 2019-02-25 09:12:02 +0100 | [diff] [blame] | 53 | WEBRTC_RTCSTATS_IMPL(RTCTestStats3, RTCStats, "test-stats-3", &string) |
hbos | 6d183ac | 2016-08-29 07:20:33 -0700 | [diff] [blame] | 54 | |
| 55 | TEST(RTCStatsReport, AddAndGetStats) { |
hbos | 6ded190 | 2016-11-01 01:50:46 -0700 | [diff] [blame] | 56 | rtc::scoped_refptr<RTCStatsReport> report = RTCStatsReport::Create(1337); |
| 57 | EXPECT_EQ(report->timestamp_us(), 1337u); |
hbos | 6d183ac | 2016-08-29 07:20:33 -0700 | [diff] [blame] | 58 | EXPECT_EQ(report->size(), static_cast<size_t>(0)); |
hbos | 0e6758d | 2016-08-31 07:57:36 -0700 | [diff] [blame] | 59 | report->AddStats(std::unique_ptr<RTCStats>(new RTCTestStats1("a0", 1))); |
| 60 | report->AddStats(std::unique_ptr<RTCStats>(new RTCTestStats1("a1", 2))); |
| 61 | report->AddStats(std::unique_ptr<RTCStats>(new RTCTestStats2("b0", 4))); |
| 62 | report->AddStats(std::unique_ptr<RTCStats>(new RTCTestStats2("b1", 8))); |
| 63 | report->AddStats(std::unique_ptr<RTCStats>(new RTCTestStats1("a2", 16))); |
| 64 | report->AddStats(std::unique_ptr<RTCStats>(new RTCTestStats2("b2", 32))); |
hbos | 6d183ac | 2016-08-29 07:20:33 -0700 | [diff] [blame] | 65 | EXPECT_EQ(report->size(), static_cast<size_t>(6)); |
| 66 | |
| 67 | EXPECT_EQ(report->Get("missing"), nullptr); |
| 68 | EXPECT_EQ(report->Get("a0")->id(), "a0"); |
| 69 | EXPECT_EQ(report->Get("b2")->id(), "b2"); |
| 70 | |
| 71 | std::vector<const RTCTestStats1*> a = report->GetStatsOfType<RTCTestStats1>(); |
| 72 | EXPECT_EQ(a.size(), static_cast<size_t>(3)); |
hbos | 0e6758d | 2016-08-31 07:57:36 -0700 | [diff] [blame] | 73 | int64_t mask = 0; |
hbos | 6d183ac | 2016-08-29 07:20:33 -0700 | [diff] [blame] | 74 | for (const RTCTestStats1* stats : a) |
hbos | 0e6758d | 2016-08-31 07:57:36 -0700 | [diff] [blame] | 75 | mask |= stats->timestamp_us(); |
| 76 | EXPECT_EQ(mask, static_cast<int64_t>(1 | 2 | 16)); |
hbos | 6d183ac | 2016-08-29 07:20:33 -0700 | [diff] [blame] | 77 | |
| 78 | std::vector<const RTCTestStats2*> b = report->GetStatsOfType<RTCTestStats2>(); |
| 79 | EXPECT_EQ(b.size(), static_cast<size_t>(3)); |
| 80 | mask = 0; |
| 81 | for (const RTCTestStats2* stats : b) |
hbos | 0e6758d | 2016-08-31 07:57:36 -0700 | [diff] [blame] | 82 | mask |= stats->timestamp_us(); |
| 83 | EXPECT_EQ(mask, static_cast<int64_t>(4 | 8 | 32)); |
hbos | 6d183ac | 2016-08-29 07:20:33 -0700 | [diff] [blame] | 84 | |
| 85 | EXPECT_EQ(report->GetStatsOfType<RTCTestStats3>().size(), |
| 86 | static_cast<size_t>(0)); |
| 87 | } |
| 88 | |
| 89 | TEST(RTCStatsReport, StatsOrder) { |
hbos | 6ded190 | 2016-11-01 01:50:46 -0700 | [diff] [blame] | 90 | rtc::scoped_refptr<RTCStatsReport> report = RTCStatsReport::Create(1337); |
| 91 | EXPECT_EQ(report->timestamp_us(), 1337u); |
hbos | 0e6758d | 2016-08-31 07:57:36 -0700 | [diff] [blame] | 92 | report->AddStats(std::unique_ptr<RTCStats>(new RTCTestStats1("C", 2))); |
| 93 | report->AddStats(std::unique_ptr<RTCStats>(new RTCTestStats1("D", 3))); |
| 94 | report->AddStats(std::unique_ptr<RTCStats>(new RTCTestStats2("B", 1))); |
| 95 | report->AddStats(std::unique_ptr<RTCStats>(new RTCTestStats2("A", 0))); |
| 96 | report->AddStats(std::unique_ptr<RTCStats>(new RTCTestStats2("E", 4))); |
| 97 | report->AddStats(std::unique_ptr<RTCStats>(new RTCTestStats2("F", 5))); |
| 98 | report->AddStats(std::unique_ptr<RTCStats>(new RTCTestStats2("G", 6))); |
| 99 | int64_t i = 0; |
hbos | 6d183ac | 2016-08-29 07:20:33 -0700 | [diff] [blame] | 100 | for (const RTCStats& stats : *report) { |
hbos | 0e6758d | 2016-08-31 07:57:36 -0700 | [diff] [blame] | 101 | EXPECT_EQ(stats.timestamp_us(), i); |
hbos | 6d183ac | 2016-08-29 07:20:33 -0700 | [diff] [blame] | 102 | ++i; |
| 103 | } |
hbos | 0e6758d | 2016-08-31 07:57:36 -0700 | [diff] [blame] | 104 | EXPECT_EQ(i, static_cast<int64_t>(7)); |
hbos | 6d183ac | 2016-08-29 07:20:33 -0700 | [diff] [blame] | 105 | } |
| 106 | |
Henrik Boström | b619936 | 2018-03-12 10:27:55 +0100 | [diff] [blame] | 107 | TEST(RTCStatsReport, Take) { |
| 108 | rtc::scoped_refptr<RTCStatsReport> report = RTCStatsReport::Create(0); |
| 109 | report->AddStats(std::unique_ptr<RTCStats>(new RTCTestStats1("A", 1))); |
| 110 | report->AddStats(std::unique_ptr<RTCStats>(new RTCTestStats1("B", 2))); |
| 111 | EXPECT_TRUE(report->Get("A")); |
| 112 | EXPECT_EQ(report->size(), 2u); |
| 113 | auto a = report->Take("A"); |
| 114 | EXPECT_TRUE(a); |
| 115 | EXPECT_EQ(report->size(), 1u); |
| 116 | EXPECT_FALSE(report->Get("A")); |
| 117 | EXPECT_FALSE(report->Take("A")); |
| 118 | } |
| 119 | |
hbos | 6d183ac | 2016-08-29 07:20:33 -0700 | [diff] [blame] | 120 | TEST(RTCStatsReport, TakeMembersFrom) { |
hbos | 6ded190 | 2016-11-01 01:50:46 -0700 | [diff] [blame] | 121 | rtc::scoped_refptr<RTCStatsReport> a = RTCStatsReport::Create(1337); |
| 122 | EXPECT_EQ(a->timestamp_us(), 1337u); |
hbos | 0e6758d | 2016-08-31 07:57:36 -0700 | [diff] [blame] | 123 | a->AddStats(std::unique_ptr<RTCStats>(new RTCTestStats1("B", 1))); |
| 124 | a->AddStats(std::unique_ptr<RTCStats>(new RTCTestStats1("C", 2))); |
| 125 | a->AddStats(std::unique_ptr<RTCStats>(new RTCTestStats1("E", 4))); |
hbos | 6ded190 | 2016-11-01 01:50:46 -0700 | [diff] [blame] | 126 | rtc::scoped_refptr<RTCStatsReport> b = RTCStatsReport::Create(1338); |
| 127 | EXPECT_EQ(b->timestamp_us(), 1338u); |
hbos | 0e6758d | 2016-08-31 07:57:36 -0700 | [diff] [blame] | 128 | b->AddStats(std::unique_ptr<RTCStats>(new RTCTestStats1("A", 0))); |
| 129 | b->AddStats(std::unique_ptr<RTCStats>(new RTCTestStats1("D", 3))); |
| 130 | b->AddStats(std::unique_ptr<RTCStats>(new RTCTestStats1("F", 5))); |
hbos | 6d183ac | 2016-08-29 07:20:33 -0700 | [diff] [blame] | 131 | |
| 132 | a->TakeMembersFrom(b); |
| 133 | EXPECT_EQ(b->size(), static_cast<size_t>(0)); |
hbos | 0e6758d | 2016-08-31 07:57:36 -0700 | [diff] [blame] | 134 | int64_t i = 0; |
hbos | 6d183ac | 2016-08-29 07:20:33 -0700 | [diff] [blame] | 135 | for (const RTCStats& stats : *a) { |
hbos | 0e6758d | 2016-08-31 07:57:36 -0700 | [diff] [blame] | 136 | EXPECT_EQ(stats.timestamp_us(), i); |
hbos | 6d183ac | 2016-08-29 07:20:33 -0700 | [diff] [blame] | 137 | ++i; |
| 138 | } |
hbos | 0e6758d | 2016-08-31 07:57:36 -0700 | [diff] [blame] | 139 | EXPECT_EQ(i, static_cast<int64_t>(6)); |
hbos | 6d183ac | 2016-08-29 07:20:33 -0700 | [diff] [blame] | 140 | } |
| 141 | |
| 142 | } // namespace webrtc |