mirror of
https://github.com/xtream1101/humblebundle-downloader.git
synced 2024-11-20 16:59:16 +01:00
86 lines
2.5 KiB
Python
86 lines
2.5 KiB
Python
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
|