mirror of
https://github.com/kennethreitz-archive/pystache.git
synced 2026-06-05 23:40:16 +00:00
Carl Whittaker
53 lines
1.5 KiB
Python
53 lines
1.5 KiB
Python
import glob
|
|
import os.path
|
|
import pystache
|
|
from pystache import Loader
|
|
import unittest
|
|
import yaml
|
|
|
|
def code_constructor(loader, node):
|
|
value = loader.construct_mapping(node)
|
|
return eval(value['python'], {})
|
|
|
|
yaml.add_constructor(u'!code', code_constructor)
|
|
|
|
specs = os.path.join(os.path.dirname(__file__), '..', 'ext', 'spec', 'specs')
|
|
specs = glob.glob(os.path.join(specs, '*.yml'))
|
|
|
|
class MustacheSpec(unittest.TestCase):
|
|
pass
|
|
|
|
def buildTest(testData, spec):
|
|
def test(self):
|
|
template = testData['template']
|
|
partials = testData.has_key('partials') and testData['partials'] or {}
|
|
expected = testData['expected']
|
|
data = testData['data']
|
|
files = []
|
|
|
|
try:
|
|
for key in partials.keys():
|
|
filename = "%s.%s" % (key, Loader.template_extension)
|
|
files.append(os.path.join(Loader.template_path, filename))
|
|
p = open(files[-1], 'w')
|
|
p.write(partials[key])
|
|
p.close()
|
|
self.assertEquals(pystache.render(template, data), expected)
|
|
finally:
|
|
[os.remove(f) for f in files]
|
|
|
|
test.__doc__ = testData['desc']
|
|
test.__name__ = 'test %s (%s)' % (testData['name'], spec)
|
|
return test
|
|
|
|
for spec in specs:
|
|
name = os.path.basename(spec).replace('.yml', '')
|
|
|
|
for test in yaml.load(open(spec))['tests']:
|
|
test = buildTest(test, name)
|
|
print name
|
|
setattr(MustacheSpec, test.__name__, test)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|