fix some bug that doesn't let user to sign in and connect with discord, added some test

2026-10-10 23:50:32 -04:00 · 2026-04-22 03:44:29 -04:00
parent c6257c5260
commit 0a972f6a3f
13 changed files with 262 additions and 18 deletions
@@ -0,0 +1,120 @@
+import unittest
+import asyncio
+from unittest.mock import AsyncMock, MagicMock, patch
+import discord
+from commands.login import LoginCog
+from commands.profile import ProfileCog
+from commands.recent import RecentCog
+
+class TestCommands(unittest.IsolatedAsyncioTestCase):
+    def setUp(self):
+        self.bot = MagicMock()
+        self.login_cog = LoginCog(self.bot)
+        self.profile_cog = ProfileCog(self.bot)
+        self.recent_cog = RecentCog(self.bot)
+
+    async def test_login_success(self):
+        interaction = AsyncMock(spec=discord.Interaction)
+        interaction.user.id = 12345
+        interaction.response.defer = AsyncMock()
+        interaction.followup.send = AsyncMock()
+        
+        with patch("commands.login.save_user") as mock_save, \
+             patch("aiohttp.ClientSession.get") as mock_get:
+            
+            # Mock Riot Account API response
+            mock_resp = AsyncMock()
+            mock_resp.status = 200
+            mock_resp.json.return_value = {"puuid": "fake_puuid"}
+            mock_get.return_value.__aenter__.return_value = mock_resp
+            
+            await self.login_cog.login.callback(self.login_cog, interaction, "TestUser#TAG")
+            
+            interaction.response.defer.assert_called_once()
+            mock_save.assert_called_once_with("12345", "fake_puuid", "TestUser", "TAG")
+            interaction.followup.send.assert_called_once()
+            self.assertIn("Linked **TestUser#TAG**", interaction.followup.send.call_args[0][0])
+
+    async def test_login_not_found(self):
+        interaction = AsyncMock(spec=discord.Interaction)
+        interaction.response.defer = AsyncMock()
+        interaction.followup.send = AsyncMock()
+        
+        with patch("aiohttp.ClientSession.get") as mock_get:
+            mock_resp = AsyncMock()
+            mock_resp.status = 404
+            mock_get.return_value.__aenter__.return_value = mock_resp
+            
+            await self.login_cog.login.callback(self.login_cog, interaction, "NonExistent#TAG")
+            
+            interaction.followup.send.assert_called_once()
+            self.assertIn("Summoner not found", interaction.followup.send.call_args[0][0])
+
+    async def test_profile_not_linked(self):
+        interaction = AsyncMock(spec=discord.Interaction)
+        interaction.user.id = 12345
+        interaction.response.defer = AsyncMock()
+        interaction.followup.send = AsyncMock()
+        
+        with patch("commands.profile.get_user", return_value=None):
+            await self.profile_cog.profile.callback(self.profile_cog, interaction)
+            
+            interaction.followup.send.assert_called_once()
+            self.assertIn("haven't linked your Riot account", interaction.followup.send.call_args[0][0])
+
+    async def test_profile_success(self):
+        interaction = AsyncMock(spec=discord.Interaction)
+        interaction.user.id = 12345
+        interaction.response.defer = AsyncMock()
+        interaction.followup.send = AsyncMock()
+        
+        # (discord_id, puuid, game_name, tag_line)
+        mock_user = ("12345", "fake_puuid", "TestUser", "TAG")
+        
+        with patch("commands.profile.get_user", return_value=mock_user), \
+             patch("commands.profile.fetch") as mock_fetch:
+            
+            # Mock Summoner Data
+            mock_fetch.side_effect = [
+                {"id": "summoner_id", "summonerLevel": 100}, # Summoner info
+                [{"queueType": "RANKED_SOLO_5x5", "tier": "GOLD", "rank": "I", "leaguePoints": 50, "wins": 10, "losses": 5}] # League info
+            ]
+            
+            await self.profile_cog.profile.callback(self.profile_cog, interaction)
+            
+            interaction.followup.send.assert_called_once()
+            embed = interaction.followup.send.call_args[1]["embed"]
+            self.assertEqual(embed.title, "📊 TestUser#TAG")
+            self.assertIn("Gold I", embed.fields[1].value)
+
+    async def test_recent_success(self):
+        interaction = AsyncMock(spec=discord.Interaction)
+        interaction.user.id = 12345
+        interaction.response.defer = AsyncMock()
+        interaction.followup.send = AsyncMock()
+        
+        mock_user = ("12345", "fake_puuid", "TestUser", "TAG")
+        
+        with patch("commands.recent.get_user", return_value=mock_user), \
+             patch("commands.recent.fetch") as mock_fetch, \
+             patch("commands.recent.fetch_all") as mock_fetch_all:
+            
+            mock_fetch.return_value = ["match1"]
+            mock_fetch_all.return_value = [{
+                "metadata": {"matchId": "match1"},
+                "info": {
+                    "participants": [{"puuid": "fake_puuid", "win": True, "championName": "Lux", "kills": 10, "deaths": 2, "assists": 15, "totalMinionsKilled": 100, "neutralMinionsKilled": 10}],
+                    "gameDuration": 1800,
+                    "queueId": 420
+                }
+            }]
+            
+            await self.recent_cog.recent.callback(self.recent_cog, interaction, count=1)
+            
+            interaction.followup.send.assert_called_once()
+            embeds = interaction.followup.send.call_args[1]["embeds"]
+            self.assertEqual(len(embeds), 1)
+            self.assertIn("Victory — Lux", embeds[0].title)
+
+if __name__ == "__main__":
+    unittest.main()