diff --git a/commands/shiba/random_shiba.py b/commands/shiba/random_shiba.py index d3cea92..5c552c3 100644 --- a/commands/shiba/random_shiba.py +++ b/commands/shiba/random_shiba.py @@ -1,26 +1,30 @@ -import discord -from ..modules.async_utils import _async_get_json -from random import choice - -POSSIBLE_BREEDS = ( - "https://dog.ceo/api/breed/shiba/images/random", - "https://dog.ceo/api/breed/akita/images/random", -) - - -def register_commands(tree, this_guild: discord.Object): - @tree.command( - name="doge", - description="Random Shiba or Akita image", - guild=this_guild, - ) - async def random_shiba( - interaction: discord.Interaction, - ): - link = choice(POSSIBLE_BREEDS) - shiba_json = await _async_get_json(link) - if shiba_json is None or shiba_json["status"] != "success": # ): - return await interaction.response.send_message( - "Sorry, we couldn't find any dogs ):" - ) - await interaction.response.send_message(shiba_json["message"]) +import discord +from ..modules.async_utils import _async_get_json +from random import choice + +POSSIBLE_BREEDS = ( + "https://dog.ceo/api/breed/shiba/images/random", + "https://dog.ceo/api/breed/akita/images/random", +) + + +async def get_shiba(link): + return await _async_get_json(link) + + +def register_commands(tree, this_guild: discord.Object): + @tree.command( + name="doge", + description="Random Shiba or Akita image", + guild=this_guild, + ) + async def random_shiba( + interaction: discord.Interaction, + ): + link = choice(POSSIBLE_BREEDS) + shiba_json = await get_shiba(link) + if shiba_json is None or shiba_json["status"] != "success": # ): + return await interaction.response.send_message( + "Sorry, we couldn't find any dogs ):" + ) + await interaction.response.send_message(shiba_json["message"]) diff --git a/tests/test_shiba.py b/tests/test_shiba.py new file mode 100644 index 0000000..e9c2f80 --- /dev/null +++ b/tests/test_shiba.py @@ -0,0 +1,12 @@ +from commands.shiba import random_shiba +import requests + + +async def test_dog_api(): + """ + If something is wrong with their API, we either need + to adjust it or get rid of it entirely. + """ + for link in random_shiba.POSSIBLE_BREEDS: + result = await random_shiba.get_shiba(link) + assert result["status"] == "success"