aboutsummaryrefslogtreecommitdiffstats
path: root/tests/cogs/test_assistant.py
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--tests/cogs/test_assistant.py20
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
7def assistant_cog(): 7def 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
54async def test_assistant_on_message_keyword_match(assistant_cog): 54async 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)