Add Test Cases

Author: cmaureir

intbot/core/bot/main.py

@@ -66,7 +66,7 @@ async def close(ctx):
 
         # Send notification to the thread
         await channel.send(f"# This was marked as done by {author.mention}")
-        
+
         # We need to archive after adding tags in case it was a forum.
         await channel.edit(archived=True)
 

intbot/tests/test_bot/test_main.py

@@ -6,7 +6,7 @@
 
 import pytest
 from asgiref.sync import sync_to_async
-from core.bot.main import ping, poll_database, qlen, source, version, wiki
+from core.bot.main import ping, poll_database, qlen, source, version, wiki, close
 from core.models import DiscordMessage
 from django.utils import timezone
 
@@ -100,6 +100,24 @@ async def test_wiki_command():
         suppress_embeds=True,
     )
 
+@pytest.mark.asyncio
+async def test_close_command():
+    # Mock context
+    ctx = AsyncMock()
+    ctx.channel = discord.Thread()
+    ctx.channel.parent = discord.ForumChannel()
+    ctx.author = discord.Member()
+    ctx.author.mention = "TestUser"
+
+    # Call the command
+    await close(ctx)
+
+    # Assert that the command sent the expected message
+    ctx.send.assert_called_once_with(
+        "# This was marked as done by {ctx.author.mention}",
+        suppress_embeds=True,
+    )
+
 
 @pytest.mark.asyncio
 async def test_version_command():