1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
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)))
@pytest.mark.parametrize(
'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
@pytest.mark.parametrize(
('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
)
|
