cookiecutter/cookiecutter-django
1
1
Fork 0
mirror of https://github.com/cookiecutter/cookiecutter-django.git synced 2025-02-24 07:20:40 +03:00
Code Issues Packages Projects Releases Wiki Activity

Change RequestFactory to APIRequestFactory in tests for API views (#4110)

Browse Source
This commit is contained in:
Bruno Alla 2023-02-06 18:33:08 +00:00 committed by GitHub
parent 8810677961
commit 5bc4bed952
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/test_drf_views.py
View File

@ -1,22 +1,27 @@
from django.test import RequestFactory
import pytest
from rest_framework.test import APIRequestFactory
from {{ cookiecutter.project_slug }}.users.api.views import UserViewSet
from {{ cookiecutter.project_slug }}.users.models import User
class TestUserViewSet:
def test_get_queryset(self, user: User, rf: RequestFactory):
@pytest.fixture
def api_rf(self) -> APIRequestFactory:
return APIRequestFactory()
def test_get_queryset(self, user: User, api_rf: APIRequestFactory):
view = UserViewSet()
request = rf.get("/fake-url/")
request = api_rf.get("/fake-url/")
request.user = user
view.request = request
assert user in view.get_queryset()
def test_me(self, user: User, rf: RequestFactory):
def test_me(self, user: User, api_rf: APIRequestFactory):
view = UserViewSet()
request = rf.get("/fake-url/")
request = api_rf.get("/fake-url/")
request.user = user
view.request = request