Fixed types and created tests
This commit is contained in:
79
test/llm-grok.test.mjs
Normal file
79
test/llm-grok.test.mjs
Normal file
@@ -0,0 +1,79 @@
|
||||
// Grok provider — unit tests
|
||||
// Uses Node.js built-in test runner (node:test) — no extra dependencies
|
||||
|
||||
import { describe, it, mock, beforeEach } from 'node:test';
|
||||
import assert from 'node:assert/strict';
|
||||
import { GrokProvider } from '../lib/llm/grok.mjs';
|
||||
import { createLLMProvider } from '../lib/llm/index.mjs';
|
||||
|
||||
// ─── Unit Tests ───
|
||||
|
||||
describe('GrokProvider', () => {
|
||||
it('should set defaults correctly', () => {
|
||||
const provider = new GrokProvider({ apiKey: 'sk-test' });
|
||||
assert.equal(provider.name, 'grok');
|
||||
assert.equal(provider.model, 'grok-3');
|
||||
assert.equal(provider.isConfigured, true);
|
||||
});
|
||||
|
||||
it('should accept custom model', () => {
|
||||
const provider = new GrokProvider({ apiKey: 'sk-test', model: 'grok-2' });
|
||||
assert.equal(provider.model, 'grok-2');
|
||||
});
|
||||
|
||||
it('should report not configured without API key', () => {
|
||||
const provider = new GrokProvider({});
|
||||
assert.equal(provider.isConfigured, false);
|
||||
});
|
||||
|
||||
it('should throw on API error', async () => {
|
||||
const provider = new GrokProvider({ apiKey: 'sk-test' });
|
||||
const originalFetch = globalThis.fetch;
|
||||
globalThis.fetch = mock.fn(() =>
|
||||
Promise.resolve({ ok: false, status: 401, text: () => Promise.resolve('Unauthorized') })
|
||||
);
|
||||
try {
|
||||
await assert.rejects(
|
||||
() => provider.complete('system', 'user'),
|
||||
(err) => {
|
||||
assert.match(err.message, /Grok API 401/);
|
||||
return true;
|
||||
}
|
||||
);
|
||||
} finally {
|
||||
globalThis.fetch = originalFetch;
|
||||
}
|
||||
});
|
||||
|
||||
it('should parse successful response', async () => {
|
||||
const provider = new GrokProvider({ apiKey: 'sk-test' });
|
||||
const mockResponse = {
|
||||
choices: [{ message: { content: 'Hello world' } }],
|
||||
usage: { prompt_tokens: 10, completion_tokens: 5 },
|
||||
model: 'grok-3'
|
||||
};
|
||||
const originalFetch = globalThis.fetch;
|
||||
globalThis.fetch = mock.fn(() =>
|
||||
Promise.resolve({ ok: true, json: () => Promise.resolve(mockResponse) })
|
||||
);
|
||||
try {
|
||||
const result = await provider.complete('system', 'user');
|
||||
assert.equal(result.text, 'Hello world');
|
||||
assert.equal(result.usage.inputTokens, 10);
|
||||
assert.equal(result.usage.outputTokens, 5);
|
||||
assert.equal(result.model, 'grok-3');
|
||||
} finally {
|
||||
globalThis.fetch = originalFetch;
|
||||
}
|
||||
});
|
||||
});
|
||||
|
||||
// ─── Factory Tests ───
|
||||
|
||||
describe('createLLMProvider', () => {
|
||||
it('should create Grok provider', () => {
|
||||
const provider = createLLMProvider({ provider: 'grok', apiKey: 'sk-test' });
|
||||
assert.ok(provider instanceof GrokProvider);
|
||||
assert.equal(provider.isConfigured, true);
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user