1 // Copyright 2022 The Chromium Authors
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "base/metrics/ranges_manager.h"
6
7 #include <vector>
8
9 #include "testing/gmock/include/gmock/gmock.h"
10 #include "testing/gtest/include/gtest/gtest.h"
11
12 namespace base {
13
14 using testing::UnorderedElementsAre;
15
TEST(RangesManagerTest,GetOrRegisterCanonicalRanges)16 TEST(RangesManagerTest, GetOrRegisterCanonicalRanges) {
17 RangesManager ranges_manager;
18
19 // Create some BucketRanges. We call |ResetChecksum| to calculate and set
20 // their checksums. Checksums are used to validate integrity (and test for
21 // non-equivalence) and should be computed after a BucketRanges is fully
22 // initialized. Note that BucketRanges are initialized with 0 for all ranges,
23 // i.e., all buckets will be [0, 0).
24 BucketRanges* ranges1 = new BucketRanges(3);
25 ranges1->ResetChecksum();
26 BucketRanges* ranges2 = new BucketRanges(4);
27 ranges2->ResetChecksum();
28
29 // Register new ranges.
30 EXPECT_EQ(ranges1, ranges_manager.GetOrRegisterCanonicalRanges(ranges1));
31 EXPECT_EQ(ranges2, ranges_manager.GetOrRegisterCanonicalRanges(ranges2));
32 EXPECT_THAT(ranges_manager.GetBucketRanges(),
33 UnorderedElementsAre(ranges1, ranges2));
34
35 // Register |ranges1| again. The registered BucketRanges set should not change
36 // as |ranges1| is already registered.
37 EXPECT_EQ(ranges1, ranges_manager.GetOrRegisterCanonicalRanges(ranges1));
38 EXPECT_THAT(ranges_manager.GetBucketRanges(),
39 UnorderedElementsAre(ranges1, ranges2));
40
41 // Make sure |ranges1| still exists, and is the same as what we expect (all
42 // ranges are 0).
43 ASSERT_EQ(3u, ranges1->size());
44 EXPECT_EQ(0, ranges1->range(0));
45 EXPECT_EQ(0, ranges1->range(1));
46 EXPECT_EQ(0, ranges1->range(2));
47
48 // Register a new |ranges3| that is equivalent to |ranges1| (same ranges). If
49 // GetOrRegisterCanonicalRanges() returns a different object than the param
50 // (as asserted here), we are responsible for deleting the object (below).
51 BucketRanges* ranges3 = new BucketRanges(3);
52 ranges3->ResetChecksum();
53 ASSERT_EQ(ranges1, ranges_manager.GetOrRegisterCanonicalRanges(ranges3));
54 delete ranges3;
55 EXPECT_THAT(ranges_manager.GetBucketRanges(),
56 UnorderedElementsAre(ranges1, ranges2));
57 }
58
TEST(RangesManagerTest,ReleaseBucketRangesOnDestroy)59 TEST(RangesManagerTest, ReleaseBucketRangesOnDestroy) {
60 std::unique_ptr<RangesManager> ranges_manager =
61 std::make_unique<RangesManager>();
62
63 // Create a BucketRanges. We call |ResetChecksum| to calculate and set its
64 // checksum. Checksums are used to validate integrity (and test for
65 // non-equivalence) and should be computed after a BucketRanges is fully
66 // initialized. Note that BucketRanges are initialized with 0 for all ranges,
67 // i.e., all buckets will be [0, 0).
68 BucketRanges* ranges = new BucketRanges(1);
69 ranges->ResetChecksum();
70
71 // Register new range.
72 EXPECT_EQ(ranges, ranges_manager->GetOrRegisterCanonicalRanges(ranges));
73 EXPECT_THAT(ranges_manager->GetBucketRanges(), UnorderedElementsAre(ranges));
74
75 // Explicitly destroy |ranges_manager|.
76 ranges_manager.reset();
77
78 // LeakSanitizer (lsan) bots will verify that |ranges| will be properly
79 // released after destroying |ranges_manager|.
80 }
81
82 } // namespace base
83