From b343381fea3eab26e334a1b3bba202d715f9ea2e Mon Sep 17 00:00:00 2001 From: salmonik Date: Sun, 19 Apr 2026 15:44:33 +0200 Subject: test: improve test assertions and edge cases --- tests/cogs/test_assistant.py | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) (limited to 'tests/cogs/test_assistant.py') 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 def assistant_cog(): with patch('os.getenv', return_value="test_keyword"): from cogs.assistant import Jarvis - + mock_bot = MagicMock() mock_bot.user = MagicMock() mock_bot.user.id = 111 - + cog = Jarvis(mock_bot) return cog @@ -25,10 +25,10 @@ async def test_assistant_on_message_ignored_bot_message(assistant_cog): mock_author = MagicMock() mock_author.id = 456 mock_author.bot = True - + mock_message = MagicMock() mock_message.author = mock_author - + with patch.object(assistant_cog, 'assistant', new_callable=AsyncMock) as mock_assistant: await assistant_cog.on_message(mock_message) mock_assistant.assert_not_called() @@ -39,12 +39,12 @@ async def test_assistant_on_message_mentions_bot(assistant_cog): mock_message = MagicMock() mock_message.author = MagicMock() mock_message.author.id = 123 - + mock_mention = MagicMock() mock_mention.id = 111 mock_message.mentions = [mock_mention] mock_message.content = "test" - + with patch.object(assistant_cog, 'assistant', new_callable=AsyncMock) as mock_assistant: await assistant_cog.on_message(mock_message) mock_assistant.assert_called_once_with(mock_message) @@ -53,14 +53,14 @@ async def test_assistant_on_message_mentions_bot(assistant_cog): @pytest.mark.asyncio async def test_assistant_on_message_keyword_match(assistant_cog): assistant_cog.keyword = "test" - + mock_message = MagicMock() mock_message.author = MagicMock() mock_message.author.id = 123 - + mock_message.mentions = [] mock_message.content = "test message with test keyword" - + with patch.object(assistant_cog, 'assistant', new_callable=AsyncMock) as mock_assistant: await assistant_cog.on_message(mock_message) - mock_assistant.assert_called_once_with(mock_message) \ No newline at end of file + mock_assistant.assert_called_once_with(mock_message) -- cgit v1.2.3