| 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 |