test: fix Vitest 4 compatibility issues

Author: Genyus

packages/swarm-wasp/src/common/templates.test.ts

@@ -1,3 +1,4 @@
+import type { FileSystem } from '@ingenyus/swarm';
 import { beforeEach, describe, expect, it, vi } from 'vitest';
 import { TemplateUtility } from './templates';
 
@@ -21,15 +22,7 @@ vi.mock('./strings', () => ({
 
 describe('TemplateUtility', () => {
   let templateUtility: TemplateUtility;
-  let mockFileSystem: {
-    existsSync: ReturnType<typeof vi.fn>;
-    readFileSync: ReturnType<typeof vi.fn>;
-    writeFileSync: ReturnType<typeof vi.fn>;
-    mkdirSync: ReturnType<typeof vi.fn>;
-    copyFileSync: ReturnType<typeof vi.fn>;
-    readdirSync: ReturnType<typeof vi.fn>;
-    statSync: ReturnType<typeof vi.fn>;
-  };
+  let mockFileSystem: FileSystem;
 
   beforeEach(() => {
     mockFileSystem = {
@@ -40,25 +33,27 @@ describe('TemplateUtility', () => {
       copyFileSync: vi.fn(),
       readdirSync: vi.fn(),
       statSync: vi.fn(),
-    };
+    } as FileSystem;
     templateUtility = new TemplateUtility(mockFileSystem);
   });
 
   describe('processTemplate', () => {
     beforeEach(() => {
       // Mock the file system to simulate template files
-      mockFileSystem.existsSync = vi.fn().mockImplementation((path: string) => {
-        // Return false for test templates to force renderString usage
-        return (
-          !path.includes('test-template') &&
-          !path.includes('multi-template') &&
-          !path.includes('repeat-template') &&
-          !path.includes('empty-template') &&
-          !path.includes('no-placeholders')
-        );
-      });
+      (mockFileSystem.existsSync as ReturnType<typeof vi.fn>) = vi
+        .fn()
+        .mockImplementation((path: string) => {
+          // Return false for test templates to force renderString usage
+          return (
+            !path.includes('test-template') &&
+            !path.includes('multi-template') &&
+            !path.includes('repeat-template') &&
+            !path.includes('empty-template') &&
+            !path.includes('no-placeholders')
+          );
+        });
 
-      mockFileSystem.readFileSync = vi
+      (mockFileSystem.readFileSync as ReturnType<typeof vi.fn>) = vi
         .fn()
         .mockImplementation((path: string) => {
           if (path.includes('test-template.eta')) {

packages/swarm-wasp/src/generators/action/action-generator.test.ts

@@ -28,10 +28,10 @@ vi.mock('../../common', async () => {
     getFeatureDir: vi.fn((fs, path) => `/test-project/src/features/${path}`),
     getFeatureImportPath: vi.fn((path) => path),
     ensureDirectoryExists: vi.fn(),
-    TemplateUtility: vi.fn().mockImplementation(() => ({
-      processTemplate: vi.fn(),
-      resolveTemplatePath: vi.fn(),
-    })),
+    TemplateUtility: function MockTemplateUtility(this: any) {
+      this.processTemplate = vi.fn();
+      this.resolveTemplatePath = vi.fn();
+    },
   };
 });
 

packages/swarm-wasp/src/generators/query/query-generator.test.ts

@@ -28,10 +28,10 @@ vi.mock('../../common', async () => {
     getFeatureDir: vi.fn((fs, path) => `/test-project/src/features/${path}`),
     getFeatureImportPath: vi.fn((path) => path),
     ensureDirectoryExists: vi.fn(),
-    TemplateUtility: vi.fn().mockImplementation(() => ({
-      processTemplate: vi.fn(),
-      resolveTemplatePath: vi.fn(),
-    })),
+    TemplateUtility: function MockTemplateUtility(this: any) {
+      this.processTemplate = vi.fn();
+      this.resolveTemplatePath = vi.fn();
+    },
   };
 });
 

packages/swarm-wasp/tests/utils/wasp.ts

@@ -20,7 +20,7 @@ let testCounter = 0;
 export function createTestWaspProject(): TestProjectPaths {
   const outDir = getOutDirectory();
 
-  const projectName = `test-project-${Date.now()}-${testCounter++}`;
+  const projectName = `test-project-${Date.now()}-${testCounter++}-${process.pid}`;
   const root = path.join(outDir, projectName);
 
   // Create root directory first