From a0423aedb4abaf12b4008c0b229ec0ecad4ddfd8 Mon Sep 17 00:00:00 2001
From: Chris McDonough <chrism@agendaless.com>
Date: Thu, 31 Jul 2008 06:17:31 +0000
Subject:   - Add event sends for INewRequest and INewResponse.  See the    
 events.rst chapter in the documentation's ``api`` directory.

---
 repoze/bfg/tests/test_router.py | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)

(limited to 'repoze/bfg/tests/test_router.py')

diff --git a/repoze/bfg/tests/test_router.py b/repoze/bfg/tests/test_router.py
index 4b1e914b7..d2907ef8a 100644
--- a/repoze/bfg/tests/test_router.py
+++ b/repoze/bfg/tests/test_router.py
@@ -33,6 +33,11 @@ class RouterTests(unittest.TestCase, PlacelessSetup):
         from repoze.bfg.interfaces import ISecurityPolicy
         gsm.registerUtility(secpol, ISecurityPolicy)
 
+    def _registerEventListener(self, listener, iface):
+        import zope.component
+        gsm = zope.component.getGlobalSiteManager()
+        gsm.registerHandler(listener, (iface,))
+
     def _getTargetClass(self):
         from repoze.bfg.router import Router
         return Router
@@ -232,6 +237,34 @@ class RouterTests(unittest.TestCase, PlacelessSetup):
         self.failUnless('permission' in result[0])
         self.assertEqual(permissionfactory.checked_with, secpol)
 
+    def test_call_eventsends(self):
+        rootpolicy = make_rootpolicy(None)
+        context = DummyContext()
+        traversalfactory = make_traversal_factory(context, '', [])
+        response = DummyResponse()
+        response.app_iter = ['Hello world']
+        view = make_view(response)
+        environ = self._makeEnviron()
+        self._registerTraverserFactory(traversalfactory, '', None, None)
+        self._registerView(view, '', None, None)
+        from repoze.bfg.interfaces import INewRequest
+        from repoze.bfg.interfaces import INewResponse
+        request_events = []
+        response_events = []
+        def handle_request(event):
+            request_events.append(event)
+        def handle_response(event):
+            response_events.append(event)
+        self._registerEventListener(handle_request, INewRequest)
+        self._registerEventListener(handle_response, INewResponse)
+        router = self._makeOne(rootpolicy, None)
+        start_response = DummyStartResponse()
+        result = router(environ, start_response)
+        self.assertEqual(len(request_events), 1)
+        self.assertEqual(request_events[0].request.environ, environ)
+        self.assertEqual(len(response_events), 1)
+        self.assertEqual(response_events[0].response, response)
+
 class MakeAppTests(unittest.TestCase, PlacelessSetup):
     def setUp(self):
         PlacelessSetup.setUp(self)
-- 
cgit v1.2.3