third_party/WebKit/Source/core/frame/UseCounterTest.cpp

   1 // Copyright 2015 The Chromium Authors. All rights reserved.
   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 "config.h"
   6 #include "core/frame/UseCounter.h"
   7
   8 #include <gtest/gtest.h>
   9
  10 namespace blink {
  11
  12 class UseCounterTest : public ::testing::Test {
  13 protected:
  14     bool hasRecordedMeasurement(const UseCounter& useCounter, UseCounter::Feature feature)
  15     {
  16         return useCounter.hasRecordedMeasurement(feature);
  17     }
  18
  19     void recordMeasurement(UseCounter& useCounter, UseCounter::Feature feature)
  20     {
  21         useCounter.recordMeasurement(feature);
  22     }
  23
  24     void setUseCounterMuted(UseCounter& useCounter, bool muteCount)
  25     {
  26         UseCounter::m_muteCount = muteCount;
  27     }
  28 };
  29
  30 TEST_F(UseCounterTest, RecordingMeasurements)
  31 {
  32     UseCounter useCounter;
  33     for (unsigned feature = 0; feature < UseCounter::NumberOfFeatures; feature++) {
  34         if (feature != UseCounter::Feature::PageDestruction) {
  35             EXPECT_FALSE(hasRecordedMeasurement(useCounter, static_cast<UseCounter::Feature>(feature)));
  36             recordMeasurement(useCounter, static_cast<UseCounter::Feature>(feature));
  37             EXPECT_TRUE(hasRecordedMeasurement(useCounter, static_cast<UseCounter::Feature>(feature)));
  38         }
  39     }
  40 }
  41
  42 TEST_F(UseCounterTest, MultipleMeasurements)
  43 {
  44     UseCounter useCounter;
  45     for (unsigned feature = 0; feature < UseCounter::NumberOfFeatures; feature++) {
  46         if (feature != UseCounter::Feature::PageDestruction) {
  47             recordMeasurement(useCounter, static_cast<UseCounter::Feature>(feature));
  48             recordMeasurement(useCounter, static_cast<UseCounter::Feature>(feature));
  49             EXPECT_TRUE(hasRecordedMeasurement(useCounter, static_cast<UseCounter::Feature>(feature)));
  50         }
  51     }
  52 }
  53
  54 TEST_F(UseCounterTest, InspectorDisablesMeasurement)
  55 {
  56     UseCounter useCounter;
  57
  58     // The specific feature we use here isn't important.
  59     UseCounter::Feature feature = UseCounter::Feature::SVGSMILElementInDocument;
  60
  61     EXPECT_FALSE(hasRecordedMeasurement(useCounter, feature));
  62
  63     UseCounter::muteForInspector();
  64     recordMeasurement(useCounter, feature);
  65     EXPECT_FALSE(hasRecordedMeasurement(useCounter, feature));
  66
  67     UseCounter::muteForInspector();
  68     recordMeasurement(useCounter, feature);
  69     EXPECT_FALSE(hasRecordedMeasurement(useCounter, feature));
  70
  71     UseCounter::unmuteForInspector();
  72     recordMeasurement(useCounter, feature);
  73     EXPECT_FALSE(hasRecordedMeasurement(useCounter, feature));
  74
  75     UseCounter::unmuteForInspector();
  76     recordMeasurement(useCounter, feature);
  77     EXPECT_TRUE(hasRecordedMeasurement(useCounter, feature));
  78 }
  79
  80 } // namespace blink