From eba6392983192cc16ba4e73e90aec991bd8c6f92 Mon Sep 17 00:00:00 2001
From: Victor Giers <mail@victorgiers.com>
Date: Sun, 3 May 2026 13:22:50 +0200
Subject: [PATCH] Improve mock implementation for foliage model loader in tests

---
 .../domain/foliage-instanced-renderer.test.ts | 27 ++++++++++++-------
 1 file changed, 17 insertions(+), 10 deletions(-)

diff --git a/tests/domain/foliage-instanced-renderer.test.ts b/tests/domain/foliage-instanced-renderer.test.ts
index b5992097..34afcf2e 100644
--- a/tests/domain/foliage-instanced-renderer.test.ts
+++ b/tests/domain/foliage-instanced-renderer.test.ts
@@ -19,18 +19,25 @@ const loaderState = vi.hoisted(() => ({
   loadCalls: [] as string[]
 }));
 
-vi.mock("../../src/foliage/bundled-foliage-model-loader", () => ({
-  loadBundledFoliageModelTemplate: async (bundledPath: string) => {
-    loaderState.loadCalls.push(bundledPath);
+vi.mock("../../src/foliage/bundled-foliage-model-loader", async () => {
+  const three = await vi.importActual<typeof import("three")>("three");
 
-    const template = new Group();
-    template.add(
-      new Mesh(new BoxGeometry(1, 1, 1), new MeshBasicMaterial())
-    );
+  return {
+    loadBundledFoliageModelTemplate: async (bundledPath: string) => {
+      loaderState.loadCalls.push(bundledPath);
 
-    return template;
-  }
-}));
+      const template = new three.Group();
+      template.add(
+        new three.Mesh(
+          new three.BoxGeometry(1, 1, 1),
+          new three.MeshBasicMaterial()
+        )
+      );
+
+      return template;
+    }
+  };
+});
 
 import { FoliageInstancedRenderer } from "../../src/foliage/foliage-instanced-renderer";