CodeMirror/bulk-downloader-for-reddit
1
0
Fork 0
mirror of synced 2024-06-26 18:10:26 +12:00
Code Issues Releases Wiki Activity

Generalise test

Browse source
This commit is contained in:
Serene-Arc 2021-03-17 22:28:31 +10:00 committed by Ali Parlakci
parent f13e029e47
commit a9028434c4
1 changed files with 11 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
bulkredditdownloader/tests/test_downloader.py
View file

@ -271,17 +271,25 @@ def test_get_user_saved(downloader_mock: MagicMock, authenticated_reddit_instanc
@pytest.mark.online
@pytest.mark.reddit
def test_download_submission(downloader_mock: MagicMock, reddit_instance: praw.Reddit, tmp_path: Path):
@pytest.mark.parametrize(('test_submission_id', 'expected_files_len'), (
('ljyy27', 4),
))
def test_download_submission(
test_submission_id: str,
expected_files_len: int,
downloader_mock: MagicMock,
reddit_instance: praw.Reddit,
tmp_path: Path):
downloader_mock.reddit_instance = reddit_instance
downloader_mock.download_filter.check_url.return_value = True
downloader_mock.args.set_folder_scheme = ''
downloader_mock.file_name_formatter = RedditDownloader._create_file_name_formatter(downloader_mock)
downloader_mock.download_directory = tmp_path
downloader_mock.master_hash_list = []
submission = downloader_mock.reddit_instance.submission(id='ljyy27')
submission = downloader_mock.reddit_instance.submission(id=test_submission_id)
RedditDownloader._download_submission(downloader_mock, submission)
folder_contents = list(tmp_path.iterdir())
assert len(folder_contents) == 4
assert len(folder_contents) == expected_files_len
@pytest.mark.online