refact test_config()

Author: MaxNumerique

src/opengeodeweb_viewer/config.py

@@ -32,48 +32,34 @@ def test_config(path):
     tmp_data_root = tempfile.mkdtemp(prefix="ogw_test_data_")
     os.environ["DATA_FOLDER_PATH"] = tmp_data_root
 
-    original_data_path = os.path.join(path, "data")
-    if not os.path.exists(original_data_path):
-        raise FileNotFoundError(f"Test data folder not found: {original_data_path}")
-
-    valid_extensions = {".vtp", ".vti", ".vtu", ".vtm", ".png", ".jpeg", ".jpg"}
+    src_data = os.path.join(path, "data")
+    if not os.path.isdir(src_data):
+        raise FileNotFoundError(f"Test data folder not found: {src_data}")
 
     test_ids = ["123456789", "12345678"]
+    valid_exts = {".vtp", ".vti", ".vtu", ".vtm", ".png", ".jpeg", ".jpg"}
 
-    for test_id in test_ids:
-        test_id_dir = os.path.join(tmp_data_root, test_id)
-        os.makedirs(test_id_dir, exist_ok=True)
-
-    test_project_uuid = "test-project-uuid"
-    test_data_uuid = "test-data-uuid"
-    new_structure_dir = os.path.join(tmp_data_root, test_project_uuid, test_data_uuid)
-    os.makedirs(new_structure_dir, exist_ok=True)
+    project_uuid = "test-project-uuid"
+    data_uuid = "test-data-uuid"
+    uploads_directory = os.path.join(tmp_data_root, project_uuid, "uploads")
+    structure_directory = os.path.join(tmp_data_root, project_uuid, data_uuid)
 
-    uploads_dir = os.path.join(tmp_data_root, test_project_uuid, "uploads")
-    os.makedirs(uploads_dir, exist_ok=True)
+    for directory in [*test_ids, uploads_directory, structure_directory]: # create directories for tests
+        os.makedirs(os.path.join(tmp_data_root, directory) if isinstance(directory, str) else directory, exist_ok=True)
 
-    for root, dirs, files in os.walk(original_data_path):
-        for d in dirs:
-            src_dir = os.path.join(root, d)
-            dst_dir = os.path.join(tmp_data_root, test_ids[0], d)
-            if not os.path.exists(dst_dir):
-                copytree(src_dir, dst_dir, dirs_exist_ok=True)
+    for root, directories, files in os.walk(src_data): 
+        for directory in directories:
+            dst = os.path.join(tmp_data_root, test_ids[0], directory)
+            copytree(os.path.join(root, directory), dst, dirs_exist_ok=True)
 
-        for file_name in files:
-            ext = os.path.splitext(file_name)[1].lower()
-            if ext not in valid_extensions:
+        for file in files:
+            if os.path.splitext(file)[1].lower() not in valid_exts:
                 continue
 
-            full_path = os.path.join(root, file_name)
-
+            src = os.path.join(root, file)
             for test_id in test_ids:
-                test_id_dst = os.path.join(tmp_data_root, test_id, file_name)
-                copyfile(full_path, test_id_dst)
-
-            new_structure_dst = os.path.join(new_structure_dir, file_name)
-            copyfile(full_path, new_structure_dst)
-
-            uploads_dst = os.path.join(uploads_dir, file_name)
-            copyfile(full_path, uploads_dst)
+                copyfile(src, os.path.join(tmp_data_root, test_id, file))
+            copyfile(src, os.path.join(structure_directory, file))
+            copyfile(src, os.path.join(uploads_directory, file))
 
-    print(f"\n✅ DATA_FOLDER_PATH set to: {tmp_data_root}", flush=True)
+    print(f"\nDATA_FOLDER_PATH set to: {tmp_data_root}", flush=True)