Skip to content
Snippets Groups Projects
conftest.py 554 B
Newer Older
James Pic's avatar
James Pic committed
import pytest

from django.contrib.auth.models import AnonymousUser
from django.contrib.sessions.backends.base import SessionBase
from django.test.client import RequestFactory as drf


class RequestFactory(drf):
    def __init__(self, user):
        self.user = user
        super().__init__()

    def generic(self, *args, **kwargs):
        request = super().generic(*args, **kwargs)
        request.session = SessionBase()
        request.user = self.user
        return request


@pytest.fixture
def srf():
    return RequestFactory(AnonymousUser())