Xiaohan-Tian
78 lines
2.6 KiB
TypeScript
78 lines
2.6 KiB
TypeScript
import { beforeEach, describe, expect, it, vi } from 'vitest';
|
|
import { CreateNewTrackTool } from './CreateNewTrackTool';
|
|
import { KGCore } from '../../core/KGCore';
|
|
import { KGProject } from '../../core/KGProject';
|
|
import { KGAudioInterface } from '../../core/audio-interface/KGAudioInterface';
|
|
import { KGMidiTrack } from '../../core/track/KGMidiTrack';
|
|
|
|
vi.mock('../../stores/projectStore', () => ({
|
|
useProjectStore: {
|
|
getState: () => ({
|
|
activeRegionId: null,
|
|
selectedRegionIds: [],
|
|
selectedTrackId: null,
|
|
}),
|
|
},
|
|
}));
|
|
|
|
function mockCore(project: KGProject) {
|
|
vi.spyOn(KGCore, 'instance').mockReturnValue({
|
|
getCurrentProject: () => project,
|
|
executeCommand: (command: { execute(): void }) => command.execute(),
|
|
} as unknown as KGCore);
|
|
}
|
|
|
|
describe('CreateNewTrackTool', () => {
|
|
beforeEach(() => {
|
|
vi.restoreAllMocks();
|
|
vi.spyOn(KGAudioInterface, 'instance').mockReturnValue({
|
|
createTrackSynth: vi.fn(),
|
|
} as unknown as KGAudioInterface);
|
|
});
|
|
|
|
it('creates a new track and returns the exact output shape', async () => {
|
|
const project = new KGProject('create-track-project');
|
|
project.setTracks([new KGMidiTrack('Lead', 1, 'trumpet')]);
|
|
mockCore(project);
|
|
|
|
const tool = new CreateNewTrackTool();
|
|
const result = await tool.execute({
|
|
track_name: 'Bass',
|
|
instrument: 'Electric Bass (finger)',
|
|
});
|
|
|
|
expect(result).toEqual({
|
|
success: true,
|
|
result: 'New track created:\ntrack_id: 2\ntrack_name: Bass\ninstrument: Electric Bass (finger)',
|
|
});
|
|
expect(project.getTracks()).toHaveLength(2);
|
|
expect(project.getTracks()[1]).toBeInstanceOf(KGMidiTrack);
|
|
expect((project.getTracks()[1] as KGMidiTrack).getInstrument()).toBe('electric_bass_finger');
|
|
expect(tool.isReadOnlyTool()).toBe(false);
|
|
expect(tool.isAvailableInEfficientMode()).toBe(false);
|
|
expect(tool.buildToolResultDisplayContent({
|
|
track_name: 'Bass',
|
|
instrument: 'Electric Bass (finger)',
|
|
}, result)).toBe(
|
|
'New track created:\n- track_id: 2\n- track_name: Bass\n- instrument: Electric Bass (finger)',
|
|
);
|
|
});
|
|
|
|
it('rejects an invalid instrument name', async () => {
|
|
const project = new KGProject('invalid-instrument-project');
|
|
mockCore(project);
|
|
|
|
const tool = new CreateNewTrackTool();
|
|
const result = await tool.execute({
|
|
track_name: 'Bass',
|
|
instrument: 'electric_bass_finger',
|
|
});
|
|
|
|
expect(result).toEqual({
|
|
success: false,
|
|
result: 'Invalid instrument "electric_bass_finger". Use the exact English name from list_all_available_instruments.',
|
|
});
|
|
expect(project.getTracks()).toHaveLength(0);
|
|
});
|
|
});
|