PrivacyScanData/archive/new2/bulk-downloader-for-reddit-development/tests/site_downloaders/test_vidble.py

#!/usr/bin/env python3
# coding=utf-8
from unittest.mock import Mock

import pytest

from bdfr.resource import Resource
from bdfr.site_downloaders.vidble import Vidble


@pytest.mark.parametrize(('test_url', 'expected'), (
    ('/RDFbznUvcN_med.jpg', '/RDFbznUvcN.jpg'),
))
def test_change_med_url(test_url: str, expected: str):
    result = Vidble.change_med_url(test_url)
    assert result == expected


@pytest.mark.online
@pytest.mark.parametrize(('test_url', 'expected'), (
    ('https://www.vidble.com/show/UxsvAssYe5', {
        'https://www.vidble.com/UxsvAssYe5.gif',
    }),
    ('https://vidble.com/show/RDFbznUvcN', {
        'https://www.vidble.com/RDFbznUvcN.jpg',
    }),
    ('https://vidble.com/album/h0jTLs6B', {
        'https://www.vidble.com/XG4eAoJ5JZ.jpg',
        'https://www.vidble.com/IqF5UdH6Uq.jpg',
        'https://www.vidble.com/VWuNsnLJMD.jpg',
        'https://www.vidble.com/sMmM8O650W.jpg',
    }),
    ('https://www.vidble.com/pHuwWkOcEb', {
        'https://www.vidble.com/pHuwWkOcEb.jpg',
    }),
))
def test_get_links(test_url: str, expected: set[str]):
    results = Vidble.get_links(test_url)
    assert results == expected


@pytest.mark.online
@pytest.mark.parametrize(('test_url', 'expected_hashes'), (
    ('https://www.vidble.com/show/UxsvAssYe5', {
        '0ef2f8e0e0b45936d2fb3e6fbdf67e28',
    }),
    ('https://vidble.com/show/RDFbznUvcN', {
        'c2dd30a71e32369c50eed86f86efff58',
    }),
    ('https://vidble.com/album/h0jTLs6B', {
        '3b3cba02e01c91f9858a95240b942c71',
        'dd6ecf5fc9e936f9fb614eb6a0537f99',
        'b31a942cd8cdda218ed547bbc04c3a27',
        '6f77c570b451eef4222804bd52267481',
    }),
    ('https://www.vidble.com/pHuwWkOcEb', {
        '585f486dd0b2f23a57bddbd5bf185bc7',
    }),
))
def test_find_resources(test_url: str, expected_hashes: set[str]):
    mock_download = Mock()
    mock_download.url = test_url
    downloader = Vidble(mock_download)
    results = downloader.find_resources()
    assert all([isinstance(res, Resource) for res in results])
    [res.download() for res in results]
    hashes = set([res.hash.hexdigest() for res in results])
    assert hashes == set(expected_hashes)