Add histograms for data usage

chrome/browser/android/data_usage/data_use_matcher_unittest.cc

 // Copyright 2015 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 "chrome/browser/android/data_usage/data_use_matcher.h"
 
 #include <stddef.h>
 #include <stdint.h>
 
 #include <string>
 #include <vector>
 
 #include "base/macros.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/stringprintf.h"
+#include "base/test/histogram_tester.h"
 #include "base/time/tick_clock.h"
 #include "base/time/time.h"
 #include "chrome/browser/android/data_usage/external_data_use_observer.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/test/test_browser_thread_bundle.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "url/gurl.h"
 
 namespace {
 
+const char kUMAMatchingRulesCountValidHistogram[] =
+    "DataUsage.MatchingRulesCount.Valid";
+const char kUMAMatchingRulesCountInvalidHistogram[] =
+    "DataUsage.MatchingRulesCount.Invalid";
+const char kUMAURLRegexMatchDurationHistogram[] =
+    "DataUsage.Perf.URLRegexMatchDuration";
+
 const char kRegexFoo[] = "http://foo.com/";
 const char kLabelFoo[] = "label_foo";
 const char kAppFoo[] = "com.example.foo";
 
 const uint32_t kDefaultMatchingRuleExpirationDurationSeconds =
     60 * 60 * 24;  // 24 hours.
 
 class NowTestTickClock : public base::TickClock {
  public:
   NowTestTickClock() {}
@@ skipping 51 matching lines @@
   content::TestBrowserThreadBundle thread_bundle_;
   DataUseMatcher data_use_matcher_;
   DISALLOW_COPY_AND_ASSIGN(DataUseMatcherTest);
 };
 
 TEST_F(DataUseMatcherTest, SingleRegex) {
   const struct {
     std::string url;
     std::string regex;
     bool expect_match;
+    int expect_count_valid_rules;
+    int expect_count_url_match_duration_samples;
   } tests[] = {
-      {"http://www.google.com", "http://www.google.com/", true},
-      {"http://www.Google.com", "http://www.google.com/", true},
-      {"http://www.googleacom", "http://www.google.com/", true},
-      {"http://www.googleaacom", "http://www.google.com/", false},
-      {"http://www.google.com", "https://www.google.com/", false},
+      {"http://www.google.com", "http://www.google.com/", true, 1, 1},
+      {"http://www.Google.com", "http://www.google.com/", true, 1, 1},
+      {"http://www.googleacom", "http://www.google.com/", true, 1, 1},
+      {"http://www.googleaacom", "http://www.google.com/", false, 1, 1},
+      {"http://www.google.com", "https://www.google.com/", false, 1, 1},
       {"http://www.google.com", "{http|https}://www[.]google[.]com/search.*",
-       false},
+       false, 1, 1},
       {"https://www.google.com/search=test",
-       "https://www[.]google[.]com/search.*", true},
+       "https://www[.]google[.]com/search.*", true, 1, 1},
       {"https://www.googleacom/search=test",
-       "https://www[.]google[.]com/search.*", false},
+       "https://www[.]google[.]com/search.*", false, 1, 1},
       {"https://www.google.com/Search=test",
-       "https://www[.]google[.]com/search.*", true},
-      {"www.google.com", "http://www.google.com", false},
-      {"www.google.com:80", "http://www.google.com", false},
-      {"http://www.google.com:80", "http://www.google.com", false},
-      {"http://www.google.com:80/", "http://www.google.com/", true},
-      {"", "http://www.google.com", false},
-      {"", "", false},
-      {"https://www.google.com", "http://www.google.com", false},
+       "https://www[.]google[.]com/search.*", true, 1, 1},
+      {"www.google.com", "http://www.google.com", false, 1, 0},
+      {"www.google.com:80", "http://www.google.com", false, 1, 1},
+      {"http://www.google.com:80", "http://www.google.com", false, 1, 1},
+      {"http://www.google.com:80/", "http://www.google.com/", true, 1, 1},
+      {"", "http://www.google.com", false, 1, 0},
+      {"", "", false, 0, 0},
+      {"https://www.google.com", "http://www.google.com", false, 1, 1},
+      {"https://www.google.com", "[", false, 0},
+      {"https://www.google.com", "]", false, 1, 1},
   };
 
   for (size_t i = 0; i < arraysize(tests); ++i) {
+    base::HistogramTester histogram_tester;
     std::string label("");
     RegisterURLRegexes(
         // App package name not specified in the matching rule.
         std::vector<std::string>(1, std::string()),
         std::vector<std::string>(1, tests[i].regex),
         std::vector<std::string>(1, "label"));
+    histogram_tester.ExpectUniqueSample(kUMAMatchingRulesCountValidHistogram,
+                                        tests[i].expect_count_valid_rules, 1);
+    histogram_tester.ExpectUniqueSample(kUMAMatchingRulesCountInvalidHistogram,
+                                        1 - tests[i].expect_count_valid_rules,
+                                        1);
     EXPECT_EQ(tests[i].expect_match,
               data_use_matcher()->MatchesURL(GURL(tests[i].url), &label))
         << i;
+    histogram_tester.ExpectTotalCount(
+        kUMAURLRegexMatchDurationHistogram,
+        tests[i].expect_count_url_match_duration_samples);
 
     // Verify label matches the expected label.
     std::string expected_label = "";
     if (tests[i].expect_match)
       expected_label = "label";
 
     EXPECT_EQ(expected_label, label);
     EXPECT_FALSE(data_use_matcher()->MatchesAppPackageName(
         "com.example.helloworld", &label))
         << i;
     // Empty package name should not match against empty package name in the
     // matching rule.
     EXPECT_FALSE(
         data_use_matcher()->MatchesAppPackageName(std::string(), &label))
         << i;
   }
 }
 
 TEST_F(DataUseMatcherTest, TwoRegex) {
   const struct {
     std::string url;
     std::string regex1;
     std::string regex2;
     bool expect_match;
+    int expect_count_valid_rules;
+    int expect_count_url_match_duration_samples;
   } tests[] = {
       {"http://www.google.com", "http://www.google.com/",
-       "https://www.google.com/", true},
+       "https://www.google.com/", true, 1, 1},
       {"http://www.googleacom", "http://www.google.com/",
-       "http://www.google.com/", true},
+       "http://www.google.com/", true, 1, 1},
       {"https://www.google.com", "http://www.google.com/",
-       "https://www.google.com/", true},
+       "https://www.google.com/", true, 1, 1},
       {"https://www.googleacom", "http://www.google.com/",
-       "https://www.google.com/", true},
+       "https://www.google.com/", true, 1, 1},
       {"http://www.google.com", "{http|https}://www[.]google[.]com/search.*",
-       "", false},
+       "", false, 1, 1},
       {"http://www.google.com/search=test",
        "http://www[.]google[.]com/search.*",
-       "https://www[.]google[.]com/search.*", true},
+       "https://www[.]google[.]com/search.*", true, 1, 1},
       {"https://www.google.com/search=test",
        "http://www[.]google[.]com/search.*",
-       "https://www[.]google[.]com/search.*", true},
+       "https://www[.]google[.]com/search.*", true, 1, 1},
       {"http://google.com/search=test", "http://www[.]google[.]com/search.*",
-       "https://www[.]google[.]com/search.*", false},
+       "https://www[.]google[.]com/search.*", false, 1, 1},
       {"https://www.googleacom/search=test", "",
-       "https://www[.]google[.]com/search.*", false},
+       "https://www[.]google[.]com/search.*", false, 1, 1},
       {"https://www.google.com/Search=test", "",
-       "https://www[.]google[.]com/search.*", true},
-      {"www.google.com", "http://www.google.com", "", false},
-      {"www.google.com:80", "http://www.google.com", "", false},
-      {"http://www.google.com:80", "http://www.google.com", "", false},
-      {"", "http://www.google.com", "", false},
-      {"https://www.google.com", "http://www.google.com", "", false},
+       "https://www[.]google[.]com/search.*", true, 1, 1},
+      {"www.google.com", "http://www.google.com", "", false, 1, 0},
+      {"www.google.com:80", "http://www.google.com", "", false, 1, 1},
+      {"http://www.google.com:80", "http://www.google.com", "", false, 1, 1},
+      {"", "http://www.google.com", "", false, 1, 0},
+      {"https://www.google.com", "http://www.google.com", "", false, 1, 1},
   };
 
   for (size_t i = 0; i < arraysize(tests); ++i) {
+    base::HistogramTester histogram_tester;
     std::string got_label("");
     std::vector<std::string> url_regexes;
     url_regexes.push_back(tests[i].regex1 + "|" + tests[i].regex2);
     const std::string label("label");
     RegisterURLRegexes(
         std::vector<std::string>(url_regexes.size(), "com.example.helloworld"),
         url_regexes, std::vector<std::string>(url_regexes.size(), label));
+    histogram_tester.ExpectTotalCount(kUMAMatchingRulesCountValidHistogram, 1);
+    histogram_tester.ExpectTotalCount(kUMAMatchingRulesCountInvalidHistogram,
+                                      1);
+    histogram_tester.ExpectUniqueSample(kUMAMatchingRulesCountValidHistogram,
+                                        tests[i].expect_count_valid_rules, 1);
+    histogram_tester.ExpectUniqueSample(kUMAMatchingRulesCountInvalidHistogram,
+                                        1 - tests[i].expect_count_valid_rules,
+                                        1);
     EXPECT_EQ(tests[i].expect_match,
               data_use_matcher()->MatchesURL(GURL(tests[i].url), &got_label))
         << i;
+    histogram_tester.ExpectTotalCount(
+        kUMAURLRegexMatchDurationHistogram,
+        tests[i].expect_count_url_match_duration_samples);
     const std::string expected_label =
         tests[i].expect_match ? label : std::string();
     EXPECT_EQ(expected_label, got_label);
 
     EXPECT_TRUE(data_use_matcher()->MatchesAppPackageName(
         "com.example.helloworld", &got_label))
         << i;
     EXPECT_EQ(label, got_label);
   }
 }
 
 TEST_F(DataUseMatcherTest, MultipleRegex) {
+  base::HistogramTester histogram_tester;
   std::vector<std::string> url_regexes;
   url_regexes.push_back(
       "https?://www[.]google[.]com/#q=.*|https?://www[.]google[.]com[.]ph/"
       "#q=.*|https?://www[.]google[.]com[.]ph/[?]gws_rd=ssl#q=.*");
   RegisterURLRegexes(
       std::vector<std::string>(url_regexes.size(), std::string()), url_regexes,
       std::vector<std::string>(url_regexes.size(), "label"));
+  histogram_tester.ExpectUniqueSample(kUMAMatchingRulesCountValidHistogram, 1,
+                                      1);
+  histogram_tester.ExpectUniqueSample(kUMAMatchingRulesCountInvalidHistogram, 0,
+                                      1);
 
   const struct {
     std::string url;
     bool expect_match;
   } tests[] = {
       {"", false},
       {"http://www.google.com", false},
       {"http://www.googleacom", false},
       {"https://www.google.com", false},
       {"https://www.googleacom", false},
@@ skipping 54 matching lines @@
       std::vector<std::string>(url_regexes.size(), std::string()), url_regexes,
       std::vector<std::string>(url_regexes.size(), "label"));
   EXPECT_FALSE(data_use_matcher()->MatchesURL(GURL(""), &label));
   EXPECT_FALSE(data_use_matcher()->MatchesURL(
       GURL("http://www.google.com#q=abc"), &label));
   EXPECT_TRUE(data_use_matcher()->MatchesURL(
       GURL("http://www.google.co.in#q=abc"), &label));
 }
 
 TEST_F(DataUseMatcherTest, MultipleAppPackageName) {
+  base::HistogramTester histogram_tester;
   std::vector<std::string> url_regexes;
   url_regexes.push_back(
       "http://www[.]foo[.]com/#q=.*|https://www[.]foo[.]com/#q=.*");
   url_regexes.push_back(
       "http://www[.]bar[.]com/#q=.*|https://www[.]bar[.]com/#q=.*");
   url_regexes.push_back("");
 
   std::vector<std::string> labels;
   const char kLabelBar[] = "label_bar";
   const char kLabelBaz[] = "label_baz";
   labels.push_back(kLabelFoo);
   labels.push_back(kLabelBar);
   labels.push_back(kLabelBaz);
 
   std::vector<std::string> app_package_names;
   const char kAppBar[] = "com.example.bar";
   const char kAppBaz[] = "com.example.baz";
   app_package_names.push_back(kAppFoo);
   app_package_names.push_back(kAppBar);
   app_package_names.push_back(kAppBaz);
 
   RegisterURLRegexes(app_package_names, url_regexes, labels);
+  histogram_tester.ExpectUniqueSample(kUMAMatchingRulesCountValidHistogram, 3,
+                                      1);
+  histogram_tester.ExpectUniqueSample(kUMAMatchingRulesCountInvalidHistogram, 0,
+                                      1);
 
   // Test if labels are matched properly for app package names.
   std::string got_label;
   EXPECT_TRUE(data_use_matcher()->MatchesAppPackageName(kAppFoo, &got_label));
   EXPECT_EQ(kLabelFoo, got_label);
 
   EXPECT_TRUE(data_use_matcher()->MatchesAppPackageName(kAppBar, &got_label));
   EXPECT_EQ(kLabelBar, got_label);
 
   EXPECT_TRUE(data_use_matcher()->MatchesAppPackageName(kAppBaz, &got_label));
@@ skipping 60 matching lines @@
         << test.app_package_name;
     DCHECK_EQ(base::TimeTicks::UnixEpoch() + test.expected_expiration_duration,
               expiration)
         << test.app_package_name;
   }
 }
 
 // Tests if the expiration time encoded as milliseconds since epoch is parsed
 // correctly.
 TEST_F(DataUseMatcherTest, EncodeExpirationTimeInPackageName) {
+  base::HistogramTester histogram_tester;
   NowTestTickClock* tick_clock = new NowTestTickClock();
 
   // |tick_clock| will be owned by |data_use_matcher_|.
   data_use_matcher()->tick_clock_.reset(tick_clock);
 
   std::vector<std::string> url_regexes, labels, app_package_names;
   url_regexes.push_back(kRegexFoo);
   labels.push_back(kLabelFoo);
 
-  // Set current time to to Epoch.
+  // Set current time to Epoch.
   tick_clock->set_now_ticks(base::TimeTicks::UnixEpoch());
 
   app_package_names.push_back(base::StringPrintf("%s|%d", kAppFoo, 10000));
   RegisterURLRegexes(app_package_names, url_regexes, labels);
   EXPECT_FALSE(IsExpired(0));
+  histogram_tester.ExpectUniqueSample(kUMAMatchingRulesCountValidHistogram, 1,
+                                      1);
+  histogram_tester.ExpectUniqueSample(kUMAMatchingRulesCountInvalidHistogram, 0,
+                                      1);
   // Fast forward 10 seconds, and matching rule expires.
   tick_clock->set_now_ticks(base::TimeTicks::UnixEpoch() +
                             base::TimeDelta::FromMilliseconds(10000 + 1));
   EXPECT_TRUE(IsExpired(0));
 
   // Empty app package name.
   app_package_names.clear();
   app_package_names.push_back(base::StringPrintf("|%d", 20000));
   RegisterURLRegexes(app_package_names, url_regexes, labels);
   EXPECT_FALSE(IsExpired(0));
   // Fast forward 20 seconds, and matching rule expires.
   tick_clock->set_now_ticks(base::TimeTicks::UnixEpoch() +
                             base::TimeDelta::FromMilliseconds(20000 + 1));
   EXPECT_TRUE(IsExpired(0));
 }
 
 // Tests if the expiration time encoded in Java format is parsed correctly.
 TEST_F(DataUseMatcherTest, EncodeJavaExpirationTimeInPackageName) {
+  base::HistogramTester histogram_tester;
   std::vector<std::string> url_regexes, labels, app_package_names;
   url_regexes.push_back(kRegexFoo);
   labels.push_back(kLabelFoo);
 
   base::TimeTicks start_ticks = base::TimeTicks::Now();
   base::Time expiration_time =
       base::Time::Now() + base::TimeDelta::FromMilliseconds(10000);
 
   app_package_names.push_back(base::StringPrintf(
       "%s|%lld", kAppFoo,
       static_cast<long long int>(expiration_time.ToJavaTime())));
   RegisterURLRegexes(app_package_names, url_regexes, labels);
   EXPECT_FALSE(IsExpired(0));
+  histogram_tester.ExpectUniqueSample(kUMAMatchingRulesCountValidHistogram, 1,
+                                      1);
+  histogram_tester.ExpectUniqueSample(kUMAMatchingRulesCountInvalidHistogram, 0,
+                                      1);
 
   // Check if expiration duration is close to 10 seconds.
   EXPECT_GE(base::TimeDelta::FromMilliseconds(10001),
             data_use_matcher()->matching_rules_[0]->expiration() -
                 data_use_matcher()->tick_clock_->NowTicks());
   EXPECT_LE(base::TimeDelta::FromMilliseconds(9999),
             data_use_matcher()->matching_rules_[0]->expiration() - start_ticks);
 }
 
 // Tests that expired matching rules are ignored by MatchesURL and
 // MatchesAppPackageName.
 TEST_F(DataUseMatcherTest, MatchesIgnoresExpiredRules) {
+  base::HistogramTester histogram_tester;
   std::vector<std::string> url_regexes, labels, app_package_names;
   std::string got_label;
   NowTestTickClock* tick_clock = new NowTestTickClock();
 
   // |tick_clock| will be owned by |data_use_matcher_|.
   data_use_matcher()->tick_clock_.reset(tick_clock);
   tick_clock->set_now_ticks(base::TimeTicks::UnixEpoch());
 
   url_regexes.push_back(kRegexFoo);
   labels.push_back(kLabelFoo);
   app_package_names.push_back(base::StringPrintf("%s|%d", kAppFoo, 10000));
   RegisterURLRegexes(app_package_names, url_regexes, labels);
+  histogram_tester.ExpectUniqueSample(kUMAMatchingRulesCountValidHistogram, 1,
+                                      1);
+  histogram_tester.ExpectUniqueSample(kUMAMatchingRulesCountInvalidHistogram, 0,
+                                      1);
 
   tick_clock->set_now_ticks(base::TimeTicks::UnixEpoch() +
                             base::TimeDelta::FromMilliseconds(1));
 
   EXPECT_FALSE(IsExpired(0));
   EXPECT_TRUE(data_use_matcher()->MatchesURL(GURL(kRegexFoo), &got_label));
   EXPECT_EQ(kLabelFoo, got_label);
   EXPECT_TRUE(data_use_matcher()->MatchesAppPackageName(kAppFoo, &got_label));
   EXPECT_EQ(kLabelFoo, got_label);
 
   // Advance time to make it expired.
   tick_clock->set_now_ticks(base::TimeTicks::UnixEpoch() +
                             base::TimeDelta::FromMilliseconds(10001));
 
   EXPECT_TRUE(IsExpired(0));
   EXPECT_FALSE(data_use_matcher()->MatchesURL(GURL(kRegexFoo), &got_label));
   EXPECT_FALSE(data_use_matcher()->MatchesAppPackageName(kAppFoo, &got_label));
 }
 
 }  // namespace android
 
 }  // namespace chrome