test modified to fit with new folder_path (refacto needed)

Author: MaxNumerique

.gitignore

@@ -9,4 +9,5 @@ __pycache__
 .vscode
 uploads
 node_modules
-schemas.json
+schemas.json
+tmp

tests/conftest.py

@@ -1,9 +1,11 @@
 # Standard library imports
 import time
 import shutil
+import os
 
 # Third party imports
 import pytest
+import uuid
 
 # Local application imports
 from app import app
@@ -31,3 +33,23 @@ def client():
 def app_context():
     with app.app_context():
         yield
+
+@pytest.fixture
+def uuid_project_structure():
+    uuid_project = uuid.uuid4().hex
+    uuid_data = uuid.uuid4().hex
+
+    base_path = os.path.join("tmp", "vease", uuid_project)
+    data_path = os.path.join(base_path, uuid_data)
+    uploads_path = os.path.join(base_path, "uploads")
+
+    os.makedirs(data_path, exist_ok=True)
+    os.makedirs(uploads_path, exist_ok=True)
+
+    return {
+        "uuid_project": uuid_project,
+        "uuid_data": uuid_data,
+        "base_path": base_path,
+        "data_path": data_path,
+        "uploads_path": uploads_path,
+    }

tests/test_models_routes.py

@@ -1,30 +1,58 @@
-def test_model_mesh_components(client):
-    route = f"/models/vtm_component_indices"
-    get_full_data = lambda: {"id": "cube"}
-    json = get_full_data()
-    response = client.post(route, json=json)
-    assert response.status_code == 200
+import shutil
+import os
 
-    uuid_dict = response.json["uuid_to_flat_index"]
-    assert isinstance(uuid_dict, dict)
+def test_model_mesh_components(client, uuid_project_structure):
+    route = "/models/vtm_component_indices"
+    uuid_data = uuid_project_structure["uuid_data"]
+    data_path = uuid_project_structure["data_path"]
 
-    indices = list(uuid_dict.values())
-    indices.sort()
-    assert all(indices[i] > indices[i - 1] for i in range(1, len(indices)))
-    for uuid in uuid_dict.keys():
-        assert isinstance(uuid, str)
+    shutil.copy("./tests/data/cube.vtm", os.path.join(data_path, "viewable.vtm"))
 
+    original_path = client.application.config["DATA_FOLDER_PATH"]
+    client.application.config["DATA_FOLDER_PATH"] = uuid_project_structure["base_path"]
 
-def test_extract_brep_uuids(client):
+    try:
+        response = client.post(route, json={"id": uuid_data})
+        assert response.status_code == 200
+        uuid_dict = response.json["uuid_to_flat_index"]
+        assert isinstance(uuid_dict, dict)
+
+        indices = list(uuid_dict.values())
+        indices.sort()
+        assert all(indices[i] > indices[i - 1] for i in range(1, len(indices)))
+        for uuid_key in uuid_dict.keys():
+            assert isinstance(uuid_key, str)
+    finally:
+        client.application.config["DATA_FOLDER_PATH"] = original_path
+
+
+
+def test_extract_brep_uuids(client, uuid_project_structure):
     route = "/models/mesh_components"
-    json_data = {"filename": "cube.og_brep", "geode_object": "BRep"}
-    response = client.post(route, json=json_data)
-
-    assert response.status_code == 200
-    uuid_dict = response.json["uuid_dict"]
-    assert isinstance(uuid_dict, dict)
-    expected_keys = {"Block", "Line", "Surface", "Corner"}
-    assert any(key in uuid_dict for key in expected_keys)
-    for key, value in uuid_dict.items():
-        assert isinstance(value, list)
-        assert all(isinstance(v, str) for v in value)
+    uuid_data = uuid_project_structure["uuid_data"]
+    data_path = uuid_project_structure["data_path"]
+
+    shutil.copy("./tests/data/cube.og_brep", os.path.join(data_path, "cube.og_brep"))
+
+    original_path = client.application.config["DATA_FOLDER_PATH"]
+    client.application.config["DATA_FOLDER_PATH"] = uuid_project_structure["base_path"]
+
+    try:
+        json_data = {
+            "filename": "cube.og_brep",
+            "geode_object": "BRep",
+            "id": uuid_data
+        }
+        response = client.post(route, json=json_data)
+        assert response.status_code == 200
+
+        uuid_dict = response.json["uuid_dict"]
+        assert isinstance(uuid_dict, dict)
+        expected_keys = {"Block", "Line", "Surface", "Corner"}
+        assert any(key in uuid_dict for key in expected_keys)
+        for key, value in uuid_dict.items():
+            assert isinstance(value, list)
+            assert all(isinstance(v, str) for v in value)
+    finally:
+        client.application.config["DATA_FOLDER_PATH"] = original_path
+

tests/test_utils_functions.py

@@ -71,22 +71,22 @@ def test_handle_exception(client):
     assert type(data["code"]) is int
 
 
-def test_generate_native_viewable_and_light_viewable():
-    geode_object = "BRep"
-    folder_absolute_path = os.path.abspath("./data")
-    data = geode_functions.load(
-        geode_object, os.path.join(folder_absolute_path, "test.og_brep")
-    )
-    folder_absolute_path = "None"
-    result = utils_functions.generate_native_viewable_and_light_viewable(
-        geode_object, data, folder_absolute_path
-    )
+def test_generate_native_viewable_and_light_viewable(client):
+    app = client.application
+    with app.app_context():
+        geode_object = "BRep"
+        data = geode_functions.load(geode_object, "./tests/data/test.og_brep")
+
+        result = utils_functions.generate_native_viewable_and_light_viewable(
+            geode_object, data
+        )
+
     assert type(result) is dict
     assert type(result["name"]) is str
     assert type(result["native_file_name"]) is str
-    assert re.match(r"[0-9a-f]{32}\.[a-zA-Z0-9]+", result["native_file_name"])
+    assert result["native_file_name"] == "native.og_brep"
     assert type(result["viewable_file_name"]) is str
-    assert re.match(r"[0-9a-f]{32}\.[a-zA-Z0-9]+", result["viewable_file_name"])
+    assert result["viewable_file_name"] == "viewable.vtm"
     assert type(result["id"]) is str
     assert re.match(r"[0-9a-f]{32}", result["id"])
     assert type(result["object_type"]) is str