update e2e tests

Author: JulienChampagnol

tests/test_functions.py

@@ -1,4 +1,5 @@
 import os
+import uuid
 from src.opengeodeweb_back import geode_functions, geode_objects
 
 
@@ -50,44 +51,75 @@ def test_missing_files():
             assert type(additional_files) is list
 
 
-def test_load():
+def test_geode_objects_output_extensions():
     for geode_object, value in geode_objects.geode_objects_dict().items():
-        print(f"{geode_object=}")
-        input_extensions = geode_functions.geode_object_input_extensions(geode_object)
-        missing_files = geode_functions.missing_files(
-            geode_object, f"tests/data/test.{input_extension}"
-        )
-        has_missing_files = missing_files.has_missing_files()
-        if has_missing_files:
-            mandatory_files = geode_functions.mandatory_files(missing_files)
-            print(f"{mandatory_files=}")
-            additional_files = geode_functions.additional_files(missing_files)
-            print(f"{additional_files=}")
-        file_apsolute_path = os.path.abspath(f"tests/data/test.{input_extension}")
-
-        data = geode_functions.load(geode_object, file_apsolute_path)
-        output_extensions = geode_functions.get_geode_object_output_extensions(
-            geode_object
+        geode_objects_and_output_extensions_list = (
+            geode_functions.geode_objects_output_extensions(geode_object)
         )
+        assert type(geode_objects_and_output_extensions_list) is list
+        for (
+            geode_object_and_output_extensions
+        ) in geode_objects_and_output_extensions_list:
+            assert type(geode_object_and_output_extensions) is dict
+            assert "geode_object" in geode_object_and_output_extensions
+            assert "output_extension" in geode_object_and_output_extensions
 
-        for output_extension in output_extensions:
-            print(f"{output_extension=}")
-            uu_id = str(uuid.uuid4()).replace("-", "")
-            geode_functions.save(
-                geode_object,
-                data,
-                os.path.abspath(f"output"),
-                f"{uu_id}.{output_extension}",
-            )
 
-            if "save_viewable" in value:
-                uu_id = str(uuid.uuid4()).replace("-", "")
-                geode_functions.save_viewable(
-                    geode_object,
-                    data,
-                    os.path.abspath(f"output"),
-                    uu_id,
+def test_load():
+    for geode_object, value in geode_objects.geode_objects_dict().items():
+        print(f"\n{geode_object=}")
+        input_extensions = geode_functions.geode_object_input_extensions(geode_object)
+        for input_extension in input_extensions:
+            if input_extension not in ["grdecl", "og_img3d", "vti", "vo"]:
+                print(f"\t{input_extension=}")
+                missing_files = geode_functions.missing_files(
+                    geode_object, f"tests/data/test.{input_extension}"
                 )
+                has_missing_files = missing_files.has_missing_files()
+                if has_missing_files:
+                    mandatory_files = missing_files.mandatory_files
+                    print(f"\t\t{mandatory_files=}")
+                    additional_files = missing_files.additional_files
+                    print(f"\t\t{additional_files=}")
+                file_apsolute_path = os.path.abspath(
+                    f"tests/data/test.{input_extension}"
+                )
+
+                data = geode_functions.load(geode_object, file_apsolute_path)
+                geode_functions.geode_objects_output_extensions(geode_object)
+                for (
+                    geode_object_and_output_extensions
+                ) in geode_functions.geode_objects_output_extensions(geode_object):
+                    output_geode_object = geode_object_and_output_extensions[
+                        "geode_object"
+                    ]
+                    print(f"\t\t{output_geode_object=}")
+                    output_extensions = geode_functions.geode_object_output_extensions(
+                        geode_object
+                    )
+
+                    for output_extension in output_extensions:
+                        print(f"\t\t\t{output_extension=}")
+                        uu_id = str(uuid.uuid4()).replace("-", "")
+                        filename = f"{uu_id}.{output_extension}"
+                        if geode_functions.is_saveable(geode_object, data, filename):
+                            file_path = geode_functions.save(
+                                geode_object,
+                                data,
+                                os.path.abspath(f"output"),
+                                f"{uu_id}.{output_extension}",
+                            )
+
+                            if "save_viewable" in value:
+                                uu_id = str(uuid.uuid4()).replace("-", "")
+                                viewable_file_path = geode_functions.save_viewable(
+                                    geode_object,
+                                    data,
+                                    os.path.abspath(f"output"),
+                                    uu_id,
+                                )
+
+                                os.remove(viewable_file_path)
 
 
 def test_is_model():