From 423155a846923c377ff7eac192a8e4358fb6d0f2 Mon Sep 17 00:00:00 2001 From: Serene-Arc Date: Wed, 17 Mar 2021 20:42:35 +1000 Subject: [PATCH] Add some more Gallery tests --- .../tests/downloaders/test_gallery.py | 20 +++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/bulkredditdownloader/tests/downloaders/test_gallery.py b/bulkredditdownloader/tests/downloaders/test_gallery.py index 9694964..49e3e83 100644 --- a/bulkredditdownloader/tests/downloaders/test_gallery.py +++ b/bulkredditdownloader/tests/downloaders/test_gallery.py @@ -8,6 +8,26 @@ from bulkredditdownloader.resource import Resource from bulkredditdownloader.site_downloaders.gallery import Gallery +@pytest.mark.online +@pytest.mark.parametrize(('test_url', 'expected'), ( + ('https://www.reddit.com/gallery/m6lvrh', { + 'https://preview.redd.it/18nzv9ch0hn61.jpg', + 'https://preview.redd.it/jqkizcch0hn61.jpg', + 'https://preview.redd.it/k0fnqzbh0hn61.jpg', + 'https://preview.redd.it/m3gamzbh0hn61.jpg' + }), + ('https://www.reddit.com/gallery/ljyy27', { + 'https://preview.redd.it/04vxj25uqih61.png', + 'https://preview.redd.it/0fnx83kpqih61.png', + 'https://preview.redd.it/7zkmr1wqqih61.png', + 'https://preview.redd.it/u37k5gxrqih61.png' + }), +)) +def test_gallery_get_links(test_url: str, expected: set[str]): + results = Gallery._get_links(test_url) + assert set(results) == expected + + @pytest.mark.online @pytest.mark.reddit @pytest.mark.parametrize(('test_submission_id', 'expected_len'), (