remove tests out of date
This commit is contained in:
bobo.yang
parent
908b18acc6
commit
741ee9fcbf
@ -1,181 +0,0 @@
|
|||||||
import { expect } from 'chai';
|
|
||||||
// import { describe, it } from 'mocha';
|
|
||||||
import sinon from 'sinon';
|
|
||||||
import * as path from 'path';
|
|
||||||
import { parseMessage, parseMessageAndSetOptions, processChatResponse, sendMessageBase, stopDevChatBase } from '../../src/handler/sendMessageBase';
|
|
||||||
import { ChatResponse } from '../../src/toolwrapper/devchat';
|
|
||||||
import { UiUtilWrapper } from '../../src/util/uiUtil';
|
|
||||||
|
|
||||||
import * as dotenv from 'dotenv';
|
|
||||||
|
|
||||||
const envPath = path.join(__dirname, '../../', '.env');
|
|
||||||
dotenv.config({ path: envPath });
|
|
||||||
|
|
||||||
describe('sendMessageBase', () => {
|
|
||||||
let workspaceFoldersFirstPathStub: sinon.SinonStub;
|
|
||||||
let getConfigurationStub: sinon.SinonStub;
|
|
||||||
|
|
||||||
beforeEach(() => {
|
|
||||||
workspaceFoldersFirstPathStub = sinon.stub(UiUtilWrapper, 'workspaceFoldersFirstPath');
|
|
||||||
getConfigurationStub = sinon.stub(UiUtilWrapper, 'getConfiguration');
|
|
||||||
});
|
|
||||||
|
|
||||||
afterEach(() => {
|
|
||||||
workspaceFoldersFirstPathStub.restore();
|
|
||||||
getConfigurationStub.restore();
|
|
||||||
});
|
|
||||||
|
|
||||||
describe('parseMessage', () => {
|
|
||||||
it('should parse message correctly', () => {
|
|
||||||
const message = '[context|path/to/context] [instruction|path/to/instruction] [reference|path/to/reference] Hello, world!';
|
|
||||||
const result = parseMessage(message);
|
|
||||||
|
|
||||||
expect(result.context).to.deep.equal(['path/to/context']);
|
|
||||||
expect(result.instruction).to.deep.equal(['path/to/instruction']);
|
|
||||||
expect(result.reference).to.deep.equal(['path/to/reference']);
|
|
||||||
expect(result.text).to.equal('Hello, world!');
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe('parseMessageAndSetOptions', () => {
|
|
||||||
it('should parse message and set options correctly', async () => {
|
|
||||||
const message = {
|
|
||||||
text: '[context|path/to/context] [instruction|path/to/instruction] [reference|path/to/reference] Hello, world!'
|
|
||||||
};
|
|
||||||
|
|
||||||
const [result, chatOptions] = await parseMessageAndSetOptions(message);
|
|
||||||
|
|
||||||
expect(result.context).to.deep.equal(['path/to/context']);
|
|
||||||
expect(result.instruction).to.deep.equal(['path/to/instruction']);
|
|
||||||
expect(result.reference).to.deep.equal(['path/to/reference']);
|
|
||||||
expect(result.text).to.equal('Hello, world!');
|
|
||||||
expect(chatOptions.context).to.deep.equal(['path/to/context']);
|
|
||||||
expect(chatOptions.header).to.deep.equal([]);
|
|
||||||
expect(chatOptions.reference).to.deep.equal(['path/to/reference']);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe('processChatResponse', () => {
|
|
||||||
it('should handle response text correctly when isError is false', async () => {
|
|
||||||
const partialDataText = 'Partial data';
|
|
||||||
const chatResponse: ChatResponse = {
|
|
||||||
"finish_reason": "",
|
|
||||||
response: 'Hello, user!',
|
|
||||||
isError: false,
|
|
||||||
user: 'user',
|
|
||||||
date: '2022-01-01T00:00:00.000Z',
|
|
||||||
'prompt-hash': 'responsehash'
|
|
||||||
};
|
|
||||||
|
|
||||||
const result = await processChatResponse(chatResponse);
|
|
||||||
expect(result).to.equal('Hello, user!');
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should handle response text correctly when isError is true', async () => {
|
|
||||||
const partialDataText = 'Partial data';
|
|
||||||
const chatResponse: ChatResponse = {
|
|
||||||
"finish_reason": "",
|
|
||||||
response: 'Error occurred!',
|
|
||||||
isError: true,
|
|
||||||
user: 'user',
|
|
||||||
date: '2022-01-01T00:00:00.000Z',
|
|
||||||
'prompt-hash': 'responsehash'
|
|
||||||
};
|
|
||||||
|
|
||||||
const result = await processChatResponse(chatResponse);
|
|
||||||
expect(result).to.equal('Error occurred!');
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe('sendMessageBase', () => {
|
|
||||||
it('should send message correct with DevChat access key', async function() {
|
|
||||||
const message = {
|
|
||||||
text: 'Hello, world!'
|
|
||||||
};
|
|
||||||
const handlePartialData = (data: { command: string, text: string, user: string, date: string }) => {
|
|
||||||
// Handle partial data
|
|
||||||
};
|
|
||||||
|
|
||||||
workspaceFoldersFirstPathStub.returns('./');
|
|
||||||
|
|
||||||
getConfigurationStub.withArgs('DevChat', 'Access_Key_DevChat').returns(process.env.TEST_DEVCHAT_KEY);
|
|
||||||
getConfigurationStub.withArgs('DevChat', 'OpenAI.temperature').returns(0);
|
|
||||||
getConfigurationStub.withArgs('DevChat', 'OpenAI.stream').returns('true');
|
|
||||||
|
|
||||||
const result = await sendMessageBase(message, handlePartialData);
|
|
||||||
expect(result).to.be.an('object');
|
|
||||||
expect(result!.command).to.equal('receiveMessage');
|
|
||||||
expect(result!.text).to.be.a('string');
|
|
||||||
expect(result!.hash).to.be.a('string');
|
|
||||||
expect(result!.user).to.be.a('string');
|
|
||||||
expect(result!.date).to.be.a('string');
|
|
||||||
// TODO fix
|
|
||||||
// Need to mock more config setting
|
|
||||||
// expect(result!.isError).to.be.false;
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should send message error with invalid api key', async function() {
|
|
||||||
const message = {
|
|
||||||
text: 'Hello, world!'
|
|
||||||
};
|
|
||||||
const handlePartialData = (data: { command: string, text: string, user: string, date: string }) => {
|
|
||||||
// Handle partial data
|
|
||||||
};
|
|
||||||
|
|
||||||
workspaceFoldersFirstPathStub.returns('./');
|
|
||||||
|
|
||||||
getConfigurationStub.withArgs('DevChat', 'Access_Key_DevChat').returns('sk-KvH7ZCtHmFDCBTqH0jUv');
|
|
||||||
getConfigurationStub.withArgs('DevChat', 'OpenAI.temperature').returns('0');
|
|
||||||
getConfigurationStub.withArgs('DevChat', 'OpenAI.stream').returns('true');
|
|
||||||
|
|
||||||
const result = await sendMessageBase(message, handlePartialData);
|
|
||||||
expect(result).to.be.an('object');
|
|
||||||
expect(result!.command).to.equal('receiveMessage');
|
|
||||||
expect(result!.text).to.be.a('string');
|
|
||||||
expect(result!.hash).to.be.a('string');
|
|
||||||
expect(result!.user).to.be.a('string');
|
|
||||||
expect(result!.date).to.be.a('string');
|
|
||||||
expect(result!.isError).to.be.true;
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe('stopDevChatBase', () => {
|
|
||||||
it('should stop sendMessageBase correctly', async () => {
|
|
||||||
const message = {
|
|
||||||
text: 'Hello, world!'
|
|
||||||
};
|
|
||||||
const handlePartialData = (data: { command: string, text: string, user: string, date: string }) => {
|
|
||||||
// Handle partial data
|
|
||||||
};
|
|
||||||
|
|
||||||
workspaceFoldersFirstPathStub.returns('./');
|
|
||||||
|
|
||||||
getConfigurationStub.withArgs('DevChat', 'Access_Key_DevChat').returns(process.env.TEST_DEVCHAT_KEY);
|
|
||||||
getConfigurationStub.withArgs('DevChat', 'OpenAI.temperature').returns(0);
|
|
||||||
getConfigurationStub.withArgs('DevChat', 'OpenAI.stream').returns('true');
|
|
||||||
|
|
||||||
|
|
||||||
// Start sendMessageBase in a separate Promise
|
|
||||||
const sendMessagePromise = sendMessageBase(message, handlePartialData);
|
|
||||||
|
|
||||||
// Wait for a short period to ensure sendMessageBase has started
|
|
||||||
await new Promise((resolve) => setTimeout(resolve, 100));
|
|
||||||
|
|
||||||
// Call stopDevChatBase
|
|
||||||
const stopMessage = {
|
|
||||||
text: 'stop'
|
|
||||||
};
|
|
||||||
await stopDevChatBase(stopMessage);
|
|
||||||
|
|
||||||
// Check if sendMessageBase has been stopped and returns an error
|
|
||||||
// TODO fix
|
|
||||||
// Need to mock more config setting
|
|
||||||
// try {
|
|
||||||
// const result = await sendMessagePromise;
|
|
||||||
// expect(result).to.undefined;
|
|
||||||
// } catch (error) {
|
|
||||||
// expect(error).to.be.an('error');
|
|
||||||
// }
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@ -1,71 +0,0 @@
|
|||||||
import { expect } from 'chai';
|
|
||||||
// import { describe, it } from 'mocha';
|
|
||||||
import sinon from 'sinon';
|
|
||||||
import DevChat, { ChatOptions } from '../../src/toolwrapper/devchat';
|
|
||||||
import { CommandRun } from '../../src/util/commonUtil';
|
|
||||||
import { UiUtilWrapper } from '../../src/util/uiUtil';
|
|
||||||
import { ApiKeyManager } from '../../src/util/apiKey';
|
|
||||||
|
|
||||||
// TODO: 删除devchat.js时，删除此测试文件
|
|
||||||
// TODO: 同时为 DevChatCLI & DevChatClient 添加测试
|
|
||||||
describe('DevChat', () => {
|
|
||||||
let devChat: DevChat;
|
|
||||||
let spawnAsyncStub: sinon.SinonStub;
|
|
||||||
let workspaceFoldersFirstPathStub: sinon.SinonStub;
|
|
||||||
let apiKeyManagerStub: sinon.SinonStub;
|
|
||||||
|
|
||||||
beforeEach(() => {
|
|
||||||
devChat = new DevChat();
|
|
||||||
spawnAsyncStub = sinon.stub(CommandRun.prototype, 'spawnAsync');
|
|
||||||
workspaceFoldersFirstPathStub = sinon.stub(UiUtilWrapper, 'workspaceFoldersFirstPath');
|
|
||||||
apiKeyManagerStub = sinon.stub(ApiKeyManager, 'llmModel');
|
|
||||||
});
|
|
||||||
|
|
||||||
afterEach(() => {
|
|
||||||
spawnAsyncStub.restore();
|
|
||||||
workspaceFoldersFirstPathStub.restore();
|
|
||||||
apiKeyManagerStub.restore();
|
|
||||||
});
|
|
||||||
|
|
||||||
|
|
||||||
describe('chat', () => {
|
|
||||||
it('should return a ChatResponse object with isError false when the chat is successful', async () => {
|
|
||||||
const content = 'Test chat content';
|
|
||||||
const options: ChatOptions = {
|
|
||||||
// Provide mock values for the options
|
|
||||||
parent: 'parent_value',
|
|
||||||
reference: ['ref1', 'ref2'],
|
|
||||||
header: ['header1', 'header2'],
|
|
||||||
context: ['context1', 'context2'],
|
|
||||||
};
|
|
||||||
const mockResponse = {
|
|
||||||
exitCode: 0,
|
|
||||||
stdout: 'User: Test user\nDate: 2022-01-01\nTest chat response\nprompt-hash: 12345',
|
|
||||||
stderr: '',
|
|
||||||
};
|
|
||||||
const mockWorkspacePath = './';
|
|
||||||
const llmModelResponse = {
|
|
||||||
"model": "gpt-3.5-turbo",
|
|
||||||
"api_key": "DC.1234567890"
|
|
||||||
}
|
|
||||||
|
|
||||||
spawnAsyncStub.resolves(mockResponse);
|
|
||||||
workspaceFoldersFirstPathStub.returns(mockWorkspacePath);
|
|
||||||
apiKeyManagerStub.resolves(llmModelResponse);
|
|
||||||
|
|
||||||
const response = await devChat.chat(content, options, (data)=>{}, false);
|
|
||||||
|
|
||||||
expect(response).to.have.property('prompt-hash', '');
|
|
||||||
expect(response).to.have.property('user', '');
|
|
||||||
expect(response).to.have.property('date', '');
|
|
||||||
expect(response).to.have.property('response', 'Test chat response');
|
|
||||||
expect(response).to.have.property('isError', false);
|
|
||||||
expect(spawnAsyncStub.calledOnce).to.be.true;
|
|
||||||
expect(workspaceFoldersFirstPathStub.calledOnce).to.be.true;
|
|
||||||
});
|
|
||||||
|
|
||||||
// Add more test cases for the chat method here
|
|
||||||
});
|
|
||||||
|
|
||||||
// ... other test cases
|
|
||||||
});
|
|
||||||
Loading…
x
Reference in New Issue
Block a user