1 files changed, 155 insertions, 0 deletions
diff --git a/tests/test_scaffolds/test_template.py b/tests/test_scaffolds/test_template.py
new file mode 100644
index 000000000..98f2daf73
--- /dev/null
+++ b/tests/test_scaffolds/test_template.py
@@ -0,0 +1,155 @@
+import unittest
+
+from pyramid.compat import bytes_
+
+class TestTemplate(unittest.TestCase):
+    def _makeOne(self, name='whatever'):
+        from pyramid.scaffolds.template import Template
+        return Template(name)
+
+    def test_render_template_success(self):
+        inst = self._makeOne()
+        result = inst.render_template('{{a}} {{b}}', {'a':'1', 'b':'2'})
+        self.assertEqual(result, bytes_('1 2'))
+
+    def test_render_template_expr_failure(self):
+        inst = self._makeOne()
+        self.assertRaises(AttributeError, inst.render_template,
+                          '{{a.foo}}', {'a':'1', 'b':'2'})
+
+    def test_render_template_expr_success(self):
+        inst = self._makeOne()
+        result = inst.render_template('{{a.lower()}}', {'a':'A'})
+        self.assertEqual(result, b'a')
+
+    def test_render_template_expr_success_via_pipe(self):
+        inst = self._makeOne()
+        result = inst.render_template('{{b|c|a.lower()}}', {'a':'A'})
+        self.assertEqual(result, b'a')
+
+    def test_render_template_expr_success_via_pipe2(self):
+        inst = self._makeOne()
+        result = inst.render_template('{{b|a.lower()|c}}', {'a':'A'})
+        self.assertEqual(result, b'a')
+
+    def test_render_template_expr_value_is_None(self):
+        inst = self._makeOne()
+        result = inst.render_template('{{a}}', {'a':None})
+        self.assertEqual(result, b'')
+
+    def test_render_template_with_escaped_double_braces(self):
+        inst = self._makeOne()
+        result = inst.render_template('{{a}} {{b}} \{\{a\}\} \{\{c\}\}', {'a':'1', 'b':'2'})
+        self.assertEqual(result, bytes_('1 2 {{a}} {{c}}'))
+
+    def test_render_template_with_breaking_escaped_braces(self):
+        inst = self._makeOne()
+        result = inst.render_template('{{a}} {{b}} \{\{a\} \{b\}\}', {'a':'1', 'b':'2'})
+        self.assertEqual(result, bytes_('1 2 \{\{a\} \{b\}\}'))
+
+    def test_render_template_with_escaped_single_braces(self):
+        inst = self._makeOne()
+        result = inst.render_template('{{a}} {{b}} \{a\} \{b', {'a':'1', 'b':'2'})
+        self.assertEqual(result, bytes_('1 2 \{a\} \{b'))
+
+    def test_module_dir(self):
+        import sys
+        import pkg_resources
+        package = sys.modules['pyramid.scaffolds.template']
+        path = pkg_resources.resource_filename(package.__name__, '')
+        inst = self._makeOne()
+        result = inst.module_dir()
+        self.assertEqual(result, path)
+
+    def test_template_dir__template_dir_is_None(self):
+        inst = self._makeOne()
+        self.assertRaises(AssertionError, inst.template_dir)
+
+    def test_template_dir__template_dir_is_tuple(self):
+        inst = self._makeOne()
+        inst._template_dir = ('a', 'b')
+        self.assertEqual(inst.template_dir(), ('a', 'b'))
+
+    def test_template_dir__template_dir_is_not_None(self):
+        import os
+        import sys
+        import pkg_resources
+        package = sys.modules['pyramid.scaffolds.template']
+        path = pkg_resources.resource_filename(package.__name__, '')
+        inst = self._makeOne()
+        inst._template_dir ='foo'
+        result = inst.template_dir()
+        self.assertEqual(result, os.path.join(path, 'foo'))
+
+    def test_write_files_path_exists(self):
+        import os
+        import sys
+        import pkg_resources
+        package = sys.modules['pyramid.scaffolds.template']
+        path = pkg_resources.resource_filename(package.__name__, '')
+        inst = self._makeOne()
+        inst._template_dir = 'foo'
+        inst.exists = lambda *arg: True
+        copydir = DummyCopydir()
+        inst.copydir = copydir
+        command = DummyCommand()
+        inst.write_files(command, 'output dir', {'a':1})
+        self.assertEqual(copydir.template_dir, os.path.join(path, 'foo'))
+        self.assertEqual(copydir.output_dir, 'output dir')
+        self.assertEqual(copydir.vars, {'a':1})
+        self.assertEqual(copydir.kw,
+                         {'template_renderer':inst.render_template,
+                          'indent':1,
+                          'verbosity':1,
+                          'simulate':False,
+                          'overwrite':False,
+                          'interactive':False,
+                          })
+
+    def test_write_files_path_missing(self):
+        L = []
+        inst = self._makeOne()
+        inst._template_dir = 'foo'
+        inst.exists = lambda *arg: False
+        inst.out = lambda *arg: None
+        inst.makedirs = lambda dir: L.append(dir)
+        copydir = DummyCopydir()
+        inst.copydir = copydir
+        command = DummyCommand()
+        inst.write_files(command, 'output dir', {'a':1})
+        self.assertEqual(L, ['output dir'])
+
+    def test_run(self):
+        L = []
+        inst = self._makeOne()
+        inst._template_dir = 'foo'
+        inst.exists = lambda *arg: False
+        inst.out = lambda *arg: None
+        inst.makedirs = lambda dir: L.append(dir)
+        copydir = DummyCopydir()
+        inst.copydir = copydir
+        command = DummyCommand()
+        inst.run(command, 'output dir', {'a':1})
+        self.assertEqual(L, ['output dir'])
+
+    def test_check_vars(self):
+        inst = self._makeOne()
+        self.assertRaises(RuntimeError, inst.check_vars, 'one', 'two')
+
+class DummyCopydir(object):
+    def copy_dir(self, template_dir, output_dir, vars, **kw):
+        self.template_dir = template_dir
+        self.output_dir = output_dir
+        self.vars = vars
+        self.kw = kw
+
+class DummyArgs(object):
+    simulate = False
+    overwrite = False
+    interactive = False
+
+class DummyCommand(object):
+    args = DummyArgs()
+    verbosity = 1
+
+