mirror of
https://github.com/lrnselfreliance/wrolpi.git
synced 2026-10-26 19:01:39 -04:00
41 lines
1.9 KiB
Python
41 lines
1.9 KiB
Python
import pytest
|
|
|
|
from modules.videos.common import get_youtube_channel_id
|
|
from modules.videos.normalize_video_url import normalize_video_url
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
'url,expected',
|
|
[
|
|
('https://example.com', 'https://example.com'),
|
|
('https://www.youtube.com/watch?v=0xfMLNVFq2Y', 'https://www.youtube.com/watch?v=0xfMLNVFq2Y'),
|
|
('https://www.youtube.com/shorts/0xfMLNVFq2Y', 'https://www.youtube.com/watch?v=0xfMLNVFq2Y'),
|
|
('https://youtu.be/0xfMLNVFq2Y', 'https://www.youtube.com/watch?v=0xfMLNVFq2Y'),
|
|
('https://youtu.be/0xfMLNVFq2Y', 'https://www.youtube.com/watch?v=0xfMLNVFq2Y'),
|
|
(
|
|
'https://www.youtube.com/watch?v=0xfMLNVFq2Y&list=PLMRyOtjhfFJUMPtB122lfeXebndQaePSF&index=2',
|
|
'https://www.youtube.com/watch?v=0xfMLNVFq2Y'
|
|
),
|
|
('https://youtu.be/0xfMLNVFq2Y?si=YALbJCpBgQcoY_Dp', 'https://www.youtube.com/watch?v=0xfMLNVFq2Y')
|
|
]
|
|
)
|
|
def test_normalize_video_url(url: str, expected: str):
|
|
assert normalize_video_url(url) == expected
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
'url,expected',
|
|
[
|
|
('https://www.youtube.com/channel/UC4t8bw1besFTyjW7ZBCOIrw', 'UC4t8bw1besFTyjW7ZBCOIrw'),
|
|
('https://www.youtube.com/channel/UC4t8bw1besFTyjW7ZBCOIrw/videos', 'UC4t8bw1besFTyjW7ZBCOIrw'),
|
|
('https://www.youtube.com/channel/UC4t8bw1besFTyjW7ZBCOIrw/', 'UC4t8bw1besFTyjW7ZBCOIrw'),
|
|
('https://m.youtube.com/channel/UC4t8bw1besFTyjW7ZBCOIrw', 'UC4t8bw1besFTyjW7ZBCOIrw'),
|
|
('https://www.youtube.com/@wrolpi', None),
|
|
('https://www.youtube.com/@wrolpi/featured', None),
|
|
('https://www.youtube.com/@wrolpi/videos', None),
|
|
('https://www.youtube.com/@wrolpi/playlists', None),
|
|
('https://example.com/channel/something', None),
|
|
]
|
|
)
|
|
def test_get_youtube_channel_id(url: str, expected):
|
|
assert get_youtube_channel_id(url) == expected
|