33 lines
1.2 KiB
Python
33 lines
1.2 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
# coding=utf-8
|
||
|
|
|
||
|
|
import praw
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from bulkredditdownloader.archive_entry import ArchiveEntry
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.online
|
||
|
|
@pytest.mark.reddit
|
||
|
|
@pytest.mark.parametrize(('test_submission_id', 'min_comments'), (
|
||
|
|
('m3reby', 27),
|
||
|
|
))
|
||
|
|
def test_get_comments(test_submission_id: str, min_comments: int, reddit_instance: praw.Reddit):
|
||
|
|
test_submission = reddit_instance.submission(id=test_submission_id)
|
||
|
|
test_archive_entry = ArchiveEntry(test_submission)
|
||
|
|
test_archive_entry._get_comments()
|
||
|
|
assert len(test_archive_entry.comments) >= min_comments
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.online
|
||
|
|
@pytest.mark.reddit
|
||
|
|
@pytest.mark.parametrize(('test_submission_id', 'expected_dict'), (
|
||
|
|
('m3reby', {'author': 'sinjen-tos', 'id': 'm3reby', 'link_flair_text': 'image'}),
|
||
|
|
('m3kua3', {'author': 'DELETED'}),
|
||
|
|
))
|
||
|
|
def test_get_post_details(test_submission_id: str, expected_dict: dict, reddit_instance: praw.Reddit):
|
||
|
|
test_submission = reddit_instance.submission(id=test_submission_id)
|
||
|
|
test_archive_entry = ArchiveEntry(test_submission)
|
||
|
|
test_archive_entry._get_post_details()
|
||
|
|
assert all([test_archive_entry.post_details[key] == expected_dict[key] for key in expected_dict.keys()])
|