Add necessary tests
This commit is contained in:
parent
247fe3e6f7
commit
49b0fac7c7
2 changed files with 11 additions and 2 deletions
|
@ -26,7 +26,7 @@ from bulkredditdownloader.site_downloaders.youtube import Youtube
|
||||||
('lu29zn', SelfPost),
|
('lu29zn', SelfPost),
|
||||||
('lu2ykk', Direct), # Imgur direct link
|
('lu2ykk', Direct), # Imgur direct link
|
||||||
('luh2pd', Direct), # Reddit direct link
|
('luh2pd', Direct), # Reddit direct link
|
||||||
('luo9eo', Direct), # Imgur direct link gif
|
('luo9eo', Imgur), # Imgur .gifv link
|
||||||
('lumulo', Direct), # Imgur direct link gif
|
('lumulo', Direct), # Imgur direct link gif
|
||||||
('lui5t3', Imgur),
|
('lui5t3', Imgur),
|
||||||
('lu93m7', Gallery),
|
('lu93m7', Gallery),
|
||||||
|
|
|
@ -28,7 +28,7 @@ from bulkredditdownloader.site_downloaders.imgur import Imgur
|
||||||
{'hash': 'TSAkikk', 'ext': '.jpg', 'title': ''},
|
{'hash': 'TSAkikk', 'ext': '.jpg', 'title': ''},
|
||||||
]),
|
]),
|
||||||
))
|
))
|
||||||
def test_get_data(test_url: str, expected_gen_dict: dict, expected_image_dict: list[dict]):
|
def test_get_data_album(test_url: str, expected_gen_dict: dict, expected_image_dict: list[dict]):
|
||||||
result = Imgur._get_data(test_url)
|
result = Imgur._get_data(test_url)
|
||||||
assert all([result.get(key) == expected_gen_dict[key] for key in expected_gen_dict.keys()])
|
assert all([result.get(key) == expected_gen_dict[key] for key in expected_gen_dict.keys()])
|
||||||
|
|
||||||
|
@ -36,6 +36,15 @@ def test_get_data(test_url: str, expected_gen_dict: dict, expected_image_dict: l
|
||||||
assert any([all([image.get(key) == image_dict[key] for key in image_dict.keys()])
|
assert any([all([image.get(key) == image_dict[key] for key in image_dict.keys()])
|
||||||
for image_dict in expected_image_dict for image in result['album_images']['images']])
|
for image_dict in expected_image_dict for image in result['album_images']['images']])
|
||||||
|
|
||||||
|
@pytest.mark.online
|
||||||
|
@pytest.mark.parametrize(('test_url', 'expected_image_dict'), (
|
||||||
|
('https://i.imgur.com/dLk3FGY.gifv',
|
||||||
|
{'hash': 'dLk3FGY', 'title': '', 'ext': '.mp4', 'animated': True}),
|
||||||
|
))
|
||||||
|
def test_get_data_image(test_url: str, expected_image_dict: dict):
|
||||||
|
result = Imgur._get_data(test_url)
|
||||||
|
assert all([result.get(key) == expected_image_dict[key] for key in expected_image_dict.keys()])
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize('test_extension',
|
@pytest.mark.parametrize('test_extension',
|
||||||
('.gif', '.png', '.jpg', '.mp4')
|
('.gif', '.png', '.jpg', '.mp4')
|
||||||
|
|
Loading…
Reference in a new issue