diff --git a/fixtures/sudo_testutils.py b/fixtures/sudo_testutils.py index 577994409d24..7b3a7b781c50 100644 --- a/fixtures/sudo_testutils.py +++ b/fixtures/sudo_testutils.py @@ -1,22 +1,27 @@ -from django.contrib.auth.models import AnonymousUser, User +from typing import Any + +from django.contrib.auth.models import AnonymousUser +from django.http import HttpRequest from django.test import RequestFactory, TestCase +from sentry.users.models.user import User + class BaseTestCase(TestCase): - def setUp(self): + def setUp(self) -> None: self.request = self.get("/foo") - self.request.session = {} + self.request.session = self.client.session self.setUser(AnonymousUser()) - def get(self, *args, **kwargs): + def get(self, *args: Any, **kwargs: Any) -> HttpRequest: return RequestFactory().get(*args, **kwargs) - def post(self, *args, **kwargs): + def post(self, *args: Any, **kwargs: Any) -> HttpRequest: return RequestFactory().post(*args, **kwargs) - def setUser(self, user): + def setUser(self, user: AnonymousUser | User) -> None: self.user = self.request.user = user - def login(self, user_class=User): + def login(self, user_class: type[User] = User) -> None: user = user_class() self.setUser(user) diff --git a/pyproject.toml b/pyproject.toml index 3481d666bacb..273dc8da4404 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -444,7 +444,6 @@ module = [ "fixtures.sdk_crash_detection.crash_event_dotnet", "fixtures.sdk_crash_detection.crash_event_native", "fixtures.sdk_crash_detection.crash_event_react_native", - "fixtures.sudo_testutils", "fixtures.vsts", "sentry", "sentry.__main__",