From f716d982b0b75ebbacf7c06a7a006ccc313cea30 Mon Sep 17 00:00:00 2001 From: Serene-Arc Date: Wed, 20 Oct 2021 10:51:28 +1000 Subject: [PATCH] Update Erome tests --- tests/site_downloaders/test_erome.py | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/tests/site_downloaders/test_erome.py b/tests/site_downloaders/test_erome.py index d27cbb7..e06fab5 100644 --- a/tests/site_downloaders/test_erome.py +++ b/tests/site_downloaders/test_erome.py @@ -1,6 +1,6 @@ #!/usr/bin/env python3 # coding=utf-8 - +import re from unittest.mock import MagicMock import pytest @@ -11,21 +11,21 @@ from bdfr.site_downloaders.erome import Erome @pytest.mark.online @pytest.mark.parametrize(('test_url', 'expected_urls'), ( ('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://s15.erome.com/355/ORhX0FZz/9IYQocM9_480p.mp4', - 'https://s15.erome.com/355/ORhX0FZz/9eEDc8xm_480p.mp4', - 'https://s15.erome.com/355/ORhX0FZz/EvApC7Rp_480p.mp4', - 'https://s15.erome.com/355/ORhX0FZz/LruobtMs_480p.mp4', - 'https://s15.erome.com/355/ORhX0FZz/TJNmSUU5_480p.mp4', - 'https://s15.erome.com/355/ORhX0FZz/X11Skh6Z_480p.mp4', - 'https://s15.erome.com/355/ORhX0FZz/bjlTkpn7_480p.mp4' + r'https://s\d+.erome.com/355/ORhX0FZz/9IYQocM9_480p.mp4', + r'https://s\d+.erome.com/355/ORhX0FZz/9eEDc8xm_480p.mp4', + r'https://s\d+.erome.com/355/ORhX0FZz/EvApC7Rp_480p.mp4', + r'https://s\d+.erome.com/355/ORhX0FZz/LruobtMs_480p.mp4', + r'https://s\d+.erome.com/355/ORhX0FZz/TJNmSUU5_480p.mp4', + r'https://s\d+.erome.com/355/ORhX0FZz/X11Skh6Z_480p.mp4', + r'https://s\d+.erome.com/355/ORhX0FZz/bjlTkpn7_480p.mp4' )), )) def test_get_link(test_url: str, expected_urls: tuple[str]): 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