Google App Engine + Python WebApp2 的本地单元测试

Question

我有一个非常简单的网络应用程序。所有重要的事情都发生在 index.py 中：

from google.appengine.api import users
import webapp2
import os
import jinja2

JINJA_ENVIRONMENT = jinja2.Environment(
    loader=jinja2.FileSystemLoader(os.path.dirname(__file__)),
    extensions=['jinja2.ext.autoescape'],
    autoescape=True)

def get_user():
  user = {}
  user['email'] = str(users.get_current_user())
  user['name'], user['domain'] = user['email'].split('@')
  user['logout_link'] = users.create_logout_url('/')
  return user

class BaseHandler(webapp2.RequestHandler):

  def dispatch(self):
    user = get_user()
    template_values = {'user': user}
    if user['domain'] != 'foo.com':
      template_values['page_title'] = 'Access Denied'
      template = '403'
    else:
      template_values['page_title'] = 'Home'
      template = 'index'
    template_engine = JINJA_ENVIRONMENT.get_template('%s.html' % template)
    self.response.write(template_engine.render(template_values))

app = webapp2.WSGIApplication([
    ('/', BaseHandler),
], debug=True)

我正在努力成为一个好人并编写一些本地单元测试，但是 - 在查看 documentation 之后 - 我完全超出了我的深度。我想要的只是一个基本框架，我可以在其中执行以下操作：

python test_security.py 

并模拟两个用户访问域 - 一个应该获得索引模板的 @foo.com 和一个应该获得 403 模板的 @bar.com。

到目前为止，这是我所做的：

import sys

# I don't want to talk about it, let's just ignore this block
sys.path.append('C:\Program Files (x86)\Google\google_appengine\lib\webapp2-2.5.2')
sys.path.append('C:\Program Files (x86)\Google\google_appengine\lib\webob-1.2.3')
sys.path.append('C:\Program Files (x86)\Google\google_appengine\lib\jinja2-2.6')
sys.path.append('C:\Program Files (x86)\Google\google_appengine\lib\yaml-3.10')
sys.path.append('C:\Program Files (x86)\Google\google_appengine\lib\jinja2-2.6')
sys.path.append('C:\Program Files (x86)\Google\google_appengine')
sys.path.append('C:\pytest')


# A few proper imports
import unittest
import webapp2
from google.appengine.ext import testbed

# Import the module I'd like to test
import index

class TestHandlers(unittest.TestCase):
   def test_hello(self):

       self.testbed = testbed.Testbed()
       self.testbed.init_user_stub()
       self.testbed.setup_env(USER_EMAIL='test@foo.com',USER_ID='1', USER_IS_ADMIN='0')

       request = webapp2.Request.blank('/')
       response = request.get_response(main.app)

       print "running test"

       self.assertEqual(response.status_int, 200)
       self.assertEqual(response.body, 'Hello, world!')

可以预见的是，这根本行不通。我错过了什么？我是不是太高估了这应该有多简单？

Answer 1

如果您打算使用“python test_security.py”来调用它，您正在寻找的神奇词汇是：

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

这将使您的单元测试运行 - 目前您所做的只是定义它。

另请注意，您需要将 request.get_response 从“main.app”更改为“index.app”。

Answer 2

我怀疑（主要基于函数名称）您应该调用self.testbed.init_user_stub()之前调用self.testbed.setup_env()，而不是之后。

此外，您似乎缺少一个初始的 self.testbed = testbed.Testbed() 和一个 testbed.activate() 呼叫。

你可能想看看这个答案：https://stackoverflow.com/a/21139805/4495081