def test_unexisting_phase(self):
handler = Handlers()
with self.assertRaises(Exception) as cm:
handler.handlers('foo')
exception = cm.exception
self.assertTrue(exception)
pass
def test_unexisting_phase(self):
handler = Handlers()
with self.assertRaises(Exception) as cm:
handler.handlers('foo')
exception = cm.exception
self.assertTrue(exception)
pass
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_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_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 = Client()
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_prepare(self):
request = Request('FOO', 'http')
client = Client()
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_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_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)