Index: tools/chrome_proxy/common/chrome_proxy_metrics_unittest.py |
diff --git a/tools/chrome_proxy/common/chrome_proxy_metrics_unittest.py b/tools/chrome_proxy/common/chrome_proxy_metrics_unittest.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..35c9e1bfb91045554350488f119fac7089cd64f0 |
--- /dev/null |
+++ b/tools/chrome_proxy/common/chrome_proxy_metrics_unittest.py |
@@ -0,0 +1,44 @@ |
+# 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 base64 |
+import unittest |
+ |
+from common import chrome_proxy_metrics as metrics |
+from common import network_metrics_unittest as network_unittest |
+ |
+ |
+class ChromeProxyMetricTest(unittest.TestCase): |
+ |
+ def testChromeProxyResponse(self): |
+ # An https non-proxy response. |
+ resp = metrics.ChromeProxyResponse( |
+ network_unittest.NetworkMetricTest.MakeNetworkTimelineEvent( |
+ url='https://test.url', |
+ response_headers={ |
+ 'Content-Type': 'text/html', |
+ 'Content-Length': str(len(network_unittest.HTML_BODY)), |
+ 'Via': 'some other via', |
+ }, |
+ body=network_unittest.HTML_BODY)) |
+ self.assertFalse(resp.ShouldHaveChromeProxyViaHeader()) |
+ self.assertFalse(resp.HasChromeProxyViaHeader()) |
+ self.assertTrue(resp.IsValidByViaHeader()) |
+ |
+ # A proxied JPEG image response |
+ resp = metrics.ChromeProxyResponse( |
+ network_unittest.NetworkMetricTest.MakeNetworkTimelineEvent( |
+ url='http://test.image', |
+ response_headers={ |
+ 'Content-Type': 'image/jpeg', |
+ 'Content-Encoding': 'gzip', |
+ 'Via': '1.1 ' + metrics.CHROME_PROXY_VIA_HEADER, |
+ 'X-Original-Content-Length': str(network_unittest.IMAGE_OCL), |
+ }, |
+ body=base64.b64encode(network_unittest.IMAGE_BODY), |
+ base64_encoded_body=True)) |
+ self.assertTrue(resp.ShouldHaveChromeProxyViaHeader()) |
+ self.assertTrue(resp.HasChromeProxyViaHeader()) |
+ self.assertTrue(resp.IsValidByViaHeader()) |
+ |