Add a test to verify the use of getVar and write inside of a PSP

1 files changed, 17 insertions, 1 deletions

diff --git a/cheetah/Tests/SyntaxAndOutput.py b/cheetah/Tests/SyntaxAndOutput.py
index 66e8bd7..92c129b 100644
--- a/cheetah/Tests/SyntaxAndOutput.py
+++ b/cheetah/Tests/SyntaxAndOutput.py
@@ -162,7 +162,8 @@ Template output mismatch:
                 **extraKwArgs
                 )
             moduleCode = templateClass._CHEETAH_generatedModuleCode
-            self.template = templateObj = templateClass(searchList=self.searchList())
+            searchList = self.searchList() or self._searchList
+            self.template = templateObj = templateClass(searchList=searchList)
         else:
             self.template = templateObj = Template(
                 input,
@@ -2344,6 +2345,8 @@ class UnlessDirective(OutputTest):
         self.verify("#unless 0: 1234\n"*2, "1234\n"*2)
         
 class PSP(OutputTest):
+    def searchList(self):
+        return None
     
     def test1(self):
         """simple <%= [int] %>"""
@@ -2382,6 +2385,19 @@ class PSP(OutputTest):
         self.verify("""<% for i in range(5):
     i=i*2$%><%=i%>-<%end%>""", "0-2-4-6-8-")
 
+    def test10(self):
+        """ Using getVar and write within a PSP """
+        self._searchList = [{'me' : 1}]
+        template = '''This is my template
+<%
+me = self.getVar('me')
+if isinstance(me, int):
+    write('Bork')
+else:
+    write('Nork')
+%>'''
+        self.verify(template, 'This is my template\nBork')
+
 
 class WhileDirective(OutputTest):
     def test1(self):