Unittest Jinja2和Webapp2:找不到模板

时间:2013-10-16 14:25:03

标签: google-app-engine jinja2 webapp2

我在GAE项目中使用带有Webapp2的Jinja2。

我有RequestHandler中描述的基础webapp2_extras.jinja2

import webapp2

from webapp2_extras import jinja2


def jinja2_factory(app):
    """Set configuration environment for Jinja."""
    config = {my config...}
    j = jinja2.Jinja2(app, config=config)
    return j


class BaseHandler(webapp2.RequestHandler):

    @webapp2.cached_property
    def jinja2(self):
        # Returns a Jinja2 renderer cached in the app registry.
        return jinja2.get_jinja2(factory=jinja2_factory, app=self.app)

    def render_response(self, _template, **context):
        # Renders a template and writes the result to the response.
        rv = self.jinja2.render_template(_template, **context)
        self.response.write(rv)

视图处理程序为:

class MyHandler(BaseHandler):
    def get(self):
        context = {'message': 'Hello, world!'}
        self.render_response('my_template.html', **context)

我的模板位于默认位置(templates)。

该应用程序在开发服务器上运行良好,模板正确呈现。

但是当我尝试使用

进行单元测试MyHandler
import unittest
import webapp2
import webstest

class MyHandlerTest(unittest.TestCase):

    def setUp(self):
        application = webapp2.WSGIApplication([('/', MyHandler)])
        self.testapp = webtest.TestApp(application)

    def test_response(self):
        response = application.get_response('/')
        ...

application.get_response('/my-view')提出异常:TemplateNotFound: my_template.html

我错过了什么吗?像jinja2环境或模板加载器配置?

1 个答案:

答案 0 :(得分:3)

问题来源: Jinja2默认加载程序搜索相对./templates/目录中的文件。在开发服务器上运行GAE应用程序时,此路径相对于应用程序的根目录。但是,当您运行单元测试时,此路径与您的unittest文件相关。

<强>解决方案: 这不是一个理想的解决方案,但这是我解决问题的一个技巧。

我更新了jinja2工厂以添加动态模板路径,在app config中设置:

def jinja2_factory(app):
    """Set configuration environment for Jinja."""
    config = {'template_path': app.config.get('templates_path', 'templates'),}
    j = jinja2.Jinja2(app, config=config)
    return j

我在我的单元测试的setUp中设置了模板的绝对路径:

class MyHandlerTest(unittest.TestCase):

    def setUp(self):
        # Set template path for loader
        start = os.path.dirname(__file__)
        rel_path = os.path.join(start, '../../templates') # Path to my template
        abs_path = os.path.realpath(rel_path)
        application.config.update({'templates_path': abs_path})