more tests and bugfix

cupp.py

@@ -276,7 +276,7 @@ def improve_dictionary(file_to_open):
 
     unique_list_finished = [x for x in unique_list if len(x) > wcfrom and len(x) < wcto]
 
-    print_to_file(sys.argv[2] + ".cupp.txt", unique_list_finished)
+    print_to_file(file_to_open + ".cupp.txt", unique_list_finished)
 
     fajl.close()
 

test_cupp.py

@@ -70,7 +70,6 @@ def test_parser(self):
         download_wordlist_http("16")
 
         filename = "dictionaries/hindi/hindu-names.gz"
-
         self.assertTrue(os.path.isfile(filename), "file " + filename + "exists")
 
     def test_print_cow(self):
@@ -90,6 +89,21 @@ def test_alectodb_download(self):
             "file alectodb-passwords.txt exists",
         )
 
+    def test_improve_dictionary(self):
+
+        filename = "improveme.txt"
+        open(filename, "a").write("password123\n2018password\npassword\n")
+
+        __builtins__.input = lambda _: "Y"  # Mock
+        improve_dictionary(filename)
+
+    def test_download_wordlist(self):
+        """ Download wordlists via menu """
+        __builtins__.input = lambda _: "31"  # Mock
+        download_wordlist()
+        filename = "dictionaries/russian/russian.lst.gz"
+        self.assertTrue(os.path.isfile(filename), "file " + filename + "exists")
+
 
 if __name__ == "__main__":
     unittest.main()