sentinel/tests/sentry/utils/test_urls.py

from unittest import TestCase

import pytest

from sentry.utils.urls import (
    add_params_to_url,
    non_standard_url_join,
    parse_link,
    urlsplit_best_effort,
)


@pytest.mark.parametrize(
    "base,to_join,expected",
    [
        ("http://example.com/foo", "bar", "http://example.com/bar"),
        ("http://example.com/foo", "/bar", "http://example.com/bar"),
        ("https://example.com/foo", "/bar", "https://example.com/bar"),
        ("http://example.com/foo/baz", "bar", "http://example.com/foo/bar"),
        ("http://example.com/foo/baz", "/bar", "http://example.com/bar"),
        ("aps://example.com/foo", "/bar", "aps://example.com/bar"),
        ("apsunknown://example.com/foo", "/bar", "apsunknown://example.com/bar"),
        ("apsunknown://example.com/foo", "//aha/uhu", "apsunknown://aha/uhu"),
    ],
)
def test_non_standard_url_join(base, to_join, expected):
    assert non_standard_url_join(base, to_join) == expected


class AddParamsToUrlTest(TestCase):
    def test_basic(self):
        url = "https://sentry.io?myparam=value#hash-param"
        new_url = add_params_to_url(url, {"new_param": "another"})
        assert new_url == "https://sentry.io?myparam=value&new_param=another#hash-param"


class ParseLinkTest(TestCase):
    def test_parse_link(self):
        assert (
            parse_link(
                "https://meowlificent.ngrok.io/organizations/sentry/issues/167/?project=2&query=is%3Aunresolved"
            )
            == "organizations/{organization}/issues/{issue_id}/project=%7Bproject%7D&query=%5B%27is%3Aunresolved%27%5D"
        )
        assert (
            parse_link(
                "https://meowlificent.ngrok.io/organizations/sentry/issues/1/events/2d113519854c4f7a85bae8b69c7404ad/?project=2"
            )
            == "organizations/{organization}/issues/{issue_id}/events/{event_id}/project=%7Bproject%7D"
        )
        assert (
            parse_link(
                "https://meowlificent.ngrok.io/organizations/sentry/issues/9998089891/events/198e93sfa99d41b993ac8ae5dc384642/events/"
            )
            == "organizations/{organization}/issues/{issue_id}/events/{event_id}/events/"
        )


@pytest.mark.parametrize(
    ("s", "expected"),
    (
        pytest.param(
            "https://example.com:123/path?query",
            ("https", "example.com:123", "/path", "query"),
            id="normal, valid url",
        ),
        pytest.param(
            "https://[not-a-url/path?query",
            ("https", "[not-a-url", "/path", "query"),
            id="invalid url",
        ),
        pytest.param(
            "https://[ip]:3456/path",
            ("https", "[ip]:3456", "/path", ""),
            id="invalid url in python 3.11+",
        ),
    ),
)
def test_urlsplit_best_effort(s: str, expected: tuple[str, str, str, str]) -> None:
    assert urlsplit_best_effort(s) == expected