healthchecks/hc/accounts/tests/test_team_access_middleware.py

from django.contrib.auth.models import User
from django.test import TestCase
from hc.accounts.models import Profile


class TeamAccessMiddlewareTestCase(TestCase):

    def test_it_handles_missing_profile(self):
        user = User(username="ned", email="[email protected]")
        user.set_password("password")
        user.save()

        self.client.login(username="[email protected]", password="password")
        r = self.client.get("/docs/")
        self.assertEqual(r.status_code, 200)

        self.assertEqual(Profile.objects.count(), 1)