add tests for packer template

tests/packerlicious/test_template.py

@@ -0,0 +1,99 @@
+import json
+
+from packerlicious import UserVar, Template
+from packerlicious import builder, post_processor, provisioner
+
+
+class TestPackerTemplate(object):
+    def test_template(self):
+        t = Template(description="test description", min_packer_version="0.9.0")
+
+        d = t.to_dict()
+        assert d['description'] == "test description"
+        assert d['min_packer_version'] == "0.9.0"
+
+        t.add_description("new description")
+        t.add_min_packer_version("1.0.0")
+        d = t.to_dict()
+        assert d['description'] == "new description"
+        assert d['min_packer_version'] == "1.0.0"
+
+    def test_template_variables(self):
+        t = Template()
+
+        t.add_variable(UserVar("my_var", "my_value"))
+
+        d = t.to_dict()
+        assert d['variables'] == {'my_var': 'my_value'}
+
+    def test_template_builders(self):
+        expected_json = """
+        {
+          "builders": [
+            {
+              "source": "/source/path",
+              "target": "/target/path",
+              "type": "file"
+            }
+          ]
+        }
+        """
+
+        t = Template()
+        t.add_builder(builder.File(
+            target="/target/path",
+            source="/source/path",
+        ))
+
+        to_json = t.to_json()
+        assert to_json == json.dumps(json.loads(expected_json), sort_keys=True, indent=2,
+                                     separators=(',', ': '))
+
+    def test_template_provisioners(self):
+        expected_json = """
+        {
+          "provisioners": [
+            {
+              "source": "/src/path",
+              "destination": "/dest/path",
+              "direction": "upload",
+              "type": "file"
+            }
+          ]
+        }
+        """
+
+        t = Template()
+        t.add_provisioner(provisioner.File(
+            source="/src/path",
+            destination="/dest/path",
+            direction=provisioner.File.Upload,
+        ))
+
+        to_json = t.to_json()
+        assert to_json == json.dumps(json.loads(expected_json), sort_keys=True, indent=2,
+                                     separators=(',', ': '))
+
+    def test_template_post_processors(self):
+        expected_json = """
+        {
+          "post-processors": [
+            {
+              "script": "/my/post/script",
+              "type": "shell-local"
+            }
+          ]
+        }
+        """
+
+        t = Template()
+        t.add_post_processor(post_processor.ShellLocal(
+            script="/my/post/script",
+        ))
+
+        to_json = t.to_json()
+        assert to_json == json.dumps(json.loads(expected_json), sort_keys=True, indent=2,
+                                     separators=(',', ': '))
+
+
+

tests/packerlicious/test_variables.py

@@ -1,6 +1,6 @@
 import pytest
 
-from packerlicious import Ref,EnvVar, TemplateVar, UserVar
+from packerlicious import Ref, EnvVar, TemplateVar, UserVar
 
 
 class TestPackerVariables(object):