Add UMA to determine how often we offline a page with previews.

chrome/browser/offline_pages/background_loader_offliner_unittest.cc

 // Copyright 2016 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/offline_pages/background_loader_offliner.h"
 
 #include "base/bind.h"
 #include "base/run_loop.h"
 #include "base/test/histogram_tester.h"
 #include "base/test/scoped_feature_list.h"
 #include "base/test/scoped_mock_time_message_loop_task_runner.h"
 #include "base/threading/thread_task_runner_handle.h"
 #include "chrome/browser/android/offline_pages/offliner_helper.h"
+#include "chrome/browser/loader/chrome_navigation_data.h"
 #include "chrome/browser/net/prediction_options.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/test/base/testing_profile.h"
 #include "components/content_settings/core/common/pref_names.h"
 #include "components/offline_pages/content/background_loader/background_loader_contents_stub.h"
 #include "components/offline_pages/core/background/load_termination_listener.h"
 #include "components/offline_pages/core/background/offliner.h"
 #include "components/offline_pages/core/background/offliner_policy.h"
 #include "components/offline_pages/core/background/save_page_request.h"
 #include "components/offline_pages/core/offline_page_feature.h"
@@ skipping 539 matching lines @@
   offliner()->DidFinishNavigation(handle.get());
   // NavigationHandle is always destroyed after finishing navigation.
   handle.reset();
   CompleteLoading();
   PumpLoop();
 
   EXPECT_TRUE(completion_callback_called());
   EXPECT_EQ(Offliner::RequestStatus::LOADING_FAILED_NO_NEXT, request_status());
 }
 
+TEST_F(BackgroundLoaderOfflinerTest, OffliningPreviewsStatusOffHistogram) {
+  base::Time creation_time = base::Time::Now();
+  SavePageRequest request(kRequestId, kHttpUrl, kClientId, creation_time,
+                          kUserRequested);
+  EXPECT_TRUE(offliner()->LoadAndSave(request, completion_callback(),
+                                      progress_callback()));
+
+  // Called after calling LoadAndSave so we have web_contents to work with.
+  std::unique_ptr<content::NavigationHandle> handle(
+      content::NavigationHandle::CreateNavigationHandleForTesting(
+          kHttpUrl, offliner()->web_contents()->GetMainFrame(), true,
+          net::Error::OK));
+  // Set up ChromeNavigationData on the handle.
+  std::unique_ptr<ChromeNavigationData> chrome_navigation_data(
+      new ChromeNavigationData());
+  chrome_navigation_data->set_previews_state(
+      content::PreviewsTypes::PREVIEWS_NO_TRANSFORM);
+  std::unique_ptr<content::NavigationData> navigation_data(
+      chrome_navigation_data.release());
+  // Call DidFinishNavigation with handle.
+  offliner()->DidFinishNavigation(handle.get());
+
+  histograms().ExpectBucketCount(
+      "OfflinePages.Background.OffliningPreviewStatus.async_loading",
+      0,  // Previews Disabled
+      1);
+}
+
+TEST_F(BackgroundLoaderOfflinerTest, OffliningPreviewsStatusOnHistogram) {
+  base::Time creation_time = base::Time::Now();
+  SavePageRequest request(kRequestId, kHttpUrl, kClientId, creation_time,
+                          kUserRequested);
+  EXPECT_TRUE(offliner()->LoadAndSave(request, completion_callback(),
+                                      progress_callback()));
+
+  // Called after calling LoadAndSave so we have web_contents to work with.
+  std::unique_ptr<content::NavigationHandle> handle(
+      content::NavigationHandle::CreateNavigationHandleForTesting(
+          kHttpUrl, offliner()->web_contents()->GetMainFrame(), true,
+          net::Error::OK));
+  // Set up ChromeNavigationData on the handle.
+  std::unique_ptr<ChromeNavigationData> chrome_navigation_data(
+      new ChromeNavigationData());
+  chrome_navigation_data->set_previews_state(
+      content::PreviewsTypes::CLIENT_LOFI_ON);
+  std::unique_ptr<content::NavigationData> navigation_data(
+      chrome_navigation_data.release());
+  offliner()->web_contents_tester()->SetNavigationData(
+      handle.get(), std::move(navigation_data));
+  // Call DidFinishNavigation with handle.
+  offliner()->DidFinishNavigation(handle.get());
+
+  histograms().ExpectBucketCount(
+      "OfflinePages.Background.OffliningPreviewStatus.async_loading",
+      1,  // Previews Enabled
+      1);
+}
+
 TEST_F(BackgroundLoaderOfflinerTest, OnlySavesOnceOnMultipleLoads) {
   base::Time creation_time = base::Time::Now();
   SavePageRequest request(kRequestId, kHttpUrl, kClientId, creation_time,
                           kUserRequested);
   EXPECT_TRUE(offliner()->LoadAndSave(request, completion_callback(),
                                       progress_callback()));
   // First load
   CompleteLoading();
   // Second load
   CompleteLoading();
@@ skipping 114 matching lines @@
   CompleteLoading();
   PumpLoop();
 
   // One extra part should be added if the flag is on.
   content::MHTMLExtraParts* extra_parts =
       content::MHTMLExtraParts::FromWebContents(offliner()->web_contents());
   EXPECT_EQ(extra_parts->size(), 1);
 }
 
 }  // namespace offline_pages