about summary refs log tree commit diff
path: root/absl/container/internal/hashtablez_force_sampling_test.cc
blob: 9ff1046a9ad303eebab35068658361088027c1a8 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
// Copyright 2018 The Abseil Authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#include <cstddef>

#include "gmock/gmock.h"
#include "gtest/gtest.h"
#include "absl/container/internal/hashtablez_sampler.h"

namespace absl {
namespace container_internal {

class HashtablezInfoHandlePeer {
 public:
  static bool IsSampled(const HashtablezInfoHandle& h) {
    return h.info_ != nullptr;
  }
};

namespace {

bool samples[3]{true, true, true};

// We do this test in a global object to test that this works even before main.
struct Global {
  Global() {
    // By default it is sampled.
    samples[0] = HashtablezInfoHandlePeer::IsSampled(Sample());

    // Even with a large parameter, it is sampled.
    SetHashtablezSampleParameter(100);
    samples[1] = HashtablezInfoHandlePeer::IsSampled(Sample());

    // Even if we turn it off, it is still sampled.
    SetHashtablezEnabled(false);
    samples[2] = HashtablezInfoHandlePeer::IsSampled(Sample());
  }
} global;

TEST(kAbslContainerInternalSampleEverything, Works) {
  EXPECT_THAT(samples, testing::Each(true));
  EXPECT_TRUE(kAbslContainerInternalSampleEverything);
  // One more after main()
  EXPECT_TRUE(HashtablezInfoHandlePeer::IsSampled(Sample()));
}

}  // namespace
}  // namespace container_internal
}  // namespace absl