From 39c08e94f198db6c6628593c8103a025893915c9 Mon Sep 17 00:00:00 2001
From: Valentin Palkovic <valentin@chromatic.com>
Date: Wed, 8 Jan 2025 15:38:47 +0100
Subject: [PATCH] Refactor extension handling for workspace file

---
 code/addons/test/src/node/vitest-manager.ts | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/code/addons/test/src/node/vitest-manager.ts b/code/addons/test/src/node/vitest-manager.ts
index 8696126fd920..5e23088dfcbb 100644
--- a/code/addons/test/src/node/vitest-manager.ts
+++ b/code/addons/test/src/node/vitest-manager.ts
@@ -25,6 +25,7 @@ import { StorybookReporter } from './reporter';
 import type { TestManager } from './test-manager';
 
 const VITEST_CONFIG_FILE_EXTENSIONS = ['mts', 'mjs', 'cts', 'cjs', 'ts', 'tsx', 'js', 'jsx'];
+const VITEST_WORKSPACE_FILE_EXTENSION = ['ts', 'js', 'json'];
 
 type TagsFilter = {
   include: string[];
@@ -72,9 +73,7 @@ export class VitestManager {
     ) as CoverageOptions;
 
     const vitestWorkspaceConfig = await findUp([
-      'vitest.workspace.ts',
-      'vitest.workspace.js',
-      'vitest.workspace.json',
+      ...VITEST_WORKSPACE_FILE_EXTENSION.map((ext) => `vitest.workspace.${ext}`),
       ...VITEST_CONFIG_FILE_EXTENSIONS.map((ext) => `vitest.config.${ext}`),
     ]);