sentinel/tests/sentry/auth/test_email.py

from unittest import mock

import pytest

from sentry.auth.email import AmbiguousUserFromEmail, resolve_email_to_user
from sentry.testutils.cases import TestCase
from sentry.testutils.silo import control_silo_test


@control_silo_test
class EmailResolverTest(TestCase):
    def setUp(self) -> None:
        self.user1 = self.create_user()
        self.user2 = self.create_user()

    def test_no_match(self):
        result = resolve_email_to_user("[email protected]")
        assert result is None

    def test_single_match(self):
        result = resolve_email_to_user(self.user1.email)
        assert result == self.user1

    @mock.patch("sentry.auth.email.metrics")
    def test_ambiguous_match(self, mock_metrics):
        for user in (self.user1, self.user2):
            self.create_useremail(user=user, email="[email protected]")

        with pytest.raises(AmbiguousUserFromEmail) as excinfo:
            resolve_email_to_user("[email protected]")
        assert set(excinfo.value.users) == {self.user1, self.user2}
        assert mock_metrics.incr.call_args.args == ("auth.email_resolution.no_resolution",)

    @mock.patch("sentry.auth.email.metrics")
    def test_prefers_verified_email(self, mock_metrics):
        org = self.create_organization()
        self.create_useremail(user=self.user1, email="[email protected]", is_verified=True)
        self.create_useremail(user=self.user2, email="[email protected]", is_verified=False)
        self.create_member(organization=org, user_id=self.user2.id)

        result = resolve_email_to_user("[email protected]", organization=org)
        assert result == self.user1

        assert mock_metrics.incr.call_args.args == ("auth.email_resolution.by_verification",)

    @mock.patch("sentry.auth.email.metrics")
    def test_prefers_org_member(self, mock_metrics):
        org = self.create_organization()
        self.create_useremail(user=self.user1, email="[email protected]", is_verified=True)
        self.create_useremail(user=self.user2, email="[email protected]", is_verified=True)
        self.create_member(organization=org, user_id=self.user2.id)

        result = resolve_email_to_user("[email protected]", organization=org)
        assert result == self.user2

        assert mock_metrics.incr.call_args.args == ("auth.email_resolution.by_org_membership",)

    @mock.patch("sentry.auth.email.metrics")
    def test_prefers_primary_email(self, mock_metrics):
        self.create_useremail(user=self.user1, email=self.user2.email, is_verified=True)

        result = resolve_email_to_user(self.user2.email)
        assert result == self.user2

        assert mock_metrics.incr.call_args.args == ("auth.email_resolution.by_primary_email",)