predictors: Add prefetcher histograms for speculative prefetch.

chrome/browser/predictors/resource_prefetcher_unittest.cc

 // Copyright 2014 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/predictors/resource_prefetcher.h"
 
 #include <stddef.h>
 
 #include <memory>
 #include <string>
 #include <utility>
 
 #include "base/macros.h"
 #include "base/memory/ref_counted.h"
 #include "base/message_loop/message_loop.h"
+#include "base/test/histogram_tester.h"
 #include "chrome/browser/predictors/resource_prefetch_predictor_test_util.h"
 #include "chrome/browser/predictors/resource_prefetcher_manager.h"
 #include "chrome/test/base/testing_profile.h"
 #include "content/public/test/test_browser_thread.h"
 #include "net/base/load_flags.h"
 #include "net/url_request/redirect_info.h"
 #include "net/url_request/url_request.h"
 #include "net/url_request/url_request_test_util.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
@@ skipping 12 matching lines @@
                          const GURL& main_frame_url,
                          const std::vector<GURL>& urls)
       : ResourcePrefetcher(delegate, config, main_frame_url, urls) {}
 
   ~TestResourcePrefetcher() override {}
 
   MOCK_METHOD1(StartURLRequest, void(net::URLRequest* request));
 
   void ReadFullResponse(net::URLRequest* request) override {
     EXPECT_TRUE(request->load_flags() & net::LOAD_PREFETCH);
+    RequestComplete(request);
     FinishRequest(request);
   }
 
  private:
   DISALLOW_COPY_AND_ASSIGN(TestResourcePrefetcher);
 };
 
 
 // Delegate for ResourcePrefetcher.
 class TestResourcePrefetcherDelegate : public ResourcePrefetcher::Delegate {
@@ skipping 100 matching lines @@
                             GURL("http://yahoo.com/resource2.png"),
                             GURL("http://yahoo.com/resource3.png"),
                             GURL("http://m.google.com/resource1.jpg"),
                             GURL("http://www.google.com/resource3.html"),
                             GURL("http://m.google.com/resource2.html"),
                             GURL("http://m.google.com/resource3.css"),
                             GURL("http://m.google.com/resource4.png"),
                             GURL("http://yahoo.com/resource4.png"),
                             GURL("http://yahoo.com/resource5.png")};
 
-  NavigationID navigation_id = CreateNavigationID(1, main_frame_url.spec());
-
   prefetcher_.reset(new TestResourcePrefetcher(&prefetcher_delegate_, config_,
                                                main_frame_url, urls));
 
   // Starting the prefetcher maxes out the number of possible requests.
   AddStartUrlRequestExpectation("http://www.google.com/resource1.html");
   AddStartUrlRequestExpectation("http://www.google.com/resource2.png");
   AddStartUrlRequestExpectation("http://yahoo.com/resource1.png");
   AddStartUrlRequestExpectation("http://yahoo.com/resource2.png");
   AddStartUrlRequestExpectation("http://m.google.com/resource1.jpg");
 
@@ skipping 50 matching lines @@
 
 TEST_F(ResourcePrefetcherTest, TestPrefetcherStopped) {
   GURL main_frame_url("http://www.google.com");
   std::vector<GURL> urls = {GURL("http://www.google.com/resource1.html"),
                             GURL("http://www.google.com/resource2.png"),
                             GURL("http://yahoo.com/resource1.png"),
                             GURL("http://yahoo.com/resource2.png"),
                             GURL("http://yahoo.com/resource3.png"),
                             GURL("http://m.google.com/resource1.jpg")};
 
-  NavigationID navigation_id = CreateNavigationID(1, main_frame_url.spec());
-
   prefetcher_.reset(new TestResourcePrefetcher(&prefetcher_delegate_, config_,
                                                main_frame_url, urls));
 
   // Starting the prefetcher maxes out the number of possible requests.
   AddStartUrlRequestExpectation("http://www.google.com/resource1.html");
   AddStartUrlRequestExpectation("http://www.google.com/resource2.png");
   AddStartUrlRequestExpectation("http://yahoo.com/resource1.png");
   AddStartUrlRequestExpectation("http://yahoo.com/resource2.png");
   AddStartUrlRequestExpectation("http://m.google.com/resource1.jpg");
 
@@ skipping 15 matching lines @@
   CheckPrefetcherState(1, 1, 1);
 
   // Expect the final call.
   EXPECT_CALL(prefetcher_delegate_,
               ResourcePrefetcherFinished(Eq(prefetcher_.get())));
 
   OnResponse("http://m.google.com/resource1.jpg");
   CheckPrefetcherState(0, 1, 0);
 }
 
+TEST_F(ResourcePrefetcherTest, TestHistogramsCollected) {
+  base::HistogramTester histogram_tester;
+  GURL main_frame_url("http://www.google.com");
+  std::vector<GURL> urls = {GURL("http://www.google.com/resource1.png"),
+                            GURL("http://www.google.com/resource2.png"),
+                            GURL("http://www.google.com/resource3.png"),
+                            GURL("http://www.google.com/resource4.png"),
+                            GURL("http://www.google.com/resource5.png"),
+                            GURL("http://www.google.com/resource6.png")};
+
+  prefetcher_ = base::MakeUnique<TestResourcePrefetcher>(
+      &prefetcher_delegate_, config_, main_frame_url, urls);
+
+  // Starting the prefetcher maxes out the number of possible requests.
+  AddStartUrlRequestExpectation("http://www.google.com/resource1.png");
+  AddStartUrlRequestExpectation("http://www.google.com/resource2.png");
+  AddStartUrlRequestExpectation("http://www.google.com/resource3.png");
+
+  prefetcher_->Start();
+
+  AddStartUrlRequestExpectation("http://www.google.com/resource4.png");
+  OnResponse("http://www.google.com/resource1.png");
+  histogram_tester.ExpectTotalCount(
+      internal::kResourcePrefetchPredictorCachePatternHistogram, 1);
+
+  // Failed prefetches aren't counted.
+  AddStartUrlRequestExpectation("http://www.google.com/resource5.png");
+  OnReceivedRedirect("http://www.google.com/resource2.png");
+
+  AddStartUrlRequestExpectation("http://www.google.com/resource6.png");
+  OnAuthRequired("http://www.google.com/resource3.png");
+
+  OnCertificateRequested("http://www.google.com/resource4.png");
+
+  OnSSLCertificateError("http://www.google.com/resource5.png");
+  histogram_tester.ExpectTotalCount(
+      internal::kResourcePrefetchPredictorCachePatternHistogram, 1);
+
+  // Expect the final call.
+  EXPECT_CALL(prefetcher_delegate_,
+              ResourcePrefetcherFinished(Eq(prefetcher_.get())));
+
+  OnResponse("http://www.google.com/resource6.png");
+  histogram_tester.ExpectTotalCount(
+      internal::kResourcePrefetchPredictorCachePatternHistogram, 2);
+  histogram_tester.ExpectBucketCount(
+      internal::kResourcePrefetchPredictorPrefetchedCountHistogram, 2, 1);
+  histogram_tester.ExpectTotalCount(
+      internal::kResourcePrefetchPredictorPrefetchedSizeHistogram, 1);
+}
+
 }  // namespace predictors