Makes PreviewsService IsPreviewsTypeEnabled aware of new CPAT Feature

chrome/browser/previews/previews_service_unittest.cc

 // Copyright 2017 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_service.h"
 
 #include <initializer_list>
 #include <map>
 #include <memory>
 #include <string>
 
 #include "base/files/file_path.h"
 #include "base/memory/ptr_util.h"
 #include "base/message_loop/message_loop.h"
 #include "base/metrics/field_trial.h"
 #include "base/metrics/field_trial_params.h"
 #include "base/run_loop.h"
 #include "base/single_thread_task_runner.h"
+#include "base/test/scoped_feature_list.h"
+#include "components/data_reduction_proxy/core/common/data_reduction_proxy_features.h"
 #include "components/previews/core/previews_io_data.h"
 #include "components/previews/core/previews_ui_service.h"
 #include "components/variations/variations_associated_data.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/test/test_browser_thread_bundle.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
 
 void CreateFieldTrialWithParams(
@@ skipping 30 matching lines @@
  private:
   previews::PreviewsIsEnabledCallback enabled_callback_;
 };
 
 // Class to test the validity of the callback passed to PreviewsIOData from
 // PreviewsService.
 class PreviewsServiceTest : public testing::Test {
  public:
   PreviewsServiceTest()
 
-      : field_trial_list_(nullptr) {}
+      : field_trial_list_(nullptr), scoped_feature_list_() {}
 
   void SetUp() override {
     io_data_ = base::MakeUnique<TestPreviewsIOData>();
 
     service_ = base::MakeUnique<PreviewsService>();
     base::FilePath file_path;
     service_->Initialize(io_data_.get(),
                          content::BrowserThread::GetTaskRunnerForThread(
                              content::BrowserThread::UI),
                          file_path);
+    scoped_feature_list_.InitAndDisableFeature(
+        data_reduction_proxy::features::kDataReductionProxyDecidesTransform);
   }
 
   void TearDown() override { variations::testing::ClearAllVariationParams(); }
 
   ~PreviewsServiceTest() override {}
 
   TestPreviewsIOData* io_data() const { return io_data_.get(); }
 
  private:
   content::TestBrowserThreadBundle threads_;
   base::FieldTrialList field_trial_list_;
   std::unique_ptr<TestPreviewsIOData> io_data_;
   std::unique_ptr<PreviewsService> service_;
+  base::test::ScopedFeatureList scoped_feature_list_;
 };
 
 }  // namespace
 
 TEST_F(PreviewsServiceTest, TestOfflineFieldTrialEnabled) {
   CreateFieldTrialWithParams("ClientSidePreviews", "Enabled",
                              {{"show_offline_pages", "true"}});
   EXPECT_TRUE(io_data()->IsPreviewEnabled(previews::PreviewsType::OFFLINE));
 }
 
@@ skipping 36 matching lines @@
   base::FieldTrialList::CreateFieldTrial("DataCompressionProxyLoFi",
                                          "Disabled");
   EXPECT_FALSE(io_data()->IsPreviewEnabled(previews::PreviewsType::LOFI));
 }
 
 TEST_F(PreviewsServiceTest, TestServerLoFiFieldTrialNotSet) {
   EXPECT_FALSE(io_data()->IsPreviewEnabled(previews::PreviewsType::LOFI));
 }
 
 TEST_F(PreviewsServiceTest, TestLitePageFieldTrialEnabledPreview) {
+  base::test::ScopedFeatureList scoped_feature_list;
+  scoped_feature_list.InitAndDisableFeature(
+      data_reduction_proxy::features::kDataReductionProxyDecidesTransform);
   base::FieldTrialList::CreateFieldTrial("DataCompressionProxyLoFi",
                                          "Enabled_Preview");
   EXPECT_TRUE(io_data()->IsPreviewEnabled(previews::PreviewsType::LITE_PAGE));
 }
 
 TEST_F(PreviewsServiceTest, TestLitePageFieldTrialEnabled) {
   base::FieldTrialList::CreateFieldTrial("DataCompressionProxyLoFi", "Enabled");
   EXPECT_FALSE(io_data()->IsPreviewEnabled(previews::PreviewsType::LITE_PAGE));
 }
 
 TEST_F(PreviewsServiceTest, TestLitePageFieldTrialDisabled) {
   base::FieldTrialList::CreateFieldTrial("DataCompressionProxyLoFi",
                                          "Disabled");
   EXPECT_FALSE(io_data()->IsPreviewEnabled(previews::PreviewsType::LITE_PAGE));
 }
 
 TEST_F(PreviewsServiceTest, TestLitePageFieldTrialNotSet) {
   EXPECT_FALSE(io_data()->IsPreviewEnabled(previews::PreviewsType::LITE_PAGE));
 }
+
+TEST_F(PreviewsServiceTest, TestServerLoFiProxyDecidesTransform) {
+  base::test::ScopedFeatureList scoped_feature_list;
+  scoped_feature_list.InitAndEnableFeature(
+      data_reduction_proxy::features::kDataReductionProxyDecidesTransform);
+  EXPECT_TRUE(io_data()->IsPreviewEnabled(previews::PreviewsType::LITE_PAGE));
+}
+
+TEST_F(PreviewsServiceTest, TestLitePageProxyDecidesTransform) {
+  base::test::ScopedFeatureList scoped_feature_list;
+  scoped_feature_list.InitAndEnableFeature(
+      data_reduction_proxy::features::kDataReductionProxyDecidesTransform);
+  EXPECT_TRUE(io_data()->IsPreviewEnabled(previews::PreviewsType::LITE_PAGE));
+}