Add tests pre-dpytest patch for GuildIterator

 - These are broken without a working patch

Author: Unknownlocal

tests/test_Insights.py

@@ -12,51 +12,47 @@
 import random
 
 # Libs
+import discord.ext.commands
 import discord.ext.test as dpytest
 import mock
 import pytest
 
 # Own modules
 import KoalaBot
 from cogs import Insights, BaseCog
-from tests.utils_testing import LastCtxCog
+from tests.utils_testing import LastCtxCog, TestUtils
 
 
 # Constants
 
 # Variables
 
-@pytest.fixture(scope="function", autouse=True)
-def utils_cog(bot):
+@pytest.fixture(scope="function", autouse=False)
+def setup(bot):
     utils_cog = LastCtxCog.LastCtxCog(bot)
+    insights_cog = Insights.Insights(bot)
     bot.add_cog(utils_cog)
-    dpytest.configure(bot)
-    print("Tests starting")
-    return utils_cog
-
-
-@pytest.fixture(scope="function", autouse=True)
-def base_cog(bot):
-    base_cog = BaseCog.BaseCog(bot)
-    bot.add_cog(base_cog)
-    dpytest.configure(bot)
-    print("Tests starting")
-    return base_cog
-
+    bot.add_cog(insights_cog)
+    print("Tests starting (setup)")
+    return bot
 
-@pytest.fixture(scope="function", autouse=True)
-async def insights_cog(bot):
+@pytest.fixture(scope="function", autouse=False)
+def setup_no_conf(bot_no_configure):
+    bot: discord.ext.commands.Bot = bot_no_configure
+    utils_cog = LastCtxCog.LastCtxCog(bot)
     insights_cog = Insights.Insights(bot)
+    bot.add_cog(utils_cog)
     bot.add_cog(insights_cog)
-    dpytest.configure(bot)
-    print("Tests starting")
-    return insights_cog
+    print("Tests starting (setup)")
+    return bot
 
 
 @pytest.mark.asyncio
 @pytest.mark.parametrize("num_guilds, num_members",
-                         [(1, 1), (1, 2), (1, 10), (2, 2), (2, 5), (2, 20), (5, 100), (100, 10000), (20, 20000)])
-async def test_insights(num_guilds, num_members):
+                         [(1, 1), (1, 2), (1, 10), (2, 2), (2, 5), (2, 20), (5, 100), (100, 500), (20, 2000)])
+async def test_insights(setup_no_conf, num_guilds, num_members):
+    bot = setup_no_conf
+    dpytest.configure(bot,num_guilds,1,num_members)
     test_config = dpytest.get_config()
 
     for i in range(num_guilds):
@@ -77,3 +73,59 @@ async def test_insights(num_guilds, num_members):
     await dpytest.message(KoalaBot.COMMAND_PREFIX + "insights")
     assert dpytest.verify().message().content(
         f"KoalaBot is in {len(dpytest.get_config().guilds)} servers with a member total of {expected_users}.")
+
+
+@pytest.mark.asyncio
+@pytest.mark.parametrize("num_guilds", [1,2,3,4,5,6,7,8,9,10,20,50,100])
+async def test_servers_no_args(setup_no_conf, num_guilds):
+    bot = setup_no_conf
+    dpytest.configure(bot, num_guilds, 1, 1)
+    test_config = dpytest.get_config()
+
+    expected_str = [""]
+    expected_index = 0
+
+    for g in test_config.guilds:
+        if len(expected_str[expected_index]) + 2 + len(g.name) > 2000:
+            expected_index += 1
+            expected_str.append("")
+        expected_str[expected_index] += g.name + ", "
+
+    expected_str[-1] = expected_str[-1][:-2]
+    await dpytest.message(KoalaBot.COMMAND_PREFIX + "servers")
+    for m in expected_str:
+        assert dpytest.verify().message().content(m), print(dpytest.sent_queue)
+
+
+@pytest.mark.asyncio
+@pytest.mark.parametrize("num_guilds, num_valid", [(1,0), (1,1), (1,10), (2,0), (2, 1), (2,10), (5,0), (5,1), (5,10), (10, 0), (10, 1), (10, 50)])
+async def test_servers_args(setup_no_conf, num_guilds, num_valid):
+    bot = setup_no_conf
+    dpytest.configure(bot, num_guilds, 1, 1)
+    test_config = dpytest.get_config()
+
+    arg_name = TestUtils.fake_custom_emoji_name_str()
+
+    for i in range(num_valid):
+        g = dpytest.back.make_guild(arg_name + " " + str(i), test_config.members)
+        print(g)
+
+    if num_valid != 0:
+        expected_str = [""]
+        expected_index = 0
+
+        for g in test_config.guilds:
+            if arg_name in g.name:
+                if len(expected_str[expected_index]) + 2 + len(g.name) > 2000:
+                    expected_index += 1
+                    expected_str.append("")
+                expected_str[expected_index] += g.name + ", "
+
+        expected_str[-1] = expected_str[-1][:-2]
+        await dpytest.message(KoalaBot.COMMAND_PREFIX + "servers")
+        for m in expected_str:
+            assert dpytest.verify().message().content(m), print(dpytest.sent_queue)
+    else:
+        await dpytest.message(KoalaBot.COMMAND_PREFIX + "servers " + arg_name)
+        assert dpytest.verify().message().content(f"No servers with {arg_name} in their name!")
+