Client config and QUIC transaction integration tests

tools/chrome_proxy/webdriver/smoke.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
+import time
 from common import TestDriver
 from common import IntegrationTest
 from common import NotAndroid
 
 
 class Smoke(IntegrationTest):
 
   # Ensure Chrome does not use DataSaver in Incognito mode.
   # Clank does not honor the --incognito flag.
   @NotAndroid
@@ skipping 10 matching lines @@
     with TestDriver() as t:
       t.AddChromeArg('--enable-spdy-proxy-auth')
       t.AddChromeArg('--enable-quic')
       t.AddChromeArg('--data-reduction-proxy-http-proxies=https://proxy.googlezip.net:443')
       t.AddChromeArg('--force-fieldtrials=DataReductionProxyUseQuic/Enabled')
       t.LoadURL('http://check.googlezip.net/test.html')
       responses = t.GetHTTPResponses()
       self.assertEqual(2, len(responses))
       for response in responses:
         self.assertHasChromeProxyViaHeader(response)
+
+      # Verify that histogram DataReductionProxy.Quic.ProxyStatus has at least 1
+      # sample. This sample must be in bucket 0 (QUIC_PROXY_STATUS_AVAILABLE).
+      proxy_status = t.GetHistogram('DataReductionProxy.Quic.ProxyStatus')
+      self.assertLessEqual(1, proxy_status['count'])
+      self.assertEqual(0, proxy_status['sum'])
+
+      # Navigate to one more page to ensure that established QUIC connection
+      # is used for the next request. Give 3 seconds extra headroom for the QUIC
+      # connection to be established.
+      time.sleep(3)
+      t.LoadURL('http://check.googlezip.net/test.html')
+      proxy_usage = t.GetHistogram('Net.QuicAlternativeProxy.Usage')
+      # Bucket ALTERNATIVE_PROXY_USAGE_NO_RACE should have at least onesample.
+      self.assertLessEqual(1, proxy_usage['buckets'][0]['count'])
   
   # Ensure Chrome uses DataSaver in normal mode.
   def testCheckPageWithNormalMode(self):
     with TestDriver() as t:
       t.AddChromeArg('--enable-spdy-proxy-auth')
       t.LoadURL('http://check.googlezip.net/test.html')
       responses = t.GetHTTPResponses()
       self.assertNotEqual(0, len(responses))
       for response in responses:
         self.assertHasChromeProxyViaHeader(response)
 
   # Ensure pageload metric pingback with DataSaver.
   def testPingback(self):
     with TestDriver() as t:
       t.AddChromeArg('--enable-spdy-proxy-auth')
       t.AddChromeArg('--enable-data-reduction-proxy-force-pingback')
       t.LoadURL('http://check.googlezip.net/test.html')
       t.LoadURL('http://check.googlezip.net/test.html')
       t.SleepUntilHistogramHasEntry("DataReductionProxy.Pingback.Succeeded")
       # Verify one pingback attempt that was successful.
       attempted = t.GetHistogram('DataReductionProxy.Pingback.Attempted')
       self.assertEqual(1, attempted['count'])
       succeeded = t.GetHistogram('DataReductionProxy.Pingback.Succeeded')
       self.assertEqual(1, succeeded['count'])
 
+  # Ensure client config is fetched at the start of the Chrome session, and the
+  # session ID is correctly set in the chrome-proxy request header.
+  def testClientConfig(self):
+    with TestDriver() as t:
+      t.AddChromeArg('--enable-spdy-proxy-auth')
+      t.SleepUntilHistogramHasEntry(
+        'DataReductionProxy.ConfigService.FetchResponseCode')
+      t.LoadURL('http://check.googlezip.net/test.html')
+      responses = t.GetHTTPResponses()
+      self.assertEqual(2, len(responses))
+      for response in responses:
+        chrome_proxy_header = response.request_headers['chrome-proxy']
+        self.assertIn('s=', chrome_proxy_header)
+        self.assertNotIn('ps=', chrome_proxy_header)
+        self.assertNotIn('sid=', chrome_proxy_header)
+        # Verify that the proxy server honored the session ID.
+        self.assertHasChromeProxyViaHeader(response)
+        self.assertEqual(200, response.status)
+
 if __name__ == '__main__':
   IntegrationTest.RunAllTests()