mirror of
https://github.com/QwenLM/qwen-code.git
synced 2025-12-21 17:27:54 +00:00
100 lines
3.1 KiB
TypeScript
100 lines
3.1 KiB
TypeScript
/**
|
|
* @license
|
|
* Copyright 2025 Google LLC
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
|
|
// Patch: Unset NO_COLOR at the very top before any imports
|
|
if (process.env['NO_COLOR'] !== undefined) {
|
|
delete process.env['NO_COLOR'];
|
|
}
|
|
|
|
import { describe, it, expect, beforeEach } from 'vitest';
|
|
import { themeManager, DEFAULT_THEME } from './theme-manager.js';
|
|
import { CustomTheme } from './theme.js';
|
|
|
|
const validCustomTheme: CustomTheme = {
|
|
type: 'custom',
|
|
name: 'MyCustomTheme',
|
|
Background: '#000000',
|
|
Foreground: '#ffffff',
|
|
LightBlue: '#89BDCD',
|
|
AccentBlue: '#3B82F6',
|
|
AccentPurple: '#8B5CF6',
|
|
AccentCyan: '#06B6D4',
|
|
AccentGreen: '#3CA84B',
|
|
AccentYellow: 'yellow',
|
|
AccentRed: 'red',
|
|
DiffAdded: 'green',
|
|
DiffRemoved: 'red',
|
|
Comment: 'gray',
|
|
Gray: 'gray',
|
|
};
|
|
|
|
describe('ThemeManager', () => {
|
|
beforeEach(() => {
|
|
// Reset themeManager state
|
|
themeManager.loadCustomThemes({});
|
|
themeManager.setActiveTheme(DEFAULT_THEME.name);
|
|
});
|
|
|
|
it('should load valid custom themes', () => {
|
|
themeManager.loadCustomThemes({ MyCustomTheme: validCustomTheme });
|
|
expect(themeManager.getCustomThemeNames()).toContain('MyCustomTheme');
|
|
expect(themeManager.isCustomTheme('MyCustomTheme')).toBe(true);
|
|
});
|
|
|
|
it('should set and get the active theme', () => {
|
|
expect(themeManager.getActiveTheme().name).toBe(DEFAULT_THEME.name);
|
|
themeManager.setActiveTheme('Ayu');
|
|
expect(themeManager.getActiveTheme().name).toBe('Ayu');
|
|
});
|
|
|
|
it('should set and get a custom active theme', () => {
|
|
themeManager.loadCustomThemes({ MyCustomTheme: validCustomTheme });
|
|
themeManager.setActiveTheme('MyCustomTheme');
|
|
expect(themeManager.getActiveTheme().name).toBe('MyCustomTheme');
|
|
});
|
|
|
|
it('should return false when setting a non-existent theme', () => {
|
|
expect(themeManager.setActiveTheme('NonExistentTheme')).toBe(false);
|
|
expect(themeManager.getActiveTheme().name).toBe(DEFAULT_THEME.name);
|
|
});
|
|
|
|
it('should list available themes including custom themes', () => {
|
|
themeManager.loadCustomThemes({ MyCustomTheme: validCustomTheme });
|
|
const available = themeManager.getAvailableThemes();
|
|
expect(
|
|
available.some(
|
|
(t: { name: string; isCustom?: boolean }) =>
|
|
t.name === 'MyCustomTheme' && t.isCustom,
|
|
),
|
|
).toBe(true);
|
|
});
|
|
|
|
it('should get a theme by name', () => {
|
|
expect(themeManager.getTheme('Ayu')).toBeDefined();
|
|
themeManager.loadCustomThemes({ MyCustomTheme: validCustomTheme });
|
|
expect(themeManager.getTheme('MyCustomTheme')).toBeDefined();
|
|
});
|
|
|
|
it('should fall back to default theme if active theme is invalid', () => {
|
|
(themeManager as unknown as { activeTheme: unknown }).activeTheme = {
|
|
name: 'NonExistent',
|
|
type: 'custom',
|
|
};
|
|
expect(themeManager.getActiveTheme().name).toBe(DEFAULT_THEME.name);
|
|
});
|
|
|
|
it('should return NoColorTheme if NO_COLOR is set', () => {
|
|
const original = process.env['NO_COLOR'];
|
|
process.env['NO_COLOR'] = '1';
|
|
expect(themeManager.getActiveTheme().name).toBe('NoColor');
|
|
if (original === undefined) {
|
|
delete process.env['NO_COLOR'];
|
|
} else {
|
|
process.env['NO_COLOR'] = original;
|
|
}
|
|
});
|
|
});
|