From d4d595ed6877874e125738130b0ee9bf846380a5 Mon Sep 17 00:00:00 2001
From: Yohann D'ANELLO <ynerant@crans.org>
Date: Tue, 26 Jan 2021 17:07:12 +0100
Subject: [PATCH] Ajout d'un fichier de test avec pytest

Signed-off-by: Yohann D'ANELLO <ynerant@crans.org>
---
 .gitlab-ci.yml |  2 +-
 main.py        |  8 ++++++--
 main_test.py   | 24 ++++++++++++++++++++++++
 3 files changed, 31 insertions(+), 3 deletions(-)
 mode change 100644 => 100755 main.py
 create mode 100755 main_test.py

diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index d2f52f8..cf68258 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -24,4 +24,4 @@ pylint:
 test:
   stage: test
   image: python:3-alpine
-  script: python main.py
+  script: pytest --show-locals .
diff --git a/main.py b/main.py
old mode 100644
new mode 100755
index 286f7c0..4394901
--- a/main.py
+++ b/main.py
@@ -1,4 +1,4 @@
-#!/usr/bin/env python
+#!/usr/bin/env python3
 
 """
 Ce script est utilisé en guise d'appui pour le séminaire Crans d'introduction
@@ -24,7 +24,11 @@ def main():
     while True:
         command = input("> ")
         args = command.split(" ")
-        print(globals()[args[0]](*args[1:]))
+        print(commande(args[0], *args[1:]))
+
+
+def commande(name: str, *args):
+    return globals()[name](*args)
 
 
 def aide():
diff --git a/main_test.py b/main_test.py
new file mode 100755
index 0000000..4c2a21c
--- /dev/null
+++ b/main_test.py
@@ -0,0 +1,24 @@
+#!/usr/bin/env python3
+
+"""
+Exécution des tests du script.
+"""
+
+import unittest
+
+import main
+
+
+def test_aide():
+    """
+    On essaie d'afficher l'aide, et on vérifie si ça affiche la bonne chose.
+    """
+    res = main.commande("aide")
+    lines = res.split("\n")
+    assert len(lines) == 6
+    assert lines[0].startswith("aide")
+    assert lines[1].startswith("seminaire")
+    assert lines[2].startswith("blague")
+    assert lines[3].startswith("calcul")
+    assert lines[4].startswith("tri")
+    assert lines[5].startswith("stop")