122 lines
3.4 KiB
TypeScript
122 lines
3.4 KiB
TypeScript
import { expect } from 'chai';
|
|
import { describe, it } from 'mocha';
|
|
import CommandManager, { Command } from '../../src/command/commandManager';
|
|
import CustomCommands, { Command as CCommand } from '../../src/command/customCommand';
|
|
|
|
describe('CommandManager', () => {
|
|
let commandManager: CommandManager;
|
|
|
|
beforeEach(() => {
|
|
commandManager = CommandManager.getInstance();
|
|
});
|
|
|
|
afterEach(() => {
|
|
// Reset the command list after each test
|
|
commandManager['commands'] = [];
|
|
});
|
|
|
|
it('should register a command', () => {
|
|
const command: Command = {
|
|
name: 'test',
|
|
pattern: 'test',
|
|
description: 'Test command',
|
|
args: 0,
|
|
handler: async (commandName: string, userInput: string) => {
|
|
return 'Test result';
|
|
},
|
|
};
|
|
|
|
commandManager.registerCommand(command);
|
|
expect(commandManager['commands']).to.include(command);
|
|
});
|
|
|
|
it('should return the command list', () => {
|
|
const command: Command = {
|
|
name: 'test',
|
|
pattern: 'test',
|
|
description: 'Test command',
|
|
args: 0,
|
|
handler: async (commandName: string, userInput: string) => {
|
|
return 'Test result';
|
|
},
|
|
};
|
|
|
|
commandManager.registerCommand(command);
|
|
expect(commandManager.getCommandList()).to.include(command);
|
|
});
|
|
|
|
it('should process text with a command', async () => {
|
|
const command: Command = {
|
|
name: 'test',
|
|
pattern: 'test',
|
|
description: 'Test command',
|
|
args: 0,
|
|
handler: async (commandName: string, userInput: string) => {
|
|
return 'Test result';
|
|
},
|
|
};
|
|
|
|
commandManager.registerCommand(command);
|
|
const result = await commandManager.processText('/test');
|
|
expect(result).to.equal('Test result');
|
|
});
|
|
|
|
it('should process text with a custom command', async () => {
|
|
const customCommand: CCommand = {
|
|
name: 'customTest',
|
|
pattern: 'customTest',
|
|
description: 'Custom test command',
|
|
message: 'Custom test result',
|
|
args: 0,
|
|
show: true,
|
|
default: false,
|
|
instructions: []
|
|
};
|
|
|
|
CustomCommands.getInstance().regCommand(customCommand);
|
|
const result = await commandManager.processText('/customTest');
|
|
expect(result).to.equal(' Custom test result');
|
|
});
|
|
|
|
it('should match /xxx with space or newline, but not with other characters', async () => {
|
|
const command: Command = {
|
|
name: 'xxx',
|
|
pattern: 'xxx',
|
|
description: 'Test command',
|
|
args: 0,
|
|
handler: async (commandName: string, userInput: string) => {
|
|
return 'Matched';
|
|
},
|
|
};
|
|
|
|
commandManager.registerCommand(command);
|
|
|
|
const result1 = await commandManager.processText('/xxx someother text');
|
|
expect(result1).to.equal('Matched someother text');
|
|
|
|
const result2 = await commandManager.processText('/xxx\n');
|
|
expect(result2).to.equal('Matched\n');
|
|
|
|
const result3 = await commandManager.processText('/xxx-123');
|
|
expect(result3).to.equal('/xxx-123');
|
|
|
|
const result4 = await commandManager.processText('/xxx123');
|
|
expect(result4).to.equal('/xxx123');
|
|
});
|
|
|
|
it('should process text with a command containing two arguments', async () => {
|
|
const command: Command = {
|
|
name: 'test',
|
|
pattern: 'xxx {{prompt}}',
|
|
description: 'Test command with two arguments ${1} and ${2}',
|
|
args: 2,
|
|
handler: async (commandName: string, userInput: string) => {
|
|
return `Test result with argument: ["arg1", "arg2"]`;
|
|
},
|
|
};
|
|
|
|
commandManager.registerCommand(command);
|
|
const result = await commandManager.processText('/xxx {{["arg1", "arg2"]}}');
|
|
expect(result).to.equal('Test result with argument: ["arg1", "arg2"]');
|
|
});
|
|
}); |