Add a method to create an entry from a rendered template

1 files changed, 71 insertions, 0 deletions

diff --git a/tests/test_collection.py b/tests/test_collection.py
index d05415f..2308633 100644
--- a/tests/test_collection.py
+++ b/tests/test_collection.py
@@ -1,4 +1,5 @@
 import datetime
+import decimal
 import logging
 import os.path
 import shutil
@@ -294,6 +295,76 @@ class testSimpleCollection(unittest.TestCase):
             {'value': None, 'frequency': 1},
         ])
 
+    def test_entry_from_template(self):
+        # TODO: make finding the templates less prone to breaking and
+        # then remove the cwd change from here
+        old_cwd = os.getcwd()
+        os.chdir(self.tmpdir)
+        data = {
+            "name": "This is a name",
+        }
+        entry = self.collection.entry_from_rendered_template(
+            "templates/new_entry_from_data.yaml",
+            data
+        )
+        os.chdir(old_cwd)
+        self.assertIsInstance(entry, lesana.Entry)
+        self.assertEqual(entry.data["name"], "This is a name")
+
+    def test_entry_from_template_multiple_data_sources(self):
+        # TODO: make finding the templates less prone to breaking and
+        # then remove the cwd change from here
+        old_cwd = os.getcwd()
+        os.chdir(self.tmpdir)
+        data = {
+            "name": "This is a name",
+        }
+        values = {
+            "quantity": 5,
+            "cost": decimal.Decimal("3.5"),
+        }
+        entry = self.collection.entry_from_rendered_template(
+            "templates/new_entry_from_multiple_data.yaml",
+            {
+                "data": data,
+                "values": values
+            }
+        )
+        os.chdir(old_cwd)
+        self.assertIsInstance(entry, lesana.Entry)
+        self.assertEqual(entry.data["name"], "This is a name")
+        self.assertEqual(entry.data["quantity"], 5)
+
+    def test_entry_from_bad_template(self):
+        # TODO: make finding the templates less prone to breaking and
+        # then remove the cwd change from here
+        old_cwd = os.getcwd()
+        os.chdir(self.tmpdir)
+        data = {
+            "name": "This is a name",
+        }
+        with self.assertRaises(lesana.collection.TemplatingError):
+            self.collection.entry_from_rendered_template(
+                "templates/new_entry_from_data_broken.yaml",
+                data
+            )
+        os.chdir(old_cwd)
+
+    def test_entry_from_bad_yaml(self):
+        # TODO: make finding the templates less prone to breaking and
+        # then remove the cwd change from here
+        old_cwd = os.getcwd()
+        os.chdir(self.tmpdir)
+        data = {
+            "name": "This is a name",
+        }
+        with self.assertRaises(lesana.collection.TemplatingError):
+            self.collection.entry_from_rendered_template(
+                "templates/new_entry_from_data_invalid_yaml.yaml",
+                data
+            )
+        os.chdir(old_cwd)
+
 
 class testComplexCollection(unittest.TestCase):
     def setUp(self):