diff options
author | 2019-10-25 15:24:33 +0200 | |
---|---|---|
committer | 2019-10-25 15:24:33 +0200 | |
commit | ccbc6a6c218056629af0210f101468fa39bc312e (patch) | |
tree | 5ade16d0cf00c2cc876c511f73b12237bc31f603 /tests/rules/test_attachments.py | |
parent | Substract one to now calculation (diff) | |
parent | Merge pull request #543 from atmishra/moderator-channel-check (diff) |
Merge branch 'master' into reminder-up
Diffstat (limited to 'tests/rules/test_attachments.py')
-rw-r--r-- | tests/rules/test_attachments.py | 52 |
1 files changed, 0 insertions, 52 deletions
diff --git a/tests/rules/test_attachments.py b/tests/rules/test_attachments.py deleted file mode 100644 index 6f025b3cb..000000000 --- a/tests/rules/test_attachments.py +++ /dev/null @@ -1,52 +0,0 @@ -import asyncio -from dataclasses import dataclass -from typing import Any, List - -import pytest - -from bot.rules import attachments - - -# Using `MagicMock` sadly doesn't work for this usecase -# since it's __eq__ compares the MagicMock's ID. We just -# want to compare the actual attributes we set. -@dataclass -class FakeMessage: - author: str - attachments: List[Any] - - -def msg(total_attachments: int): - return FakeMessage(author='lemon', attachments=list(range(total_attachments))) - - - 'messages', - ( - (msg(0), msg(0), msg(0)), - (msg(2), msg(2)), - (msg(0),), - ) -) -def test_allows_messages_without_too_many_attachments(messages): - last_message, *recent_messages = messages - coro = attachments.apply(last_message, recent_messages, {'max': 5}) - assert asyncio.run(coro) is None - - - ('messages', 'relevant_messages', 'total'), - ( - ((msg(4), msg(0), msg(6)), [msg(4), msg(6)], 10), - ((msg(6),), [msg(6)], 6), - ((msg(1),) * 6, [msg(1)] * 6, 6), - ) -) -def test_disallows_messages_with_too_many_attachments(messages, relevant_messages, total): - last_message, *recent_messages = messages - coro = attachments.apply(last_message, recent_messages, {'max': 5}) - assert asyncio.run(coro) == ( - f"sent {total} attachments in 5s", - ('lemon',), - relevant_messages - ) |