Unittest Jinja2 和 Webapp2：找不到模板

Question

我在一个 GAE 项目中使用 Jinja2 和 Webapp2。

我有一个基础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 环境或模板加载器配置？

Answer 1

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

解决方案： 不是一个理想的解决方案，但这是我解决问题的一个技巧。

我更新了 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})