mirror of
https://github.com/Rainyy21/lol_discordbot
synced 2026-10-10 23:50:32 -04:00
Rainy21
121 lines
5.2 KiB
Python
121 lines
5.2 KiB
Python
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()
|