def get_engine(template_type):
if template_type == DEFAULT_TEMPLATE_TYPE:
return Engine
else:
try:
external_engine = load_external_engine(template_type)
except ImportError:
raise NotImplementedError("No such template support")
return external_engine.get_engine(template_type)
def test_load_external_engine():
load_external_engine("unknown_template_type")
def test_load_mako_engine(): # fake mako
module = load_external_engine("mako")
engine_class = module.get_engine("mako")
assert engine_class == Engine
