humblebundle-downloader/tests/test_download_library.py
2021-05-21 12:55:34 +00:00

79 lines
2.3 KiB
Python

from humblebundle_downloader.download_library import DownloadLibrary
###
# _should_download_file_type
###
def test_include_logic_has_values():
dl = DownloadLibrary(
'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_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_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_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_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_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_library_path',
platform_include=['audio'],
)
assert dl._should_download_platform('ebook') is False
assert dl._should_download_platform('audio') is True