def test_render_from_string(self): """Test that the templating module render_from_string works ok""" template = ">>$one<< >>$two<< >>${three}<<" # render with all variables of different types d = {'one': 1, 'two': 'two', 'three': [3]} self.assertEqual([b">>1<< >>two<< >>[3]<<"], templating.render_from_string(template, d)) # with missing values d = {'one': 1} self.assertEqual([b">>1<< >>$two<< >>${three}<<"], templating.render_from_string(template, d))
def test_render(self): """Test that the templating render works rendering from a file""" template = ">>$one<< >>$two<< >>${three}<<" (handle, filepath) = tempfile.mkstemp(dir=templating.TEMPLATE_DIR) self.addCleanup(lambda: os.unlink(filepath)) os.write(handle, template.encode()) os.close(handle) templatename = os.path.basename(filepath) # render with all variables d = {'one': 1, 'two': 'two', 'three': [3]} self.assertEqual([b">>1<< >>two<< >>[3]<<"], templating.render(templatename, d)) # with missing values d = {'one': 1} self.assertEqual([b">>1<< >>$two<< >>${three}<<"], templating.render_from_string(template, d))