seminaire-ci/main_test.py

#!/usr/bin/env python3

"""
Exécution des tests du script.
"""

import unittest

import main


class TestMain(unittest.TestCase):
    """
    Cette classe permet d'executer l'ensemble des scripts
    """

    def test_aide(self):
        """
        On essaie d'afficher l'aide, et on vérifie si ça affiche la bonne chose.
        """
        res = main.commande("aide")
        lines = res.split("\n")
        self.assertEqual(len(lines), 6)
        self.assertTrue(lines[0].startswith("aide"))
        self.assertTrue(lines[1].startswith("seminaire"))
        self.assertTrue(lines[2].startswith("blague"))
        self.assertTrue(lines[3].startswith("calcul"))
        self.assertTrue(lines[4].startswith("tri"))
        self.assertTrue(lines[5].startswith("stop"))