def test_findCachedLocation(self):
t = Template(self.zp)
exists = os.path.exists
os.path.exists = MagicMock(return_value=True)
t.dest_file = 'foo.py'
t.tfile = 'foo.py.tmpl'
self.source_template = 'foo.tmpl'
t.findTemplate()
self.assertEqual(t.tfile, t.TemplateCacheLocation())
os.path.exists = exists
def test_findingTemplate(self):
t = Template(self.zp)
t.dest_file = 'foo.py'
t.tfile = 'foo.py.tmpl'
self.source_template = 'foo.tmpl'
t.findTemplate()
self.assertEqual(t.tfile, "%s/Templates/%s" %
("/".join(inspect.getfile(zpg).split('/')[:-1]), t.source_template))
def testCaching(self):
m = mock_open()
os.makedirs = MagicMock(return_value=True)
with patch('__builtin__.open', mock_open(read_data='Source Template'), create=True) as m:
t = Template(self.zp)
t.dest_file = 'foo.py'
t.tfile = 'foo.py.tmpl'
t.cacheTemplate()
# Write File Handle
wfh = m.return_value.__enter__.return_value
self.assertEqual(wfh.write.call_args_list, [
call('## Source Template foo.py.tmpl \n'), call('Source Template')])
def test_templateCacheLocation(self):
t = Template(self.zp)
t.dest_file = 'foo.py'
top = defaults.get('prefix', os.getcwd())
self.assertEqual(t.TemplateCacheLocation(),
'%s/%s/Templates/foo.py.tmpl' % (top, DEFAULT_NAME2))
def test_processTemplate(self):
t = Template(self.zp)
self.write(t, '${zenpack.id}\n${zenpack.version}\n')
name = DEFAULT_NAME
version = defaults.get('version')
self.assertEqual(self.results[-1], call(u'%s\n%s\n' % (name, version)))
