Add "q=preview" directive on main frame requests on poor networks

components/data_reduction_proxy/content/browser/content_lofi_decider_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 "components/data_reduction_proxy/content/browser/content_lofi_decider.h"
 
 #include <string>
 
 #include "base/command_line.h"
 #include "base/message_loop/message_loop.h"
 #include "base/metrics/field_trial.h"
 #include "base/strings/string_util.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_network_delegate.h"
 #include "components/data_reduction_proxy/core/browser/data_reduction_proxy_test_utils.h"
 #include "components/data_reduction_proxy/core/common/data_reduction_proxy_headers.h"
 #include "components/data_reduction_proxy/core/common/data_reduction_proxy_params.h"
 #include "components/data_reduction_proxy/core/common/data_reduction_proxy_params_test_utils.h"
 #include "components/data_reduction_proxy/core/common/data_reduction_proxy_switches.h"
 #include "content/public/browser/resource_request_info.h"
+#include "net/base/load_flags.h"
 #include "net/base/network_delegate_impl.h"
 #include "net/http/http_request_headers.h"
 #include "net/proxy/proxy_info.h"
 #include "net/socket/socket_test_util.h"
 #include "net/url_request/url_request.h"
 #include "net/url_request/url_request_test_util.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace data_reduction_proxy {
 
@@ skipping 88 matching lines @@
   static void VerifyLoFiHeader(bool expected_lofi_used,
                                const net::HttpRequestHeaders& headers) {
     EXPECT_TRUE(headers.HasHeader(chrome_proxy_header()));
     std::string header_value;
     headers.GetHeader(chrome_proxy_header(), &header_value);
     EXPECT_EQ(
         expected_lofi_used,
         header_value.find(chrome_proxy_lo_fi_directive()) != std::string::npos);
   }
 
+  static void VerifyLoFiPreviewHeader(bool expected_lofi_preview_used,
+                                      const net::HttpRequestHeaders& headers) {
+    EXPECT_TRUE(headers.HasHeader(chrome_proxy_header()));
+    std::string header_value;
+    headers.GetHeader(chrome_proxy_header(), &header_value);
+    EXPECT_EQ(expected_lofi_preview_used,
+              header_value.find(chrome_proxy_lo_fi_preview_directive()) !=
+                  std::string::npos);
+  }
+
   static void VerifyLoFiExperimentHeader(
       bool expected_lofi_experiment_used,
       const net::HttpRequestHeaders& headers) {
     EXPECT_TRUE(headers.HasHeader(chrome_proxy_header()));
     std::string header_value;
     headers.GetHeader(chrome_proxy_header(), &header_value);
     EXPECT_EQ(expected_lofi_experiment_used,
               header_value.find(chrome_proxy_lo_fi_experiment_directive()) !=
                   std::string::npos);
   }
 
  protected:
   base::MessageLoopForIO message_loop_;
   net::MockClientSocketFactory mock_socket_factory_;
   net::TestURLRequestContext context_;
   net::TestDelegate delegate_;
   scoped_ptr<DataReductionProxyTestContext> test_context_;
   scoped_ptr<DataReductionProxyNetworkDelegate>
       data_reduction_proxy_network_delegate_;
 };
 
 TEST_F(ContentLoFiDeciderTest, LoFiFlags) {
   // Enable Lo-Fi.
   const struct {
     bool is_using_lofi;
+    bool is_using_previews;
   } tests[] = {
-      {false}, {true},
+      {false, false}, {true, false}, {false, true}, {true, true},
   };
 
   for (size_t i = 0; i < arraysize(tests); ++i) {
     scoped_ptr<net::URLRequest> request = CreateRequest(tests[i].is_using_lofi);
     base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
     command_line->InitFromArgv(command_line->argv());
+    if (tests[i].is_using_previews) {
+      command_line->AppendSwitch(
+          switches::kEnableDataReductionProxyLoFiPreview);
+    }
 
     // No flags or field trials. The Lo-Fi header should not be added.
     net::HttpRequestHeaders headers;
     NotifyBeforeSendProxyHeaders(&headers, request.get());
     VerifyLoFiHeader(false, headers);
+    VerifyLoFiPreviewHeader(false, headers);
 
-    // The Lo-Fi flag is "always-on" and Lo-Fi is being used. Lo-Fi header
-    // should
-    // be added.
+    // The Lo-Fi flag is "always-on", Lo-Fi is being used, and it's a main frame
+    // request. Lo-Fi preview header should be added.
     command_line->AppendSwitchASCII(
         switches::kDataReductionProxyLoFi,
         switches::kDataReductionProxyLoFiValueAlwaysOn);
+    request->SetLoadFlags(request->load_flags() | net::LOAD_MAIN_FRAME);
+    headers.Clear();
+    NotifyBeforeSendProxyHeaders(&headers, request.get());
+    VerifyLoFiHeader(tests[i].is_using_lofi && !tests[i].is_using_previews,
+                     headers);
+    VerifyLoFiPreviewHeader(
+        tests[i].is_using_lofi && tests[i].is_using_previews, headers);
+
+    // The Lo-Fi flag is "always-on" and Lo-Fi is being used. Lo-Fi header
+    // should be added.
+    request->SetLoadFlags(0);
     headers.Clear();
     NotifyBeforeSendProxyHeaders(&headers, request.get());
     VerifyLoFiHeader(tests[i].is_using_lofi, headers);
+    VerifyLoFiPreviewHeader(false, headers);
 
     // The Lo-Fi flag is "cellular-only" and Lo-Fi is being used. Lo-Fi header
     // should be added.
     command_line->AppendSwitchASCII(
         switches::kDataReductionProxyLoFi,
         switches::kDataReductionProxyLoFiValueCellularOnly);
     headers.Clear();
     NotifyBeforeSendProxyHeaders(&headers, request.get());
     VerifyLoFiHeader(tests[i].is_using_lofi, headers);
+    VerifyLoFiPreviewHeader(false, headers);
 
     // The Lo-Fi flag is "slow-connections-only" and Lo-Fi is being used. Lo-Fi
     // header should be added.
     command_line->AppendSwitchASCII(
         switches::kDataReductionProxyLoFi,
         switches::kDataReductionProxyLoFiValueSlowConnectionsOnly);
     headers.Clear();
     NotifyBeforeSendProxyHeaders(&headers, request.get());
     VerifyLoFiHeader(tests[i].is_using_lofi, headers);
+    VerifyLoFiPreviewHeader(false, headers);
   }
 }
 
 TEST_F(ContentLoFiDeciderTest, LoFiEnabledFieldTrial) {
   base::FieldTrialList field_trial_list(nullptr);
   base::FieldTrialList::CreateFieldTrial(params::GetLoFiFieldTrialName(),
                                          "Enabled");
   // Enable Lo-Fi.
   const struct {
     bool is_using_lofi;
   } tests[] = {
       {false}, {true},
   };
 
   for (size_t i = 0; i < arraysize(tests); ++i) {
     scoped_ptr<net::URLRequest> request = CreateRequest(tests[i].is_using_lofi);
     net::HttpRequestHeaders headers;
     NotifyBeforeSendProxyHeaders(&headers, request.get());
     VerifyLoFiHeader(tests[i].is_using_lofi, headers);
+    VerifyLoFiPreviewHeader(false, headers);
   }
 }
 
 TEST_F(ContentLoFiDeciderTest, LoFiControlFieldTrial) {
   base::FieldTrialList field_trial_list(nullptr);
   base::FieldTrialList::CreateFieldTrial(params::GetLoFiFieldTrialName(),
                                          "Control");
   // Enable Lo-Fi.
   const struct {
     bool is_using_lofi;
   } tests[] = {
       {false}, {true},
   };
 
   for (size_t i = 0; i < arraysize(tests); ++i) {
     scoped_ptr<net::URLRequest> request = CreateRequest(tests[i].is_using_lofi);
     net::HttpRequestHeaders headers;
     NotifyBeforeSendProxyHeaders(&headers, request.get());
     VerifyLoFiHeader(false, headers);
+    VerifyLoFiPreviewHeader(false, headers);
   }
 }
 
