如何单元测试从webapp2的请求处理程序传递给Jinja2的模板中的模板变量(how to uni

我以前从未做过单元测试。 我想掌握它。

我想测试我的webapp2的处理程序。 要做到这一点，我认为这将是发送到处理程序如的请求是一个好主意：

request = webapp2.Request.blank('/')
# Get a response for that request.
response = request.get_response(main.app)

问题是，响应大多只是一堆HTML等。

我想看看什么被传递给从处理我的Jinja2模板它变成HTML之前。

我希望我的测试，以获得在处理类代码中的状态。 我wan't能够看到某些变量看起来像在响应处理程序，然后我想看到的字典模板是什么样子，然后才传递给render_to_response（）这个

我想测试这些变量的值正确。

这里是我的测试代码，到目前为止，但我坚持，因为响应= request.get_response（）只是给了我一堆的HTML，而不是原始变量。

import unittest
#from google.appengine.ext import db
#from google.appengine.ext import testbed
#from google.appengine.datastore import datastore_stub_util
import main
import webapp2

class DemoTestCase(unittest.TestCase):
    def setUp(self):
        pass

    def tearDown(self):
        pass

    def testNothing(self):
        self.assertEqual(42, 21 + 21)

    def testHomeHandler(self):
        # Build a request object passing the URI path to be tested.
        # You can also pass headers, query arguments etc.
        request = webapp2.Request.blank('/')
        # Get a response for that request.
        response = request.get_response(main.app)

        # Let's check if the response is correct.
        self.assertEqual(response.status_int, 200)
        self.assertEqual(response.body, 'Hello, world!')


if __name__ == '__main__':
    unittest.main()

这里是我的处理程序：

class HomeHandler(BaseHandler):
    def get(self, file_name_filter=None, category_filter=None):
        file_names = os.listdir('blog_posts')
        blogs = []

        get_line = lambda file_: file_.readline().strip().replace("<!--","").replace("-->","")

        for fn in file_names:
            with open('blog_posts/%s' % fn) as file_:
                heading = get_line(file_)
                link_name = get_line(file_)
                category = get_line(file_)

            date_ = datetime.strptime(fn.split("_")[0], "%Y%m%d")

            blog_dict = {'date': date_, 'heading': heading,
                         'link_name': link_name,
                         'category': category,
                         'filename': fn.replace(".html", ""),
                         'raw_file_name': fn}

            blogs.append(blog_dict)

        categories = Counter(d['category'] for d in blogs)
        templates = {'categories': categories,
                     'blogs': blogs,
                     'file_name_filter': file_name_filter,
                     'category_filter': category_filter}

        assert(len(file_names) == len(set(d['link_name'] for d in blogs)))

        self.render_template('home.html', **templates)

这里是我basehandler：

class BaseHandler(webapp2.RequestHandler):
    @webapp2.cached_property
    def jinja2(self):
        return jinja2.get_jinja2(app=self.app)

    def render_template(self, filename, **kwargs):
        #kwargs.update({})
        #TODO() datastore caching here for caching of (handlername, handler parameters, changeable parameters, app_upload_date)
        #TODO() write rendered page to its own html file, and just serve that whole file. (includes all posts). JQuery can show/hide posts.
        self.response.write(self.jinja2.render_template(filename, **kwargs))

也许我有一个如何做单元测试错误的想法，也许我应该写的方式，使得它更容易测试我的代码？ 或者是有让我的代码的状态的某种方式？

此外，如果有人要重新编写代码并更改变量名，则测试将打破..

请告知我的情况：X

Answer 1:

你可以嘲笑BaseHandler.render_template方法和测试它的参数。

见这个问题对于流行的Python嘲弄框架的列表。

Answer 2:

由于proppy的建议，最后我用一个模拟。

http://www.voidspace.org.uk/python/mock/

（模拟被包括作为在Python 3部分或unittest.mock）

因此，这里是我的main.py代码是类似于我在webapp2的：

的注意，而不是BaseHandler.render_template我有BaseHandler.say_yo

__author__ = 'Robert'

print "hello from main"

class BaseHandler():
    def say_yo(self,some_number=99):
        print "yo"
        return "sup"

class TheHandler(BaseHandler):
    def get(self, my_number=42):
        print "in TheHandler's get()"
        print self.say_yo(my_number)
        return "TheHandler's return string"

和atest.py：

__author__ = 'Robert'

import unittest
import main
from mock import patch

class DemoTestCase(unittest.TestCase):
    def setUp(self):
        pass

    def tearDown(self):
        pass

    def testNothing(self):
        self.assertEqual(42, 21 + 21)

    def testSomeRequests(self):
        print "hi"
        bh = main.BaseHandler()
        print bh.say_yo()

        print "1111111"

        with patch('main.BaseHandler.say_yo') as patched_bh:
            print dir(patched_bh)
            patched_bh.return_value = 'double_sup'
            bh2 = main.BaseHandler()
            print bh2.say_yo()
            print "222222"

        bh3 = main.BaseHandler()
        print bh3.say_yo()

        print "3333"

        th = main.TheHandler()
        print th.get()

        print "44444"
        with patch('main.BaseHandler.say_yo') as patched_bh:
            patched_bh.return_value = 'last_sup'
            th = main.TheHandler()
            print th.get()
            print th.get(123)
            print "---"
            print patched_bh.called
            print patched_bh.call_args_list
            print "555555"



if __name__ == '__main__':
    unittest.main()

此代码提供大量输出，这里是一个例子：

44444
in TheHandler's get()
last_sup
TheHandler's return string
in TheHandler's get()
last_sup
TheHandler's return string
---
True
[call(42), call(123)]
555555

文章来源: how to unittest the template variables passed to jinja2 template from webapp2 request handler