New tests

MZshnik · Jul 8, 2024 · 0c646c7 · 0c646c7
1 parent 9a6d477
commit 0c646c7
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 3 deletions.
diff --git a/src/MZscript/functions_handler.py b/src/MZscript/functions_handler.py
@@ -76,7 +76,7 @@ async def check_ifs(self, chunks: list):
         if ifs < elses:
             raise SyntaxError("The amount of $else must be equal or lower to the amount of $if")
 
-    async def get_args(self, entry: str, ctx=None):  # ctx not needed but many entries provide ctx
+    async def get_args(self, entry: str, ctx = None):  # ctx not needed but many entries provide ctx
         """
         ## Gets args from function
         ### Example:

diff --git a/src/MZscript/main.py b/src/MZscript/main.py
@@ -95,7 +95,7 @@ def add_slash(
         options: list = None,
         onlyguild: bool = False,
         isnsfw: bool = False,
-    ):
+        ):
         """
         ## Add slash command
 
@@ -116,8 +116,8 @@ def add_slash(
                 dm_permission=onlyguild,
                 nsfw=isnsfw,
                 auto_sync=True,
+                )
             )
-        )
         self.user_slash_commands.append([name, code])
 
     def load_command(self, path: str):

diff --git a/tests/testhandler.py b/tests/testhandler.py
@@ -52,5 +52,20 @@ async def test_chunks_and_ifs(self):
         self.assertEqual(result, ["$console[Okay]"])
         self.assertEqual(await check_ifs(result), None)
 
+    async def test_get_args(self):
+        funcs = FunctionsHandler()
+        get_args = funcs.get_args
+        self.assertEqual(await get_args("1234567890987654321;Hello World!"), ["1234567890987654321", "Hello World!"])
+        self.assertEqual(await get_args("$sendMessage[];Hello World!"), ["$sendMessage[]", "Hello World!"])
+        self.assertEqual(await get_args("$sendMessage;Hello World!"), ["$sendMessage", "Hello World!"])
+        self.assertEqual(await get_args("Hello World!;$sendMessage[]"), ["Hello World!", "$sendMessage[]"])
+        self.assertEqual(await get_args("Hello World!;$sendMessage"), ["Hello World!", "$sendMessage"])
+        self.assertEqual(await get_args("Hello World!;$sendMessage[]!"), ["Hello World!", "$sendMessage[]!"])
+        self.assertEqual(await get_args("Hello World!;$sendMessage!"), ["Hello World!", "$sendMessage!"])
+        self.assertEqual(await get_args("Hello World!;$sendMessage;;"), ["Hello World!", "$sendMessage", ""])
+        self.assertEqual(await get_args(";;Hello World!;$sendMessage"), ["", "", "Hello World!", "$sendMessage"])
+        self.assertEqual(await get_args(";Hello World!;$sendMessage!;"), ["", "Hello World!", "$sendMessage!"])
+        self.assertEqual(await get_args(" ;Hello World!;$sendMessage!; "), [" ", "Hello World!", "$sendMessage!", " "])
+
 if __name__ == "__main__":
     unittest.main()