Implement the LitePage Integration Test with ChromeDriver

tools/chrome_proxy/webdriver/lofi.py

Index: tools/chrome_proxy/webdriver/lofi.py
diff --git a/tools/chrome_proxy/webdriver/lofi.py b/tools/chrome_proxy/webdriver/lofi.py
index 0f89d6233eee6322afe4e61bf3e8ef650c4463f7..bd6e7cb559780cbe6a08df4fe7afab4aec133a66 100644
--- a/tools/chrome_proxy/webdriver/lofi.py
+++ b/tools/chrome_proxy/webdriver/lofi.py
@@ -42,5 +42,36 @@ class LoFi(IntegrationTest):
       # 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.AddChromeArg('--disable-quic')
+
+      test_driver.LoadURL('http://check.googlezip.net/test.html')
+
+      lite_page_responses = 0
+      for response in test_driver.GetHTTPResponses():
+        if '/csi?' in response.url:

[Robert Ogden, 2017/02/23 23:47:48]

Why is this needed? Maybe leave a comment.

[megjablon, 2017/02/23 23:58:18]

Done.

+          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.assertTrue('exp=ignore_preview_blacklist' in

[Robert Ogden, 2017/02/23 23:47:48]

Use assertIn
https://docs.python.org/2/library/unittest.html#unittest.TestCase.assertIn

[megjablon, 2017/02/23 23:58:18]

Done.

+          chrome_proxy_request)
+        if ('lite-page' in cpct_response):
+          lite_page_responses = lite_page_responses + 1
+          self.assertTrue('lite-page' in 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()