diff options
Diffstat (limited to 'tests/cogs/test_assistant.py')
| -rw-r--r-- | tests/cogs/test_assistant.py | 20 |
1 files changed, 10 insertions, 10 deletions
diff --git a/tests/cogs/test_assistant.py b/tests/cogs/test_assistant.py index 09c3934..9a33232 100644 --- a/tests/cogs/test_assistant.py +++ b/tests/cogs/test_assistant.py | |||
| @@ -7,11 +7,11 @@ from tests.conftest import MockApplicationContext | |||
| 7 | def assistant_cog(): | 7 | def assistant_cog(): |
| 8 | with patch('os.getenv', return_value="test_keyword"): | 8 | with patch('os.getenv', return_value="test_keyword"): |
| 9 | from cogs.assistant import Jarvis | 9 | from cogs.assistant import Jarvis |
| 10 | 10 | ||
| 11 | mock_bot = MagicMock() | 11 | mock_bot = MagicMock() |
| 12 | mock_bot.user = MagicMock() | 12 | mock_bot.user = MagicMock() |
| 13 | mock_bot.user.id = 111 | 13 | mock_bot.user.id = 111 |
| 14 | 14 | ||
| 15 | cog = Jarvis(mock_bot) | 15 | cog = Jarvis(mock_bot) |
| 16 | return cog | 16 | return cog |
| 17 | 17 | ||
| @@ -25,10 +25,10 @@ async def test_assistant_on_message_ignored_bot_message(assistant_cog): | |||
| 25 | mock_author = MagicMock() | 25 | mock_author = MagicMock() |
| 26 | mock_author.id = 456 | 26 | mock_author.id = 456 |
| 27 | mock_author.bot = True | 27 | mock_author.bot = True |
| 28 | 28 | ||
| 29 | mock_message = MagicMock() | 29 | mock_message = MagicMock() |
| 30 | mock_message.author = mock_author | 30 | mock_message.author = mock_author |
| 31 | 31 | ||
| 32 | with patch.object(assistant_cog, 'assistant', new_callable=AsyncMock) as mock_assistant: | 32 | with patch.object(assistant_cog, 'assistant', new_callable=AsyncMock) as mock_assistant: |
| 33 | await assistant_cog.on_message(mock_message) | 33 | await assistant_cog.on_message(mock_message) |
| 34 | mock_assistant.assert_not_called() | 34 | mock_assistant.assert_not_called() |
| @@ -39,12 +39,12 @@ async def test_assistant_on_message_mentions_bot(assistant_cog): | |||
| 39 | mock_message = MagicMock() | 39 | mock_message = MagicMock() |
| 40 | mock_message.author = MagicMock() | 40 | mock_message.author = MagicMock() |
| 41 | mock_message.author.id = 123 | 41 | mock_message.author.id = 123 |
| 42 | 42 | ||
| 43 | mock_mention = MagicMock() | 43 | mock_mention = MagicMock() |
| 44 | mock_mention.id = 111 | 44 | mock_mention.id = 111 |
| 45 | mock_message.mentions = [mock_mention] | 45 | mock_message.mentions = [mock_mention] |
| 46 | mock_message.content = "test" | 46 | mock_message.content = "test" |
| 47 | 47 | ||
| 48 | with patch.object(assistant_cog, 'assistant', new_callable=AsyncMock) as mock_assistant: | 48 | with patch.object(assistant_cog, 'assistant', new_callable=AsyncMock) as mock_assistant: |
| 49 | await assistant_cog.on_message(mock_message) | 49 | await assistant_cog.on_message(mock_message) |
| 50 | mock_assistant.assert_called_once_with(mock_message) | 50 | mock_assistant.assert_called_once_with(mock_message) |
| @@ -53,14 +53,14 @@ async def test_assistant_on_message_mentions_bot(assistant_cog): | |||
| 53 | @pytest.mark.asyncio | 53 | @pytest.mark.asyncio |
| 54 | async def test_assistant_on_message_keyword_match(assistant_cog): | 54 | async def test_assistant_on_message_keyword_match(assistant_cog): |
| 55 | assistant_cog.keyword = "test" | 55 | assistant_cog.keyword = "test" |
| 56 | 56 | ||
| 57 | mock_message = MagicMock() | 57 | mock_message = MagicMock() |
| 58 | mock_message.author = MagicMock() | 58 | mock_message.author = MagicMock() |
| 59 | mock_message.author.id = 123 | 59 | mock_message.author.id = 123 |
| 60 | 60 | ||
| 61 | mock_message.mentions = [] | 61 | mock_message.mentions = [] |
| 62 | mock_message.content = "test message with test keyword" | 62 | mock_message.content = "test message with test keyword" |
| 63 | 63 | ||
| 64 | with patch.object(assistant_cog, 'assistant', new_callable=AsyncMock) as mock_assistant: | 64 | with patch.object(assistant_cog, 'assistant', new_callable=AsyncMock) as mock_assistant: |
| 65 | await assistant_cog.on_message(mock_message) | 65 | await assistant_cog.on_message(mock_message) |
| 66 | mock_assistant.assert_called_once_with(mock_message) \ No newline at end of file | 66 | mock_assistant.assert_called_once_with(mock_message) |
