Use the Previews Black List for server previews

chrome/browser/previews/previews_infobar_delegate_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/previews/previews_infobar_delegate.h"
 
 #include <memory>
 #include <string>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/memory/ptr_util.h"
+#include "base/metrics/field_trial.h"
 #include "base/optional.h"
 #include "base/test/histogram_tester.h"
 #include "base/threading/thread_task_runner_handle.h"
 #include "chrome/browser/android/android_theme_resources.h"
 #include "chrome/browser/infobars/infobar_service.h"
 #include "chrome/browser/loader/chrome_navigation_data.h"
 #include "chrome/browser/net/spdyproxy/data_reduction_proxy_chrome_settings.h"
 #include "chrome/browser/net/spdyproxy/data_reduction_proxy_chrome_settings_factory.h"
 #include "chrome/browser/previews/previews_infobar_tab_helper.h"
 #include "chrome/grit/generated_resources.h"
 #include "chrome/test/base/chrome_render_view_host_test_harness.h"
+#include "components/data_reduction_proxy/core/browser/data_reduction_proxy_config_test_utils.h"
 #include "components/data_reduction_proxy/core/browser/data_reduction_proxy_data.h"
 #include "components/data_reduction_proxy/core/browser/data_reduction_proxy_pingback_client.h"
 #include "components/data_reduction_proxy/core/browser/data_reduction_proxy_service.h"
 #include "components/data_reduction_proxy/core/browser/data_reduction_proxy_settings.h"
 #include "components/data_reduction_proxy/core/browser/data_reduction_proxy_test_utils.h"
 #include "components/data_reduction_proxy/core/common/data_reduction_proxy_pref_names.h"
 #include "components/infobars/core/confirm_infobar_delegate.h"
 #include "components/infobars/core/infobar.h"
 #include "components/infobars/core/infobar_delegate.h"
 #include "components/prefs/pref_registry_simple.h"
 #include "components/proxy_config/proxy_config_pref_names.h"
 #include "content/public/browser/navigation_entry.h"
 #include "content/public/browser/navigation_handle.h"
+#include "content/public/browser/reload_type.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/browser/web_contents_observer.h"
 #include "content/public/browser/web_contents_user_data.h"
 #include "content/public/common/referrer.h"
 #include "content/public/test/test_renderer_host.h"
 #include "content/public/test/web_contents_tester.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "ui/base/page_transition_types.h"
 #include "ui/base/window_open_disposition.h"
 
@@ skipping 213 matching lines @@
 
   tester.ExpectBucketCount(
       kUMAPreviewsInfoBarActionLoFi,
       PreviewsInfoBarDelegate::INFOBAR_DISMISSED_BY_TAB_CLOSURE, 1);
   EXPECT_EQ(0, drp_test_context_->pref_service()->GetInteger(
                    data_reduction_proxy::prefs::kLoFiLoadImagesPerSession));
   EXPECT_FALSE(user_opt_out_.value());
 }
 
 TEST_F(PreviewsInfoBarDelegateUnitTest, InfobarTestClickLinkLoFi) {
-  base::HistogramTester tester;
+  NavigateAndCommit(GURL(kTestUrl));
+  const struct {
+    bool is_transitioned;

[megjablon, 2017/05/10 19:11:23]

using_previews_black_list? or something that mentions the blacklist. thanks for
combining these.

[RyanSturm, 2017/05/10 19:57:02]

Done.

+  } tests[] = {
+      {true}, {false},
+  };
+  for (const auto test : tests) {
+    drp_test_context_->config()->ResetLoFiStatusForTest();
+    base::FieldTrialList field_trial_list(nullptr);
+    if (test.is_transitioned) {
+      base::FieldTrialList::CreateFieldTrial(
+          "DataReductionProxyPreviewsBlackListTransition", "Enabled_");
+    }
+    base::HistogramTester tester;
+    // Call Reload and CommitPendingNavigation to force DidFinishNavigation.
+    web_contents()->GetController().Reload(content::ReloadType::NORMAL, true);
+    content::WebContentsTester::For(web_contents())->CommitPendingNavigation();
 
-  NavigateAndCommit(GURL(kTestUrl));
+    ConfirmInfoBarDelegate* infobar = CreateInfoBar(
+        PreviewsInfoBarDelegate::LOFI, true /* is_data_saver_user */);
 
-  ConfirmInfoBarDelegate* infobar = CreateInfoBar(
-      PreviewsInfoBarDelegate::LOFI, true /* is_data_saver_user */);
+    // Simulate clicking the infobar link.
+    if (infobar->LinkClicked(WindowOpenDisposition::CURRENT_TAB))
+      infobar_service()->infobar_at(0)->RemoveSelf();
+    EXPECT_EQ(0U, infobar_service()->infobar_count());
 
-  // Simulate clicking the infobar link.
-  if (infobar->LinkClicked(WindowOpenDisposition::CURRENT_TAB))
-    infobar_service()->infobar_at(0)->RemoveSelf();
-  EXPECT_EQ(0U, infobar_service()->infobar_count());
+    tester.ExpectBucketCount(
+        kUMAPreviewsInfoBarActionLoFi,
+        PreviewsInfoBarDelegate::INFOBAR_LOAD_ORIGINAL_CLICKED, 1);
+    EXPECT_EQ(test.is_transitioned ? 0 : 1,
+              drp_test_context_->pref_service()->GetInteger(
+                  data_reduction_proxy::prefs::kLoFiLoadImagesPerSession));
+    EXPECT_TRUE(user_opt_out_.value());
 
-  tester.ExpectBucketCount(
-      kUMAPreviewsInfoBarActionLoFi,
-      PreviewsInfoBarDelegate::INFOBAR_LOAD_ORIGINAL_CLICKED, 1);
-  EXPECT_EQ(1, drp_test_context_->pref_service()->GetInteger(
-                   data_reduction_proxy::prefs::kLoFiLoadImagesPerSession));
-  EXPECT_TRUE(user_opt_out_.value());
-
-  auto* data_reduction_proxy_settings =
-      DataReductionProxyChromeSettingsFactory::GetForBrowserContext(
-          web_contents()->GetBrowserContext());
-
-  EXPECT_EQ(1u, data_reduction_proxy_settings->data_reduction_proxy_service()
-                    ->pingback_client()
-                    ->OptOutsSizeForTesting());
+    auto* data_reduction_proxy_settings =
+        DataReductionProxyChromeSettingsFactory::GetForBrowserContext(
+            web_contents()->GetBrowserContext());
+    EXPECT_EQ(1u, data_reduction_proxy_settings->data_reduction_proxy_service()
+                      ->pingback_client()
+                      ->OptOutsSizeForTesting());
+  }
 }
 
 TEST_F(PreviewsInfoBarDelegateUnitTest, InfobarTestClickLinkLitePage) {
   base::HistogramTester tester;
 
   NavigateAndCommit(GURL(kTestUrl));
 
   ConfirmInfoBarDelegate* infobar = CreateInfoBar(
       PreviewsInfoBarDelegate::LITE_PAGE, true /* is_data_saver_user */);
 
@@ skipping 211 matching lines @@
   EXPECT_EQ(1u, data_reduction_proxy_settings->data_reduction_proxy_service()
                     ->pingback_client()
                     ->OptOutsSizeForTesting());
 
   drp_test_context_->RunUntilIdle();
 
   EXPECT_EQ(0u, data_reduction_proxy_settings->data_reduction_proxy_service()
                     ->pingback_client()
                     ->OptOutsSizeForTesting());
 }