From 6225a24982dfaeffbc53f85d214159c08a0dbfc2 Mon Sep 17 00:00:00 2001
From: Chris McDonough <chrism@agendaless.com>
Date: Sat, 19 Dec 2009 18:40:19 +0000
Subject: - Add a ``custom_predicates`` argument to the ``Configurator``  
 ``add_view`` method, the ``bfg_view`` decorator and the attribute   list of
 the ZCML ``view`` directive.  If ``custom_predicates`` is   specified, it
 must be a sequence of predicate callables (a predicate   callable accepts two
 arguments: ``context`` and ``request`` and   returns ``True`` or ``False``). 
 The associated view callable will   only be invoked if all custom predicates
 return ``True``.  Use one   or more custom predicates when no existing
 predefined predicate is   useful.  Predefined and custom predicates can be
 mixed freely.

- Add a ``custom_predicates`` argument to the ``Configurator``
  ``add_route`` and the attribute list of the ZCML ``route``
  directive.  If ``custom_predicates`` is specified, it must be a
  sequence of predicate callables (a predicate callable accepts two
  arguments: ``context`` and ``request`` and returns ``True`` or
  ``False``).  The associated route will match will only be invoked if
  all custom predicates return ``True``, else route matching
  continues.  Use one or more custom predicates when no existing
  predefined predicate is useful.  Predefined and custom predicates
  can be mixed freely.
---
 repoze/bfg/tests/test_zcml.py | 43 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 43 insertions(+)

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

diff --git a/repoze/bfg/tests/test_zcml.py b/repoze/bfg/tests/test_zcml.py
index f3e0ed80f..cbbbe664c 100644
--- a/repoze/bfg/tests/test_zcml.py
+++ b/repoze/bfg/tests/test_zcml.py
@@ -92,6 +92,31 @@ class TestViewDirective(unittest.TestCase):
         regview = reg.adapters.lookup((IDummy, IRequest), IView, name='')
         self.assertEqual(regview(None, None).body, 'OK')
 
+    def test_with_custom_predicates(self):
+        from repoze.bfg.threadlocal import get_current_registry
+        from repoze.bfg.interfaces import IView
+        from repoze.bfg.interfaces import IRequest
+        context = DummyContext()
+        reg = get_current_registry()
+        view = lambda *arg: 'OK'
+        def pred1(context, request):
+            return True
+        def pred2(context, request):
+            return True
+        preds = (pred1, pred2)
+        self._callFUT(context, 'repoze.view', IDummy, view=view,
+                      custom_predicates=preds)
+        actions = context.actions
+        self.assertEqual(len(actions), 1)
+        discrim = ('view', IDummy, '', None, IView, None, None, None, None,
+                   None, False, None, None, None)
+        discrim = discrim + tuple(sorted(preds))
+        self.assertEqual(actions[0]['discriminator'], discrim)
+        register = actions[0]['callable']
+        register()
+        regview = reg.adapters.lookup((IDummy, IRequest), IView, name='')
+        self.assertEqual(regview(None, None), 'OK')
+
 class TestNotFoundDirective(unittest.TestCase):
     def setUp(self):
         testing.setUp()
@@ -490,6 +515,24 @@ class TestRouteDirective(unittest.TestCase):
         result = wrapped(None, request)
         self.assertEqual(result.body, 'OK')
 
+    def test_with_custom_predicates(self):
+        def pred1(context, request): pass
+        def pred2(context, request): pass
+        preds = tuple(sorted([pred1, pred2]))
+
+        context = DummyContext()
+        self._callFUT(context, 'name', 'path', custom_predicates=(pred1, pred2))
+        actions = context.actions
+        self.assertEqual(len(actions), 1)
+
+        route_action = actions[0]
+        route_action['callable']()
+        route_discriminator = route_action['discriminator']
+        self.assertEqual(
+            route_discriminator,
+            ('route', 'name', False, None, None, None, None,None) + preds)
+        self._assertRoute('name', 'path', 2)
+
 class TestStaticDirective(unittest.TestCase):
     def setUp(self):
         testing.setUp()
-- 
cgit v1.2.3