mirror of
https://github.com/tiennm99/ccs.git
synced 2026-10-15 10:14:36 +00:00
Merge pull request #577 from kaitranntt/feat/576-cursor-battle-test-integration
feat(cursor): harden daemon integration and model discovery
This commit is contained in:
16 files changed
+1682
-349
No files matched your search
@@ -431,11 +431,23 @@ describe('autoDetectTokens', () => {
|
||||
return;
|
||||
}
|
||||
|
||||
const result = autoDetectTokens();
|
||||
const originalHome = process.env.HOME;
|
||||
const isolatedHome = path.join(tempDir, 'no-cursor-home');
|
||||
process.env.HOME = isolatedHome;
|
||||
|
||||
// Should fail because Cursor database doesn't exist in test environment
|
||||
expect(result.found).toBe(false);
|
||||
expect(result.error).toBeDefined();
|
||||
try {
|
||||
const result = autoDetectTokens();
|
||||
|
||||
// Should fail because isolated test home has no Cursor database
|
||||
expect(result.found).toBe(false);
|
||||
expect(result.error).toBeDefined();
|
||||
} finally {
|
||||
if (originalHome !== undefined) {
|
||||
process.env.HOME = originalHome;
|
||||
} else {
|
||||
delete process.env.HOME;
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
it('should have found property in return type', () => {
|
||||
|
||||
@@ -19,7 +19,7 @@ import {
|
||||
} from '../../../src/cursor/cursor-daemon';
|
||||
import { getCcsDir } from '../../../src/utils/config-manager';
|
||||
import { handleCursorCommand } from '../../../src/commands/cursor-command';
|
||||
import { loadCredentials } from '../../../src/cursor/cursor-auth';
|
||||
import { loadCredentials, saveCredentials } from '../../../src/cursor/cursor-auth';
|
||||
|
||||
// Test isolation
|
||||
let originalCcsHome: string | undefined;
|
||||
@@ -153,6 +153,13 @@ describe('startDaemon', () => {
|
||||
const running = await isDaemonRunning(port);
|
||||
expect(running).toBe(true);
|
||||
|
||||
// Verify models endpoint exists and is OpenAI-compatible list shape
|
||||
const modelsResponse = await fetch(`http://127.0.0.1:${port}/v1/models`);
|
||||
expect(modelsResponse.status).toBe(200);
|
||||
const modelsJson = (await modelsResponse.json()) as { object?: string; data?: unknown[] };
|
||||
expect(modelsJson.object).toBe('list');
|
||||
expect(Array.isArray(modelsJson.data)).toBe(true);
|
||||
|
||||
// Verify chat endpoint exists (requires auth, should not be 404)
|
||||
const chatResponse = await fetch(`http://127.0.0.1:${port}/v1/chat/completions`, {
|
||||
method: 'POST',
|
||||
@@ -174,6 +181,93 @@ describe('startDaemon', () => {
|
||||
},
|
||||
35000
|
||||
);
|
||||
|
||||
it('returns 404 for unknown routes', async () => {
|
||||
const port = 10000 + Math.floor(Math.random() * 50000);
|
||||
const result = await startDaemon({ port, ghost_mode: true });
|
||||
expect(result.success).toBe(true);
|
||||
|
||||
try {
|
||||
const response = await fetch(`http://127.0.0.1:${port}/unknown`);
|
||||
expect(response.status).toBe(404);
|
||||
} finally {
|
||||
await stopDaemon();
|
||||
}
|
||||
});
|
||||
|
||||
it('returns 401 when credentials are expired', async () => {
|
||||
const port = 10000 + Math.floor(Math.random() * 50000);
|
||||
const expiredAt = new Date(Date.now() - 26 * 60 * 60 * 1000).toISOString();
|
||||
|
||||
saveCredentials({
|
||||
accessToken: 'a'.repeat(60),
|
||||
machineId: '1234567890abcdef1234567890abcdef',
|
||||
authMethod: 'manual',
|
||||
importedAt: expiredAt,
|
||||
});
|
||||
|
||||
const result = await startDaemon({ port, ghost_mode: true });
|
||||
expect(result.success).toBe(true);
|
||||
|
||||
try {
|
||||
const response = await fetch(`http://127.0.0.1:${port}/v1/chat/completions`, {
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({
|
||||
model: 'gpt-4.1',
|
||||
messages: [{ role: 'user', content: 'hello' }],
|
||||
}),
|
||||
});
|
||||
|
||||
expect(response.status).toBe(401);
|
||||
const body = (await response.json()) as { error?: { message?: string } };
|
||||
expect(body.error?.message).toContain('expired');
|
||||
} finally {
|
||||
await stopDaemon();
|
||||
}
|
||||
});
|
||||
|
||||
it('validates invalid JSON, invalid message schema, and oversized body', async () => {
|
||||
const port = 10000 + Math.floor(Math.random() * 50000);
|
||||
const result = await startDaemon({ port, ghost_mode: true });
|
||||
expect(result.success).toBe(true);
|
||||
|
||||
try {
|
||||
const invalidJson = await fetch(`http://127.0.0.1:${port}/v1/chat/completions`, {
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: '{invalid-json',
|
||||
});
|
||||
expect(invalidJson.status).toBe(400);
|
||||
|
||||
const invalidSchema = await fetch(`http://127.0.0.1:${port}/v1/chat/completions`, {
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({
|
||||
model: 'gpt-4.1',
|
||||
messages: { role: 'user', content: 'hello' },
|
||||
}),
|
||||
});
|
||||
expect(invalidSchema.status).toBe(400);
|
||||
|
||||
const oversized = await fetch(`http://127.0.0.1:${port}/v1/chat/completions`, {
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({
|
||||
model: 'gpt-4.1',
|
||||
messages: [
|
||||
{
|
||||
role: 'user',
|
||||
content: 'x'.repeat(10 * 1024 * 1024 + 1024),
|
||||
},
|
||||
],
|
||||
}),
|
||||
});
|
||||
expect(oversized.status).toBe(413);
|
||||
} finally {
|
||||
await stopDaemon();
|
||||
}
|
||||
});
|
||||
});
|
||||
|
||||
describe('isDaemonRunning', () => {
|
||||
|
||||
@@ -2,7 +2,8 @@
|
||||
* Unit tests for Cursor models module
|
||||
*/
|
||||
|
||||
import { describe, it, expect } from 'bun:test';
|
||||
import { describe, it, expect, beforeEach } from 'bun:test';
|
||||
import * as http from 'http';
|
||||
import {
|
||||
DEFAULT_CURSOR_MODELS,
|
||||
DEFAULT_CURSOR_PORT,
|
||||
@@ -11,6 +12,9 @@ import {
|
||||
detectProvider,
|
||||
formatModelName,
|
||||
fetchModelsFromDaemon,
|
||||
fetchModelsFromCursorApi,
|
||||
getModelsForDaemon,
|
||||
clearCursorModelsCache,
|
||||
} from '../../../src/cursor/cursor-models';
|
||||
|
||||
describe('DEFAULT_CURSOR_MODELS', () => {
|
||||
@@ -95,10 +99,322 @@ describe('formatModelName', () => {
|
||||
|
||||
describe('fetchModelsFromDaemon', () => {
|
||||
it('falls back to DEFAULT_CURSOR_MODELS when daemon is unreachable', async () => {
|
||||
// Use a port that nothing is listening on
|
||||
const unreachablePort = 9999;
|
||||
const models = await fetchModelsFromDaemon(unreachablePort);
|
||||
|
||||
expect(models).toEqual(DEFAULT_CURSOR_MODELS);
|
||||
});
|
||||
|
||||
it('falls back to defaults when daemon returns invalid JSON', async () => {
|
||||
const server = http.createServer((_req, res) => {
|
||||
res.writeHead(200, { 'Content-Type': 'application/json' });
|
||||
res.end('{not-valid-json');
|
||||
});
|
||||
|
||||
await new Promise<void>((resolve) => server.listen(0, '127.0.0.1', resolve));
|
||||
const address = server.address();
|
||||
if (!address || typeof address === 'string') {
|
||||
throw new Error('Unable to resolve test server port');
|
||||
}
|
||||
|
||||
try {
|
||||
const models = await fetchModelsFromDaemon(address.port);
|
||||
expect(models).toEqual(DEFAULT_CURSOR_MODELS);
|
||||
} finally {
|
||||
await new Promise<void>((resolve) => server.close(() => resolve()));
|
||||
}
|
||||
});
|
||||
|
||||
it('falls back to defaults when daemon response exceeds max body size', async () => {
|
||||
const oversizedPayload = 'x'.repeat(1024 * 1024 + 1024);
|
||||
const server = http.createServer((_req, res) => {
|
||||
res.writeHead(200, { 'Content-Type': 'application/json' });
|
||||
res.end(oversizedPayload);
|
||||
});
|
||||
|
||||
await new Promise<void>((resolve) => server.listen(0, '127.0.0.1', resolve));
|
||||
const address = server.address();
|
||||
if (!address || typeof address === 'string') {
|
||||
throw new Error('Unable to resolve test server port');
|
||||
}
|
||||
|
||||
try {
|
||||
const models = await fetchModelsFromDaemon(address.port);
|
||||
expect(models).toEqual(DEFAULT_CURSOR_MODELS);
|
||||
} finally {
|
||||
await new Promise<void>((resolve) => server.close(() => resolve()));
|
||||
}
|
||||
});
|
||||
});
|
||||
|
||||
describe('fetchModelsFromCursorApi', () => {
|
||||
it('parses model list from API response', async () => {
|
||||
const server = http.createServer((_req, res) => {
|
||||
res.writeHead(200, { 'Content-Type': 'application/json' });
|
||||
res.end(
|
||||
JSON.stringify({
|
||||
data: [
|
||||
{ id: 'gpt-5.3-codex', name: 'GPT-5.3 Codex', provider: 'openai' },
|
||||
{ id: 'claude-4.6-opus', name: 'Claude 4.6 Opus', provider: 'anthropic' },
|
||||
],
|
||||
})
|
||||
);
|
||||
});
|
||||
|
||||
await new Promise<void>((resolve) => server.listen(0, '127.0.0.1', resolve));
|
||||
const address = server.address();
|
||||
if (!address || typeof address === 'string') {
|
||||
throw new Error('Unable to resolve test server port');
|
||||
}
|
||||
|
||||
try {
|
||||
const models = await fetchModelsFromCursorApi(
|
||||
{
|
||||
accessToken: 'test-token-123',
|
||||
machineId: '1234567890abcdef1234567890abcdef',
|
||||
},
|
||||
{
|
||||
endpoint: `http://127.0.0.1:${address.port}/v1/models`,
|
||||
timeoutMs: 2000,
|
||||
}
|
||||
);
|
||||
|
||||
expect(models).not.toBeNull();
|
||||
expect(models?.[0].id).toBe('gpt-5.3-codex');
|
||||
expect(models?.[1].id).toBe('claude-4.6-opus');
|
||||
} finally {
|
||||
await new Promise<void>((resolve) => server.close(() => resolve()));
|
||||
}
|
||||
});
|
||||
|
||||
it('returns null for non-200 responses', async () => {
|
||||
const server = http.createServer((_req, res) => {
|
||||
res.writeHead(403, { 'Content-Type': 'application/json' });
|
||||
res.end(JSON.stringify({ error: 'forbidden' }));
|
||||
});
|
||||
|
||||
await new Promise<void>((resolve) => server.listen(0, '127.0.0.1', resolve));
|
||||
const address = server.address();
|
||||
if (!address || typeof address === 'string') {
|
||||
throw new Error('Unable to resolve test server port');
|
||||
}
|
||||
|
||||
try {
|
||||
const models = await fetchModelsFromCursorApi(
|
||||
{
|
||||
accessToken: 'test-token-123',
|
||||
machineId: '1234567890abcdef1234567890abcdef',
|
||||
},
|
||||
{
|
||||
endpoint: `http://127.0.0.1:${address.port}/v1/models`,
|
||||
timeoutMs: 2000,
|
||||
}
|
||||
);
|
||||
|
||||
expect(models).toBeNull();
|
||||
} finally {
|
||||
await new Promise<void>((resolve) => server.close(() => resolve()));
|
||||
}
|
||||
});
|
||||
|
||||
it('parses response.models and filters invalid records', async () => {
|
||||
const server = http.createServer((_req, res) => {
|
||||
res.writeHead(200, { 'Content-Type': 'application/json' });
|
||||
res.end(
|
||||
JSON.stringify({
|
||||
models: [
|
||||
{ id: 'gpt-5.3-codex', name: 'GPT-5.3 Codex' },
|
||||
{ id: '', name: 'invalid-empty-id' },
|
||||
{ id: 123, name: 'invalid-type-id' },
|
||||
],
|
||||
})
|
||||
);
|
||||
});
|
||||
|
||||
await new Promise<void>((resolve) => server.listen(0, '127.0.0.1', resolve));
|
||||
const address = server.address();
|
||||
if (!address || typeof address === 'string') {
|
||||
throw new Error('Unable to resolve test server port');
|
||||
}
|
||||
|
||||
try {
|
||||
const models = await fetchModelsFromCursorApi(
|
||||
{
|
||||
accessToken: 'test-token-123',
|
||||
machineId: '1234567890abcdef1234567890abcdef',
|
||||
},
|
||||
{
|
||||
endpoint: `http://127.0.0.1:${address.port}/v1/models`,
|
||||
timeoutMs: 2000,
|
||||
}
|
||||
);
|
||||
|
||||
expect(models).not.toBeNull();
|
||||
expect(models).toHaveLength(1);
|
||||
expect(models?.[0].id).toBe('gpt-5.3-codex');
|
||||
} finally {
|
||||
await new Promise<void>((resolve) => server.close(() => resolve()));
|
||||
}
|
||||
});
|
||||
|
||||
it('returns null when required credentials are missing', async () => {
|
||||
const models = await fetchModelsFromCursorApi(
|
||||
{
|
||||
accessToken: '',
|
||||
machineId: '1234567890abcdef1234567890abcdef',
|
||||
},
|
||||
{
|
||||
endpoint: 'http://127.0.0.1:9/v1/models',
|
||||
timeoutMs: 50,
|
||||
}
|
||||
);
|
||||
|
||||
expect(models).toBeNull();
|
||||
});
|
||||
|
||||
it('returns null on timeout/abort', async () => {
|
||||
const server = http.createServer((_req, _res) => {
|
||||
// Intentionally no response within timeout.
|
||||
});
|
||||
|
||||
await new Promise<void>((resolve) => server.listen(0, '127.0.0.1', resolve));
|
||||
const address = server.address();
|
||||
if (!address || typeof address === 'string') {
|
||||
throw new Error('Unable to resolve test server port');
|
||||
}
|
||||
|
||||
try {
|
||||
const models = await fetchModelsFromCursorApi(
|
||||
{
|
||||
accessToken: 'test-token-123',
|
||||
machineId: '1234567890abcdef1234567890abcdef',
|
||||
},
|
||||
{
|
||||
endpoint: `http://127.0.0.1:${address.port}/v1/models`,
|
||||
timeoutMs: 25,
|
||||
}
|
||||
);
|
||||
|
||||
expect(models).toBeNull();
|
||||
} finally {
|
||||
await new Promise<void>((resolve) => server.close(() => resolve()));
|
||||
}
|
||||
});
|
||||
});
|
||||
|
||||
describe('getModelsForDaemon', () => {
|
||||
beforeEach(() => {
|
||||
clearCursorModelsCache();
|
||||
});
|
||||
|
||||
it('falls back to defaults without credentials', async () => {
|
||||
const models = await getModelsForDaemon();
|
||||
expect(models).toEqual(DEFAULT_CURSOR_MODELS);
|
||||
});
|
||||
|
||||
it('uses cached live models when endpoint becomes unavailable', async () => {
|
||||
const liveModelId = 'test-live-model';
|
||||
const server = http.createServer((_req, res) => {
|
||||
res.writeHead(200, { 'Content-Type': 'application/json' });
|
||||
res.end(
|
||||
JSON.stringify({
|
||||
data: [{ id: liveModelId, name: 'Live Model', provider: 'openai' }],
|
||||
})
|
||||
);
|
||||
});
|
||||
|
||||
await new Promise<void>((resolve) => server.listen(0, '127.0.0.1', resolve));
|
||||
const address = server.address();
|
||||
if (!address || typeof address === 'string') {
|
||||
throw new Error('Unable to resolve test server port');
|
||||
}
|
||||
|
||||
const endpoint = `http://127.0.0.1:${address.port}/v1/models`;
|
||||
|
||||
try {
|
||||
const first = await getModelsForDaemon({
|
||||
credentials: {
|
||||
accessToken: 'test-token-123',
|
||||
machineId: '1234567890abcdef1234567890abcdef',
|
||||
},
|
||||
endpoint,
|
||||
timeoutMs: 2000,
|
||||
});
|
||||
|
||||
expect(first[0]?.id).toBe(liveModelId);
|
||||
} finally {
|
||||
await new Promise<void>((resolve) => server.close(() => resolve()));
|
||||
}
|
||||
|
||||
const second = await getModelsForDaemon({
|
||||
endpoint: 'http://127.0.0.1:9/v1/models',
|
||||
timeoutMs: 250,
|
||||
});
|
||||
|
||||
expect(second[0]?.id).toBe(liveModelId);
|
||||
});
|
||||
|
||||
it('clears cache after auth failures and falls back to defaults', async () => {
|
||||
const liveModelId = 'test-live-model-auth-cache';
|
||||
const okServer = http.createServer((_req, res) => {
|
||||
res.writeHead(200, { 'Content-Type': 'application/json' });
|
||||
res.end(
|
||||
JSON.stringify({
|
||||
data: [{ id: liveModelId, name: 'Live Model', provider: 'openai' }],
|
||||
})
|
||||
);
|
||||
});
|
||||
|
||||
await new Promise<void>((resolve) => okServer.listen(0, '127.0.0.1', resolve));
|
||||
const okAddress = okServer.address();
|
||||
if (!okAddress || typeof okAddress === 'string') {
|
||||
throw new Error('Unable to resolve test server port');
|
||||
}
|
||||
|
||||
try {
|
||||
const first = await getModelsForDaemon({
|
||||
credentials: {
|
||||
accessToken: 'test-token-123',
|
||||
machineId: '1234567890abcdef1234567890abcdef',
|
||||
},
|
||||
endpoint: `http://127.0.0.1:${okAddress.port}/v1/models`,
|
||||
timeoutMs: 2000,
|
||||
});
|
||||
|
||||
expect(first[0]?.id).toBe(liveModelId);
|
||||
} finally {
|
||||
await new Promise<void>((resolve) => okServer.close(() => resolve()));
|
||||
}
|
||||
|
||||
const forbiddenServer = http.createServer((_req, res) => {
|
||||
res.writeHead(403, { 'Content-Type': 'application/json' });
|
||||
res.end(JSON.stringify({ error: 'forbidden' }));
|
||||
});
|
||||
|
||||
await new Promise<void>((resolve) => forbiddenServer.listen(0, '127.0.0.1', resolve));
|
||||
const forbiddenAddress = forbiddenServer.address();
|
||||
if (!forbiddenAddress || typeof forbiddenAddress === 'string') {
|
||||
throw new Error('Unable to resolve test server port');
|
||||
}
|
||||
|
||||
try {
|
||||
const forbidden = await fetchModelsFromCursorApi(
|
||||
{
|
||||
accessToken: 'test-token-123',
|
||||
machineId: '1234567890abcdef1234567890abcdef',
|
||||
},
|
||||
{
|
||||
endpoint: `http://127.0.0.1:${forbiddenAddress.port}/v1/models`,
|
||||
timeoutMs: 2000,
|
||||
}
|
||||
);
|
||||
|
||||
expect(forbidden).toBeNull();
|
||||
} finally {
|
||||
await new Promise<void>((resolve) => forbiddenServer.close(() => resolve()));
|
||||
}
|
||||
|
||||
const afterAuthFailure = await getModelsForDaemon();
|
||||
expect(afterAuthFailure).toEqual(DEFAULT_CURSOR_MODELS);
|
||||
});
|
||||
});
|
||||
@@ -371,18 +371,12 @@ describe('Request Encoding', () => {
|
||||
it('should handle multi-frame buffer', () => {
|
||||
const executor = new CursorExecutor();
|
||||
|
||||
// Create two simple frames
|
||||
const frame1 = wrapConnectRPCFrame(
|
||||
encodeField(FIELD.ChatResponse.TEXT, WIRE_TYPE.LEN, 'Frame 1'),
|
||||
false
|
||||
);
|
||||
const frame2 = wrapConnectRPCFrame(
|
||||
encodeField(FIELD.ChatResponse.TEXT, WIRE_TYPE.LEN, ' Frame 2'),
|
||||
false
|
||||
);
|
||||
// Build two valid response frames (top-level Response.RESPONSE wrapper).
|
||||
const frame1 = buildTextFrame('Frame 1');
|
||||
const frame2 = buildTextFrame(' Frame 2');
|
||||
|
||||
// Concatenate them
|
||||
const multiFrame = Buffer.concat([Buffer.from(frame1), Buffer.from(frame2)]);
|
||||
const multiFrame = Buffer.concat([frame1, frame2]);
|
||||
|
||||
const result = executor.transformProtobufToJSON(multiFrame, 'gpt-4', {
|
||||
messages: [],
|
||||
@@ -457,6 +451,29 @@ describe('CursorExecutor', () => {
|
||||
expect(headers.authorization).toBe('Bearer actual-token');
|
||||
});
|
||||
|
||||
it('should throw when token becomes empty after delimiter parsing', () => {
|
||||
const credentials = {
|
||||
accessToken: 'prefix::',
|
||||
machineId: 'test-machine-id',
|
||||
};
|
||||
|
||||
expect(() => executor.buildHeaders(credentials)).toThrow('Access token is empty');
|
||||
});
|
||||
|
||||
it('should include normalized platform and timezone headers', () => {
|
||||
const credentials = {
|
||||
accessToken: 'test-token',
|
||||
machineId: 'test-machine-id',
|
||||
};
|
||||
|
||||
const headers = executor.buildHeaders(credentials);
|
||||
|
||||
expect(['windows', 'macos', 'linux']).toContain(headers['x-cursor-client-os']);
|
||||
expect(['aarch64', 'x64']).toContain(headers['x-cursor-client-arch']);
|
||||
expect(typeof headers['x-cursor-timezone']).toBe('string');
|
||||
expect(headers['x-cursor-timezone'].length).toBeGreaterThan(0);
|
||||
});
|
||||
|
||||
it('should respect ghostMode flag', () => {
|
||||
const credentialsGhost = {
|
||||
accessToken: 'test-token',
|
||||
@@ -531,6 +548,55 @@ describe('CursorExecutor', () => {
|
||||
const body = JSON.parse(bodyText);
|
||||
expect(body.error.type).toBe('rate_limit_error');
|
||||
});
|
||||
|
||||
it('should surface reasoning_content when thinking payload is present', async () => {
|
||||
const textContent = 'Final answer';
|
||||
const thinkingContent = 'Internal reasoning trail';
|
||||
const thinkingField = encodeField(FIELD.Thinking.TEXT, WIRE_TYPE.LEN, thinkingContent);
|
||||
const chatResponse = concatArrays(
|
||||
encodeField(FIELD.ChatResponse.TEXT, WIRE_TYPE.LEN, textContent),
|
||||
encodeField(FIELD.ChatResponse.THINKING, WIRE_TYPE.LEN, thinkingField)
|
||||
);
|
||||
const responseMsg = encodeField(FIELD.Response.RESPONSE, WIRE_TYPE.LEN, chatResponse);
|
||||
const frame = wrapConnectRPCFrame(responseMsg, false);
|
||||
|
||||
const result = executor.transformProtobufToJSON(Buffer.from(frame), 'gpt-4', {
|
||||
messages: [],
|
||||
});
|
||||
|
||||
expect(result.status).toBe(200);
|
||||
const bodyText = await result.text();
|
||||
const body = JSON.parse(bodyText);
|
||||
expect(body.choices[0].message.content).toBe(textContent);
|
||||
expect(body.choices[0].message.reasoning_content).toBe(thinkingContent);
|
||||
});
|
||||
|
||||
it('should merge fragmented tool call arguments and set tool_calls finish reason', async () => {
|
||||
const frame1 = buildToolCallFrame({
|
||||
id: 'call_123',
|
||||
name: 'search_docs',
|
||||
args: '{"q":"hel',
|
||||
isLast: false,
|
||||
});
|
||||
const frame2 = buildToolCallFrame({
|
||||
id: 'call_123',
|
||||
name: 'search_docs',
|
||||
args: 'lo"}',
|
||||
isLast: true,
|
||||
});
|
||||
const combined = Buffer.concat([frame1, frame2]);
|
||||
|
||||
const result = executor.transformProtobufToJSON(combined, 'gpt-4', {
|
||||
messages: [],
|
||||
});
|
||||
|
||||
expect(result.status).toBe(200);
|
||||
const body = JSON.parse(await result.text());
|
||||
expect(body.choices[0].finish_reason).toBe('tool_calls');
|
||||
expect(body.choices[0].message.tool_calls[0].id).toBe('call_123');
|
||||
expect(body.choices[0].message.tool_calls[0].function.name).toBe('search_docs');
|
||||
expect(body.choices[0].message.tool_calls[0].function.arguments).toBe('{"q":"hello"}');
|
||||
});
|
||||
});
|
||||
|
||||
describe('transformProtobufToSSE', () => {
|
||||
@@ -572,6 +638,49 @@ describe('CursorExecutor', () => {
|
||||
const body = JSON.parse(bodyText);
|
||||
expect(body.error.type).toBe('rate_limit_error');
|
||||
});
|
||||
|
||||
it('should emit reasoning_content deltas for thinking payloads', async () => {
|
||||
const thinkingContent = 'Deliberate reasoning';
|
||||
const thinkingField = encodeField(FIELD.Thinking.TEXT, WIRE_TYPE.LEN, thinkingContent);
|
||||
const chatResponse = encodeField(FIELD.ChatResponse.THINKING, WIRE_TYPE.LEN, thinkingField);
|
||||
const responseMsg = encodeField(FIELD.Response.RESPONSE, WIRE_TYPE.LEN, chatResponse);
|
||||
const frame = wrapConnectRPCFrame(responseMsg, false);
|
||||
|
||||
const result = executor.transformProtobufToSSE(Buffer.from(frame), 'gpt-4', {
|
||||
messages: [],
|
||||
});
|
||||
|
||||
expect(result.status).toBe(200);
|
||||
const bodyText = await result.text();
|
||||
expect(bodyText).toContain('reasoning_content');
|
||||
expect(bodyText).toContain(thinkingContent);
|
||||
});
|
||||
|
||||
it('should emit tool call deltas and end with finish_reason tool_calls', async () => {
|
||||
const frame1 = buildToolCallFrame({
|
||||
id: 'call_abc',
|
||||
name: 'search_docs',
|
||||
args: '{"q":"foo',
|
||||
isLast: false,
|
||||
});
|
||||
const frame2 = buildToolCallFrame({
|
||||
id: 'call_abc',
|
||||
name: 'search_docs',
|
||||
args: '"}',
|
||||
isLast: true,
|
||||
});
|
||||
const combined = Buffer.concat([frame1, frame2]);
|
||||
|
||||
const result = executor.transformProtobufToSSE(combined, 'gpt-4', {
|
||||
messages: [],
|
||||
});
|
||||
|
||||
expect(result.status).toBe(200);
|
||||
const bodyText = await result.text();
|
||||
expect(bodyText).toContain('tool_calls');
|
||||
expect(bodyText).toContain('search_docs');
|
||||
expect(bodyText).toContain('"finish_reason":"tool_calls"');
|
||||
});
|
||||
});
|
||||
|
||||
describe('decompressPayload error handling', () => {
|
||||
@@ -676,6 +785,35 @@ function buildTextFrame(text: string): Buffer {
|
||||
return buildFrame(responseMsg);
|
||||
}
|
||||
|
||||
/**
|
||||
* Helper: build a protobuf thinking response frame
|
||||
*/
|
||||
function buildThinkingFrame(thinking: string): Buffer {
|
||||
const thinkingField = encodeField(FIELD.Thinking.TEXT, WIRE_TYPE.LEN, thinking);
|
||||
const responseField = encodeField(FIELD.ChatResponse.THINKING, WIRE_TYPE.LEN, thinkingField);
|
||||
const responseMsg = encodeField(FIELD.Response.RESPONSE, WIRE_TYPE.LEN, responseField);
|
||||
return buildFrame(responseMsg);
|
||||
}
|
||||
|
||||
/**
|
||||
* Helper: build a protobuf tool call response frame
|
||||
*/
|
||||
function buildToolCallFrame(options: {
|
||||
id: string;
|
||||
name: string;
|
||||
args: string;
|
||||
isLast: boolean;
|
||||
}): Buffer {
|
||||
const toolCallPayload = concatArrays(
|
||||
encodeField(FIELD.ToolCall.ID, WIRE_TYPE.LEN, options.id),
|
||||
encodeField(FIELD.ToolCall.NAME, WIRE_TYPE.LEN, options.name),
|
||||
encodeField(FIELD.ToolCall.RAW_ARGS, WIRE_TYPE.LEN, options.args),
|
||||
encodeField(FIELD.ToolCall.IS_LAST, WIRE_TYPE.VARINT, options.isLast ? 1 : 0)
|
||||
);
|
||||
const responseMsg = encodeField(FIELD.Response.TOOL_CALL, WIRE_TYPE.LEN, toolCallPayload);
|
||||
return buildFrame(responseMsg);
|
||||
}
|
||||
|
||||
describe('StreamingFrameParser', () => {
|
||||
it('should parse a complete single frame', () => {
|
||||
const parser = new StreamingFrameParser();
|
||||
@@ -784,6 +922,51 @@ describe('StreamingFrameParser', () => {
|
||||
}
|
||||
});
|
||||
|
||||
it('should parse thinking frames', () => {
|
||||
const parser = new StreamingFrameParser();
|
||||
const frame = buildThinkingFrame('Think step by step');
|
||||
const results = parser.push(frame);
|
||||
|
||||
expect(results.length).toBe(1);
|
||||
expect(results[0].type).toBe('thinking');
|
||||
if (results[0].type === 'thinking') {
|
||||
expect(results[0].text).toBe('Think step by step');
|
||||
}
|
||||
});
|
||||
|
||||
it('should parse tool call frames', () => {
|
||||
const parser = new StreamingFrameParser();
|
||||
const frame = buildToolCallFrame({
|
||||
id: 'call_parser',
|
||||
name: 'search_docs',
|
||||
args: '{"q":"docs"}',
|
||||
isLast: true,
|
||||
});
|
||||
const results = parser.push(frame);
|
||||
|
||||
expect(results.length).toBe(1);
|
||||
expect(results[0].type).toBe('toolCall');
|
||||
if (results[0].type === 'toolCall') {
|
||||
expect(results[0].toolCall.id).toBe('call_parser');
|
||||
expect(results[0].toolCall.function.name).toBe('search_docs');
|
||||
expect(results[0].toolCall.function.arguments).toBe('{"q":"docs"}');
|
||||
expect(results[0].toolCall.isLast).toBe(true);
|
||||
}
|
||||
});
|
||||
|
||||
it('should classify malformed protobuf payload as server error', () => {
|
||||
const parser = new StreamingFrameParser();
|
||||
const malformedFrame = buildFrame(new Uint8Array([0xff, 0xff, 0xff]));
|
||||
const results = parser.push(malformedFrame);
|
||||
|
||||
expect(results.length).toBe(1);
|
||||
expect(results[0].type).toBe('error');
|
||||
if (results[0].type === 'error') {
|
||||
expect(results[0].errorType).toBe('server_error');
|
||||
expect(results[0].message).toContain('Malformed protobuf response');
|
||||
}
|
||||
});
|
||||
|
||||
it('should report hasPartial() correctly', () => {
|
||||
const parser = new StreamingFrameParser();
|
||||
expect(parser.hasPartial()).toBe(false);
|
||||
|
||||
@@ -1,7 +1,7 @@
|
||||
/**
|
||||
* Unit tests for process-utils.ts
|
||||
*/
|
||||
import { describe, it, expect, beforeEach, afterEach, jest } from 'bun:test';
|
||||
import { describe, it, expect, jest } from 'bun:test';
|
||||
import { EventEmitter } from 'events';
|
||||
import { killWithEscalation } from '../../../src/utils/process-utils';
|
||||
import type { ChildProcess } from 'child_process';
|
||||
@@ -20,15 +20,11 @@ function createMockProcess(exitCode: number | null = null): ChildProcess {
|
||||
return proc as ChildProcess;
|
||||
}
|
||||
|
||||
function wait(ms: number): Promise<void> {
|
||||
return new Promise((resolve) => setTimeout(resolve, ms));
|
||||
}
|
||||
|
||||
describe('killWithEscalation', () => {
|
||||
beforeEach(() => {
|
||||
jest.useFakeTimers();
|
||||
});
|
||||
|
||||
afterEach(() => {
|
||||
jest.useRealTimers();
|
||||
});
|
||||
|
||||
it('should send SIGTERM immediately', () => {
|
||||
const proc = createMockProcess();
|
||||
killWithEscalation(proc);
|
||||
@@ -37,36 +33,34 @@ describe('killWithEscalation', () => {
|
||||
expect(proc.kill).toHaveBeenCalledTimes(1);
|
||||
});
|
||||
|
||||
it('should send SIGKILL after grace period if process still running', () => {
|
||||
it('should send SIGKILL after grace period if process still running', async () => {
|
||||
const proc = createMockProcess(null); // exitCode null = still running
|
||||
killWithEscalation(proc, 3000);
|
||||
killWithEscalation(proc, 10);
|
||||
|
||||
// SIGTERM sent immediately
|
||||
expect(proc.kill).toHaveBeenCalledWith('SIGTERM');
|
||||
expect(proc.kill).toHaveBeenCalledTimes(1);
|
||||
|
||||
// Advance time by grace period
|
||||
jest.advanceTimersByTime(3000);
|
||||
await wait(40);
|
||||
|
||||
// SIGKILL sent after grace period
|
||||
expect(proc.kill).toHaveBeenCalledWith('SIGKILL');
|
||||
expect(proc.kill).toHaveBeenCalledTimes(2);
|
||||
});
|
||||
|
||||
it('should NOT send SIGKILL if process exits before grace period', () => {
|
||||
it('should NOT send SIGKILL if process exits before grace period', async () => {
|
||||
const proc = createMockProcess(null);
|
||||
killWithEscalation(proc, 3000);
|
||||
killWithEscalation(proc, 40);
|
||||
|
||||
expect(proc.kill).toHaveBeenCalledWith('SIGTERM');
|
||||
expect(proc.kill).toHaveBeenCalledTimes(1);
|
||||
|
||||
// Simulate process exit after 1 second
|
||||
jest.advanceTimersByTime(1000);
|
||||
proc.exitCode = 0; // Process exited
|
||||
// Simulate process exit before grace timeout
|
||||
await wait(10);
|
||||
proc.exitCode = 0;
|
||||
proc.emit('exit', 0);
|
||||
|
||||
// Advance remaining time
|
||||
jest.advanceTimersByTime(2000);
|
||||
await wait(60);
|
||||
|
||||
// SIGKILL should NOT have been sent
|
||||
expect(proc.kill).toHaveBeenCalledTimes(1);
|
||||
@@ -75,56 +69,95 @@ describe('killWithEscalation', () => {
|
||||
|
||||
it('should use default grace period of 3000ms', () => {
|
||||
const proc = createMockProcess(null);
|
||||
killWithEscalation(proc); // No grace period argument
|
||||
const originalSetTimeout = globalThis.setTimeout;
|
||||
const fakeTimer = {
|
||||
unref: () => fakeTimer,
|
||||
ref: () => fakeTimer,
|
||||
hasRef: () => false,
|
||||
refresh: () => fakeTimer,
|
||||
} as unknown as ReturnType<typeof setTimeout>;
|
||||
|
||||
expect(proc.kill).toHaveBeenCalledWith('SIGTERM');
|
||||
let observedDelay: number | undefined;
|
||||
|
||||
// Advance by default 3000ms
|
||||
jest.advanceTimersByTime(3000);
|
||||
globalThis.setTimeout = ((handler: TimerHandler, timeout?: number) => {
|
||||
observedDelay = timeout;
|
||||
void handler; // avoid executing callback in this assertion-only test
|
||||
return fakeTimer;
|
||||
}) as typeof globalThis.setTimeout;
|
||||
|
||||
expect(proc.kill).toHaveBeenCalledWith('SIGKILL');
|
||||
try {
|
||||
killWithEscalation(proc);
|
||||
expect(observedDelay).toBe(3000);
|
||||
expect(proc.kill).toHaveBeenCalledWith('SIGTERM');
|
||||
expect(proc.kill).toHaveBeenCalledTimes(1);
|
||||
} finally {
|
||||
globalThis.setTimeout = originalSetTimeout;
|
||||
}
|
||||
});
|
||||
|
||||
it('should respect custom grace period', () => {
|
||||
const proc = createMockProcess(null);
|
||||
killWithEscalation(proc, 5000); // Custom 5 second grace period
|
||||
const originalSetTimeout = globalThis.setTimeout;
|
||||
const fakeTimer = {
|
||||
unref: () => fakeTimer,
|
||||
ref: () => fakeTimer,
|
||||
hasRef: () => false,
|
||||
refresh: () => fakeTimer,
|
||||
} as unknown as ReturnType<typeof setTimeout>;
|
||||
|
||||
expect(proc.kill).toHaveBeenCalledWith('SIGTERM');
|
||||
let observedDelay: number | undefined;
|
||||
|
||||
// Advance by less than grace period
|
||||
jest.advanceTimersByTime(4999);
|
||||
expect(proc.kill).toHaveBeenCalledTimes(1); // Still only SIGTERM
|
||||
globalThis.setTimeout = ((handler: TimerHandler, timeout?: number) => {
|
||||
observedDelay = timeout;
|
||||
void handler;
|
||||
return fakeTimer;
|
||||
}) as typeof globalThis.setTimeout;
|
||||
|
||||
// Advance to grace period
|
||||
jest.advanceTimersByTime(1);
|
||||
expect(proc.kill).toHaveBeenCalledWith('SIGKILL');
|
||||
try {
|
||||
killWithEscalation(proc, 5000);
|
||||
expect(observedDelay).toBe(5000);
|
||||
expect(proc.kill).toHaveBeenCalledWith('SIGTERM');
|
||||
expect(proc.kill).toHaveBeenCalledTimes(1);
|
||||
} finally {
|
||||
globalThis.setTimeout = originalSetTimeout;
|
||||
}
|
||||
});
|
||||
|
||||
it('should clear timer when process exits', () => {
|
||||
it('should clear timer when process exits', async () => {
|
||||
const proc = createMockProcess(null);
|
||||
killWithEscalation(proc, 3000);
|
||||
const originalClearTimeout = globalThis.clearTimeout;
|
||||
let clearCalled = false;
|
||||
|
||||
// Simulate immediate exit
|
||||
proc.exitCode = 0;
|
||||
proc.emit('exit', 0);
|
||||
globalThis.clearTimeout = ((id: ReturnType<typeof setTimeout>) => {
|
||||
clearCalled = true;
|
||||
return originalClearTimeout(id);
|
||||
}) as typeof globalThis.clearTimeout;
|
||||
|
||||
// Advance way past grace period
|
||||
jest.advanceTimersByTime(10000);
|
||||
try {
|
||||
killWithEscalation(proc, 50);
|
||||
|
||||
// Should only have SIGTERM, timer was cleared
|
||||
expect(proc.kill).toHaveBeenCalledTimes(1);
|
||||
expect(proc.kill).toHaveBeenCalledWith('SIGTERM');
|
||||
// Simulate immediate exit
|
||||
proc.exitCode = 0;
|
||||
proc.emit('exit', 0);
|
||||
|
||||
await wait(70);
|
||||
|
||||
// Should only have SIGTERM, timer was cleared
|
||||
expect(clearCalled).toBe(true);
|
||||
expect(proc.kill).toHaveBeenCalledTimes(1);
|
||||
expect(proc.kill).toHaveBeenCalledWith('SIGTERM');
|
||||
} finally {
|
||||
globalThis.clearTimeout = originalClearTimeout;
|
||||
}
|
||||
});
|
||||
|
||||
it('should handle process that already exited', () => {
|
||||
it('should handle process that already exited', async () => {
|
||||
const proc = createMockProcess(0); // Already exited
|
||||
killWithEscalation(proc, 3000);
|
||||
killWithEscalation(proc, 10);
|
||||
|
||||
expect(proc.kill).toHaveBeenCalledWith('SIGTERM');
|
||||
|
||||
// Even though exitCode is not null, timer still fires
|
||||
// (because we check exitCode at timer callback time)
|
||||
jest.advanceTimersByTime(3000);
|
||||
await wait(30);
|
||||
|
||||
// SIGKILL should NOT be sent because exitCode is not null
|
||||
expect(proc.kill).toHaveBeenCalledTimes(1);
|
||||
|
||||
@@ -1,10 +1,158 @@
|
||||
/**
|
||||
* Cursor Routes Tests
|
||||
* Tests for daemon start precondition validation logic.
|
||||
* Endpoint contract tests without module-level mocks.
|
||||
*/
|
||||
|
||||
import { describe, it, expect } from 'bun:test';
|
||||
import { getDaemonStartPreconditionError } from '../../../src/web-server/routes/cursor-routes';
|
||||
import { describe, it, expect, beforeAll, beforeEach, afterAll } from 'bun:test';
|
||||
import express from 'express';
|
||||
import * as fs from 'fs';
|
||||
import * as os from 'os';
|
||||
import * as path from 'path';
|
||||
import type { Server } from 'http';
|
||||
import { execFileSync, spawnSync } from 'child_process';
|
||||
|
||||
let server: Server;
|
||||
let baseUrl = '';
|
||||
let tempDir = '';
|
||||
let originalCcsHome: string | undefined;
|
||||
|
||||
let setGlobalConfigDir: (dir: string | undefined) => void;
|
||||
let getCcsDir: () => string;
|
||||
let loadOrCreateUnifiedConfig: () => {
|
||||
cursor?: {
|
||||
enabled?: boolean;
|
||||
port?: number;
|
||||
auto_start?: boolean;
|
||||
ghost_mode?: boolean;
|
||||
model?: string;
|
||||
};
|
||||
};
|
||||
let saveUnifiedConfig: (config: {
|
||||
cursor?: {
|
||||
enabled?: boolean;
|
||||
port?: number;
|
||||
auto_start?: boolean;
|
||||
ghost_mode?: boolean;
|
||||
model?: string;
|
||||
};
|
||||
}) => void;
|
||||
let saveCredentials: (credentials: {
|
||||
accessToken: string;
|
||||
machineId: string;
|
||||
authMethod: 'manual' | 'auto-detect';
|
||||
importedAt: string;
|
||||
}) => void;
|
||||
let deleteCredentials: () => boolean;
|
||||
let checkAuthStatus: () => {
|
||||
authenticated: boolean;
|
||||
expired?: boolean;
|
||||
credentials?: {
|
||||
authMethod?: 'manual' | 'auto-detect';
|
||||
machineId?: string;
|
||||
};
|
||||
};
|
||||
let getTokenStoragePath: () => string;
|
||||
let getDaemonStartPreconditionError: (
|
||||
input: { enabled: boolean; authenticated: boolean; tokenExpired?: boolean }
|
||||
) => { status: number; error: string } | null;
|
||||
|
||||
function seedCursorConfig(overrides: {
|
||||
enabled?: boolean;
|
||||
port?: number;
|
||||
auto_start?: boolean;
|
||||
ghost_mode?: boolean;
|
||||
model?: string;
|
||||
} = {}): void {
|
||||
const config = loadOrCreateUnifiedConfig();
|
||||
config.cursor = {
|
||||
enabled: overrides.enabled ?? true,
|
||||
port: overrides.port ?? 20129,
|
||||
auto_start: overrides.auto_start ?? false,
|
||||
ghost_mode: overrides.ghost_mode ?? true,
|
||||
model: overrides.model ?? 'gpt-5.3-codex',
|
||||
};
|
||||
saveUnifiedConfig(config);
|
||||
}
|
||||
|
||||
function seedCredentials(expired: boolean): void {
|
||||
saveCredentials({
|
||||
accessToken: 'a'.repeat(60),
|
||||
machineId: '1234567890abcdef1234567890abcdef',
|
||||
authMethod: 'manual',
|
||||
importedAt: expired
|
||||
? new Date(Date.now() - 26 * 60 * 60 * 1000).toISOString()
|
||||
: new Date().toISOString(),
|
||||
});
|
||||
}
|
||||
|
||||
beforeAll(async () => {
|
||||
originalCcsHome = process.env.CCS_HOME;
|
||||
tempDir = fs.mkdtempSync(path.join(os.tmpdir(), 'ccs-cursor-routes-test-'));
|
||||
process.env.CCS_HOME = tempDir;
|
||||
|
||||
const configManager = await import('../../../src/utils/config-manager');
|
||||
setGlobalConfigDir = configManager.setGlobalConfigDir;
|
||||
getCcsDir = configManager.getCcsDir;
|
||||
setGlobalConfigDir(undefined);
|
||||
|
||||
const unifiedConfig = await import('../../../src/config/unified-config-loader');
|
||||
loadOrCreateUnifiedConfig = unifiedConfig.loadOrCreateUnifiedConfig;
|
||||
saveUnifiedConfig = unifiedConfig.saveUnifiedConfig;
|
||||
|
||||
const cursorAuth = await import('../../../src/cursor/cursor-auth');
|
||||
saveCredentials = cursorAuth.saveCredentials;
|
||||
deleteCredentials = cursorAuth.deleteCredentials;
|
||||
checkAuthStatus = cursorAuth.checkAuthStatus;
|
||||
getTokenStoragePath = cursorAuth.getTokenStoragePath;
|
||||
|
||||
const cursorRoutesModule = await import('../../../src/web-server/routes/cursor-routes');
|
||||
getDaemonStartPreconditionError = cursorRoutesModule.getDaemonStartPreconditionError;
|
||||
|
||||
const app = express();
|
||||
app.use(express.json());
|
||||
app.use('/api/cursor', cursorRoutesModule.default);
|
||||
|
||||
server = app.listen(0, '127.0.0.1');
|
||||
await new Promise<void>((resolve) => server.on('listening', () => resolve()));
|
||||
const address = server.address();
|
||||
if (!address || typeof address === 'string') {
|
||||
throw new Error('Unable to resolve test server port');
|
||||
}
|
||||
baseUrl = `http://127.0.0.1:${address.port}`;
|
||||
});
|
||||
|
||||
beforeEach(() => {
|
||||
fs.rmSync(tempDir, { recursive: true, force: true });
|
||||
fs.mkdirSync(tempDir, { recursive: true });
|
||||
|
||||
setGlobalConfigDir(undefined);
|
||||
const ccsDir = getCcsDir();
|
||||
if (!fs.existsSync(ccsDir)) {
|
||||
fs.mkdirSync(ccsDir, { recursive: true });
|
||||
}
|
||||
|
||||
seedCursorConfig();
|
||||
|
||||
// Ensure clean auth state for each test.
|
||||
deleteCredentials();
|
||||
});
|
||||
|
||||
afterAll(async () => {
|
||||
if (server) {
|
||||
await new Promise<void>((resolve) => server.close(() => resolve()));
|
||||
}
|
||||
|
||||
if (originalCcsHome !== undefined) {
|
||||
process.env.CCS_HOME = originalCcsHome;
|
||||
} else {
|
||||
delete process.env.CCS_HOME;
|
||||
}
|
||||
setGlobalConfigDir(undefined);
|
||||
|
||||
if (tempDir && fs.existsSync(tempDir)) {
|
||||
fs.rmSync(tempDir, { recursive: true, force: true });
|
||||
}
|
||||
});
|
||||
|
||||
describe('Cursor Routes Logic', () => {
|
||||
describe('POST /daemon/start preconditions', () => {
|
||||
@@ -57,4 +205,212 @@ describe('Cursor Routes Logic', () => {
|
||||
expect(result).toBeNull();
|
||||
});
|
||||
});
|
||||
|
||||
describe('HTTP contracts', () => {
|
||||
it('GET /api/cursor/status returns current state', async () => {
|
||||
const res = await fetch(`${baseUrl}/api/cursor/status`);
|
||||
expect(res.status).toBe(200);
|
||||
|
||||
const json = (await res.json()) as {
|
||||
enabled: boolean;
|
||||
authenticated: boolean;
|
||||
token_expired: boolean;
|
||||
daemon_running: boolean;
|
||||
port: number;
|
||||
};
|
||||
|
||||
expect(json.enabled).toBe(true);
|
||||
expect(json.authenticated).toBe(false);
|
||||
expect(json.token_expired).toBe(false);
|
||||
expect(json.daemon_running).toBe(false);
|
||||
expect(json.port).toBe(20129);
|
||||
});
|
||||
|
||||
it('POST /api/cursor/auth/import validates required fields', async () => {
|
||||
const res = await fetch(`${baseUrl}/api/cursor/auth/import`, {
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({ accessToken: 'only-token' }),
|
||||
});
|
||||
|
||||
expect(res.status).toBe(400);
|
||||
const json = (await res.json()) as { error?: string };
|
||||
expect(json.error).toContain('Missing accessToken or machineId');
|
||||
});
|
||||
|
||||
it('POST /api/cursor/auth/import rejects invalid token format', async () => {
|
||||
const res = await fetch(`${baseUrl}/api/cursor/auth/import`, {
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({
|
||||
accessToken: 'short',
|
||||
machineId: 'bad',
|
||||
}),
|
||||
});
|
||||
|
||||
expect(res.status).toBe(400);
|
||||
const json = (await res.json()) as { error?: string };
|
||||
expect(json.error).toContain('Invalid token or machine ID format');
|
||||
expect(checkAuthStatus().authenticated).toBe(false);
|
||||
});
|
||||
|
||||
it('POST /api/cursor/auth/import persists valid credentials', async () => {
|
||||
const res = await fetch(`${baseUrl}/api/cursor/auth/import`, {
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({
|
||||
accessToken: 'a'.repeat(60),
|
||||
machineId: '1234567890abcdef1234567890abcdef',
|
||||
}),
|
||||
});
|
||||
|
||||
expect(res.status).toBe(200);
|
||||
expect(checkAuthStatus().authenticated).toBe(true);
|
||||
});
|
||||
|
||||
it('POST /api/cursor/auth/auto-detect returns 404 when no token source found', async () => {
|
||||
const originalHome = process.env.HOME;
|
||||
const isolatedHome = path.join(tempDir, 'auto-detect-empty-home');
|
||||
process.env.HOME = isolatedHome;
|
||||
fs.mkdirSync(isolatedHome, { recursive: true });
|
||||
|
||||
try {
|
||||
const res = await fetch(`${baseUrl}/api/cursor/auth/auto-detect`, {
|
||||
method: 'POST',
|
||||
});
|
||||
|
||||
expect(res.status).toBe(404);
|
||||
const json = (await res.json()) as { error?: string };
|
||||
expect(typeof json.error).toBe('string');
|
||||
expect(json.error?.length).toBeGreaterThan(0);
|
||||
} finally {
|
||||
if (originalHome !== undefined) {
|
||||
process.env.HOME = originalHome;
|
||||
} else {
|
||||
delete process.env.HOME;
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
it('POST /api/cursor/auth/auto-detect persists credentials on success', async () => {
|
||||
if (process.platform === 'win32') {
|
||||
return;
|
||||
}
|
||||
|
||||
const sqliteCheck = spawnSync('sqlite3', ['--version'], { stdio: 'ignore' });
|
||||
if (sqliteCheck.status !== 0) {
|
||||
return;
|
||||
}
|
||||
|
||||
const originalHome = process.env.HOME;
|
||||
const fakeHome = path.join(tempDir, 'fake-home');
|
||||
process.env.HOME = fakeHome;
|
||||
fs.mkdirSync(fakeHome, { recursive: true });
|
||||
|
||||
const token = 'a'.repeat(60);
|
||||
const machineId = '1234567890abcdef1234567890abcdef';
|
||||
|
||||
try {
|
||||
const dbPath = getTokenStoragePath();
|
||||
fs.mkdirSync(path.dirname(dbPath), { recursive: true });
|
||||
execFileSync('sqlite3', [dbPath, 'CREATE TABLE IF NOT EXISTS itemTable (key TEXT PRIMARY KEY, value TEXT);']);
|
||||
execFileSync('sqlite3', [
|
||||
dbPath,
|
||||
`INSERT OR REPLACE INTO itemTable (key, value) VALUES ('cursorAuth/accessToken', '${token}');`,
|
||||
]);
|
||||
execFileSync('sqlite3', [
|
||||
dbPath,
|
||||
`INSERT OR REPLACE INTO itemTable (key, value) VALUES ('storage.serviceMachineId', '${machineId}');`,
|
||||
]);
|
||||
|
||||
const res = await fetch(`${baseUrl}/api/cursor/auth/auto-detect`, {
|
||||
method: 'POST',
|
||||
});
|
||||
|
||||
expect(res.status).toBe(200);
|
||||
const json = (await res.json()) as { success?: boolean };
|
||||
expect(json.success).toBe(true);
|
||||
|
||||
const auth = checkAuthStatus();
|
||||
expect(auth.authenticated).toBe(true);
|
||||
expect(auth.credentials?.authMethod).toBe('auto-detect');
|
||||
expect(auth.credentials?.machineId).toBe(machineId);
|
||||
} finally {
|
||||
if (originalHome !== undefined) {
|
||||
process.env.HOME = originalHome;
|
||||
} else {
|
||||
delete process.env.HOME;
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
it('POST /api/cursor/daemon/start returns 400 when integration is disabled', async () => {
|
||||
seedCursorConfig({ enabled: false });
|
||||
seedCredentials(false);
|
||||
|
||||
const res = await fetch(`${baseUrl}/api/cursor/daemon/start`, { method: 'POST' });
|
||||
expect(res.status).toBe(400);
|
||||
const json = (await res.json()) as { success?: boolean; error?: string };
|
||||
expect(json.success).toBe(false);
|
||||
expect(json.error).toContain('disabled');
|
||||
});
|
||||
|
||||
it('POST /api/cursor/daemon/start returns 401 when unauthenticated', async () => {
|
||||
seedCursorConfig({ enabled: true });
|
||||
|
||||
const res = await fetch(`${baseUrl}/api/cursor/daemon/start`, { method: 'POST' });
|
||||
expect(res.status).toBe(401);
|
||||
const json = (await res.json()) as { error?: string };
|
||||
expect(json.error).toContain('authentication required');
|
||||
});
|
||||
|
||||
it('POST /api/cursor/daemon/start returns 401 when token is expired', async () => {
|
||||
seedCursorConfig({ enabled: true });
|
||||
seedCredentials(true);
|
||||
|
||||
const res = await fetch(`${baseUrl}/api/cursor/daemon/start`, { method: 'POST' });
|
||||
expect(res.status).toBe(401);
|
||||
const json = (await res.json()) as { error?: string };
|
||||
expect(json.error).toContain('expired');
|
||||
});
|
||||
|
||||
it(
|
||||
'POST /api/cursor/daemon/start starts daemon and /daemon/stop stops it',
|
||||
async () => {
|
||||
const port = 15000 + Math.floor(Math.random() * 20000);
|
||||
seedCursorConfig({ enabled: true, port });
|
||||
seedCredentials(false);
|
||||
|
||||
const startRes = await fetch(`${baseUrl}/api/cursor/daemon/start`, { method: 'POST' });
|
||||
expect(startRes.status).toBe(200);
|
||||
|
||||
const startJson = (await startRes.json()) as { success?: boolean; pid?: number };
|
||||
expect(startJson.success).toBe(true);
|
||||
expect(typeof startJson.pid).toBe('number');
|
||||
|
||||
const stopRes = await fetch(`${baseUrl}/api/cursor/daemon/stop`, { method: 'POST' });
|
||||
expect(stopRes.status).toBe(200);
|
||||
const stopJson = (await stopRes.json()) as { success?: boolean };
|
||||
expect(stopJson.success).toBe(true);
|
||||
},
|
||||
35000
|
||||
);
|
||||
|
||||
it('POST /api/cursor/daemon/stop returns success when daemon is not running', async () => {
|
||||
const res = await fetch(`${baseUrl}/api/cursor/daemon/stop`, { method: 'POST' });
|
||||
expect(res.status).toBe(200);
|
||||
const json = (await res.json()) as { success?: boolean };
|
||||
expect(json.success).toBe(true);
|
||||
});
|
||||
|
||||
it('GET /api/cursor/models returns current model and list payload', async () => {
|
||||
const res = await fetch(`${baseUrl}/api/cursor/models`);
|
||||
expect(res.status).toBe(200);
|
||||
|
||||
const json = (await res.json()) as { models: Array<{ id: string }>; current: string };
|
||||
expect(Array.isArray(json.models)).toBe(true);
|
||||
expect(json.models.length).toBeGreaterThan(0);
|
||||
expect(json.current).toBe('gpt-5.3-codex');
|
||||
});
|
||||
});
|
||||
});
|
||||
Reference in new issue
Block a user