Integration test for bypassing the proxy for a single request

tools/chrome_proxy/integration_tests/chrome_proxy_metrics.py

 # 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.
 
 import datetime
 import logging
 import os
 
 from integration_tests import network_metrics
 from telemetry.page import page_test
@@ skipping 217 matching lines @@
       info = GetProxyInfoFromNetworkInternals(tab)
       if not info['enabled']:
         raise ChromeProxyMetricException, (
             'Chrome proxy should be enabled. proxy info: %s' % info)
       _, expected_bad_proxies = self.IsProxyBypassed(tab)
       self.VerifyBadProxies(info['badProxies'], expected_bad_proxies)
 
     results.AddValue(scalar.ScalarValue(
         results.current_page, 'bypass', 'count', bypass_count))
 
+  def AddResultsForBlockSingle(self, tab, results):
+    eligible_response_count = 0
+    bypass_count = 0
+    for resp in self.IterResponses(tab):
+      if resp.ShouldHaveChromeProxyViaHeader():
+        eligible_response_count += 1
+        if not resp.HasChromeProxyViaHeader():
+          bypass_count += 1
+
+    if tab:
+      info = GetProxyInfoFromNetworkInternals(tab)
+      if not info['enabled']:
+        raise ChromeProxyMetricException, (
+            'Chrome proxy should be enabled. proxy info: %s' % info)
+      self.VerifyBadProxies(info['badProxies'], [])
+
+    if eligible_response_count <= 1:

[bolian, 2014/09/16 17:04:06]

Question. Why there will be more than one?  Do you see one from DRP to instruct
bypass another from direct loading? I remember in the past, the first response
was not exposed at this level.

[sclittle, 2014/09/16 19:06:53]

There would be more than one because http://check.googlezip.net/blocksingle
loads the google 404 not found page, which has a google logo image on it. The
idea was to test that the response for http://check.googlezip.net/blocksingle
bypassed the DRP, but the image that was requested immediately after that
response was received did not bypass the DRP.

This check that more that one response was received may not be necessary, since
the test already checks that the bad proxy list is empty which implies that only
one response would bypass the DRP. The eligible_response_count check also makes
the test dependent on the 404 not found page having at least one subresource on
it, which might be bad. I can remove this eligible_response_count check if you
think it's unnecessary.

[bolian, 2014/09/16 20:56:00]

No, don't remove that. :) I wasn't aware that the page contains another
resource. Thanks for explaining this.

+      raise ChromeProxyMetricException, (
+          'There should be more than one DRP eligible response '
+          '(eligible_response_count=%d, bypass_count=%d)\n' % (
+              eligible_response_count, bypass_count))
+    elif bypass_count != 1:
+      raise ChromeProxyMetricException, (
+          'Exactly one response should be bypassed. '
+          '(eligible_response_count=%d, bypass_count=%d)\n' % (
+              eligible_response_count, bypass_count))
+    else:
+      results.AddValue(scalar.ScalarValue(
+          results.current_page, 'eligible_responses', 'count',
+          eligible_response_count))
+      results.AddValue(scalar.ScalarValue(
+          results.current_page, 'bypass', 'count', bypass_count))
+
   def AddResultsForSafebrowsing(self, tab, results):
     count = 0
     safebrowsing_count = 0
     for resp in self.IterResponses(tab):
       count += 1
       if resp.IsSafebrowsingResponse():
         safebrowsing_count += 1
       else:
         r = resp.response
         raise ChromeProxyMetricException, (
@@ skipping 30 matching lines @@
     bad_proxies = []
     if 'badProxies' in info and info['badProxies']:
       bad_proxies = [p['proxy'] for p in info['badProxies']
                      if 'proxy' in p and p['proxy']]
     if bad_proxies != expected_bad_proxies:
       raise ChromeProxyMetricException, (
           'Wrong bad proxies (%s). Expect: "%s"' % (
           str(bad_proxies), str(expected_bad_proxies)))
     results.AddValue(scalar.ScalarValue(
         results.current_page, 'http_fallback', 'boolean', True))