Renaming src -> minishell_test for package name, Renaming main.py -> __main__.py for package execution with python -m

1 files changed, 47 insertions, 0 deletions

diff --git a/minishell_test/suite/decorator.py b/minishell_test/suite/decorator.py
new file mode 100644
index 0000000..fdc7fb6
--- /dev/null
+++ b/minishell_test/suite/decorator.py
@@ -0,0 +1,47 @@
+# ############################################################################ #
+#                                                                              #
+#                                                         :::      ::::::::    #
+#    decorator.py                                       :+:      :+:    :+:    #
+#                                                     +:+ +:+         +:+      #
+#    By: charles <me@cacharle.xyz>                  +#+  +:+       +#+         #
+#                                                 +#+#+#+#+#+   +#+            #
+#    Created: 2020/09/11 12:28:00 by charles           #+#    #+#              #
+#    Updated: 2021/02/04 16:18:11 by charles          ###   ########.fr        #
+#                                                                              #
+# ############################################################################ #
+
+import inspect
+from typing import List
+
+from suite import Suite
+from test import Test
+
+
+def suite(groups: List[str] = [], bonus: bool = False):  # type: ignore
+    """Decorator generator for suites arguments"""
+
+    def suite_wrapper(origin):
+        """Decorator for a suite function (fmt: suite_[name]) """
+
+        mod = inspect.getmodule(origin)
+        if mod is None:
+            raise NotImplementedError
+        mod_name = mod.__name__[len("suites."):]
+        name = "{}/{}".format(mod_name, origin.__name__[len("suite_"):])
+        description = origin.__doc__
+        if description is None:
+            print("You should had a doc string to the {} suite".format(name))
+            description = "no description"
+        description = description.split("\n")[0].strip()
+        s = Suite(name, groups + [mod_name], bonus, description)
+
+        def test_generator():
+            def test(*args, **kwargs):
+                s.add(Test(*args, **kwargs))
+            origin(test)
+
+        s.generator_func = test_generator
+        Suite.available.append(s)
+        return test_generator
+
+    return suite_wrapper