edit models' blueprint & test

Author: SpliiT

src/opengeodeweb_back/routes/models/blueprint_models.py

@@ -54,22 +54,13 @@ def uuid_to_flat_index():
 
 def extract_model_uuids(geode_object, file_path):
     model = geode_functions.load(geode_object, file_path)
-    components = {
-        "blocks": getattr(model, "blocks", lambda: [])(),
-        "lines": getattr(model, "lines", lambda: [])(),
-        "surfaces": getattr(model, "surfaces", lambda: [])(),
-        "corners": getattr(model, "corners", lambda: [])(),
-    }
-
-    uuid_dict = {
-        key: [
-            component.id().string()
-            for component in components[key]
-            if hasattr(component, "id")
-        ]
-        for key in components
-        if components[key]
-    }
+    mesh_components = model.mesh_components()
+
+    uuid_dict = {}
+
+    for mesh_component, ids in mesh_components.items():
+        component_name = mesh_component.get()
+        uuid_dict[component_name] = [id.string() for id in ids]
 
     print(f"{uuid_dict=}", flush=True)
     return uuid_dict

tests/test_models_routes.py

@@ -28,9 +28,8 @@ def test_extract_brep_uuids(client):
     uuid_dict = response.json
     assert isinstance(uuid_dict, dict)
     assert (
-        "blocks" in uuid_dict
-        or "lines" in uuid_dict
-        or "surfaces" in uuid_dict
-        or "corners" in uuid_dict
+        "Block" in uuid_dict
+        or "Line" in uuid_dict
+        or "Surface" in uuid_dict
+        or "Corner" in uuid_dict
     )
-    assert False