Move some templating code to the collection instead of commands.

1 files changed, 17 insertions, 0 deletions

diff --git a/tests/test_collection.py b/tests/test_collection.py
index d1d6455..09ff96c 100644
--- a/tests/test_collection.py
+++ b/tests/test_collection.py
@@ -145,6 +145,14 @@ class testCollection(unittest.TestCase):
         entry = self.collection.entry_from_uid('this is not an uid')
         self.assertIsNone(entry)
 
+    def test_render_collection(self):
+        self.collection = lesana.Collection('tests/data/simple')
+        template = self.collection.get_template(
+            'tests/data/simple/templates/collection_template.txt'
+            )
+        res = template.render(entries=self.collection.get_all_documents())
+        self.assertIn('11189ee4: Another item', res)
+
 
 class testEntries(unittest.TestCase):
     def setUp(self):
@@ -243,6 +251,15 @@ class testEntries(unittest.TestCase):
         self.collection.settings['entry_label'] = '{{ uid }}: {{ name }}'
         self.assertEqual(str(entry), uid + ': Another item')
 
+    def test_render_entry(self):
+        fname = '11189ee47ddf4796b718a483b379f976.yaml'
+        with open(os.path.join(self.basepath, fname)) as fp:
+            data = ruamel.yaml.safe_load(fp)
+        entry = lesana.Entry(self.collection, data=data)
+        uid = entry.uid
+        res = entry.render('tests/data/simple/templates/trivial_template.txt')
+        self.assertIn(uid, res)
+
 
 class testComplexCollection(unittest.TestCase):
     @classmethod