Updated tests

search_that_hash/config_object.py

@@ -1,4 +1,3 @@
-import toml
 from appdirs import *
 import json
 from loguru import logger
@@ -8,6 +7,7 @@
 
 import logging
 import coloredlogs
+import os.path
 
 
 def cli_config(kwargs):
@@ -40,7 +40,7 @@ def api_config(hashes: str, sth_api: str = None):
 
 
 def default_config():
-    return {
+    config = {
         "api_keys": {"STH": "rGFbPbSXMF5ldzid2eyA81i6aCa497Z25MNgi8sa"},
         "hashcat": False,
         "api": False,
@@ -52,6 +52,33 @@ def default_config():
         "offline": False,
     }
 
+    defults = {
+
+        "hashes_dot_org":"test",
+        "sth_api":"rGFbPbSXMF5ldzid2eyA81i6aCa497Z25MNgi8sa",
+        "hashcat_exe_name":"hashcat",
+        "hashcat_folder":""
+
+    }
+
+    appname = "Search-That-Hash"
+    appauthor = "HashPals"
+
+    config_json = user_data_dir(appname, appauthor) + "\\config.json"
+
+    if not os.path.isfile(config_json):
+        os.makedirs(user_data_dir(appname, appauthor))
+        with open(config_json, "w+") as file:
+            file.write(json.dumps(defults))
+            file.close()
+
+    with open(config_json) as json_file:
+        json_contents = json.load(json_file)
+        json_file.close()
+
+    config.update(json_contents)
+
+    return config
 
 def create_hash_config(hashes):
     # Gets the results from name-that-hash

tests/main_test.py

@@ -1,7 +1,8 @@
 from search_that_hash import api
 from click.testing import CliRunner
-
+from appdirs import *
 from search_that_hash.__main__ import main
+import os.path
 
 
 def test_it_works():
@@ -150,3 +151,11 @@ def test_cli_fail_on_grep():  # Fixes #63 issue
         ["-t", "jadjsjhd9239uh80dahjdah8isdh90wq90hj0j9fj9023j0-12j-j-0fasj0a", "-g"],
     )
     assert result.exit_code == 0
+
+def test_config_is_there():
+    appname = "Search-That-Hash"
+    appauthor = "HashPals"
+
+    config_json = user_data_dir(appname, appauthor) + "\\config.json"
+
+    assert os.path.isfile(config_json) == True