在Django单元测试中使用会话对象

Kin*_*arō 7 django session

我正在编写登录视图,并希望为视图添加单元测试.我的观点如下:

def login(request):

    if request.POST:
            usrname = request.POST.get('username')
            password = request.POST.get('password')
            user = authenticate(username=usrname, password=password)

            if user is not None:
                    auth_login(request, user)
                    return redirect('/core/home/')
            else:
                    context = {'error_message': "Invalid username or password"}
                    return render(request, 'core/login.html', context)
    else:
            c = {}
            c.update(csrf(request))
            return render_to_response('core/login.html',c)

def home(request):
    if request.user.is_authenticated():
        context = {'user' : request.user}
        return render(request, 'core/home.html', context)
    else:
        return render(request, 'core/login.html')
Run Code Online (Sandbox Code Playgroud)

我的单元测试看起来像这样:

class CoreViewTests(TestCase):
    def setUp(self):
            self.factory = RequestFactory()

    def test_login_view_with_valid_user(self):
            uf = UserFactory()
            user = uf.make_user("ValidUser1", "12345", "user@abc.com")
            self.assertEqual(user.username, "ValidUser1")
            request = self.factory.post('/core/login', {"username": "ValidUser1", "password": "12345"})
            response = login(request)
            self.assertEqual(response.status_code, 200)
Run Code Online (Sandbox Code Playgroud)

单元测试崩溃,因为他们找不到会话对象.我通过定义虚拟会话字典在网站上关注几个教程,但它没有帮助.

任何人都可以为我解释如何为需要处理会话对象的视图编写单元测试?

谢谢.

Dun*_*kes 8

从RequestFactory对象的文档:

它不支持中间件.如果视图需要正常运行,则必须由测试本身提供会话和身份验证属性.

request.session正如你所说,你可以尝试手动设置为包含适当内容的字典.但是,使用老式的Django测试客户端可能会更容易.