DataReductionProxy proto change to use exp=force_lite_page directive

tools/chrome_proxy/webdriver/lite_page.py

 # 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.
 
 import common
 from common import TestDriver
 from common import IntegrationTest
 
 import time
 
 class LitePage(IntegrationTest):
 
-  # Checks that a Lite Page is served and that the ignore_preview_blacklist
-  # experiment is being used.
-  def testLitePage(self):
+  # Checks that a Lite Page is served and the force_lite_page experiment
+  # directive is provided when always-on.
+  def testLitePageForcedExperiment(self):
     # If it was attempted to run with another experiment, skip this test.
     if common.ParseFlags().browser_args and ('--data-reduction-proxy-experiment'
         in common.ParseFlags().browser_args):
       self.skipTest('This test cannot be run with other experiments.')
     with TestDriver() as test_driver:
       test_driver.AddChromeArg('--enable-spdy-proxy-auth')
       test_driver.AddChromeArg('--data-reduction-proxy-lo-fi=always-on')
       test_driver.AddChromeArg('--enable-data-reduction-proxy-lite-page')
 
       test_driver.LoadURL('http://check.googlezip.net/test.html')
 
       lite_page_responses = 0
       for response in test_driver.GetHTTPResponses():
         # Skip CSI requests when validating Lite Page headers. CSI requests
         # aren't expected to have LoFi headers.
         if '/csi?' in response.url:
           continue
         if response.url.startswith('data:'):
           continue
-        self.assertIn('exp=ignore_preview_blacklist',
-          response.request_headers['chrome-proxy'])
+        # Main resource should force lite page.
+        if response.url.endswith('html'):
+          self.assertIn('exp=force_lite_page',
+                        response.request_headers['chrome-proxy'])
         if (self.checkLitePageResponse(response)):
           lite_page_responses = lite_page_responses + 1
 
       # Verify that a Lite Page response for the main frame was seen.
       self.assertEqual(1, lite_page_responses)
 
   # Checks that a Lite Page does not have an error when scrolling to the bottom
   # of the page and is able to load all resources.
   def testLitePageBTF(self):
     # If it was attempted to run with another experiment, skip this test.
@@ skipping 10 matching lines @@
 
       # Verify that a Lite Page response for the main frame was seen.
       lite_page_responses = 0
       for response in test_driver.GetHTTPResponses():
         # Skip CSI requests when validating Lite Page headers. CSI requests
         # aren't expected to have LoFi headers.
         if '/csi?' in response.url:
           continue
         if response.url.startswith('data:'):
           continue
-        self.assertIn('exp=ignore_preview_blacklist',
-          response.request_headers['chrome-proxy'])
         if (self.checkLitePageResponse(response)):
           lite_page_responses = lite_page_responses + 1
       self.assertEqual(1, lite_page_responses)
 
       # Scroll to the bottom of the window and ensure scrollHeight increases.
       original_scroll_height = test_driver.ExecuteJavascriptStatement(
         'document.body.scrollHeight')
       test_driver.ExecuteJavascriptStatement(
         'window.scrollTo(0,Math.max(document.body.scrollHeight));')
       # Give some time for loading after scrolling.
@@ skipping 44 matching lines @@
 
         # Lo-Fi fallback is not currently supported via the client. Check that
         # no Lo-Fi response is received.
         self.checkLoFiResponse(response, False)
 
       # Verify that a Lite Page was requested.
       self.assertEqual(1, lite_page_requests)
 
 if __name__ == '__main__':
   IntegrationTest.RunAllTests()