Пример #1
0
    async def test_format_reply(self, reddit: apraw.Reddit,
                                subreddit: Subreddit, banhammer: Banhammer):
        sub = await reddit.subreddit("banhammerdemo")
        item = None

        async for s in sub.new():
            item = RedditItem(s, subreddit, "new")
            break

        assert item

        if item:
            assert isinstance(item.format_reply("Test reply."), str)
Пример #2
0
    async def check_inbox(self):
        await self.bot.wait_until_ready()

        channel = self.bot.get_channel(740496959311446056)

        user = await self.bot.reddit.user.me()
        msgs = [
            msg async for msg in user.inbox() if msg.id not in self._msg_ids
        ]

        for msg in reversed(msgs):
            self._msg_ids.add(msg.id)

            if self._skip_msgs:
                continue

            if msg.was_comment:
                comment = await self.bot.reddit.comment(msg.id)
                item = RedditItem(comment, self.bot.subreddits[0], "new")
                embed = await item.get_embed(embed_template=self.bot.embed)
            else:
                author = await msg.author()
                embed = self.bot.embed.set_author(
                    name=f"Message by /u/{msg._data['author']}",
                    url=author._data.get("icon_img", "") or Embed.Empty)
                embed.description = msg.body
                embed.timestamp = msg.created_utc

            message = await channel.send(embed=embed)
            if msg.was_comment:
                await item.add_reactions(message)

        self._skip_msgs = False
Пример #3
0
    async def test_get_reaction(self, reddit: apraw.Reddit,
                                subreddit: Subreddit):
        await subreddit.load_reactions()

        sub = await reddit.subreddit("banhammerdemo")
        item = None

        async for s in sub.new():
            item = RedditItem(s, subreddit, "new")
            break

        if item:
            emojis = [r.emoji for r in item.reactions]
            if emojis:
                reaction = item.get_reaction(emojis[0])
                assert reaction == item.reactions[0]
Пример #4
0
    async def test_get_embed(self, reddit: apraw.Reddit, subreddit: Subreddit):
        sub = await reddit.subreddit("banhammerdemo")
        item = None

        async for s in sub.new():
            item = RedditItem(s, subreddit, "new")
            break

        if item:
            assert isinstance(await item.get_embed(), discord.Embed)
    async def test_get_ban_message(self, reddit: apraw.Reddit,
                                   subreddit: Subreddit, banhammer: Banhammer):
        sub = await reddit.subreddit("banhammerdemo")
        item = None

        async for s in sub.new():
            item = RedditItem(s, subreddit, "new")
            break

        if item:
            assert isinstance(
                banhammer.message_builder.get_ban_message(item, 10), str)
Пример #6
0
    async def test_url(self, reddit: apraw.Reddit, subreddit: Subreddit):
        await subreddit.load_reactions()

        sub = await reddit.subreddit("banhammerdemo")
        item = None

        async for s in sub.new():
            item = RedditItem(s, subreddit, "new")
            break

        if item:
            assert item.url in s.url
Пример #7
0
    async def test_get_reactions(self, reddit: apraw.Reddit, subreddit: Subreddit):
        await subreddit.load_reactions()

        sub = await reddit.subreddit("banhammerdemo")
        item = None

        async for s in sub.new():
            item = RedditItem(s, subreddit, "new")
            break

        if item:
            for r in subreddit.get_reactions(item):
                assert isinstance(r, Reaction)
    async def test_get_item_embed(self, reddit: apraw.Reddit,
                                  subreddit: Subreddit, banhammer: Banhammer):
        sub = await reddit.subreddit("banhammerdemo")
        item = None

        async for s in sub.new():
            item = RedditItem(s, subreddit, "new")
            break

        if item:
            assert isinstance(
                await banhammer.message_builder.get_item_embed(item),
                discord.Embed)
Пример #9
0
    async def test_get_message(self, reddit: apraw.Reddit,
                               subreddit: Subreddit):
        sub = await reddit.subreddit("banhammerdemo")
        item = None

        async for s in sub.new():
            item = RedditItem(s, subreddit, "new")
            break

        payload = ReactionPayload()
        payload.feed(item, False, "Test", "🔥", "Test reply.")

        assert isinstance(await payload.get_message(), str)
Пример #10
0
    async def test_get_author_name(self, reddit: apraw.Reddit,
                                   subreddit: Subreddit):
        sub = await reddit.subreddit("banhammerdemo")
        item = None
        author = None

        async for s in sub.new():
            item = RedditItem(s, subreddit, "new")
            author = await s.author()
            break

        if item:
            author_name = await item.get_author_name()
            assert author_name.lower() == author.name.lower()
Пример #11
0
    async def handle_reposts(self, item: RedditItem):
        self._comment_ids.add(item.item.id)
        try:
            submission = await item.item.submission()
        except Exception as e:
            print(f"Couldn't fetch comment's parent submission: {e}")
        else:
            submission_item = RedditItem(submission, item.subreddit, "reports")
            embed = await submission_item.get_embed(embed_template=self.embed)

            embed.add_field(name="Comment by /u/RepostSleuthBot",
                            value=f"{item.body}\n\n[Comment.]({item.url})")

            msg = await self.get_channel(lc_config["reposts_channel"]).send(embed=embed)
            await submission_item.add_reactions(msg)
Пример #12
0
    async def test_feed(self, reddit: apraw.Reddit, subreddit: Subreddit):
        sub = await reddit.subreddit("banhammerdemo")
        item = None

        async for s in sub.new():
            item = RedditItem(s, subreddit, "new")
            break

        payload = ReactionPayload()
        payload.feed(item, False, "Test", "🔥", "Test reply.")

        assert payload.item == item
        assert not payload.approved
        assert payload.user == "Test"
        assert payload.reply == "Test reply."
        assert payload.emoji == "🔥"
Пример #13
0
    async def test_is_author_removed(self, reddit: apraw.Reddit,
                                     subreddit: Subreddit,
                                     banhammer: Banhammer):
        sub = await reddit.subreddit("banhammerdemo")
        item = None

        async for s in sub.new():
            item = RedditItem(s, subreddit, "new")
            break

        if item:
            assert not await item.is_author_removed()

        url = "https://www.reddit.com/r/banhammerdemo/comments/c66rdl"
        item = await banhammer.get_item(url)

        if item:
            assert await item.is_author_removed()