Integration test for “lo-fi” mode in Chrome-Proxy request header

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 logging
 import time
 
 from integration_tests import network_metrics
 from telemetry.page import page_test
 from telemetry.value import scalar
@@ skipping 53 matching lines @@
       return None
 
     chrome_proxy_request_header = self.response.request_headers['Chrome-Proxy']
     values = [v.strip() for v in chrome_proxy_request_header.split(',')]
     for value in values:
       kvp = value.split('=', 1)
       if len(kvp) == 2 and kvp[0].strip() == 'c':
         return kvp[1].strip()
     return None
 
+  def HasChromeProxyLoFi(self):
+    if 'Chrome-Proxy' not in self.response.request_headers:
+      return False
+    chrome_proxy_request_header = self.response.request_headers['Chrome-Proxy']
+    values = [v.strip() for v in chrome_proxy_request_header.split(',')]
+    for value in values:
+      if len(value) == 5 and value == 'q=low':
+        return True
+    return False
 
 class ChromeProxyMetric(network_metrics.NetworkMetric):
   """A Chrome proxy timeline metric."""
 
   def __init__(self):
     super(ChromeProxyMetric, self).__init__()
     self.compute_data_saving = True
 
   def SetEvents(self, events):
     """Used for unittest."""
@@ skipping 89 matching lines @@
           raise ChromeProxyMetricException, (
               '%s: Response missing via header. Only "%s" clients should '
               'bypass for this page, but this client is "%s".' % (
                   resp.response.url, bypass_for_client_type, client_type))
 
     results.AddValue(scalar.ScalarValue(
         results.current_page, 'via', 'count', via_count))
     results.AddValue(scalar.ScalarValue(
         results.current_page, 'bypass', 'count', bypass_count))
 
+  def AddResultsForLoFi(self, tab, results):
+    lo_fi_count = 0
+
+    for resp in self.IterResponses(tab):
+      if resp.HasChromeProxyViaHeader():
+        lo_fi_count += 1
+      else:
+        r = resp.response
+        raise ChromeProxyMetricException, (
+            '%s: LoFi not in request header.' % (r.url))
+
+      cl = resp.content_length
+      resource = resp.response.url
+    results.AddValue(scalar.ScalarValue(
+        results.current_page, 'lo_fi', 'count', lo_fi_count))
+
+    for resp in self.IterResponses(tab):
+      r = resp.response
+      cl = resp.content_length
+      ocl = resp.original_content_length
+      saving = resp.data_saving_rate * 100
+      if cl > 100:
+        raise ChromeProxyMetricException, (
+            'Image %s is %d bytes. Expecting less than 100 bytes.' %
+            (resource, cl))
+
+    results.AddValue(scalar.ScalarValue(
+        results.current_page, 'content_length', 'bytes', cl))
+    results.AddValue(scalar.ScalarValue(
+        results.current_page, 'original_content_length', 'bytes', ocl))
+    results.AddValue(scalar.ScalarValue(
+        results.current_page, 'data_saving', 'percent', saving))
+
   def AddResultsForBypass(self, tab, results):
     bypass_count = 0
 
     for resp in self.IterResponses(tab):
       if resp.HasChromeProxyViaHeader():
         r = resp.response
         raise ChromeProxyMetricException, (
             '%s: Should not have Via header (%s) (refer=%s, status=%d)' % (
                 r.url, r.GetHeader('Via'), r.GetHeader('Referer'), r.status))
       bypass_count += 1
@@ skipping 220 matching lines @@
             'Response for %s should have via header; proxy should no longer '
             'be bypassed.\nReponse: status=(%d, %s)\nHeaders:\n %s' % (
                 r.url, r.status, r.status_text, r.headers))
       else:
         via_count += 1
 
     results.AddValue(scalar.ScalarValue(
         results.current_page, 'bypass', 'count', bypass_count))
     results.AddValue(scalar.ScalarValue(
         results.current_page, 'via', 'count', via_count))