vitest.config.js

import { defineConfig } from 'vitest/config';
import { playwright } from '@vitest/browser-playwright';
import path from 'path';
import nodePolyfills from '@rolldown/plugin-node-polyfills';
import { createVtkPlugins } from './Utilities/build/plugins.mjs';

const noWebGL = !!process.env.NO_WEBGL;
const webGPU = !!process.env.WEBGPU;
const testBrowser = process.env.TEST_BROWSER || 'chromium';
const ci = !!process.env.CI;

const firefoxUserPrefs = {
  'dom.webgpu.enabled': true, // off by default on Linux Firefox
  'webgl.force-enabled': true, // override GPU blocklist (no real GPU on CI)
  'webgl.disable-fail-if-major-performance-caveat': true, // accept llvmpipe
};

const chromiumArgs = ci
  ? ['--no-sandbox', '--enable-unsafe-swiftshader', '--use-angle=swiftshader']
  : [];

if (webGPU) {
  // Headless Chromium does not expose a WebGPU adapter by default.
  chromiumArgs.push('--enable-unsafe-webgpu');
  if (ci) {
    chromiumArgs.push(
      '--enable-features=Vulkan', // Use Vulkan for the compositor.
      '--use-vulkan=swiftshader' // Back Vulkan with SwiftShader on GPU-less CI.
    );
  }
}

const firefox = {
  browser: 'firefox',
  headless: false, // supported Vitest Browser option; xvfb-run supplies the display on CI
};

function buildBrowserInstances() {
  if (testBrowser === 'firefox') return [firefox];
  return [{ browser: 'chromium' }];
}

function buildPlaywrightLaunchOptions() {
  if (testBrowser === 'firefox') return { firefoxUserPrefs };
  return chromiumArgs.length ? { args: chromiumArgs } : {};
}

export default defineConfig({
  resolve: {
    alias: {
      'vtk.js': path.resolve(import.meta.dirname),
    },
  },
  optimizeDeps: {
    include: ['webworker-promise/lib/register'],
    rolldownOptions: {
      plugins: [nodePolyfills()],
    },
  },
  css: {
    modules: {
      localsConvention: 'camelCaseOnly',
    },
  },
  define: {
    __BASE_PATH__: JSON.stringify(''),
    __VTK_TEST_NO_WEBGL__: JSON.stringify(noWebGL),
    __VTK_TEST_WEBGPU__: JSON.stringify(webGPU),
    global: 'globalThis',
  },
  plugins: createVtkPlugins({
    includeCjson: true,
    includeStaticData: true,
    staticDataRootDir: import.meta.dirname,
  }),
  test: {
    include: ['Sources/**/test*.js'],
    exclude: [
      '**/node_modules/**',
      '**/dist/**',
      'Sources/Testing/testUtils.js',
      'Sources/Testing/setupTestEnv.js',
    ],
    setupFiles: ['Sources/Testing/setupTestEnv.js'],
    testTimeout: 120000,
    reporters: ['default', 'junit'],
    outputFile: { junit: 'Utilities/TestResults/junit-report.xml' },
    fileParallelism: false, // GPU tests should run sequentially
    maxWorkers: 1, // Single worker for GPU resource management
    retry: ci ? 1 : 0,
    allowOnly: !ci,
    browser: {
      enabled: true,
      headless: true,
      screenshotDirectory: 'Utilities/TestResults/screenshots',
      provider: playwright({
        launchOptions: buildPlaywrightLaunchOptions(),
      }),
      instances: buildBrowserInstances(),
    },
  },
});