enhance: optimization from comment on PR #1852

Author: fselmo

packages/testing/src/execution_testing/client_clis/file_utils.py

@@ -7,7 +7,7 @@
 
 from pydantic import BaseModel, RootModel
 
-from execution_testing.client_clis.cli_types import LazyAlloc
+from execution_testing.client_clis.cli_types import LazyAllocJson, LazyAllocStr
 
 
 def dump_files_to_directory(output_path: str, files: Dict[str, Any]) -> None:
@@ -24,15 +24,12 @@ def dump_files_to_directory(output_path: str, files: Dict[str, Any]) -> None:
             os.makedirs(os.path.join(output_path, rel_path), exist_ok=True)
         file_path = os.path.join(output_path, file_rel_path)
         with open(file_path, "w") as f:
-            if isinstance(file_contents, LazyAlloc):
-                # Validate and serialize the lazy alloc
-                f.write(
-                    file_contents.get().model_dump_json(
-                        indent=4,
-                        exclude_none=True,
-                        by_alias=True,
-                    )
-                )
+            if isinstance(file_contents, (LazyAllocStr, LazyAllocJson)):
+                if isinstance(file_contents, LazyAllocJson):
+                    dump(file_contents.raw, f, ensure_ascii=True, indent=4)
+                else:
+                    f.write(file_contents.raw)
+
             elif isinstance(file_contents, BaseModel) or isinstance(
                 file_contents, RootModel
             ):