From ff6fc02595bfe339c9c5da9402edd1e1cf2f046c Mon Sep 17 00:00:00 2001
From: Jeff Baskin <jeff.l.baskin@gmail.com>
Date: Wed, 18 Jan 2017 19:51:16 -0500
Subject: [PATCH] Refactored the testing scripts.

---
 keepalived/templates/config.jinja | 18 +++++++++++-------
 test/test_keepalived_config.py    | 20 +++++++++++---------
 2 files changed, 22 insertions(+), 16 deletions(-)

diff --git a/keepalived/templates/config.jinja b/keepalived/templates/config.jinja
index a8dc487..c7754c3 100644
--- a/keepalived/templates/config.jinja
+++ b/keepalived/templates/config.jinja
@@ -1,7 +1,11 @@
-{% macro keepalived_config(data) %}
-{% if data is none %}
-
-{% else %}
-{{ data }}
-{% endif %}
-{% endmacro %}
+{%- macro keepalived_config(data) -%}
+  {%- if data is none -%}
+{{ '\n' }}
+  {%- elif data is mapping -%}
+    {%- for key, value in data.iteritems() -%}
+{{ key }} {{ keepalived_config(value) }}
+    {%- endfor -%}
+  {%- else -%}
+{{ data }}{{ '\n' }}
+  {%- endif -%}
+{%- endmacro -%}
diff --git a/test/test_keepalived_config.py b/test/test_keepalived_config.py
index 696fbcd..e72ebc3 100755
--- a/test/test_keepalived_config.py
+++ b/test/test_keepalived_config.py
@@ -14,28 +14,30 @@ class TestKeepalivedConfiguration(unittest.TestCase):
         self.t_conf = Environment(loader=FileSystemLoader(self.t_dir), 
                 trim_blocks=True)
 
-    def _render(self, data):
+    def renderTest(self, data, result):
         holder = self.t_conf.get_template('test_config.jinja').render(testdata=data)
-        print(repr(holder))
-        return holder
+        output = repr(holder) + ' did not equal ' + repr(result)
+        self.assertEqual(holder, result, output)
 
     def test_string(self):
         testdata = 'stuff'
         result = 'stuff\n'
-        self.assertEqual(self._render(testdata), result,
-                'A string should be returned with a line feed.')
+        self.renderTest(testdata, result)
 
     def test_number(self):
         testdata = 3
         result = '3\n'
-        self.assertEqual(self._render(testdata), result,
-                'A number should be returned with a line feed.')
+        self.renderTest(testdata, result)
 
     def test_null(self):
         testdata = None
         result = '\n'
-        self.assertEqual(self._render(testdata), result,
-                'A null should just return a line feed.')
+        self.renderTest(testdata, result)
+
+    def test_key_value_pair(self):
+        testdata = {'flintstone': 'fred'}
+        result = 'flintstone fred\n'
+        self.renderTest(testdata, result)
 
 if __name__ == '__main__':
     unittest.main()