在一些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方法。
任何帮助都会很棒的!

最佳答案

来自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

关于python - 单元测试需要 session 的webapp2 RequestHandler,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/17436690/

10-10 02:49