import { describe, it, expect } from 'vitest';

import { getResolvedTestFeatures } from '@haibun/core/build/lib/test/resolvedTestFeatures.js';

// Setup jsdom and real DOMPurify for mermaid parsing in Node.js
defineGlobalDOMPurify();

import { generateMermaidGraph } from './generateMermaidGraph.js';

describe('generateMermaidGraph', () => {
	it('should generate a comprehensive Mermaid graph', async () => {
		const features = [{
			path: '/feature-1.feature',
			content: `Backgrounds: background-1\nScenario: Feature one scenario\ngwta1\nis {env_var1}`,
		},
		{
			path: '/feature-2.feature',
			content: `Backgrounds: background-1\nScenario: Feature two scenario\ngwta2\nis var_2\nScenario: Another scenario\ngwta3\nis "var_3"`, // Uses scenario var and quoted scenario var
		},
		];
		const backgrounds = [{
			path: '/background-1.feature',
			content: `is "the background"`,
		}];
		const resolvedFeatures = await getResolvedTestFeatures(features, backgrounds);

		const lines = await generateMermaidGraph(resolvedFeatures);

		const graphContent = lines.join('\n');
		try {
			const mermaid = await import('mermaid');
			await mermaid.default.parse(graphContent);
		} catch (e) {
			console.error('Mermaid parse error:', e);
			throw e;
		}

	});
});

function defineGlobalDOMPurify() {
	if (typeof window === 'undefined' || !(globalThis as Record<string, unknown>).DOMPurify) {
		// Use require to get the CommonJS build for Node.js compatibility
		// eslint-disable-next-line @typescript-eslint/no-var-requires
		const { JSDOM } = require('jsdom');
		// eslint-disable-next-line @typescript-eslint/no-var-requires
		const createDOMPurify = require('dompurify');
		const { window } = new JSDOM('<!DOCTYPE html>');
		global.window = window;
		global.document = window.document;
		global.navigator = window.navigator;
		// The CJS build returns the correct instance for mermaid
		const domPurifyInstance = createDOMPurify(window);
		global.DOMPurify = domPurifyInstance;
		window.DOMPurify = domPurifyInstance;
	}
}