sentinel/tests/sentry/api/endpoints/test_organization_auth_providers.py

from django.urls import reverse

from sentry import auth
from sentry.auth.providers.fly.provider import FlyOAuth2Provider
from sentry.testutils.cases import APITestCase, PermissionTestCase
from sentry.testutils.silo import region_silo_test


@region_silo_test
class OrganizationAuthProvidersPermissionTest(PermissionTestCase):
    def setUp(self):
        super().setUp()
        self.path = reverse(
            "sentry-api-0-organization-auth-providers", args=[self.organization.slug]
        )

    def test_owner_can_load(self):
        with self.feature("organizations:sso-basic"):
            self.assert_owner_can_access(self.path)

    def test_member_can_get(self):
        with self.feature("organizations:sso-basic"):
            self.assert_member_can_access(self.path)


@region_silo_test
class OrganizationAuthProviders(APITestCase):
    endpoint = "sentry-api-0-organization-auth-providers"

    def setUp(self):
        super().setUp()
        self.login_as(self.user)
        auth.register("Fly IO", FlyOAuth2Provider)
        self.addCleanup(auth.unregister, "Fly IO", FlyOAuth2Provider)

    def test_get_list_of_auth_providers(self):
        with self.feature("organizations:sso-basic"):
            response = self.get_success_response(self.organization.slug)
        assert any(d["key"] == "dummy" for d in response.data)
        assert any(d["key"] == "Fly IO" for d in response.data) is False