feat(ide): add auth token support to IdeClient (#8490)

Co-authored-by: Jack Wotherspoon <jackwoth@google.com>
This commit is contained in:
Shreya Keshive
2025-09-15 19:22:31 -04:00
committed by GitHub
parent 079526fd3b
commit 2cc0c1a808
2 changed files with 40 additions and 0 deletions

View File

@@ -656,4 +656,34 @@ describe('IdeClient', () => {
expect(ideClient.isDiffingEnabled()).toBe(true);
});
});
describe('authentication', () => {
it('should connect with an auth token if provided in the discovery file', async () => {
const authToken = 'test-auth-token';
const config = { port: '8080', authToken };
vi.mocked(fs.promises.readFile).mockResolvedValue(JSON.stringify(config));
(
vi.mocked(fs.promises.readdir) as Mock<
(path: fs.PathLike) => Promise<string[]>
>
).mockResolvedValue([]);
const ideClient = await IdeClient.getInstance();
await ideClient.connect();
expect(StreamableHTTPClientTransport).toHaveBeenCalledWith(
new URL('http://localhost:8080/mcp'),
expect.objectContaining({
requestInit: {
headers: {
Authorization: `Bearer ${authToken}`,
},
},
}),
);
expect(ideClient.getConnectionStatus().status).toBe(
IDEConnectionStatus.Connected,
);
});
});
});