1
0
Fork 0
mirror of synced 2024-06-21 11:40:21 +12:00

Return banned users as not existing

This commit is contained in:
Serene-Arc 2021-04-23 20:46:28 +10:00
parent db46676dec
commit e6551bb797
2 changed files with 19 additions and 1 deletions

View file

@ -344,7 +344,7 @@ class RedditDownloader:
try:
if not user.id:
return False
except prawcore.exceptions.NotFound:
except (prawcore.exceptions.NotFound, AttributeError):
return False
return True

View file

@ -458,3 +458,21 @@ def test_read_excluded_submission_ids_from_file(downloader_mock: MagicMock, tmp_
downloader_mock.args.exclude_id_file = [test_file]
results = RedditDownloader._read_excluded_ids(downloader_mock)
assert results == {'aaaaaa', 'bbbbbb'}
@pytest.mark.online
@pytest.mark.reddit
@pytest.mark.parametrize(('test_redditor_name', 'expected'), (
('anthonyhui', True), # Real
('lhnhfkuhwreolo', False), # Fake
('Bree-Boo', False), # Banned
))
def test_check_user_existence(
test_redditor_name: str,
expected: bool,
reddit_instance: praw.Reddit,
downloader_mock: MagicMock,
):
downloader_mock.reddit_instance = reddit_instance
result = RedditDownloader._check_user_existence(downloader_mock, test_redditor_name)
assert result == expected