diff --git a/bulkredditdownloader/tests/downloaders/test_download_factory.py b/bulkredditdownloader/tests/downloaders/test_download_factory.py index 9d6624f..5e1b1d0 100644 --- a/bulkredditdownloader/tests/downloaders/test_download_factory.py +++ b/bulkredditdownloader/tests/downloaders/test_download_factory.py @@ -26,7 +26,7 @@ from bulkredditdownloader.site_downloaders.youtube import Youtube ('lu29zn', SelfPost), ('lu2ykk', Direct), # Imgur direct link ('luh2pd', Direct), # Reddit direct link - ('luo9eo', Direct), # Imgur direct link gif + ('luo9eo', Imgur), # Imgur .gifv link ('lumulo', Direct), # Imgur direct link gif ('lui5t3', Imgur), ('lu93m7', Gallery), diff --git a/bulkredditdownloader/tests/downloaders/test_imgur.py b/bulkredditdownloader/tests/downloaders/test_imgur.py index 6f6a9f4..66cbbd0 100644 --- a/bulkredditdownloader/tests/downloaders/test_imgur.py +++ b/bulkredditdownloader/tests/downloaders/test_imgur.py @@ -28,7 +28,7 @@ from bulkredditdownloader.site_downloaders.imgur import Imgur {'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) 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()]) 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', ('.gif', '.png', '.jpg', '.mp4')