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

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

2 个答案: