1 files changed, 99 insertions, 0 deletions
diff --git a/tests/test_spec.py b/tests/test_spec.py
new file mode 100644
index 0000000..02f6080
--- /dev/null
+++ b/tests/test_spec.py
@@ -0,0 +1,99 @@
+# coding: utf-8
+
+"""
+Creates a unittest.TestCase for the tests defined in the mustache spec.
+
+"""
+
+# TODO: this module can be cleaned up somewhat.
+
+try:
+    # We deserialize the json form rather than the yaml form because
+    # json libraries are available for Python 2.4.
+    import json
+except:
+    # The json module is new in Python 2.6, whereas simplejson is
+    # compatible with earlier versions.
+    import simplejson as json
+
+import glob
+import os.path
+import unittest
+
+from pystache.renderer import Renderer
+
+
+root_path = os.path.join(os.path.dirname(__file__), '..', 'ext', 'spec', 'specs')
+spec_paths = glob.glob(os.path.join(root_path, '*.json'))
+
+class MustacheSpec(unittest.TestCase):
+    pass
+
+def buildTest(testData, spec_filename):
+
+    name = testData['name']
+    description  = testData['desc']
+
+    test_name = "%s (%s)" % (name, spec_filename)
+
+    def test(self):
+        template = testData['template']
+        partials = testData.has_key('partials') and testData['partials'] or {}
+        expected = testData['expected']
+        data     = testData['data']
+
+        # Convert code strings to functions.
+        # TODO: make this section of code easier to understand.
+        new_data = {}
+        for key, val in data.iteritems():
+            if isinstance(val, dict) and val.get('__tag__') == 'code':
+                val = eval(val['python'])
+            new_data[key] = val
+
+        renderer = Renderer(partials=partials)
+        actual = renderer.render(template, new_data)
+        actual = actual.encode('utf-8')
+
+        message = """%s
+
+  Template: \"""%s\"""
+
+  Expected: %s
+  Actual:   %s
+
+  Expected: \"""%s\"""
+  Actual:   \"""%s\"""
+  """ % (description, template, repr(expected), repr(actual), expected, actual)
+
+        self.assertEquals(actual, expected, message)
+
+    # The name must begin with "test" for nosetests test discovery to work.
+    name =  'test: "%s"' % test_name
+
+    # If we don't convert unicode to str, we get the following error:
+    #   "TypeError: __name__ must be set to a string object"
+    test.__name__ = str(name)
+
+    return test
+
+for spec_path in spec_paths:
+
+    file_name  = os.path.basename(spec_path)
+
+    # We avoid use of the with keyword for Python 2.4 support.
+    f = open(spec_path, 'r')
+    try:
+        spec_data = json.load(f)
+    finally:
+        f.close()
+
+    tests = spec_data['tests']
+
+    for test in tests:
+        test = buildTest(test, file_name)
+        setattr(MustacheSpec, test.__name__, test)
+        # Prevent this variable from being interpreted as another test.
+        del(test)
+
+if __name__ == '__main__':
+    unittest.main()