+TEST_F(ContentLoFiDeciderTest, LoFiPreviewFieldTrial) {
+  base::FieldTrialList field_trial_list(nullptr);
+  base::FieldTrialList::CreateFieldTrial(params::GetLoFiFieldTrialName(),
+                                         "Enabled_Preview");
+  // Enable Lo-Fi.
+  const struct {
+    bool is_using_lofi;
+    bool is_main_frame;
+  } tests[] = {
+      {false, false}, {true, false}, {false, true}, {true, true},
+  };
+
+  for (size_t i = 0; i < arraysize(tests); ++i) {
+    scoped_ptr<net::URLRequest> request = CreateRequest(tests[i].is_using_lofi);
+    if (tests[i].is_main_frame)
+      request->SetLoadFlags(request->load_flags() | net::LOAD_MAIN_FRAME);
+    net::HttpRequestHeaders headers;
+    NotifyBeforeSendProxyHeaders(&headers, request.get());
+    VerifyLoFiHeader(tests[i].is_using_lofi && !tests[i].is_main_frame,
+                     headers);
+    VerifyLoFiPreviewHeader(tests[i].is_using_lofi && tests[i].is_main_frame,
+                            headers);
+  }
+}
+
 TEST_F(ContentLoFiDeciderTest, AutoLoFi) {
   const struct {
     bool auto_lofi_enabled_group;
     bool auto_lofi_control_group;
     bool network_prohibitively_slow;
   } tests[] = {
       {false, false, false},
       {false, false, true},
       {true, false, false},
       {true, false, true},
@@ skipping 81 matching lines @@
     scoped_ptr<net::URLRequest> request =
         CreateRequest(tests[i].network_prohibitively_slow);
     net::HttpRequestHeaders headers;
     NotifyBeforeSendProxyHeaders(&headers, request.get());
 
     VerifyLoFiHeader(expect_lofi_header, headers);
   }
 }
 
 }  // namespace data_reduction_roxy