From ad6a6706391c60dbdb66073caff1306b771da0bd Mon Sep 17 00:00:00 2001 From: Chris McDonough Date: Sun, 12 Sep 2010 22:33:16 +0000 Subject: - Add a new request API: ``request.add_finished_callback``. Finished callbacks are called by the router unconditionally near the very end of request processing. --- repoze/bfg/tests/test_request.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) (limited to 'repoze/bfg/tests/test_request.py') diff --git a/repoze/bfg/tests/test_request.py b/repoze/bfg/tests/test_request.py index 0248e10be..d8a3c09fb 100644 --- a/repoze/bfg/tests/test_request.py +++ b/repoze/bfg/tests/test_request.py @@ -184,6 +184,28 @@ class TestRequest(unittest.TestCase): self.assertEqual(response.called2, True) self.assertEqual(inst.response_callbacks, ()) + def test_add_finished_callback(self): + inst = self._makeOne({}) + self.assertEqual(inst.finished_callbacks, ()) + def callback(request): + """ """ + inst.add_finished_callback(callback) + self.assertEqual(inst.finished_callbacks, [callback]) + inst.add_finished_callback(callback) + self.assertEqual(inst.finished_callbacks, [callback, callback]) + + def test__process_finished_callbacks(self): + inst = self._makeOne({}) + def callback1(request): + request.called1 = True + def callback2(request): + request.called2 = True + inst.finished_callbacks = [callback1, callback2] + inst._process_finished_callbacks() + self.assertEqual(inst.called1, True) + self.assertEqual(inst.called2, True) + self.assertEqual(inst.finished_callbacks, ()) + class Test_route_request_iface(unittest.TestCase): def _callFUT(self, name): from repoze.bfg.request import route_request_iface -- cgit v1.2.3