refactor: Move RequestFactoryNoCsrf helper to common/

This commit is contained in:
stvn
2021-04-13 20:16:54 -07:00
parent f646971c74
commit 1b3d915af7
5 changed files with 16 additions and 17 deletions

View File

@@ -7,6 +7,7 @@ from uuid import uuid4
import factory
from django.contrib.auth.models import AnonymousUser, Group, Permission
from django.contrib.contenttypes.models import ContentType
from django.test.client import RequestFactory
from factory.django import DjangoModelFactory
from opaque_keys.edx.keys import CourseKey
from pytz import UTC
@@ -115,6 +116,16 @@ class UserFactory(DjangoModelFactory): # lint-amnesty, pylint: disable=missing-
self.groups.add(GroupFactory.simple_generate(create, name=group_name)) # lint-amnesty, pylint: disable=no-member
class RequestFactoryNoCsrf(RequestFactory):
"""
RequestFactory, which disables csrf checks.
"""
def request(self, **kwargs):
request = super().request(**kwargs)
setattr(request, '_dont_enforce_csrf_checks', True) # pylint: disable=literal-used-as-attribute
return request
class AnonymousUserFactory(factory.Factory):
class Meta:
model = AnonymousUser