| Index: tools/telemetry/telemetry/core/backends/chrome_inspector/devtools_http_unittest.py
|
| diff --git a/tools/telemetry/telemetry/core/backends/chrome_inspector/devtools_http_unittest.py b/tools/telemetry/telemetry/core/backends/chrome_inspector/devtools_http_unittest.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..6e854f7751a29aca46fc7581bbe1aed4c986b365
|
| --- /dev/null
|
| +++ b/tools/telemetry/telemetry/core/backends/chrome_inspector/devtools_http_unittest.py
|
| @@ -0,0 +1,53 @@
|
| +# 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 httplib
|
| +import socket
|
| +import unittest
|
| +import urllib2
|
| +
|
| +from telemetry.core.backends.chrome_inspector import devtools_http
|
| +
|
| +
|
| +_original_proxy_handler = urllib2.ProxyHandler
|
| +
|
| +class DevToolsHttpTest(unittest.TestCase):
|
| +
|
| + def testUrlError(self):
|
| + class FakeProxyHandler(object):
|
| + def __init__(self, *_, **__):
|
| + raise urllib2.URLError('Test')
|
| +
|
| + urllib2.ProxyHandler = FakeProxyHandler
|
| + try:
|
| + with self.assertRaises(devtools_http.DevToolsClientUrlError):
|
| + devtools_http.DevToolsHttp(1000).Request('')
|
| + finally:
|
| + urllib2.ProxyHandler = _original_proxy_handler
|
| +
|
| + def testSocketError(self):
|
| + class FakeProxyHandler(object):
|
| + def __init__(self, *_, **__):
|
| + raise socket.error
|
| +
|
| + urllib2.ProxyHandler = FakeProxyHandler
|
| + try:
|
| + with self.assertRaises(devtools_http.DevToolsClientConnectionError) as e:
|
| + devtools_http.DevToolsHttp(1000).Request('')
|
| + self.assertNotIsInstance(e, devtools_http.DevToolsClientUrlError)
|
| + finally:
|
| + urllib2.ProxyHandler = _original_proxy_handler
|
| +
|
| + def testBadStatusLine(self):
|
| + class FakeProxyHandler(object):
|
| + def __init__(self, *_, **__):
|
| + raise httplib.BadStatusLine('Test')
|
| +
|
| + urllib2.ProxyHandler = FakeProxyHandler
|
| + try:
|
| + with self.assertRaises(devtools_http.DevToolsClientConnectionError) as e:
|
| + devtools_http.DevToolsHttp(1000).Request('')
|
| + self.assertNotIsInstance(e, devtools_http.DevToolsClientUrlError)
|
| + finally:
|
| + urllib2.ProxyHandler = _original_proxy_handler
|
|
|