mirror of
https://github.com/QwenLM/qwen-code.git
synced 2025-12-21 09:17:53 +00:00
Add Google credentials provider for authenticating with MCP servers (#4748)
This commit is contained in:
67
packages/core/src/mcp/google-auth-provider.test.ts
Normal file
67
packages/core/src/mcp/google-auth-provider.test.ts
Normal file
@@ -0,0 +1,67 @@
|
||||
/**
|
||||
* @license
|
||||
* Copyright 2025 Google LLC
|
||||
* SPDX-License-Identifier: Apache-2.0
|
||||
*/
|
||||
|
||||
import { GoogleAuth } from 'google-auth-library';
|
||||
import { GoogleCredentialProvider } from './google-auth-provider.js';
|
||||
import { vi, describe, beforeEach, it, expect, Mock } from 'vitest';
|
||||
import { MCPServerConfig } from '../config/config.js';
|
||||
|
||||
vi.mock('google-auth-library');
|
||||
|
||||
describe('GoogleCredentialProvider', () => {
|
||||
it('should throw an error if no scopes are provided', () => {
|
||||
expect(() => new GoogleCredentialProvider()).toThrow(
|
||||
'Scopes must be provided in the oauth config for Google Credentials provider',
|
||||
);
|
||||
});
|
||||
|
||||
it('should use scopes from the config if provided', () => {
|
||||
const config = {
|
||||
oauth: {
|
||||
scopes: ['scope1', 'scope2'],
|
||||
},
|
||||
} as MCPServerConfig;
|
||||
new GoogleCredentialProvider(config);
|
||||
expect(GoogleAuth).toHaveBeenCalledWith({
|
||||
scopes: ['scope1', 'scope2'],
|
||||
});
|
||||
});
|
||||
|
||||
describe('with provider instance', () => {
|
||||
let provider: GoogleCredentialProvider;
|
||||
|
||||
beforeEach(() => {
|
||||
const config = {
|
||||
oauth: {
|
||||
scopes: ['scope1', 'scope2'],
|
||||
},
|
||||
} as MCPServerConfig;
|
||||
provider = new GoogleCredentialProvider(config);
|
||||
vi.clearAllMocks();
|
||||
});
|
||||
|
||||
it('should return credentials', async () => {
|
||||
const mockClient = {
|
||||
getAccessToken: vi.fn().mockResolvedValue({ token: 'test-token' }),
|
||||
};
|
||||
(GoogleAuth.prototype.getClient as Mock).mockResolvedValue(mockClient);
|
||||
|
||||
const credentials = await provider.tokens();
|
||||
|
||||
expect(credentials?.access_token).toBe('test-token');
|
||||
});
|
||||
|
||||
it('should return undefined if access token is not available', async () => {
|
||||
const mockClient = {
|
||||
getAccessToken: vi.fn().mockResolvedValue({ token: null }),
|
||||
};
|
||||
(GoogleAuth.prototype.getClient as Mock).mockResolvedValue(mockClient);
|
||||
|
||||
const credentials = await provider.tokens();
|
||||
expect(credentials).toBeUndefined();
|
||||
});
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user