Django Testing - 在RequestFactory中访问会话

Question

我在Django测试中使用RequestFactory，我找不到正确的方法来访问会话变量，当我尝试self.factory._session["zip_id"]或self.factory.session["zip_id"]时，我收到以下错误。

======================================================================
ERROR: test_middleware (dj_geo.tests.IPToZipMiddleWareTest)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "c:\dj_site_test\dj_geo\tests.py", line 36, in test_middleware
    assert self.factory._session["zip_id"] != None
AttributeError: 'RequestFactory' object has no attribute '_session'

----------------------------------------------------------------------



@override_settings(MIDDLEWARE_CLASSES=(
    'dj_geo.middleware.IPToZipMiddleWare'
))
class IPToZipMiddleWareTest(TestCase):

    def test_middleware(self):
        Zipcode.syncdb()
        assert Zipcode.objects.all().count() > 0

        self.factory = RequestFactory()
        self.request = self.factory.get('/', {}, **{'REMOTE_ADDR':'108.31.178.99'})
        assert self.factory._session["zip_id"] != None
        assert self.factory._session["zip_id"] != ""

Answer 1

你需要使用Client代替RequestFactory

self.factory = Client()

Answer 2

您可能需要使用SessionMiddleware来处理您的请求，然后将其保存以存储会话。你可以参考这个article。我也不认为直接访问工厂的受保护属性是个好主意，比如self.factory._session["zip_id"]，它会让你遇到更多问题。祝好运！

Answer 3

保存会话信息以使用您的中间件进行请求

from django.contrib.sessions.middleware import SessionMiddleware

request = RequestFactory().get('/')
middleware = SessionMiddleware()
middleware.process_request(request)
request.session.save()