e2e761fd16
- Renamed and split the API key settings in package.json into DevChat and OpenAI specific keys. - Updated the commands for setting API keys to handle OpenAI and DevChat keys separately. - Modified the ApiKeyManager to handle and validate OpenAI and DevChat keys separately. - Updated the status bar view to use the DevChat access key command. - Adjusted the tests to reflect the new handling of API keys.
91 lines
3.1 KiB
TypeScript
91 lines
3.1 KiB
TypeScript
// test/apiKey.test.ts
|
|
|
|
import { expect } from 'chai';
|
|
import { ApiKeyManager } from '../../src/util/apiKey';
|
|
import { UiUtilWrapper } from '../../src/util/uiUtil';
|
|
import sinon from 'sinon';
|
|
|
|
describe('ApiKeyManager', () => {
|
|
afterEach(() => {
|
|
sinon.restore();
|
|
delete process.env.OPENAI_API_KEY;
|
|
delete process.env.OPENAI_API_BASE;
|
|
});
|
|
|
|
describe('getApiKey', () => {
|
|
it('should return the secret storage API key', async () => {
|
|
sinon.stub(UiUtilWrapper, 'secretStorageGet').resolves('sk-secret');
|
|
sinon.stub(UiUtilWrapper, 'getConfiguration').returns(undefined);
|
|
|
|
const apiKey = await ApiKeyManager.getApiKey();
|
|
expect(apiKey).to.equal('sk-secret');
|
|
});
|
|
|
|
it('should return the configuration API key', async () => {
|
|
sinon.stub(UiUtilWrapper, 'secretStorageGet').resolves(undefined);
|
|
sinon.stub(UiUtilWrapper, 'getConfiguration').returns('sk-config');
|
|
|
|
const apiKey = await ApiKeyManager.getApiKey();
|
|
expect(apiKey).to.equal('sk-config');
|
|
});
|
|
|
|
it('should return the environment variable API key', async () => {
|
|
sinon.stub(UiUtilWrapper, 'secretStorageGet').resolves(undefined);
|
|
sinon.stub(UiUtilWrapper, 'getConfiguration').returns(undefined);
|
|
process.env.OPENAI_API_KEY = 'sk-env';
|
|
|
|
const apiKey = await ApiKeyManager.getApiKey();
|
|
expect(apiKey).to.equal('sk-env');
|
|
});
|
|
});
|
|
|
|
describe('getEndPoint', () => {
|
|
it('should return the configuration endpoint', () => {
|
|
sinon.stub(UiUtilWrapper, 'getConfiguration').returns('https://config-endpoint.com');
|
|
|
|
const endPoint = ApiKeyManager.getEndPoint('sk-key');
|
|
expect(endPoint).to.equal('https://config-endpoint.com');
|
|
});
|
|
|
|
it('should return the environment variable endpoint', () => {
|
|
sinon.stub(UiUtilWrapper, 'getConfiguration').returns(undefined);
|
|
process.env.OPENAI_API_BASE = 'https://env-endpoint.com';
|
|
|
|
const endPoint = ApiKeyManager.getEndPoint('sk-key');
|
|
expect(endPoint).to.equal('https://env-endpoint.com');
|
|
});
|
|
|
|
it('should return the default endpoint for DC keys', () => {
|
|
sinon.stub(UiUtilWrapper, 'getConfiguration').returns(undefined);
|
|
|
|
const endPoint = ApiKeyManager.getEndPoint('DC.key');
|
|
expect(endPoint).to.equal('https://api.devchat.ai/v1');
|
|
});
|
|
});
|
|
|
|
describe('getKeyType', () => {
|
|
it('should return "sk" for sk keys', () => {
|
|
const keyType = ApiKeyManager.getKeyType('sk-key');
|
|
expect(keyType).to.equal('sk');
|
|
});
|
|
|
|
it('should return "DC" for DC keys', () => {
|
|
const keyType = ApiKeyManager.getKeyType('DC.key');
|
|
expect(keyType).to.equal('DC');
|
|
});
|
|
|
|
it('should return undefined for invalid keys', () => {
|
|
const keyType = ApiKeyManager.getKeyType('invalid.key');
|
|
expect(keyType).to.be.undefined;
|
|
});
|
|
});
|
|
|
|
describe('writeApiKeySecret', () => {
|
|
it('should store the API key in secret storage', async () => {
|
|
const storeSecretStub = sinon.stub(UiUtilWrapper, 'storeSecret').resolves();
|
|
|
|
await ApiKeyManager.writeApiKeySecret('sk-secret');
|
|
expect(storeSecretStub.calledWith('openai_OPENAI_API_KEY', 'sk-secret')).to.be.true;
|
|
});
|
|
});
|
|
}); |