OLD | NEW |
(Empty) | |
| 1 # Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. |
| 4 |
| 5 import base64 |
| 6 import unittest |
| 7 |
| 8 from common import chrome_proxy_metrics as metrics |
| 9 from common import network_metrics_unittest as network_unittest |
| 10 |
| 11 |
| 12 class ChromeProxyMetricTest(unittest.TestCase): |
| 13 |
| 14 def testChromeProxyResponse(self): |
| 15 # An https non-proxy response. |
| 16 resp = metrics.ChromeProxyResponse( |
| 17 network_unittest.NetworkMetricTest.MakeNetworkTimelineEvent( |
| 18 url='https://test.url', |
| 19 response_headers={ |
| 20 'Content-Type': 'text/html', |
| 21 'Content-Length': str(len(network_unittest.HTML_BODY)), |
| 22 'Via': 'some other via', |
| 23 }, |
| 24 body=network_unittest.HTML_BODY)) |
| 25 self.assertFalse(resp.ShouldHaveChromeProxyViaHeader()) |
| 26 self.assertFalse(resp.HasChromeProxyViaHeader()) |
| 27 self.assertTrue(resp.IsValidByViaHeader()) |
| 28 |
| 29 # A proxied JPEG image response |
| 30 resp = metrics.ChromeProxyResponse( |
| 31 network_unittest.NetworkMetricTest.MakeNetworkTimelineEvent( |
| 32 url='http://test.image', |
| 33 response_headers={ |
| 34 'Content-Type': 'image/jpeg', |
| 35 'Content-Encoding': 'gzip', |
| 36 'Via': '1.1 ' + metrics.CHROME_PROXY_VIA_HEADER, |
| 37 'X-Original-Content-Length': str(network_unittest.IMAGE_OCL), |
| 38 }, |
| 39 body=base64.b64encode(network_unittest.IMAGE_BODY), |
| 40 base64_encoded_body=True)) |
| 41 self.assertTrue(resp.ShouldHaveChromeProxyViaHeader()) |
| 42 self.assertTrue(resp.HasChromeProxyViaHeader()) |
| 43 self.assertTrue(resp.IsValidByViaHeader()) |
| 44 |
OLD | NEW |