如何在 Django 测试框架中修改会话

发布于10月11日

我的站点允许个人在没有登录的情况下通过基于当前SESSION_KEY创建用户来贡献内容

我想为我的视图设置一个测试，但似乎无法修改请求.会议:

我想这样做:

from django.contrib.sessions.models import Session
s = Session()
s.expire_date = '2010-12-05'
s.session_key = 'my_session_key'
s.save()
self.client.session = s
response = self.client.get('/myview/')

但我得到的错误是:

AttributeError: can't set attribute

如何在发出get请求之前修改客户端会话？

推荐答案

这就是我是如何做到的(灵感来自于http://blog.mediaonfire.com/?p=36中的一个解决方案).

from django.test import TestCase
from django.conf import settings
from django.utils.importlib import import_module

class SessionTestCase(TestCase):
    def setUp(self):
        # http://code.djangoproject.com/ticket/10899
        settings.SESSION_ENGINE = 'django.contrib.sessions.backends.file'
        engine = import_module(settings.SESSION_ENGINE)
        store = engine.SessionStore()
        store.save()
        self.session = store
        self.client.cookies[settings.SESSION_COOKIE_NAME] = store.session_key

之后，您可以将测试创建为:

class BlahTestCase(SessionTestCase):

    def test_blah_with_session(self):
        session = self.session
        session['operator'] = 'Jimmy'
        session.save()

等等.