访问用于测试 Django DetailView 的中间件值

Question

我正在为 DetailView 编写测试，该测试通过访问中间件中设置的值来查询 get_object()。这是针对公司应用程序和公司模型的。每个用户都在一个公司中。

为了在整个项目中访问公司，我通过自定义中间件在请求中设置了当前用户的 Company.uuid。

中间件

from django.utils.deprecation import MiddlewareMixin

class DynamicCompanyUUIDMiddleware(MiddlewareMixin):
    """ Adds the current organization's UUID from the current user."""

    def process_request(self, request):
        try:
            company_uuid = request.user.company_uuid
        except:
            company_uuid = None
        request.company_uuid = company_uuid

CompanyDetailView 的 get_object() 方法通过我用于其他公司视图的 Mixin 使用了它。

混音

class CompanyMixin(LoginRequiredMixin, SetHeadlineMixin):
    model = Company

    def get_object(self):
        return get_object_or_404(
            self.model,
            uuid=self.request.user.company_uuid)

测试

我要编写的测试是：

from django.test import RequestFactory
from django.urls import reverse, resolve
from test_plus.test import TestCase
from ..models import Company
from ..views import CompanyDetailView

class BaseCompanyTestCase(TestCase):

    def setUp(self):
        self.user = self.make_user()
        self.object = Company.objects.create(owner=self.user, name="testcompany")
        self.user.company_uuid = self.object.uuid
        self.factory = RequestFactory()

class TestCompanyDetailView(BaseCompanyTestCase):

    def setUp(self):
        super(TestCompanyDetailView, self).setUp()
        self.client.login(username="testuser", password="password")
        self.view = CompanyDetailView()
        self.view.object = self.object
        request = self.factory.get(reverse('companies:detail'))
        request.user = self.user
        request.company_uuid = self.user.company_uuid
        response = CompanyDetailView.as_view()(request)
        self.assertEqual(response.status_code, 200)

    def test_get_headline(self):
        self.assertEqual(
            self.view.get_headline(),
            '%s Members' % self.object.name

结果

这会导致 404 错误，找不到测试用户的公司。

走过它：

我创建用户
为这个新的测试用户创建公司
设置user.company_uuid
这应该允许 mixin 访问 company_uuid
因此return请求中的用户所在公司

但是我并没有像 404 显示的那样 return 公司。

问题

我哪里错了？预先感谢您的帮助。

回答

我混合了 Django 的客户端和 RequestFactory。我已经更正了上面正确的代码。

Answer 1

我混合了 Django 的客户端和 RequestFactory。离开后，我在下面想通了 -

from django.test import RequestFactory
from django.urls import reverse, resolve
from test_plus.test import TestCase
from ..models import Company
from ..views import CompanyDetailView

class BaseCompanyTestCase(TestCase):

    def setUp(self):
        self.user = self.make_user()
        self.object = Company.objects.create(owner=self.user, name="testcompany")
        self.user.company_uuid = self.object.uuid
        self.factory = RequestFactory()

class TestCompanyDetailView(BaseCompanyTestCase):

    def setUp(self):
        super(TestCompanyDetailView, self).setUp()
        self.client.login(username="testuser", password="password")
        self.view = CompanyDetailView()
        self.view.object = self.object
        request = self.factory.get(reverse('companies:detail'))
        request.user = self.user
        request.company_uuid = self.user.company_uuid
        response = CompanyDetailView.as_view()(request)
        self.assertEqual(response.status_code, 200)

    def test_get_headline(self):
        self.assertEqual(
            self.view.get_headline(),
            '%s Members' % self.object.name

访问用于测试 Django DetailView 的中间件值

Accessing Middleware value for testing a Django DetailView

django

django-views

django-testing

中间件

混音

测试

结果

走过它：

问题

回答