Adding opt out and previews type information to DRP pingback

components/data_reduction_proxy/core/browser/data_reduction_proxy_network_delegate_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 "components/data_reduction_proxy/core/browser/data_reduction_proxy_network_delegate.h"
 
 #include <stddef.h>
 #include <stdint.h>
 
 #include <map>
@@ skipping 852 matching lines @@
     DataReductionProxyData* data =
         DataReductionProxyData::GetData(*request.get());
     if (!test.used_data_reduction_proxy) {
       EXPECT_FALSE(data);
     } else {
       EXPECT_TRUE(data);
       EXPECT_EQ(test.main_frame ? net::EFFECTIVE_CONNECTION_TYPE_OFFLINE
                                 : net::EFFECTIVE_CONNECTION_TYPE_UNKNOWN,
                 data->effective_connection_type());
       EXPECT_TRUE(data->used_data_reduction_proxy());
-      EXPECT_EQ(GURL(kTestURL), data->request_url());
-      EXPECT_EQ("fake-session", data->session_key());
+      EXPECT_EQ(test.main_frame ? GURL(kTestURL) : GURL(), data->request_url());
+      EXPECT_EQ(test.main_frame ? "fake-session" : "", data->session_key());
       EXPECT_EQ(test.lofi_on, data->lofi_requested());
     }
   }
 }
 
 TEST_F(DataReductionProxyNetworkDelegateTest,
        RequestDataHoldbackConfigurations) {
   Init(USE_INSECURE_PROXY, false);
   const struct {
     bool data_reduction_proxy_enabled;
@@ skipping 297 matching lines @@
         "HTTP/1.1 200 OK\r\n"
         "Date: Wed, 28 Nov 2007 09:40:09 GMT\r\n"
         "Expires: Mon, 24 Nov 2014 12:45:26 GMT\r\n"
         "Via: 1.1 Chrome-Compression-Proxy\r\n"
         "x-original-content-length: 200\r\n";
 
     if (tests[i].lofi_response)
       response_headers += "Chrome-Proxy-Content-Transform: empty-image\r\n";
 
     response_headers += "\r\n";
-    FetchURLRequest(GURL(kTestURL), nullptr, response_headers, 140, 0);
-
+    auto request =
+        FetchURLRequest(GURL(kTestURL), nullptr, response_headers, 140, 0);
+    EXPECT_EQ(tests[i].lofi_response,
+              DataReductionProxyData::GetData(*request)->lofi_received());
     VerifyDidNotifyLoFiResponse(tests[i].lofi_response);
   }
 }
 
 TEST_F(DataReductionProxyNetworkDelegateTest,
        TestLoFiTransformationTypeHistogram) {
   Init(USE_INSECURE_PROXY, false);
   const char kLoFiTransformationTypeHistogram[] =
       "DataReductionProxy.LoFi.TransformationType";
   base::HistogramTester histogram_tester;
 
   net::HttpRequestHeaders request_headers;
   request_headers.SetHeader("chrome-proxy-accept-transform", "lite-page");
   lofi_decider()->ignore_is_using_data_reduction_proxy_check();
   FetchURLRequest(GURL(kTestURL), &request_headers, std::string(), 140, 0);
   histogram_tester.ExpectBucketCount(kLoFiTransformationTypeHistogram,
                                      NO_TRANSFORMATION_LITE_PAGE_REQUESTED, 1);
 
   std::string response_headers =
       "HTTP/1.1 200 OK\r\n"
       "Chrome-Proxy-Content-Transform: lite-page\r\n"
       "Date: Wed, 28 Nov 2007 09:40:09 GMT\r\n"
       "Expires: Mon, 24 Nov 2014 12:45:26 GMT\r\n"
       "Via: 1.1 Chrome-Compression-Proxy\r\n"
       "x-original-content-length: 200\r\n";
 
   response_headers += "\r\n";
-  FetchURLRequest(GURL(kTestURL), nullptr, response_headers, 140, 0);
+  auto request =
+      FetchURLRequest(GURL(kTestURL), nullptr, response_headers, 140, 0);
+  EXPECT_TRUE(DataReductionProxyData::GetData(*request)->lite_page_received());
 
   histogram_tester.ExpectBucketCount(kLoFiTransformationTypeHistogram,
                                      LITE_PAGE, 1);
 }
 
 // Test that Brotli is not added to the accept-encoding header when it is
 // disabled globally.
 TEST_F(DataReductionProxyNetworkDelegateTest,
        BrotliAdvertisement_BrotliDisabled) {
   Init(USE_SECURE_PROXY, false /* enable_brotli_globally */);
@@ skipping 173 matching lines @@
   io_data()->request_options()->SetSecureSession("new-session");
   network_delegate()->NotifyBeforeSendHeaders(
       request.get(), data_reduction_proxy_info, proxy_retry_info, &headers);
   data = DataReductionProxyData::GetData(*request.get());
   EXPECT_EQ(1u, data->page_id().value());
 }
 
 }  // namespace
 
 }  // namespace data_reduction_proxy