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'])