Added suite description for -l option

1 files changed, 9 insertions, 4 deletions

diff --git a/src/suite/decorator.py b/src/suite/decorator.py
index d4801c3..12f58a6 100644
--- a/src/suite/decorator.py
+++ b/src/suite/decorator.py
@@ -6,7 +6,7 @@
 #    By: charles <me@cacharle.xyz>                  +#+  +:+       +#+         #
 #                                                 +#+#+#+#+#+   +#+            #
 #    Created: 2020/09/11 12:28:00 by charles           #+#    #+#              #
-#    Updated: 2020/09/11 22:08:36 by charles          ###   ########.fr        #
+#    Updated: 2020/10/08 08:34:10 by cacharle         ###   ########.fr        #
 #                                                                              #
 # ############################################################################ #
 
@@ -22,10 +22,15 @@ def suite(groups: [str] = [], bonus: bool = False):
         """Decorator for a suite function (fmt: suite_[name]) """
 
         mod_name = inspect.getmodule(origin).__name__[len("suites."):]
-        # print(mod_name)
-
         name = "{}/{}".format(mod_name, origin.__name__[len("suite_"):])
-        s = Suite(name, groups + [mod_name], bonus)
+        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):