Add a method to create an entry from a rendered template

5 files changed, 122 insertions, 0 deletions

diff --git a/tests/data/simple/templates/new_entry_from_data.yaml b/tests/data/simple/templates/new_entry_from_data.yaml
new file mode 100644
index 0000000..1f9e9a5
--- /dev/null
+++ b/tests/data/simple/templates/new_entry_from_data.yaml
@@ -0,0 +1,12 @@
+name: '{{ name }}'
+description: |
+  {{ description if description else "." | indent(width=2, first=False) }}
+position: '{{ position }}'
+# # quantity (integer): how many items are there
+quantity: {{ quantity if quantity else "0" }}
+# # value (float): how much each item is
+value: 0.0
+# # cost (decimal): how much this costs
+cost: '0'
+# # other (yaml): 
+other:
diff --git a/tests/data/simple/templates/new_entry_from_data_broken.yaml b/tests/data/simple/templates/new_entry_from_data_broken.yaml
new file mode 100644
index 0000000..73798ce
--- /dev/null
+++ b/tests/data/simple/templates/new_entry_from_data_broken.yaml
@@ -0,0 +1,14 @@
+name: '{{ name }}'
+description: |
+  {{ description if description else "." indent(width=2, first=False) }}
+  # Note that the lack of | between "." and indent is wanted, to get a proper
+  # TemplatingError.
+position: '{{ position }}'
+# quantity (integer): how many items are there
+quantity: {{ quantity if quantity else "0" }}
+# value (float): how much each item is
+value: 0.0
+# cost (decimal): how much this costs
+cost: '0'
+# other (yaml): 
+other:
diff --git a/tests/data/simple/templates/new_entry_from_data_invalid_yaml.yaml b/tests/data/simple/templates/new_entry_from_data_invalid_yaml.yaml
new file mode 100644
index 0000000..7493947
--- /dev/null
+++ b/tests/data/simple/templates/new_entry_from_data_invalid_yaml.yaml
@@ -0,0 +1,13 @@
+# A : in that position should result in invalid yaml
+name: '{{ name }}': some text
+description: |
+  {{ description if description else "." | indent(width=2, first=False) }}
+position: '{{ position }}'
+# # quantity (integer): how many items are there
+quantity: {{ quantity if quantity else "0" }}
+# # value (float): how much each item is
+value: 0.0
+# # cost (decimal): how much this costs
+cost: '0'
+# # other (yaml): 
+other:
diff --git a/tests/data/simple/templates/new_entry_from_multiple_data.yaml b/tests/data/simple/templates/new_entry_from_multiple_data.yaml
new file mode 100644
index 0000000..cfa4296
--- /dev/null
+++ b/tests/data/simple/templates/new_entry_from_multiple_data.yaml
@@ -0,0 +1,12 @@
+name: '{{ data.name }}'
+description: |
+  {{ description if description else "." | indent(width=2, first=False) }}
+position: '{{ data.position }}'
+# # quantity (integer): how many items are there
+quantity: {{ values.quantity if values.quantity else "0" }}
+# # value (float): how much each item is
+value: 0.0
+# # cost (decimal): how much this costs
+cost: '{{ values.cost if values.cost else "0.0" }}'
+# # other (yaml): 
+other:
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):