Python MockPrawSubmission.over_18の例

コード例 #1

ファイル: test_submission_filter.py プロジェクト: zacker150/DownloaderForReddit

    def test_nsfw_filter_only(self):
        user = get_user(download_nsfw=NsfwFilter.ONLY)
        submission = MockPrawSubmission(over_18=True)
        self.assertTrue(self.sub_filter.nsfw_filter(submission, user))

        submission.over_18 = False
        self.assertFalse(self.sub_filter.nsfw_filter(submission, user))

コード例 #2

ファイル: test_submission_filter.py プロジェクト: zacker150/DownloaderForReddit

    def test_nsfw_filter_exclude(self):
        user = get_user(download_nsfw=NsfwFilter.EXCLUDE)
        submission = MockPrawSubmission(over_18=False)
        self.assertTrue(self.sub_filter.nsfw_filter(submission, user))

        submission.over_18 = True
        self.assertFalse(self.sub_filter.nsfw_filter(submission, user))