sentinel/tests/sentry/web/frontend/test_account_identity.py

import pytest
from django.urls import reverse

from sentry import identity
from sentry.identity.providers.dummy import DummyProvider
from sentry.models.identity import Identity, IdentityStatus
from sentry.testutils.cases import TestCase
from sentry.testutils.silo import control_silo_test


@control_silo_test
class AccountIdentityTest(TestCase):
    @pytest.fixture(autouse=True)
    def setup_dummy_identity_provider(self):
        identity.register(DummyProvider)
        self.addCleanup(identity.unregister, DummyProvider)

    def test_associate_identity(self):
        user = self.create_user()
        organization = self.create_organization(name="foo", owner=user)
        self.create_identity_provider(type="dummy", external_id="1234")

        self.login_as(user)

        path = reverse(
            "sentry-account-associate-identity", args=[organization.slug, "dummy", "1234"]
        )
        resp = self.client.get(path)

        assert resp.status_code == 200
        assert resp.context["organization"].id == organization.id
        assert isinstance(resp.context["provider"], DummyProvider)

        resp = self.client.post(path)

        assert resp.status_code == 200
        assert resp.content == DummyProvider.TEMPLATE.encode()

        resp = self.client.post(path, data={"email": "[email protected]"})
        ident = Identity.objects.get(user=user)

        assert resp.status_code == 302
        assert ident.external_id == "[email protected]"
        assert ident.status == IdentityStatus.VALID