diff --git a/tests/test_archiver.py b/tests/test_archiver.py index c8e9076..d6f3835 100644 --- a/tests/test_archiver.py +++ b/tests/test_archiver.py @@ -26,3 +26,20 @@ def test_write_submission_json(test_submission_id: str, tmp_path: Path, test_for test_submission = reddit_instance.submission(id=test_submission_id) archiver_mock.file_name_formatter.format_path.return_value = test_path Archiver.write_entry(archiver_mock, test_submission) + +@pytest.mark.online +@pytest.mark.reddit +@pytest.mark.parametrize( + "test_submission_ids", + ( + ("1000000",), + ("https://reddit.com/comments/1000000/"), + ), +) +def test_get_submissions_from_link(test_submission_ids: list[str]) + archiver_mock = MagicMock() + archiver_mock.args.link = test_submission_ids + results = Archiver.get_submissions_from_link(archiver_mock) + assert all([isinstance(sub, praw.models.Submission) for res in results for sub in res]) + assert len(results[0]) == len(test_submission_ids) + assert results[0][0].id == test_submission_ids[0]