add extra tests for media source URI parsing (#148114)

tests/components/media_source/test_const.py

@@ -0,0 +1,80 @@
+"""Test constants for the media source component."""
+
+import pytest
+
+from homeassistant.components.media_source.const import URI_SCHEME_REGEX
+
+
+@pytest.mark.parametrize(
+    ("uri", "expected_domain", "expected_identifier"),
+    [
+        ("media-source://", None, None),
+        ("media-source://local_media", "local_media", None),
+        (
+            "media-source://local_media/some/path/file.mp3",
+            "local_media",
+            "some/path/file.mp3",
+        ),
+        ("media-source://a/b", "a", "b"),
+        (
+            "media-source://domain/file with spaces.mp4",
+            "domain",
+            "file with spaces.mp4",
+        ),
+        (
+            "media-source://domain/file-with-dashes.mp3",
+            "domain",
+            "file-with-dashes.mp3",
+        ),
+        ("media-source://domain/file.with.dots.mp3", "domain", "file.with.dots.mp3"),
+        (
+            "media-source://domain/special!@#$%^&*()chars",
+            "domain",
+            "special!@#$%^&*()chars",
+        ),
+    ],
+)
+def test_valid_uri_patterns(
+    uri: str, expected_domain: str | None, expected_identifier: str | None
+) -> None:
+    """Test various valid URI patterns."""
+    match = URI_SCHEME_REGEX.match(uri)
+    assert match is not None
+    assert match.group("domain") == expected_domain
+    assert match.group("identifier") == expected_identifier
+
+
+@pytest.mark.parametrize(
+    "uri",
+    [
+        "media-source:",  # missing //
+        "media-source:/",  # missing second /
+        "media-source:///",  # extra /
+        "media-source://domain/",  # trailing slash after domain
+        "invalid-scheme://domain",  # wrong scheme
+        "media-source//domain",  # missing :
+        "MEDIA-SOURCE://domain",  # uppercase scheme
+        "media_source://domain",  # underscore in scheme
+        "",  # empty string
+        "media-source",  # scheme only
+        "media-source://domain extra",  # extra content
+        "prefix media-source://domain",  # prefix content
+        "media-source://domain suffix",  # suffix content
+        # Invalid domain names
+        "media-source://_test",  # starts with underscore
+        "media-source://test_",  # ends with underscore
+        "media-source://_test_",  # starts and ends with underscore
+        "media-source://_",  # single underscore
+        "media-source://test-123",  # contains hyphen
+        "media-source://test.123",  # contains dot
+        "media-source://test 123",  # contains space
+        "media-source://TEST",  # uppercase letters
+        "media-source://Test",  # mixed case
+        # Identifier cannot start with slash
+        "media-source://domain//invalid",  # identifier starts with slash
+    ],
+)
+def test_invalid_uris(uri: str) -> None:
+    """Test invalid URI formats."""
+    match = URI_SCHEME_REGEX.match(uri)
+    assert match is None, f"URI '{uri}' should be invalid"