base/profiler/sample_metadata_unittest.cc

// Copyright 2019 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "base/profiler/sample_metadata.h"

#include "base/metrics/metrics_hashes.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace base {

TEST(SampleMetadataTest, ScopedSampleMetadata) {
  ProfileBuilder::MetadataItemArray items;
  ASSERT_EQ(0u, GetSampleMetadataRecorder()->CreateMetadataProvider()->GetItems(
                    &items));

  {
    ScopedSampleMetadata m("myname", 100);

    ASSERT_EQ(1u,
              GetSampleMetadataRecorder()->CreateMetadataProvider()->GetItems(
                  &items));
    EXPECT_EQ(HashMetricName("myname"), items[0].name_hash);
    EXPECT_FALSE(items[0].key.has_value());
    EXPECT_EQ(100, items[0].value);
  }

  ASSERT_EQ(0u, GetSampleMetadataRecorder()->CreateMetadataProvider()->GetItems(
                    &items));
}

TEST(SampleMetadataTest, ScopedSampleMetadataWithKey) {
  ProfileBuilder::MetadataItemArray items;
  ASSERT_EQ(0u, GetSampleMetadataRecorder()->CreateMetadataProvider()->GetItems(
                    &items));

  {
    ScopedSampleMetadata m("myname", 10, 100);

    ASSERT_EQ(1u,
              GetSampleMetadataRecorder()->CreateMetadataProvider()->GetItems(
                  &items));
    EXPECT_EQ(HashMetricName("myname"), items[0].name_hash);
    ASSERT_TRUE(items[0].key.has_value());
    EXPECT_EQ(10, *items[0].key);
    EXPECT_EQ(100, items[0].value);
  }

  ASSERT_EQ(0u, GetSampleMetadataRecorder()->CreateMetadataProvider()->GetItems(
                    &items));
}

TEST(SampleMetadataTest, SampleMetadata) {
  ProfileBuilder::MetadataItemArray items;
  ASSERT_EQ(0u, GetSampleMetadataRecorder()->CreateMetadataProvider()->GetItems(
                    &items));

  SetSampleMetadata("myname", 100);
  ASSERT_EQ(1u, GetSampleMetadataRecorder()->CreateMetadataProvider()->GetItems(
                    &items));
  EXPECT_EQ(HashMetricName("myname"), items[0].name_hash);
  EXPECT_FALSE(items[0].key.has_value());
  EXPECT_EQ(100, items[0].value);

  RemoveSampleMetadata("myname");
  ASSERT_EQ(0u, GetSampleMetadataRecorder()->CreateMetadataProvider()->GetItems(
                    &items));
}

TEST(SampleMetadataTest, SampleMetadataWithKey) {
  ProfileBuilder::MetadataItemArray items;
  ASSERT_EQ(0u, GetSampleMetadataRecorder()->CreateMetadataProvider()->GetItems(
                    &items));

  SetSampleMetadata("myname", 10, 100);
  ASSERT_EQ(1u, GetSampleMetadataRecorder()->CreateMetadataProvider()->GetItems(
                    &items));
  EXPECT_EQ(HashMetricName("myname"), items[0].name_hash);
  ASSERT_TRUE(items[0].key.has_value());
  EXPECT_EQ(10, *items[0].key);
  EXPECT_EQ(100, items[0].value);

  RemoveSampleMetadata("myname", 10);
  ASSERT_EQ(0u, GetSampleMetadataRecorder()->CreateMetadataProvider()->GetItems(
                    &items));
}

}  // namespace base