humblebundle-downloader/tests/test_download_library.py

from humblebundle_downloader.download_library import DownloadLibrary


###
# _should_download_file_type
###
def test_include_logic_has_values():
    dl = DownloadLibrary(
        'fake_cookie_path',
        'fake_library_path',
        ext_include=['pdf', 'EPub'],
    )
    assert dl._should_download_file_type('pdf') is True
    assert dl._should_download_file_type('df') is False
    assert dl._should_download_file_type('ePub') is True
    assert dl._should_download_file_type('mobi') is False


def test_include_logic_empty():
    dl = DownloadLibrary(
        'fake_cookie_path',
        'fake_library_path',
        ext_include=[],
    )
    assert dl._should_download_file_type('pdf') is True
    assert dl._should_download_file_type('df') is True
    assert dl._should_download_file_type('EPub') is True
    assert dl._should_download_file_type('mobi') is True


def test_exclude_logic_has_values():
    dl = DownloadLibrary(
        'fake_cookie_path',
        'fake_library_path',
        ext_exclude=['pdf', 'EPub'],
    )
    assert dl._should_download_file_type('pdf') is False
    assert dl._should_download_file_type('df') is True
    assert dl._should_download_file_type('ePub') is False
    assert dl._should_download_file_type('mobi') is True


def test_exclude_logic_empty():
    dl = DownloadLibrary(
        'fake_cookie_path',
        'fake_library_path',
        ext_exclude=[],
    )
    assert dl._should_download_file_type('pdf') is True
    assert dl._should_download_file_type('df') is True
    assert dl._should_download_file_type('EPub') is True
    assert dl._should_download_file_type('mobi') is True


###
# _should_download_platform
###
def test_download_platform_filter_none():
    dl = DownloadLibrary(
        'fake_cookie_path',
        'fake_library_path',
        platform_include=None,
    )
    assert dl._should_download_platform('ebook') is True
    assert dl._should_download_platform('audio') is True


def test_download_platform_filter_blank():
    dl = DownloadLibrary(
        'fake_cookie_path',
        'fake_library_path',
        platform_include=[],
    )
    assert dl._should_download_platform('ebook') is True
    assert dl._should_download_platform('audio') is True


def test_download_platform_filter_audio():
    dl = DownloadLibrary(
        'fake_cookie_path',
        'fake_library_path',
        platform_include=['audio'],
    )
    assert dl._should_download_platform('ebook') is False
    assert dl._should_download_platform('audio') is True