sentinel/tests/sentry/testutils/helpers/test_features.py

from unittest import mock

from sentry import features
from sentry.services.hybrid_cloud.organization import RpcOrganization, organization_service
from sentry.services.hybrid_cloud.organization.model import RpcOrganizationSummary
from sentry.testutils.cases import TestCase
from sentry.testutils.helpers import with_feature
from sentry.testutils.silo import region_silo_test


@region_silo_test
class TestTestUtilsFeatureHelper(TestCase):
    def setUp(self):
        self.org = self.create_organization()

    def test_without_feature(self):
        assert not features.has("organizations:global-views", self.org)

    @with_feature("organizations:global-views")
    def test_with_feature(self):
        assert features.has("organizations:global-views", self.org)

    def test_batch_has(self):
        # Test that overrides work, and if no overrides are made that we still fall back to the
        # defaults
        with mock.patch("sentry.features.default_manager._entity_handler", new=None):
            with self.feature("organizations:customer-domains"):
                # Make sure this check returns True for features that are defaulted to True and aren't
                # mocked
                ret = features.batch_has(
                    [
                        "organizations:customer-domains",
                        "organizations:advanced-search",
                        "organizations:api-keys",
                    ],
                    organization=self.org,
                )
                assert ret is not None
                results = list(ret.values())[0]
                assert results["organizations:customer-domains"]
                assert results["organizations:advanced-search"]
                assert not results["organizations:api-keys"]

    def test_feature_with_rpc_organization(self):

        with self.feature({"organizations:customer-domains": False}):
            org_context = organization_service.get_organization_by_slug(
                slug=self.org.slug, only_visible=False, user_id=None
            )
            assert org_context
            assert org_context.organization
            assert isinstance(org_context.organization, RpcOrganization)

            assert features.has("organizations:customer-domains", org_context.organization) is False

        with self.feature({"organizations:customer-domains": True}):
            org_context = organization_service.get_organization_by_slug(
                slug=self.org.slug, only_visible=False, user_id=None
            )
            assert org_context
            assert org_context.organization
            assert isinstance(org_context.organization, RpcOrganization)

            assert features.has("organizations:customer-domains", org_context.organization)

            # Works for RpcOrganizationSummary
            organization = org_context.organization
            org_summary = RpcOrganizationSummary(
                id=organization.id, slug=organization.slug, name=organization.name
            )
            assert features.has("organizations:customer-domains", org_summary)

        other_org = self.create_organization()
        with self.feature({"organizations:customer-domains": [other_org.slug]}):
            # Feature not enabled for self.org
            org_context = organization_service.get_organization_by_slug(
                slug=self.org.slug, only_visible=False, user_id=None
            )
            assert org_context
            assert org_context.organization
            assert isinstance(org_context.organization, RpcOrganization)

            assert features.has("organizations:customer-domains", org_context.organization) is False

            # Works for RpcOrganizationSummary
            organization = org_context.organization
            org_summary = RpcOrganizationSummary(
                id=organization.id, slug=organization.slug, name=organization.name
            )
            assert features.has("organizations:customer-domains", org_summary) is False

            # Feature enabled for other_org
            org_context = organization_service.get_organization_by_slug(
                slug=other_org.slug, only_visible=False, user_id=None
            )
            assert org_context
            assert org_context.organization
            assert isinstance(org_context.organization, RpcOrganization)

            assert features.has("organizations:customer-domains", org_context.organization)

            # Works for RpcOrganizationSummary
            organization = org_context.organization
            org_summary = RpcOrganizationSummary(
                id=organization.id, slug=organization.slug, name=organization.name
            )
            assert features.has("organizations:customer-domains", org_summary)