Add include_error_handler parameter.

This works like error_handler but is specific to a template
when included in another using the include tag.

Change-Id: Ie5506a8cba42c71519c703eacc82050902b9ceba
Pull-request: https://bitbucket.org/zzzeek/mako/pull-requests/22

1 files changed, 15 insertions, 0 deletions

diff --git a/test/test_template.py b/test/test_template.py
index f551230..7f21978 100644
--- a/test/test_template.py
+++ b/test/test_template.py
@@ -613,6 +613,21 @@ class IncludeTest(TemplateTest):
         """)
         assert flatten_result(lookup.get_template("c").render()) == "bar: calling bar this is a"
 
+    def test_include_error_handler(self):
+        def handle(context, error):
+            context.write('include error')
+            return True
+
+        lookup = TemplateLookup(include_error_handler=handle)
+        lookup.put_string("a", """
+            this is a.
+            <%include file="b"/>
+        """)
+        lookup.put_string("b", """
+            this is b ${1/0} end.
+        """)
+        assert flatten_result(lookup.get_template("a").render()) == "this is a. this is b include error"
+
 class UndefinedVarsTest(TemplateTest):
     def test_undefined(self):
         t = Template("""