def test_unexisting_phase(self): handler = Handlers() with self.assertRaises(Exception) as cm: handler.handlers('foo') exception = cm.exception self.assertTrue(exception) pass
def __init__(self, agent=None, timeout=60, keep_alive=1, default_headers={}, max_redirect=7, with_exceptions=False): self.timeout = timeout self.max_redirect = max_redirect self._handlers = Handlers() self._with_exceptions = with_exceptions if agent is None: self.agent = "python-httpclient/{version}".format( version=__version__) else: self.agent = agent if len(default_headers) == 0: default_headers = [( 'Connection', 'keep-alive', )] self._default_headers = Headers(default_headers) if 'User-Agent' not in default_headers: self._default_headers.add('User-Agent', self.agent) self._default_headers.add('User-Agent', 'foooo') self._poolmanager = PoolManager( maxsize=keep_alive )
def test_add_remove_handler(self): handlers = Handlers() handlers.add_handler('request_send', _cb_request_send) self.assertTrue(handlers.handlers('request_send')) handlers.remove_handler('request_send') self.assertFalse(handlers.handlers('request_send'))
def test_prepare(self): request = Request('FOO', 'http') client = HTTPClient() client.add_handler('request_prepare', _cb_request_prepare) resp = client.request(request) self.assertEqual(resp.status, 400) handlers = Handlers() handlers.add_handler('request_prepare', _cb_request_prepare_change_request) req = handlers.dispatch('request_prepare', request) self.assertEqual(req.method, 'PUT') handlers = Handlers() handlers.add_handler('request_prepare', _cb_request_prepare_do_nothing) req = handlers.dispatch('request_prepare', request) self.assertEqual(req.method, 'FOO') self.assertEqual(req.url, 'http')
def test_overwrite_handlers(self): request = Request('GET', 'http') handlers = Handlers() handlers.add_handler('request_send', lambda r: 1) resp = handlers.dispatch('request_send', request) self.assertEqual(resp, 1) handlers.add_handler('request_send', lambda r: 2) resp = handlers.dispatch('request_send', request) self.assertEqual(resp, 2)
def test_execute_handler(self): request = Request('GET', 'http') handlers = Handlers() handlers.add_handler('request_send', _cb_request_send) resp = handlers.dispatch('request_send', request) self.assertIsInstance(resp, Response)
def test_simple_handler(self): handlers = Handlers() self.assertTrue(handlers)