Total Complexity | 2 |
Total Lines | 22 |
Duplicated Lines | 0 % |
Changes | 0 |
1 | # Copyright Pincer 2021-Present |
||
|
|||
2 | # Full MIT License can be found in `LICENSE` at the project root. |
||
3 | from pincer.objects import UserMessage |
||
4 | |||
5 | |||
6 | class Message: |
||
7 | |||
8 | def __init__(self, content): |
||
9 | self.content = content |
||
10 | |||
11 | clean_content = UserMessage.clean_content |
||
12 | |||
13 | |||
14 | class TestUserMessages: |
||
15 | |||
16 | def test_clean_content(self): |
||
17 | assert Message('Hello, world!').clean_content == 'Hello, world!' |
||
18 | assert Message('_*`Hello, world!`*_').clean_content == 'Hello, world!' |
||
19 | assert Message('__~~Hello, world!~~__').clean_content == 'Hello, world!' |
||
20 | assert Message('||Hello, world!||').clean_content == 'Hello, world!' |
||
21 | assert Message('```py\nfoo\nbar\n```').clean_content == 'foo\nbar' |
||
22 |