Update Erome tests
This commit is contained in:
parent
03d0aec4f6
commit
f716d982b0
1 changed files with 10 additions and 10 deletions
|
@ -1,6 +1,6 @@
|
||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
# coding=utf-8
|
# coding=utf-8
|
||||||
|
import re
|
||||||
from unittest.mock import MagicMock
|
from unittest.mock import MagicMock
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
|
@ -11,21 +11,21 @@ from bdfr.site_downloaders.erome import Erome
|
||||||
@pytest.mark.online
|
@pytest.mark.online
|
||||||
@pytest.mark.parametrize(('test_url', 'expected_urls'), (
|
@pytest.mark.parametrize(('test_url', 'expected_urls'), (
|
||||||
('https://www.erome.com/a/vqtPuLXh', (
|
('https://www.erome.com/a/vqtPuLXh', (
|
||||||
'https://s11.erome.com/365/vqtPuLXh/KH2qBT99_480p.mp4',
|
r'https://s\d+.erome.com/365/vqtPuLXh/KH2qBT99_480p.mp4',
|
||||||
)),
|
)),
|
||||||
('https://www.erome.com/a/ORhX0FZz', (
|
('https://www.erome.com/a/ORhX0FZz', (
|
||||||
'https://s15.erome.com/355/ORhX0FZz/9IYQocM9_480p.mp4',
|
r'https://s\d+.erome.com/355/ORhX0FZz/9IYQocM9_480p.mp4',
|
||||||
'https://s15.erome.com/355/ORhX0FZz/9eEDc8xm_480p.mp4',
|
r'https://s\d+.erome.com/355/ORhX0FZz/9eEDc8xm_480p.mp4',
|
||||||
'https://s15.erome.com/355/ORhX0FZz/EvApC7Rp_480p.mp4',
|
r'https://s\d+.erome.com/355/ORhX0FZz/EvApC7Rp_480p.mp4',
|
||||||
'https://s15.erome.com/355/ORhX0FZz/LruobtMs_480p.mp4',
|
r'https://s\d+.erome.com/355/ORhX0FZz/LruobtMs_480p.mp4',
|
||||||
'https://s15.erome.com/355/ORhX0FZz/TJNmSUU5_480p.mp4',
|
r'https://s\d+.erome.com/355/ORhX0FZz/TJNmSUU5_480p.mp4',
|
||||||
'https://s15.erome.com/355/ORhX0FZz/X11Skh6Z_480p.mp4',
|
r'https://s\d+.erome.com/355/ORhX0FZz/X11Skh6Z_480p.mp4',
|
||||||
'https://s15.erome.com/355/ORhX0FZz/bjlTkpn7_480p.mp4'
|
r'https://s\d+.erome.com/355/ORhX0FZz/bjlTkpn7_480p.mp4'
|
||||||
)),
|
)),
|
||||||
))
|
))
|
||||||
def test_get_link(test_url: str, expected_urls: tuple[str]):
|
def test_get_link(test_url: str, expected_urls: tuple[str]):
|
||||||
result = Erome. _get_links(test_url)
|
result = Erome. _get_links(test_url)
|
||||||
assert set(result) == set(expected_urls)
|
assert all([any([re.match(p, r) for r in result]) for p in expected_urls])
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.online
|
@pytest.mark.online
|
||||||
|
|
Loading…
Reference in a new issue