def test_update_template(self):
with self.app.test_request_context('/'):
result = templates.update_template(self.template_id,
copy.copy(self.template2))
self.assertEqual(result, None)
result = templates.get_template(self.template_id)
self.assertEqual(result, self.template2)
def _render(name, uuid):
try:
template = get_template(name)
params = get_host_params(uuid)
return render_template_string(template, **params)
except KeyError:
return abort(404)
def script(uuid):
try:
params = get_host_params(uuid)
template = get_template(params["ipxe_script"])
return render_template_string(template, **params)
except KeyError:
return abort(404)
def test_get_template(self):
with self.app.test_request_context('/'):
result = templates.get_template(self.template_id)
self.assertEqual(result, self.template)
def _get(name):
return make_response(get_template(name))
