"""Media URL classification for the in-app viewer."""

from PySide6.QtCore import QUrl

from imwald.ui.media_viewer_dialog import classify_media_url


def test_classify_by_extension() -> None:
    assert classify_media_url(QUrl("https://cdn/x/photo.webp")) == "image"
    assert classify_media_url(QUrl("https://x.com/a/b.MP4?q=1")) == "video"
    assert classify_media_url(QUrl("https://h/audio.opus")) == "audio"


def test_classify_unknown() -> None:
    assert classify_media_url(QUrl("https://x.com/page")) is None
    assert classify_media_url(QUrl("imwald://pub/abcd")) is None