- The ``notfound`` and ``forbidden`` ZCML directives now accept the

  following addtional attributes: ``attr``, ``renderer``, and
  ``wrapper``.  These have the same meaning as they do in the context
  of a ZCML ``view`` directive.

3 files changed, 110 insertions, 18 deletions

diff --git a/repoze/bfg/includes/meta.zcml b/repoze/bfg/includes/meta.zcml
index 2eeb87314..4c9d18873 100644
--- a/repoze/bfg/includes/meta.zcml
+++ b/repoze/bfg/includes/meta.zcml
@@ -18,13 +18,13 @@
 
     <meta:directive
         name="notfound"
-        schema="repoze.bfg.zcml.INotFoundViewDirective"
+        schema="repoze.bfg.zcml.ISystemViewDirective"
         handler="repoze.bfg.zcml.notfound"
         />
 
     <meta:directive
         name="forbidden"
-        schema="repoze.bfg.zcml.IForbiddenViewDirective"
+        schema="repoze.bfg.zcml.ISystemViewDirective"
         handler="repoze.bfg.zcml.forbidden"
         />
 
diff --git a/repoze/bfg/tests/test_zcml.py b/repoze/bfg/tests/test_zcml.py
index 35584f301..7cbadd1e2 100644
--- a/repoze/bfg/tests/test_zcml.py
+++ b/repoze/bfg/tests/test_zcml.py
@@ -947,6 +947,12 @@ class TestViewDirective(unittest.TestCase):
         self.assertEqual(wrapper, view)
 
 class TestNotFoundDirective(unittest.TestCase):
+    def setUp(self):
+        cleanUp()
+
+    def tearDown(self):
+        cleanUp()
+        
     def _callFUT(self, context, view):
         from repoze.bfg.zcml import notfound
         return notfound(context, view)
@@ -972,6 +978,12 @@ class TestNotFoundDirective(unittest.TestCase):
         self.assertEqual(derived_view.__name__, view.__name__)
 
 class TestForbiddenDirective(unittest.TestCase):
+    def setUp(self):
+        cleanUp()
+
+    def tearDown(self):
+        cleanUp()
+        
     def _callFUT(self, context, view):
         from repoze.bfg.zcml import forbidden
         return forbidden(context, view)
@@ -996,6 +1008,67 @@ class TestForbiddenDirective(unittest.TestCase):
         self.assertEqual(derived_view(None, None), 'OK')
         self.assertEqual(derived_view.__name__, view.__name__)
 
+class TestViewUtility(unittest.TestCase):
+    def setUp(self):
+        cleanUp()
+
+    def tearDown(self):
+        cleanUp()
+        
+    def _callFUT(self, context, view, attr, renderer, wrapper, iface):
+        from repoze.bfg.zcml import view_utility
+        return view_utility(context, view, attr, renderer, wrapper, iface)
+
+    def test_no_view_no_renderer(self):
+        from zope.configuration.exceptions import ConfigurationError
+        context = DummyContext()
+        self.assertRaises(ConfigurationError, self._callFUT, context,
+                          None, None, None, None, None)
+    
+    def test_no_view_with_renderer(self):
+        from zope.component import getSiteManager
+        from zope.interface import Interface
+        from repoze.bfg.interfaces import IRendererFactory
+        sm = getSiteManager()
+        def renderer(path):
+            return lambda *arg: 'OK'
+        sm.registerUtility(renderer, IRendererFactory, name='dummy')
+        class IDummy(Interface):
+            pass
+        context = DummyContext()
+        self._callFUT(context, None, None, 'dummy', None, IDummy)
+        actions = context.actions
+        self.assertEqual(len(actions), 1)
+        regadapt = actions[0]
+        self.assertEqual(regadapt['discriminator'], IDummy)
+        register = regadapt['callable']
+        register()
+        derived_view = sm.getUtility(IDummy)
+        request = DummyRequest()
+        self.assertEqual(derived_view(None, request).body, 'OK')
+
+    def test_template_renderer(self):
+        from zope.component import getSiteManager
+        from zope.interface import Interface
+        from repoze.bfg.interfaces import IRendererFactory
+        sm = getSiteManager()
+        def renderer(path):
+            return lambda *arg: 'OK'
+        sm.registerUtility(renderer, IRendererFactory, name='.pt')
+        class IDummy(Interface):
+            pass
+        context = DummyContext()
+        self._callFUT(context, None, None, 'fixtures/minimal.pt', None, IDummy)
+        actions = context.actions
+        self.assertEqual(len(actions), 1)
+        regadapt = actions[0]
+        self.assertEqual(regadapt['discriminator'], IDummy)
+        register = regadapt['callable']
+        register()
+        derived_view = sm.getUtility(IDummy)
+        request = DummyRequest()
+        self.assertEqual(derived_view(None, request).body, 'OK')
+
 class TestRepozeWho1AuthenticationPolicyDirective(unittest.TestCase):
     def setUp(self):
         cleanUp()
diff --git a/repoze/bfg/zcml.py b/repoze/bfg/zcml.py
index 3faadb24b..adb1f1f31 100644
--- a/repoze/bfg/zcml.py
+++ b/repoze/bfg/zcml.py
@@ -279,29 +279,49 @@ def view(
 
 _view = view # for directives that take a view arg
 
-class INotFoundViewDirective(Interface):
+class ISystemViewDirective(Interface):
     view = GlobalObject(
         title=u"",
-        description=u"The notfound view callable",
-        required=True,
+        description=u"The view function",
+        required=False,
         )
 
-def notfound(_context, view):
-    view_utility(_context, view, INotFoundView)
+    attr = TextLine(
+        title=u'The callable attribute of the view object(default is __call__)',
+        description=u'',
+        required=False)
 
-class IForbiddenViewDirective(Interface):
-    view = GlobalObject(
-        title=u"",
-        description=u"The forbidden view callable",
-        required=True,
-        )
+    renderer = TextLine(
+        title=u'The renderer asssociated with the view',
+        description=u'',
+        required=False)
 
-def forbidden(_context, view):
-    view_utility(_context, view, IForbiddenView)
+    wrapper = TextLine(
+        title = u'The *name* of the view that acts as a wrapper for this view.',
+        description = u'',
+        required=False)
+
+def notfound(_context, view=None, attr=None, renderer=None, wrapper=None):
+    view_utility(_context, view, attr, renderer, wrapper, INotFoundView)
+
+def forbidden(_context, view=None, attr=None, renderer=None, wrapper=None):
+    view_utility(_context, view, attr, renderer, wrapper, IForbiddenView)
+
+def view_utility(_context, view, attr, renderer, wrapper, iface):
+    if not view:
+        if renderer:
+            def view(context, request):
+                return {}
+        else:
+            raise ConfigurationError('"view" attribute was not specified and '
+                                     'no renderer specified')
+
+    if renderer and '.' in renderer:
+        renderer = resource_spec(renderer, package_name(_context.resolve('.')))
 
-def view_utility(_context, view, iface):
     def register():
-        derived_view = derive_view(view)
+        derived_view = derive_view(view, attr=attr, renderer_name=renderer,
+                                   wrapper_viewname=wrapper)
         sm = getSiteManager()
         sm.registerUtility(derived_view, iface, '', _context.info)
 
@@ -310,7 +330,6 @@ def view_utility(_context, view, iface):
         callable = register,
         )
 
-
 class IResourceDirective(Interface):
     """
     Directive for specifying that one package may override resources from