File size: 2,526 Bytes
4e23b01 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 | import { mkdtemp, rm } from 'node:fs/promises';
import { tmpdir } from 'node:os';
import { join } from 'node:path';
import {
KimiOAuthToolkit,
OAuthConnectionError,
OAuthError,
OAuthUnauthorizedError,
RetryableRefreshError,
} from '@moonshot-ai/kimi-code-oauth';
import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
import { ErrorCodes, KimiError, KimiForCodingProvider } from '#/index';
import { TEST_IDENTITY } from './test-identity';
describe('KimiForCodingProvider OAuth error mapping', () => {
let homeDir: string;
beforeEach(async () => {
homeDir = await mkdtemp(join(tmpdir(), 'kimi-for-coding-provider-'));
});
afterEach(async () => {
vi.restoreAllMocks();
await rm(homeDir, { recursive: true, force: true });
});
function resolveAuth() {
const provider = new KimiForCodingProvider({ homeDir, ...TEST_IDENTITY });
return provider.resolveAuth('kimi-for-coding');
}
it('maps unauthorized token failures to auth.login_required', async () => {
vi.spyOn(KimiOAuthToolkit.prototype, 'ensureFresh').mockRejectedValue(
new OAuthUnauthorizedError('No token for "kimi-code". Run /login to authenticate.'),
);
const auth = resolveAuth();
await expect(auth(async () => 'ok')).rejects.toMatchObject({
code: ErrorCodes.AUTH_LOGIN_REQUIRED,
});
});
it('maps transient token failures to provider.connection_error', async () => {
const tokenErrors = [
new OAuthConnectionError('OAuth request to https://example.test failed: fetch failed'),
new RetryableRefreshError('Token refresh failed (HTTP 503).'),
];
for (const tokenError of tokenErrors) {
vi.spyOn(KimiOAuthToolkit.prototype, 'ensureFresh').mockRejectedValue(tokenError);
const auth = resolveAuth();
const caught = await auth(async () => 'ok').catch((error: unknown) => error);
expect(caught).toBeInstanceOf(KimiError);
expect(caught).toMatchObject({
code: ErrorCodes.PROVIDER_CONNECTION_ERROR,
message: expect.stringContaining(tokenError.message),
cause: tokenError,
});
vi.restoreAllMocks();
}
});
it('rethrows unrecognized OAuth errors raw instead of guessing a category', async () => {
const oauthError = new OAuthError('Token refresh failed (HTTP 400).');
vi.spyOn(KimiOAuthToolkit.prototype, 'ensureFresh').mockRejectedValue(oauthError);
const auth = resolveAuth();
await expect(auth(async () => 'ok')).rejects.toBe(oauthError);
});
});
|