Implement the LitePage Integration Test with ChromeDriver

tools/chrome_proxy/webdriver/lofi.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
 
 
 class LoFi(IntegrationTest):
@@ skipping 24 matching lines @@
         cpct_response = response.response_headers[
                           'chrome-proxy-content-transform']
         if ('empty-image' in cpct_response):
           lofi_responses = lofi_responses + 1
           self.assertTrue('empty-image' in cpat_request)
           self.assertTrue(int(content_length) < 100)
 
       # Verify that Lo-Fi responses were seen.
       self.assertNotEqual(0, lofi_responses)
 
+  # Checks that a Lite Page is served and that the ignore_preview_blacklist
+  # experiment is being used.
+  def testLitePage(self):
+    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
+        chrome_proxy_request = response.request_headers['chrome-proxy']
+        cpat_request = response.request_headers['chrome-proxy-accept-transform']
+        cpct_response = response.response_headers[
+                          'chrome-proxy-content-transform']
+        self.assertHasChromeProxyViaHeader(response)
+        self.assertIn('exp=ignore_preview_blacklist',
+          chrome_proxy_request)
+        if ('lite-page' in cpct_response):
+          lite_page_responses = lite_page_responses + 1
+          self.assertIn('lite-page', cpat_request)
+
+      # Verify that a Lite Page response for the main frame was seen.
+      self.assertEqual(1, lite_page_responses)
+
 if __name__ == '__main__':
   IntegrationTest.RunAllTests()