from __future__ import annotations from typing import TYPE_CHECKING, ParamSpec, TypeVar from django.contrib.auth.models import User from django.core.signing import TimestampSigner from django.test import Client, TestCase from hc.accounts.models import Member, Profile, Project if TYPE_CHECKING: # _MonkeyPatchedWSGIResponse is defined in django-stubs, # we import with a "TestHttpResponse" alias. # We list it in __all__ so subclasses can import and use it. from django.test.client import _MonkeyPatchedWSGIResponse as TestHttpResponse else: from django.http import HttpResponse as TestHttpResponse __all__ = ["BaseTestCase", "TestHttpResponse"] P = ParamSpec("P") T = TypeVar("T") class BaseTestCase(TestCase): def setUp(self) -> None: super().setUp() self.csrf_client = Client(enforce_csrf_checks=True) # Alice is a normal user for tests. Alice has team access enabled. self.alice = User(username="alice", email="alice@example.org") self.alice.set_password("password") self.alice.save() self.project = Project(owner=self.alice, api_key="X" * 32) self.project.name = "Alices Project" self.project.badge_key = self.alice.username self.project.ping_key = "p" * 22 self.project.save() self.profile = Profile(user=self.alice) self.profile.sms_limit = 50 self.profile.save() # Bob is on Alice's team and should have access to her stuff self.bob = User(username="bob", email="bob@example.org") self.bob.set_password("password") self.bob.save() self.bobs_project = Project(owner=self.bob) self.bobs_project.badge_key = self.bob.username self.bobs_project.save() self.bobs_profile = Profile(user=self.bob) self.bobs_profile.save() self.bobs_membership = Member.objects.create( user=self.bob, project=self.project, role=Member.Role.REGULAR ) # Charlie should have no access to Alice's stuff self.charlie = User(username="charlie", email="charlie@example.org") self.charlie.set_password("password") self.charlie.save() self.charlies_project = Project(owner=self.charlie) self.charlies_project.badge_key = self.charlie.username self.charlies_project.save() self.charlies_profile = Profile(user=self.charlie) self.charlies_profile.save() self.channels_url = "/projects/%s/integrations/" % self.project.code def set_sudo_flag(self) -> None: session = self.client.session session["sudo"] = TimestampSigner().sign("active") session.save()