def testWalkTemplateTree(self):
gen = generator.TemplateGenerator()
gen.SetTemplateDir(os.path.join(self._TEST_DATA_DIR, 'library'))
gen.WalkTemplateTree('templates', self._path_replacements, {}, {},
self._package)
self._package.DoneWritingArchive()
# Now read it back and verify
self.VerifyPackageContains(['foo', 'bar', 'app.yaml', 'pp/xxx'])
def testWalkTemplateTreeWithFilteredTemplate(self):
gen = generator.TemplateGenerator()
gen.SetTemplateDir(os.path.join(self._TEST_DATA_DIR, 'library'))
gen.WalkTemplateTree(
'templates',
self._path_replacements, {}, {},
self._package,
file_filter=lambda template, output: template != 'bar.tmpl')
self._package.DoneWritingArchive()
self.VerifyPackageContains(['foo'], must_not_contain=['bar'])
