sentinel/tests/sentry/utils/email/test_address.py

import pytest
from django.core.signing import BadSignature

from sentry import options
from sentry.testutils.helpers.options import override_options
from sentry.testutils.pytest.fixtures import django_db_all
from sentry.utils.email.address import (
    email_to_group_id,
    get_from_email_domain,
    group_id_to_email,
    is_valid_email_address,
    parse_email,
    parse_user_name,
)


@django_db_all
def test_get_from_email_domain():
    with override_options({"mail.from": "[email protected]"}):
        assert get_from_email_domain() == "example.com"

    with override_options({"mail.from": "root@localhost"}):
        assert get_from_email_domain() == "localhost"

    with override_options({"mail.from": "garbage"}):
        assert get_from_email_domain() == "garbage"


def test_is_valid_email_address_number_at_qqcom():
    assert is_valid_email_address("[email protected]") is False


def test_is_valid_email_address_normal_human_email_address():
    assert is_valid_email_address("[email protected]") is True


def test_parse_email_empty():
    assert parse_email("") == ""


def test_parse_email_no_email():
    assert parse_email("[email protected]") == ""


def test_parse_email_empty_email():
    assert parse_email("<>") == ""


def test_parse_email_two_emails():
    assert parse_email("<a><b>") == "a"


def test_parse_email_simple():
    assert parse_email("lauryn <[email protected]>") == "[email protected]"


def test_parse_user_name_empty():
    assert parse_user_name("") == ""


def test_parse_user_name_no_email():
    assert parse_user_name("[email protected]") == "[email protected]"


def test_parse_user_name_empty_email():
    assert parse_user_name("<>") == ""


def test_parse_user_name_simple():
    assert parse_user_name("Max Bittker <[email protected]>") == "Max Bittker"


@django_db_all
def test_group_id_to_email_backwards_compat():
    mailhost = options.get("mail.reply-hostname")
    group_id = 1234567

    signed = group_id_to_email(group_id)
    assert f"@{mailhost}" in signed
    assert f"{group_id}+" in signed

    result_group_id, org_id = email_to_group_id(signed)
    assert result_group_id == group_id
    assert org_id is None


@django_db_all
def test_group_id_to_email_with_org_id():
    mailhost = options.get("mail.reply-hostname")
    group_id = 1234567
    org_id = 9876543

    signed = group_id_to_email(group_id, org_id)
    assert f"@{mailhost}" in signed
    assert f"{group_id}.{org_id}+" in signed

    result_group_id, result_org_id = email_to_group_id(signed)
    assert result_group_id == group_id
    assert result_org_id == org_id

    with pytest.raises(BadSignature):
        assert email_to_group_id(f"trash@{mailhost}") == (None, None)

    with pytest.raises(BadSignature):
        assert email_to_group_id(f"trash:rubbish@{mailhost}") == (None, None)