Skip to content

Commit e06e35f

Browse files
authored
Merge pull request #134 from covespace/fix_unit_test
fix test cases
2 parents 9a4840c + a349d8d commit e06e35f

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

test/handler/sendMessageBase.test.ts

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@ import { expect } from 'chai';
22
import { describe, it } from 'mocha';
33
import { Context } from 'mocha';
44
import sinon from 'sinon';
5+
import * as path from 'path';
56
import { parseMessage, getInstructionFiles, parseMessageAndSetOptions, getParentHash, handleTopic, handlerResponseText, sendMessageBase, stopDevChatBase } from '../../src/handler/sendMessageBase';
67
import DevChat, { ChatResponse } from '../../src/toolwrapper/devchat';
78
import CommandManager from '../../src/command/commandManager';
@@ -10,6 +11,10 @@ import { TopicManager } from '../../src/topic/topicManager';
1011
import CustomCommands from '../../src/command/customCommand';
1112
import { UiUtilWrapper } from '../../src/util/uiUtil';
1213

14+
import * as dotenv from 'dotenv';
15+
16+
const envPath = path.join(__dirname, '../../', '.env');
17+
dotenv.config({ path: envPath });
1318

1419
describe('sendMessageBase', () => {
1520
let workspaceFoldersFirstPathStub: sinon.SinonStub;
@@ -194,7 +199,7 @@ describe('sendMessageBase', () => {
194199

195200
workspaceFoldersFirstPathStub.returns('./');
196201

197-
getConfigurationStub.withArgs('DevChat', 'API_KEY').returns('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');
202+
getConfigurationStub.withArgs('DevChat', 'API_KEY').returns(process.env.TEST_OPENAI_API_KEY);
198203
getConfigurationStub.withArgs('DevChat', 'OpenAI.model').returns('gpt-4');
199204
getConfigurationStub.withArgs('DevChat', 'OpenAI.temperature').returns(0);
200205
getConfigurationStub.withArgs('DevChat', 'OpenAI.stream').returns('true');
@@ -221,7 +226,7 @@ describe('sendMessageBase', () => {
221226

222227
workspaceFoldersFirstPathStub.returns('./');
223228

224-
getConfigurationStub.withArgs('DevChat', 'API_KEY').returns('DC.eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJvcmdfaWQiOjY2MTI0NDU1ODE2LCJqdGkiOjcyMzc4ODIzMjI3Mjc4MzI2NTJ9.qGWJ_NyWjjj66oa5mbfi3Hjawe-Yp8syCDLkpyu4yS0');
229+
getConfigurationStub.withArgs('DevChat', 'API_KEY').returns(process.env.TEST_DEVCHAT_KEY);
225230
getConfigurationStub.withArgs('DevChat', 'OpenAI.model').returns('gpt-4');
226231
getConfigurationStub.withArgs('DevChat', 'OpenAI.temperature').returns(0);
227232
getConfigurationStub.withArgs('DevChat', 'OpenAI.stream').returns('true');
@@ -277,7 +282,7 @@ describe('sendMessageBase', () => {
277282

278283
workspaceFoldersFirstPathStub.returns('./');
279284

280-
getConfigurationStub.withArgs('DevChat', 'API_KEY').returns('DC.eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJvcmdfaWQiOjY2MTI0NDU1ODE2LCJqdGkiOjcyMzc4ODIzMjI3Mjc4MzI2NTJ9.qGWJ_NyWjjj66oa5mbfi3Hjawe-Yp8syCDLkpyu4yS0');
285+
getConfigurationStub.withArgs('DevChat', 'API_KEY').returns(process.env.TEST_DEVCHAT_KEY);
281286
getConfigurationStub.withArgs('DevChat', 'OpenAI.model').returns('gpt-4');
282287
getConfigurationStub.withArgs('DevChat', 'OpenAI.temperature').returns(0);
283288
getConfigurationStub.withArgs('DevChat', 'OpenAI.stream').returns('true');

0 commit comments

Comments
 (0)