在我的一些RequestHandler视图中,我从会话中获取信息以确保用户当前已登录(自定义身份验证)
我无法在单元测试中嘲笑这个。
我的RequestHandler类中的helper方法:
@webapp2.cached_property
def get_account(self):
user_session = self.session.get('user-id')
if user_session:
user_account = helpers.get_username_data_from_session(self.session.get('user-id'))
return user_account
在已实现的(我想测试的)视图中,get方法可能如下所示(示例):
def get(self):
account = self.get_account
if not account:
self.error(302)
我无法找到一种模拟方法来测试RequestHandler类的get / post方法。
任何帮助都会很棒!
答案 0 :(得分:4)
来自http://www.recursion.org/2011/10/12/testing-webapp2-sessions-with-webtest
的摘录import webtest
import webapp2 as webapp
from webapp2_extras import sessions
from webapp2_extras.securecookie import SecureCookieSerializer
SUPER_SECRET = 'my-super-secret'
app = webapp.WSGIApplication(config={
'webapp2_extras.sessions': {
'secret_key': SUPER_SECRET
},
})
class SomeTestCase(unittest.TestCase):
def setUp(self):
self.app = webtest.TestApp(app)
self.testbed = testbed.Testbed()
self.testbed.activate()
def tearDown(self):
self.testbed.deactivate()
def test_get_index(self):
session = {'key': 'value'}
secure_cookie_serializer = SecureCookieSerializer(
SUPER_SECRET
)
serialized = secure_cookie_serializer.serialize(
'session', session
)
headers = {'Cookie': 'session=%s' % serialized}
response = self.app.get('/', headers=headers)
assert response.status_int == 200