CodeMirror/bulk-downloader-for-reddit
1
0
Fork 0
mirror of synced 2024-10-01 01:30:52 +13:00
Code Issues Releases Wiki Activity

Merge pull request #874 from OMEGARAZER/connector-test

Browse source
This commit is contained in:
Serene 2023-06-07 10:28:23 +10:00 committed by GitHub
commit 0017030419
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
tests/test_connector.py
View file

@ -3,6 +3,7 @@
from collections.abc import Iterator
from datetime import datetime, timedelta
from pathlib import Path
from typing import Union
from unittest.mock import MagicMock
import praw
@ -37,21 +38,23 @@ def downloader_mock(args: Configuration):
return downloader_mock
def assert_all_results_are_submissions(result_limit: int, results: list[Iterator]) -> list:
def assert_all_results_are_submissions(result_limit: Union[int, None], results: list[Iterator]) -> list:
results = [sub for res in results for sub in res]
assert all([isinstance(res, praw.models.Submission) for res in results])
assert not any([isinstance(m, MagicMock) for m in results])
if result_limit is not None:
assert len(results) == result_limit
assert len(results) > 0
assert len(results) <= result_limit
return results
def assert_all_results_are_submissions_or_comments(result_limit: int, results: list[Iterator]) -> list:
def assert_all_results_are_submissions_or_comments(result_limit: Union[int, None], results: list[Iterator]) -> list:
results = [sub for res in results for sub in res]
assert all([isinstance(res, (praw.models.Submission, praw.models.Comment)) for res in results])
assert not any([isinstance(m, MagicMock) for m in results])
if result_limit is not None:
assert len(results) == result_limit
assert len(results) > 0
assert len(results) <= result_limit
return results