add: Tests unitaires pour ChaCha20_Analyzer

Author: mouwaficbdr

tests/test_analyzers.py

@@ -1,14 +1,20 @@
 from unittest import TestCase, main
+import os
 import sys
-sys.path.append('.')
-sys.path.append('..')
+import hashlib
+from cryptography.hazmat.primitives.ciphers import Cipher, algorithms
+
+sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
+sys.path.append(os.path.join(os.path.dirname(__file__), '..', '..'))
+
 from src.analyzers.aes_cbc_analyzer import Aes_Cbc_Analyzer
+from src.analyzers.chacha20_analyzer import ChaCha20_Analyzer
 
-class AnalyzersTester(TestCase):
 
+class AesCbcAnalyzerTester(TestCase):
     """
-        Cette classe est principalement destinée à recueillir toutes les fonctions de test des analyseurs d'algorithme
-        de chiffrement.
+    Cette classe est principalement destinée à recueillir toutes les fonctions de test des analyseurs d'algorithme
+    de chiffrement.
     """
 
     def setUp(self):
@@ -40,6 +46,61 @@ def test_exception_dechiffrer(self):
 
         with self.assertRaises(FileNotFoundError):
             self.analyser.dechiffrer("no_file_dohi.txt", premiere_cle)
+            
+class ChaCha20AnalyzerTester(TestCase):
+
+    def setUp(self):
+        # Chemins pour ChaCha20_Analyzer
+        self.wordlist = "keys/wordlist.txt"
+        self.analyser_chacha = ChaCha20_Analyzer()
+
+        # Données de test pour ChaCha20
+        self.cle_test_chacha = hashlib.sha256(b"cle_test").digest()
+        self.nonce_test_chacha = b"\x00" * 12
+        self.texte_clair_test_chacha = b"Bonjour le monde, ceci est un test de chiffrement ChaCha20"
+        self.chemin_fichier_chacha_valide = "tests/fichiers_pour_tests/chacha20_valide.enc"
+        self.chemin_fichier_chacha_invalide = "tests/fichiers_pour_tests/chacha20_invalide.enc"
 
+        # Générer un fichier chiffré valide pour les tests de ChaCha20
+        cipher_chacha = Cipher(algorithms.ChaCha20(self.cle_test_chacha, self.nonce_test_chacha), mode=None)
+        encryptor = cipher_chacha.encryptor()
+        texte_chiffre_test = encryptor.update(self.texte_clair_test_chacha)
+        with open(self.chemin_fichier_chacha_valide, "wb") as f:
+            f.write(self.nonce_test_chacha)
+            f.write(texte_chiffre_test)
+            
+    # Ajout des tests pour ChaCha20_Analyzer
+    def test_chacha20_identifier_algo(self):
+        self.assertAlmostEqual(self.analyser_chacha.identifier_algo(self.chemin_fichier_chacha_valide), 1.0, 1)
+        self.assertAlmostEqual(self.analyser_chacha.identifier_algo(self.chemin_fichier_chacha_invalide), 0.0, 1)
+
+    def test_chacha20_generer_cles_candidates(self):
+        # Comme la fonction filtrer_dictionnaire_par_indices retourne toujours une liste vide,
+        # generer_cles_candidates doit également retourner une liste vide.
+        self.assertEqual(self.analyser_chacha.generer_cles_candidates(self.wordlist), [])
+
+    def test_chacha20_dechiffrer(self):
+        # Test de déchiffrement avec une clé et un nonce valides
+        resultat_dechiffrement = self.analyser_chacha.dechiffrer(self.chemin_fichier_chacha_valide, self.cle_test_chacha)
+        self.assertEqual(resultat_dechiffrement, self.texte_clair_test_chacha)
+
+        # Test de déchiffrement avec une clé incorrecte
+        cle_incorrecte = hashlib.sha256(b"mauvaise_cle").digest()
+        resultat_incorrect = self.analyser_chacha.dechiffrer(self.chemin_fichier_chacha_valide, cle_incorrecte)
+        self.assertNotEqual(resultat_incorrect, self.texte_clair_test_chacha)
+
+    def test_chacha20_dechiffrer_mauvaise_cle(self):
+        # Test de l'exception pour une clé de taille incorrecte
+        cle_mauvaise_taille = b"a" * 16 # La bonne taille est 32
+        with self.assertRaises(ValueError):
+            self.analyser_chacha.dechiffrer(self.chemin_fichier_chacha_valide, cle_mauvaise_taille)
+
+    def test_chacha20_dechiffrer_fichier_non_existant(self):
+        # Test de l'exception si le fichier n'existe pas
+        cle_valide = self.cle_test_chacha
+        with self.assertRaises(FileNotFoundError):
+            self.analyser_chacha.dechiffrer("chemin_invalide.enc", cle_valide)
+            
+            
 if __name__ == '__main__':
     main()