sentinel/tests/sentry/tasks/test_invite_missing_org_members.py

from unittest.mock import patch

from sentry.constants import ObjectStatus
from sentry.models.options.organization_option import OrganizationOption
from sentry.tasks.invite_missing_org_members import schedule_organizations, send_nudge_email
from sentry.testutils.cases import TestCase
from sentry.testutils.helpers import with_feature
from sentry.testutils.silo import region_silo_test


@region_silo_test
@patch(
    "sentry.notifications.notifications.missing_members_nudge.MissingMembersNudgeNotification.send",
)
@patch(
    "sentry.tasks.invite_missing_org_members.send_nudge_email",
)
class InviteMissingMembersTestCase(TestCase):
    def setUp(self):
        super().setUp()
        self.user = self.create_user(email="[email protected]")
        self.organization = self.create_organization(owner=self.user)
        self.create_member(
            email="[email protected]",
            organization=self.organization,
        )
        member = self.create_member(user=self.create_user(), organization=self.organization)
        member.user_email = "[email protected]"
        member.save()

        self.member_commit_author = self.create_commit_author(
            project=self.project, email="[email protected]"
        )
        self.nonmember_commit_author1 = self.create_commit_author(
            project=self.project, email="[email protected]"
        )
        self.nonmember_commit_author1.external_id = "github:c"
        self.nonmember_commit_author1.save()

        self.nonmember_commit_author2 = self.create_commit_author(
            project=self.project, email="[email protected]"
        )
        self.nonmember_commit_author2.external_id = "github:d"
        self.nonmember_commit_author2.save()

        nonmember_commit_author_invalid_char = self.create_commit_author(
            project=self.project, email="[email protected]"
        )
        nonmember_commit_author_invalid_char.external_id = "github:hi+1"
        nonmember_commit_author_invalid_char.save()

        nonmember_commit_author_invalid_domain = self.create_commit_author(
            project=self.project, email="[email protected]"
        )
        nonmember_commit_author_invalid_domain.external_id = "github:gmail"
        nonmember_commit_author_invalid_domain.save()

        self.repo = self.create_repo(project=self.project, provider="integrations:github")
        self.create_commit(repo=self.repo, author=self.member_commit_author)
        self.create_commit(repo=self.repo, author=self.nonmember_commit_author1)
        self.create_commit(repo=self.repo, author=self.nonmember_commit_author1)
        self.create_commit(repo=self.repo, author=self.nonmember_commit_author2)
        self.create_commit(repo=self.repo, author=nonmember_commit_author_invalid_char)
        self.create_commit(repo=self.repo, author=nonmember_commit_author_invalid_domain)

        not_shared_domain_author = self.create_commit_author(
            project=self.project, email="[email protected]"
        )
        not_shared_domain_author.external_id = "github:not"
        not_shared_domain_author.save()
        self.create_commit(repo=self.repo, author=not_shared_domain_author)

        self.login_as(self.user)

    @with_feature("organizations:integrations-gh-invite")
    def test_schedules_and_sends(self, mock_send_email, mock_send_notification):
        integration = self.create_integration(
            organization=self.organization, provider="github", name="Github", external_id="github:1"
        )
        self.repo.integration_id = integration.id
        self.repo.save()

        with self.tasks():
            schedule_organizations()

        mock_send_email.delay.assert_called_with(self.organization.id)

        send_nudge_email(org_id=self.organization.id)

        assert mock_send_notification.called

    @with_feature("organizations:integrations-gh-invite")
    @patch(
        "sentry.notifications.notifications.missing_members_nudge.MissingMembersNudgeNotification.__init__",
        return_value=None,
    )
    def test_excludes_filtered_emails(
        self, mock_init_notification, mock_send_email, mock_send_notification
    ):
        integration = self.create_integration(
            organization=self.organization, provider="github", name="Github", external_id="github:1"
        )
        self.repo.integration_id = integration.id
        self.repo.save()

        send_nudge_email(org_id=self.organization.id)

        commit_authors = [
            {"email": "[email protected]", "external_id": "c", "commit_count": 2},
            {"email": "[email protected]", "external_id": "d", "commit_count": 1},
        ]

        mock_init_notification.assert_called_once_with(
            organization=self.organization, commit_authors=commit_authors, provider="github"
        )

    def test_no_github_repos(self, mock_send_email, mock_send_notification):
        self.repo.delete()

        with self.tasks():
            schedule_organizations()

        assert not mock_send_email.delay.called

    def test_no_active_github_repos(self, mock_send_email, mock_send_notification):
        self.repo.status = ObjectStatus.DISABLED
        self.repo.save()

        with self.tasks():
            schedule_organizations()

        assert not mock_send_email.delay.called

    def test_missing_org(self, mock_send_email, mock_send_notification):
        send_nudge_email(org_id=0)

        assert not mock_send_notification.called

    def test_missing_feature_flag(self, mock_send_email, mock_send_notification):
        send_nudge_email(org_id=self.organization.id)

        assert not mock_send_notification.called

    @with_feature("organizations:integrations-gh-invite")
    def test_missing_integration(self, mock_send_email, mock_send_notification):
        send_nudge_email(org_id=self.organization.id)

        assert not mock_send_notification.called

    @with_feature("organizations:integrations-gh-invite")
    def test_missing_option(self, mock_send_email, mock_send_notification):
        OrganizationOption.objects.set_value(
            organization=self.organization, key="sentry:github_nudge_invite", value=False
        )

        integration = self.create_integration(
            organization=self.organization, provider="github", name="Github", external_id="github:1"
        )
        self.repo.integration_id = integration.id
        self.repo.save()

        send_nudge_email(org_id=self.organization.id)

        assert not mock_send_notification.called

    @with_feature("organizations:integrations-gh-invite")
    def test_missing_nonmember_commit_authors(self, mock_send_email, mock_send_notification):
        org = self.create_organization()
        project = self.create_project(organization=org)
        self.create_repo(project=project, provider="github")

        self.create_integration(
            organization=org, provider="github", name="Github", external_id="github:1"
        )

        send_nudge_email(org_id=org.id)

        assert not mock_send_notification.called