@hibanacloud/core 0.3.15

package/dist/src/code_assist/oauth2.test.js

@@ -0,0 +1,818 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { clearCachedCredentialFile, clearOauthClientCache, getOauthClient, resetOauthClientForTesting, } from './oauth2.js';
+import { UserAccountManager } from '../utils/userAccountManager.js';
+import { OAuth2Client, Compute } from 'google-auth-library';
+import * as fs from 'node:fs';
+import * as path from 'node:path';
+import http from 'node:http';
+import open from 'open';
+import crypto from 'node:crypto';
+import * as os from 'node:os';
+import { AuthType } from '../core/contentGenerator.js';
+import readline from 'node:readline';
+import { FORCE_ENCRYPTED_FILE_ENV_VAR } from '../mcp/token-storage/index.js';
+import { QWEN_DIR } from '../utils/paths.js';
+vi.mock('os', async (importOriginal) => {
+    const os = await importOriginal();
+    return {
+        ...os,
+        homedir: vi.fn(),
+    };
+});
+vi.mock('google-auth-library');
+vi.mock('http');
+vi.mock('open');
+vi.mock('crypto');
+vi.mock('node:readline');
+vi.mock('../utils/browser.js', () => ({
+    shouldAttemptBrowserLaunch: () => true,
+}));
+vi.mock('./oauth-credential-storage.js', () => ({
+    OAuthCredentialStorage: {
+        saveCredentials: vi.fn(),
+        loadCredentials: vi.fn(),
+        clearCredentials: vi.fn(),
+    },
+}));
+const mockConfig = {
+    getNoBrowser: () => false,
+    getProxy: () => 'http://test.proxy.com:8080',
+    isBrowserLaunchSuppressed: () => false,
+};
+// Mock fetch globally
+global.fetch = vi.fn();
+describe('oauth2', () => {
+    describe('with encrypted flag false', () => {
+        let tempHomeDir;
+        beforeEach(() => {
+            process.env[FORCE_ENCRYPTED_FILE_ENV_VAR] = 'false';
+            tempHomeDir = fs.mkdtempSync(path.join(os.tmpdir(), 'qwen-code-test-home-'));
+            os.homedir.mockReturnValue(tempHomeDir);
+        });
+        afterEach(() => {
+            fs.rmSync(tempHomeDir, { recursive: true, force: true });
+            vi.clearAllMocks();
+            resetOauthClientForTesting();
+            vi.unstubAllEnvs();
+        });
+        it('should perform a web login', async () => {
+            const mockAuthUrl = 'https://example.com/auth';
+            const mockCode = 'test-code';
+            const mockState = 'test-state';
+            const mockTokens = {
+                access_token: 'test-access-token',
+                refresh_token: 'test-refresh-token',
+            };
+            const mockGenerateAuthUrl = vi.fn().mockReturnValue(mockAuthUrl);
+            const mockGetToken = vi.fn().mockResolvedValue({ tokens: mockTokens });
+            const mockSetCredentials = vi.fn();
+            const mockGetAccessToken = vi
+                .fn()
+                .mockResolvedValue({ token: 'mock-access-token' });
+            const mockOAuth2Client = {
+                generateAuthUrl: mockGenerateAuthUrl,
+                getToken: mockGetToken,
+                setCredentials: mockSetCredentials,
+                getAccessToken: mockGetAccessToken,
+                credentials: mockTokens,
+                on: vi.fn(),
+            };
+            OAuth2Client.mockImplementation(() => mockOAuth2Client);
+            vi.spyOn(crypto, 'randomBytes').mockReturnValue(mockState);
+            open.mockImplementation(async () => ({ on: vi.fn() }));
+            // Mock the UserInfo API response
+            global.fetch.mockResolvedValue({
+                ok: true,
+                json: vi
+                    .fn()
+                    .mockResolvedValue({ email: 'test-google-account@gmail.com' }),
+            });
+            let requestCallback;
+            let serverListeningCallback;
+            const serverListeningPromise = new Promise((resolve) => (serverListeningCallback = resolve));
+            let capturedPort = 0;
+            const mockHttpServer = {
+                listen: vi.fn((port, _host, callback) => {
+                    capturedPort = port;
+                    if (callback) {
+                        callback();
+                    }
+                    serverListeningCallback(undefined);
+                }),
+                close: vi.fn((callback) => {
+                    if (callback) {
+                        callback();
+                    }
+                }),
+                on: vi.fn(),
+                address: () => ({ port: capturedPort }),
+            };
+            http.createServer.mockImplementation((cb) => {
+                requestCallback = cb;
+                return mockHttpServer;
+            });
+            const clientPromise = getOauthClient(AuthType.LOGIN_WITH_GOOGLE, mockConfig);
+            // wait for server to start listening.
+            await serverListeningPromise;
+            const mockReq = {
+                url: `/oauth2callback?code=${mockCode}&state=${mockState}`,
+            };
+            const mockRes = {
+                writeHead: vi.fn(),
+                end: vi.fn(),
+            };
+            await requestCallback(mockReq, mockRes);
+            const client = await clientPromise;
+            expect(client).toBe(mockOAuth2Client);
+            expect(open).toHaveBeenCalledWith(mockAuthUrl);
+            expect(mockGetToken).toHaveBeenCalledWith({
+                code: mockCode,
+                redirect_uri: `http://localhost:${capturedPort}/oauth2callback`,
+            });
+            expect(mockSetCredentials).toHaveBeenCalledWith(mockTokens);
+            // Verify Google Account was cached
+            const googleAccountPath = path.join(tempHomeDir, QWEN_DIR, 'google_accounts.json');
+            expect(fs.existsSync(googleAccountPath)).toBe(true);
+            const cachedGoogleAccount = fs.readFileSync(googleAccountPath, 'utf-8');
+            expect(JSON.parse(cachedGoogleAccount)).toEqual({
+                active: 'test-google-account@gmail.com',
+                old: [],
+            });
+            // Verify the getCachedGoogleAccount function works
+            const userAccountManager = new UserAccountManager();
+            expect(userAccountManager.getCachedGoogleAccount()).toBe('test-google-account@gmail.com');
+        });
+        it('should perform login with user code', async () => {
+            const mockConfigWithNoBrowser = {
+                getNoBrowser: () => true,
+                getProxy: () => 'http://test.proxy.com:8080',
+                isBrowserLaunchSuppressed: () => true,
+            };
+            const mockCodeVerifier = {
+                codeChallenge: 'test-challenge',
+                codeVerifier: 'test-verifier',
+            };
+            const mockAuthUrl = 'https://example.com/auth-user-code';
+            const mockCode = 'test-user-code';
+            const mockTokens = {
+                access_token: 'test-access-token-user-code',
+                refresh_token: 'test-refresh-token-user-code',
+            };
+            const mockGenerateAuthUrl = vi.fn().mockReturnValue(mockAuthUrl);
+            const mockGetToken = vi.fn().mockResolvedValue({ tokens: mockTokens });
+            const mockSetCredentials = vi.fn();
+            const mockGenerateCodeVerifierAsync = vi
+                .fn()
+                .mockResolvedValue(mockCodeVerifier);
+            const mockOAuth2Client = {
+                generateAuthUrl: mockGenerateAuthUrl,
+                getToken: mockGetToken,
+                setCredentials: mockSetCredentials,
+                generateCodeVerifierAsync: mockGenerateCodeVerifierAsync,
+                on: vi.fn(),
+            };
+            OAuth2Client.mockImplementation(() => mockOAuth2Client);
+            const mockReadline = {
+                question: vi.fn((_query, callback) => callback(mockCode)),
+                close: vi.fn(),
+            };
+            readline.createInterface.mockReturnValue(mockReadline);
+            const consoleLogSpy = vi
+                .spyOn(console, 'log')
+                .mockImplementation(() => { });
+            const client = await getOauthClient(AuthType.LOGIN_WITH_GOOGLE, mockConfigWithNoBrowser);
+            expect(client).toBe(mockOAuth2Client);
+            // Verify the auth flow
+            expect(mockGenerateCodeVerifierAsync).toHaveBeenCalled();
+            expect(mockGenerateAuthUrl).toHaveBeenCalled();
+            expect(consoleLogSpy).toHaveBeenCalledWith(expect.stringContaining(mockAuthUrl));
+            expect(mockReadline.question).toHaveBeenCalledWith('Enter the authorization code: ', expect.any(Function));
+            expect(mockGetToken).toHaveBeenCalledWith({
+                code: mockCode,
+                codeVerifier: mockCodeVerifier.codeVerifier,
+                redirect_uri: 'https://codeassist.google.com/authcode',
+            });
+            expect(mockSetCredentials).toHaveBeenCalledWith(mockTokens);
+            consoleLogSpy.mockRestore();
+        });
+        describe('in Cloud Shell', () => {
+            const mockGetAccessToken = vi.fn();
+            let mockComputeClient;
+            beforeEach(() => {
+                mockGetAccessToken.mockResolvedValue({ token: 'test-access-token' });
+                mockComputeClient = {
+                    credentials: { refresh_token: 'test-refresh-token' },
+                    getAccessToken: mockGetAccessToken,
+                };
+                Compute.mockImplementation(() => mockComputeClient);
+            });
+            it('should attempt to load cached credentials first', async () => {
+                const cachedCreds = { refresh_token: 'cached-token' };
+                const credsPath = path.join(tempHomeDir, '.qwen', 'oauth_creds.json');
+                await fs.promises.mkdir(path.dirname(credsPath), { recursive: true });
+                await fs.promises.writeFile(credsPath, JSON.stringify(cachedCreds));
+                const mockClient = {
+                    setCredentials: vi.fn(),
+                    getAccessToken: vi.fn().mockResolvedValue({ token: 'test-token' }),
+                    getTokenInfo: vi.fn().mockResolvedValue({}),
+                    on: vi.fn(),
+                };
+                // To mock the new OAuth2Client() inside the function
+                OAuth2Client.mockImplementation(() => mockClient);
+                await getOauthClient(AuthType.LOGIN_WITH_GOOGLE, mockConfig);
+                expect(mockClient.setCredentials).toHaveBeenCalledWith(cachedCreds);
+                expect(mockClient.getAccessToken).toHaveBeenCalled();
+                expect(mockClient.getTokenInfo).toHaveBeenCalled();
+                expect(Compute).not.toHaveBeenCalled(); // Should not fetch new client if cache is valid
+            });
+            it('should use Compute to get a client if no cached credentials exist', async () => {
+                await getOauthClient(AuthType.CLOUD_SHELL, mockConfig);
+                expect(Compute).toHaveBeenCalledWith({});
+                expect(mockGetAccessToken).toHaveBeenCalled();
+            });
+            it('should not cache the credentials after fetching them via ADC', async () => {
+                const newCredentials = { refresh_token: 'new-adc-token' };
+                mockComputeClient.credentials = newCredentials;
+                mockGetAccessToken.mockResolvedValue({ token: 'new-adc-token' });
+                await getOauthClient(AuthType.CLOUD_SHELL, mockConfig);
+                const credsPath = path.join(tempHomeDir, '.qwen', 'oauth_creds.json');
+                expect(fs.existsSync(credsPath)).toBe(false);
+            });
+            it('should return the Compute client on successful ADC authentication', async () => {
+                const client = await getOauthClient(AuthType.CLOUD_SHELL, mockConfig);
+                expect(client).toBe(mockComputeClient);
+            });
+            it('should throw an error if ADC fails', async () => {
+                const testError = new Error('ADC Failed');
+                mockGetAccessToken.mockRejectedValue(testError);
+                await expect(getOauthClient(AuthType.CLOUD_SHELL, mockConfig)).rejects.toThrow('Could not authenticate using Cloud Shell credentials. Please select a different authentication method or ensure you are in a properly configured environment. Error: ADC Failed');
+            });
+        });
+        describe('credential loading order', () => {
+            it('should prioritize default cached credentials over GOOGLE_APPLICATION_CREDENTIALS', async () => {
+                // Setup default cached credentials
+                const defaultCreds = { refresh_token: 'default-cached-token' };
+                const defaultCredsPath = path.join(tempHomeDir, '.qwen', 'oauth_creds.json');
+                await fs.promises.mkdir(path.dirname(defaultCredsPath), {
+                    recursive: true,
+                });
+                await fs.promises.writeFile(defaultCredsPath, JSON.stringify(defaultCreds));
+                // Setup credentials via environment variable
+                const envCreds = { refresh_token: 'env-var-token' };
+                const envCredsPath = path.join(tempHomeDir, 'env_creds.json');
+                await fs.promises.writeFile(envCredsPath, JSON.stringify(envCreds));
+                vi.stubEnv('GOOGLE_APPLICATION_CREDENTIALS', envCredsPath);
+                const mockClient = {
+                    setCredentials: vi.fn(),
+                    getAccessToken: vi.fn().mockResolvedValue({ token: 'test-token' }),
+                    getTokenInfo: vi.fn().mockResolvedValue({}),
+                    on: vi.fn(),
+                };
+                OAuth2Client.mockImplementation(() => mockClient);
+                await getOauthClient(AuthType.LOGIN_WITH_GOOGLE, mockConfig);
+                // Assert the correct credentials were used
+                expect(mockClient.setCredentials).toHaveBeenCalledWith(defaultCreds);
+                expect(mockClient.setCredentials).not.toHaveBeenCalledWith(envCreds);
+            });
+            it('should fall back to GOOGLE_APPLICATION_CREDENTIALS if default cache is missing', async () => {
+                // Setup credentials via environment variable
+                const envCreds = { refresh_token: 'env-var-token' };
+                const envCredsPath = path.join(tempHomeDir, 'env_creds.json');
+                await fs.promises.writeFile(envCredsPath, JSON.stringify(envCreds));
+                vi.stubEnv('GOOGLE_APPLICATION_CREDENTIALS', envCredsPath);
+                const mockClient = {
+                    setCredentials: vi.fn(),
+                    getAccessToken: vi.fn().mockResolvedValue({ token: 'test-token' }),
+                    getTokenInfo: vi.fn().mockResolvedValue({}),
+                    on: vi.fn(),
+                };
+                OAuth2Client.mockImplementation(() => mockClient);
+                await getOauthClient(AuthType.LOGIN_WITH_GOOGLE, mockConfig);
+                // Assert the correct credentials were used
+                expect(mockClient.setCredentials).toHaveBeenCalledWith(envCreds);
+            });
+        });
+        describe('with GCP environment variables', () => {
+            it('should use GOOGLE_CLOUD_ACCESS_TOKEN when GOOGLE_GENAI_USE_GCA is true', async () => {
+                vi.stubEnv('GOOGLE_GENAI_USE_GCA', 'true');
+                vi.stubEnv('GOOGLE_CLOUD_ACCESS_TOKEN', 'gcp-access-token');
+                const mockSetCredentials = vi.fn();
+                const mockGetAccessToken = vi
+                    .fn()
+                    .mockResolvedValue({ token: 'gcp-access-token' });
+                const mockOAuth2Client = {
+                    setCredentials: mockSetCredentials,
+                    getAccessToken: mockGetAccessToken,
+                    on: vi.fn(),
+                };
+                OAuth2Client.mockImplementation(() => mockOAuth2Client);
+                // Mock the UserInfo API response for fetchAndCacheUserInfo
+                global.fetch.mockResolvedValue({
+                    ok: true,
+                    json: vi
+                        .fn()
+                        .mockResolvedValue({ email: 'test-gcp-account@gmail.com' }),
+                });
+                const client = await getOauthClient(AuthType.LOGIN_WITH_GOOGLE, mockConfig);
+                expect(client).toBe(mockOAuth2Client);
+                expect(mockSetCredentials).toHaveBeenCalledWith({
+                    access_token: 'gcp-access-token',
+                });
+                // Verify fetchAndCacheUserInfo was effectively called
+                expect(mockGetAccessToken).toHaveBeenCalled();
+                expect(global.fetch).toHaveBeenCalledWith('https://www.googleapis.com/oauth2/v2/userinfo', {
+                    headers: {
+                        Authorization: 'Bearer gcp-access-token',
+                    },
+                });
+                // Verify Google Account was cached
+                const googleAccountPath = path.join(tempHomeDir, '.qwen', 'google_accounts.json');
+                const cachedContent = fs.readFileSync(googleAccountPath, 'utf-8');
+                expect(JSON.parse(cachedContent)).toEqual({
+                    active: 'test-gcp-account@gmail.com',
+                    old: [],
+                });
+            });
+            it('should not use GCP token if GOOGLE_CLOUD_ACCESS_TOKEN is not set', async () => {
+                vi.stubEnv('GOOGLE_GENAI_USE_GCA', 'true');
+                const mockSetCredentials = vi.fn();
+                const mockGetAccessToken = vi
+                    .fn()
+                    .mockResolvedValue({ token: 'cached-access-token' });
+                const mockGetTokenInfo = vi.fn().mockResolvedValue({});
+                const mockOAuth2Client = {
+                    setCredentials: mockSetCredentials,
+                    getAccessToken: mockGetAccessToken,
+                    getTokenInfo: mockGetTokenInfo,
+                    on: vi.fn(),
+                };
+                OAuth2Client.mockImplementation(() => mockOAuth2Client);
+                // Make it fall through to cached credentials path
+                const cachedCreds = { refresh_token: 'cached-token' };
+                const credsPath = path.join(tempHomeDir, '.qwen', 'oauth_creds.json');
+                await fs.promises.mkdir(path.dirname(credsPath), { recursive: true });
+                await fs.promises.writeFile(credsPath, JSON.stringify(cachedCreds));
+                await getOauthClient(AuthType.LOGIN_WITH_GOOGLE, mockConfig);
+                // It should be called with the cached credentials, not the GCP access token.
+                expect(mockSetCredentials).toHaveBeenCalledTimes(1);
+                expect(mockSetCredentials).toHaveBeenCalledWith(cachedCreds);
+            });
+            it('should not use GCP token if GOOGLE_GENAI_USE_GCA is not set', async () => {
+                vi.stubEnv('GOOGLE_CLOUD_ACCESS_TOKEN', 'gcp-access-token');
+                const mockSetCredentials = vi.fn();
+                const mockGetAccessToken = vi
+                    .fn()
+                    .mockResolvedValue({ token: 'cached-access-token' });
+                const mockGetTokenInfo = vi.fn().mockResolvedValue({});
+                const mockOAuth2Client = {
+                    setCredentials: mockSetCredentials,
+                    getAccessToken: mockGetAccessToken,
+                    getTokenInfo: mockGetTokenInfo,
+                    on: vi.fn(),
+                };
+                OAuth2Client.mockImplementation(() => mockOAuth2Client);
+                // Make it fall through to cached credentials path
+                const cachedCreds = { refresh_token: 'cached-token' };
+                const credsPath = path.join(tempHomeDir, '.qwen', 'oauth_creds.json');
+                await fs.promises.mkdir(path.dirname(credsPath), { recursive: true });
+                await fs.promises.writeFile(credsPath, JSON.stringify(cachedCreds));
+                await getOauthClient(AuthType.LOGIN_WITH_GOOGLE, mockConfig);
+                // It should be called with the cached credentials, not the GCP access token.
+                expect(mockSetCredentials).toHaveBeenCalledTimes(1);
+                expect(mockSetCredentials).toHaveBeenCalledWith(cachedCreds);
+            });
+        });
+        describe('error handling', () => {
+            it('should handle browser launch failure with FatalAuthenticationError', async () => {
+                const mockError = new Error('Browser launch failed');
+                open.mockRejectedValue(mockError);
+                const mockOAuth2Client = {
+                    generateAuthUrl: vi.fn().mockReturnValue('https://example.com/auth'),
+                    on: vi.fn(),
+                };
+                OAuth2Client.mockImplementation(() => mockOAuth2Client);
+                await expect(getOauthClient(AuthType.LOGIN_WITH_GOOGLE, mockConfig)).rejects.toThrow('Failed to open browser: Browser launch failed');
+            });
+            it('should handle authentication timeout with proper error message', async () => {
+                const mockAuthUrl = 'https://example.com/auth';
+                const mockOAuth2Client = {
+                    generateAuthUrl: vi.fn().mockReturnValue(mockAuthUrl),
+                    on: vi.fn(),
+                };
+                OAuth2Client.mockImplementation(() => mockOAuth2Client);
+                open.mockImplementation(async () => ({ on: vi.fn() }));
+                const mockHttpServer = {
+                    listen: vi.fn(),
+                    close: vi.fn(),
+                    on: vi.fn(),
+                    address: () => ({ port: 3000 }),
+                };
+                http.createServer.mockImplementation(() => mockHttpServer);
+                // Mock setTimeout to trigger timeout immediately
+                const originalSetTimeout = global.setTimeout;
+                global.setTimeout = vi.fn((callback) => (callback(), {}));
+                await expect(getOauthClient(AuthType.LOGIN_WITH_GOOGLE, mockConfig)).rejects.toThrow('Authentication timed out after 5 minutes. The browser tab may have gotten stuck in a loading state. Please try again or use NO_BROWSER=true for manual authentication.');
+                global.setTimeout = originalSetTimeout;
+            });
+            it('should handle OAuth callback errors with descriptive messages', async () => {
+                const mockAuthUrl = 'https://example.com/auth';
+                const mockOAuth2Client = {
+                    generateAuthUrl: vi.fn().mockReturnValue(mockAuthUrl),
+                    on: vi.fn(),
+                };
+                OAuth2Client.mockImplementation(() => mockOAuth2Client);
+                open.mockImplementation(async () => ({ on: vi.fn() }));
+                let requestCallback;
+                let serverListeningCallback;
+                const serverListeningPromise = new Promise((resolve) => (serverListeningCallback = resolve));
+                const mockHttpServer = {
+                    listen: vi.fn((_port, _host, callback) => {
+                        if (callback)
+                            callback();
+                        serverListeningCallback(undefined);
+                    }),
+                    close: vi.fn(),
+                    on: vi.fn(),
+                    address: () => ({ port: 3000 }),
+                };
+                http.createServer.mockImplementation((cb) => {
+                    requestCallback = cb;
+                    return mockHttpServer;
+                });
+                const clientPromise = getOauthClient(AuthType.LOGIN_WITH_GOOGLE, mockConfig);
+                await serverListeningPromise;
+                // Test OAuth error with description
+                const mockReq = {
+                    url: '/oauth2callback?error=access_denied&error_description=User+denied+access',
+                };
+                const mockRes = {
+                    writeHead: vi.fn(),
+                    end: vi.fn(),
+                };
+                await expect(async () => {
+                    await requestCallback(mockReq, mockRes);
+                    await clientPromise;
+                }).rejects.toThrow('Google OAuth error: access_denied. User denied access');
+            });
+            it('should handle OAuth error without description', async () => {
+                const mockAuthUrl = 'https://example.com/auth';
+                const mockOAuth2Client = {
+                    generateAuthUrl: vi.fn().mockReturnValue(mockAuthUrl),
+                    on: vi.fn(),
+                };
+                OAuth2Client.mockImplementation(() => mockOAuth2Client);
+                open.mockImplementation(async () => ({ on: vi.fn() }));
+                let requestCallback;
+                let serverListeningCallback;
+                const serverListeningPromise = new Promise((resolve) => (serverListeningCallback = resolve));
+                const mockHttpServer = {
+                    listen: vi.fn((_port, _host, callback) => {
+                        if (callback)
+                            callback();
+                        serverListeningCallback(undefined);
+                    }),
+                    close: vi.fn(),
+                    on: vi.fn(),
+                    address: () => ({ port: 3000 }),
+                };
+                http.createServer.mockImplementation((cb) => {
+                    requestCallback = cb;
+                    return mockHttpServer;
+                });
+                const clientPromise = getOauthClient(AuthType.LOGIN_WITH_GOOGLE, mockConfig);
+                await serverListeningPromise;
+                // Test OAuth error without description
+                const mockReq = {
+                    url: '/oauth2callback?error=server_error',
+                };
+                const mockRes = {
+                    writeHead: vi.fn(),
+                    end: vi.fn(),
+                };
+                await expect(async () => {
+                    await requestCallback(mockReq, mockRes);
+                    await clientPromise;
+                }).rejects.toThrow('Google OAuth error: server_error. No additional details provided');
+            });
+            it('should handle token exchange failure with descriptive error', async () => {
+                const mockAuthUrl = 'https://example.com/auth';
+                const mockCode = 'test-code';
+                const mockState = 'test-state';
+                const mockOAuth2Client = {
+                    generateAuthUrl: vi.fn().mockReturnValue(mockAuthUrl),
+                    getToken: vi
+                        .fn()
+                        .mockRejectedValue(new Error('Token exchange failed')),
+                    on: vi.fn(),
+                };
+                OAuth2Client.mockImplementation(() => mockOAuth2Client);
+                vi.spyOn(crypto, 'randomBytes').mockReturnValue(mockState);
+                open.mockImplementation(async () => ({ on: vi.fn() }));
+                let requestCallback;
+                let serverListeningCallback;
+                const serverListeningPromise = new Promise((resolve) => (serverListeningCallback = resolve));
+                const mockHttpServer = {
+                    listen: vi.fn((_port, _host, callback) => {
+                        if (callback)
+                            callback();
+                        serverListeningCallback(undefined);
+                    }),
+                    close: vi.fn(),
+                    on: vi.fn(),
+                    address: () => ({ port: 3000 }),
+                };
+                http.createServer.mockImplementation((cb) => {
+                    requestCallback = cb;
+                    return mockHttpServer;
+                });
+                const clientPromise = getOauthClient(AuthType.LOGIN_WITH_GOOGLE, mockConfig);
+                await serverListeningPromise;
+                const mockReq = {
+                    url: `/oauth2callback?code=${mockCode}&state=${mockState}`,
+                };
+                const mockRes = {
+                    writeHead: vi.fn(),
+                    end: vi.fn(),
+                };
+                await expect(async () => {
+                    await requestCallback(mockReq, mockRes);
+                    await clientPromise;
+                }).rejects.toThrow('Failed to exchange authorization code for tokens: Token exchange failed');
+            });
+            it('should handle fetchAndCacheUserInfo failure gracefully', async () => {
+                const mockAuthUrl = 'https://example.com/auth';
+                const mockCode = 'test-code';
+                const mockState = 'test-state';
+                const mockTokens = {
+                    access_token: 'test-access-token',
+                    refresh_token: 'test-refresh-token',
+                };
+                const mockOAuth2Client = {
+                    generateAuthUrl: vi.fn().mockReturnValue(mockAuthUrl),
+                    getToken: vi.fn().mockResolvedValue({ tokens: mockTokens }),
+                    setCredentials: vi.fn(),
+                    getAccessToken: vi
+                        .fn()
+                        .mockResolvedValue({ token: 'test-access-token' }),
+                    on: vi.fn(),
+                };
+                OAuth2Client.mockImplementation(() => mockOAuth2Client);
+                vi.spyOn(crypto, 'randomBytes').mockReturnValue(mockState);
+                open.mockImplementation(async () => ({ on: vi.fn() }));
+                // Mock fetch to fail
+                global.fetch.mockResolvedValue({
+                    ok: false,
+                    status: 500,
+                    statusText: 'Internal Server Error',
+                });
+                const consoleErrorSpy = vi
+                    .spyOn(console, 'error')
+                    .mockImplementation(() => { });
+                let requestCallback;
+                let serverListeningCallback;
+                const serverListeningPromise = new Promise((resolve) => (serverListeningCallback = resolve));
+                const mockHttpServer = {
+                    listen: vi.fn((_port, _host, callback) => {
+                        if (callback)
+                            callback();
+                        serverListeningCallback(undefined);
+                    }),
+                    close: vi.fn(),
+                    on: vi.fn(),
+                    address: () => ({ port: 3000 }),
+                };
+                http.createServer.mockImplementation((cb) => {
+                    requestCallback = cb;
+                    return mockHttpServer;
+                });
+                const clientPromise = getOauthClient(AuthType.LOGIN_WITH_GOOGLE, mockConfig);
+                await serverListeningPromise;
+                const mockReq = {
+                    url: `/oauth2callback?code=${mockCode}&state=${mockState}`,
+                };
+                const mockRes = {
+                    writeHead: vi.fn(),
+                    end: vi.fn(),
+                };
+                await requestCallback(mockReq, mockRes);
+                const client = await clientPromise;
+                // Authentication should succeed even if fetchAndCacheUserInfo fails
+                expect(client).toBe(mockOAuth2Client);
+                expect(consoleErrorSpy).toHaveBeenCalledWith('Failed to fetch user info:', 500, 'Internal Server Error');
+                consoleErrorSpy.mockRestore();
+            });
+            it('should handle user code authentication failure with descriptive error', async () => {
+                const mockConfigWithNoBrowser = {
+                    getNoBrowser: () => true,
+                    getProxy: () => 'http://test.proxy.com:8080',
+                    isBrowserLaunchSuppressed: () => true,
+                };
+                const mockOAuth2Client = {
+                    generateCodeVerifierAsync: vi.fn().mockResolvedValue({
+                        codeChallenge: 'test-challenge',
+                        codeVerifier: 'test-verifier',
+                    }),
+                    generateAuthUrl: vi.fn().mockReturnValue('https://example.com/auth'),
+                    getToken: vi
+                        .fn()
+                        .mockRejectedValue(new Error('Invalid authorization code')),
+                    on: vi.fn(),
+                };
+                OAuth2Client.mockImplementation(() => mockOAuth2Client);
+                const mockReadline = {
+                    question: vi.fn((_query, callback) => callback('invalid-code')),
+                    close: vi.fn(),
+                };
+                readline.createInterface.mockReturnValue(mockReadline);
+                const consoleLogSpy = vi
+                    .spyOn(console, 'log')
+                    .mockImplementation(() => { });
+                const consoleErrorSpy = vi
+                    .spyOn(console, 'error')
+                    .mockImplementation(() => { });
+                await expect(getOauthClient(AuthType.LOGIN_WITH_GOOGLE, mockConfigWithNoBrowser)).rejects.toThrow('Failed to authenticate with user code.');
+                expect(consoleErrorSpy).toHaveBeenCalledWith('Failed to authenticate with authorization code:', 'Invalid authorization code');
+                consoleLogSpy.mockRestore();
+                consoleErrorSpy.mockRestore();
+            });
+        });
+        describe('clearCachedCredentialFile', () => {
+            it('should clear cached credentials and Google account', async () => {
+                const cachedCreds = { refresh_token: 'test-token' };
+                const credsPath = path.join(tempHomeDir, '.qwen', 'oauth_creds.json');
+                await fs.promises.mkdir(path.dirname(credsPath), { recursive: true });
+                await fs.promises.writeFile(credsPath, JSON.stringify(cachedCreds));
+                const googleAccountPath = path.join(tempHomeDir, '.qwen', 'google_accounts.json');
+                const accountData = { active: 'test@example.com', old: [] };
+                await fs.promises.writeFile(googleAccountPath, JSON.stringify(accountData));
+                const userAccountManager = new UserAccountManager();
+                expect(fs.existsSync(credsPath)).toBe(true);
+                expect(fs.existsSync(googleAccountPath)).toBe(true);
+                expect(userAccountManager.getCachedGoogleAccount()).toBe('test@example.com');
+                await clearCachedCredentialFile();
+                expect(fs.existsSync(credsPath)).toBe(false);
+                expect(userAccountManager.getCachedGoogleAccount()).toBeNull();
+                const updatedAccountData = JSON.parse(fs.readFileSync(googleAccountPath, 'utf-8'));
+                expect(updatedAccountData.active).toBeNull();
+                expect(updatedAccountData.old).toContain('test@example.com');
+            });
+            it('should clear the in-memory OAuth client cache', async () => {
+                const mockSetCredentials = vi.fn();
+                const mockGetAccessToken = vi
+                    .fn()
+                    .mockResolvedValue({ token: 'test-token' });
+                const mockGetTokenInfo = vi.fn().mockResolvedValue({});
+                const mockOAuth2Client = {
+                    setCredentials: mockSetCredentials,
+                    getAccessToken: mockGetAccessToken,
+                    getTokenInfo: mockGetTokenInfo,
+                    on: vi.fn(),
+                };
+                OAuth2Client.mockImplementation(() => mockOAuth2Client);
+                // Pre-populate credentials to make getOauthClient resolve quickly
+                const credsPath = path.join(tempHomeDir, '.qwen', 'oauth_creds.json');
+                await fs.promises.mkdir(path.dirname(credsPath), { recursive: true });
+                await fs.promises.writeFile(credsPath, JSON.stringify({ refresh_token: 'token' }));
+                // First call, should create a client
+                await getOauthClient(AuthType.LOGIN_WITH_GOOGLE, mockConfig);
+                expect(OAuth2Client).toHaveBeenCalledTimes(1);
+                // Second call, should use cached client
+                await getOauthClient(AuthType.LOGIN_WITH_GOOGLE, mockConfig);
+                expect(OAuth2Client).toHaveBeenCalledTimes(1);
+                clearOauthClientCache();
+                // Third call, after clearing cache, should create a new client
+                await getOauthClient(AuthType.LOGIN_WITH_GOOGLE, mockConfig);
+                expect(OAuth2Client).toHaveBeenCalledTimes(2);
+            });
+        });
+    });
+    describe('with encrypted flag true', () => {
+        let tempHomeDir;
+        beforeEach(() => {
+            process.env[FORCE_ENCRYPTED_FILE_ENV_VAR] = 'true';
+            tempHomeDir = fs.mkdtempSync(path.join(os.tmpdir(), 'qwen-code-test-home-'));
+            os.homedir.mockReturnValue(tempHomeDir);
+        });
+        afterEach(() => {
+            fs.rmSync(tempHomeDir, { recursive: true, force: true });
+            vi.clearAllMocks();
+            resetOauthClientForTesting();
+            vi.unstubAllEnvs();
+        });
+        it('should save credentials using OAuthCredentialStorage during web login', async () => {
+            const { OAuthCredentialStorage } = await import('./oauth-credential-storage.js');
+            const mockAuthUrl = 'https://example.com/auth';
+            const mockCode = 'test-code';
+            const mockState = 'test-state';
+            const mockTokens = {
+                access_token: 'test-access-token',
+                refresh_token: 'test-refresh-token',
+            };
+            let onTokensCallback = () => { };
+            const mockOn = vi.fn((event, callback) => {
+                if (event === 'tokens') {
+                    onTokensCallback = callback;
+                }
+            });
+            const mockGetToken = vi.fn().mockImplementation(async () => {
+                onTokensCallback(mockTokens);
+                return { tokens: mockTokens };
+            });
+            const mockOAuth2Client = {
+                generateAuthUrl: vi.fn().mockReturnValue(mockAuthUrl),
+                getToken: mockGetToken,
+                setCredentials: vi.fn(),
+                getAccessToken: vi
+                    .fn()
+                    .mockResolvedValue({ token: 'mock-access-token' }),
+                on: mockOn,
+                credentials: mockTokens,
+            };
+            OAuth2Client.mockImplementation(() => mockOAuth2Client);
+            vi.spyOn(crypto, 'randomBytes').mockReturnValue(mockState);
+            open.mockImplementation(async () => ({ on: vi.fn() }));
+            global.fetch.mockResolvedValue({
+                ok: true,
+                json: vi
+                    .fn()
+                    .mockResolvedValue({ email: 'test-google-account@gmail.com' }),
+            });
+            let requestCallback;
+            let serverListeningCallback;
+            const serverListeningPromise = new Promise((resolve) => (serverListeningCallback = resolve));
+            let capturedPort = 0;
+            const mockHttpServer = {
+                listen: vi.fn((port, _host, callback) => {
+                    capturedPort = port;
+                    if (callback) {
+                        callback();
+                    }
+                    serverListeningCallback(undefined);
+                }),
+                close: vi.fn((callback) => {
+                    if (callback) {
+                        callback();
+                    }
+                }),
+                on: vi.fn(),
+                address: () => ({ port: capturedPort }),
+            };
+            http.createServer.mockImplementation((cb) => {
+                requestCallback = cb;
+                return mockHttpServer;
+            });
+            const clientPromise = getOauthClient(AuthType.LOGIN_WITH_GOOGLE, mockConfig);
+            await serverListeningPromise;
+            const mockReq = {
+                url: `/oauth2callback?code=${mockCode}&state=${mockState}`,
+            };
+            const mockRes = {
+                writeHead: vi.fn(),
+                end: vi.fn(),
+            };
+            requestCallback(mockReq, mockRes);
+            await clientPromise;
+            expect(OAuthCredentialStorage.saveCredentials).toHaveBeenCalledWith(mockTokens);
+            const credsPath = path.join(tempHomeDir, '.qwen', 'oauth_creds.json');
+            expect(fs.existsSync(credsPath)).toBe(false);
+        });
+        it('should load credentials using OAuthCredentialStorage and not from file', async () => {
+            const { OAuthCredentialStorage } = await import('./oauth-credential-storage.js');
+            const cachedCreds = { refresh_token: 'cached-encrypted-token' };
+            OAuthCredentialStorage.loadCredentials.mockResolvedValue(cachedCreds);
+            // Create a dummy unencrypted credential file.
+            // If the logic is correct, this file should be ignored.
+            const unencryptedCreds = { refresh_token: 'unencrypted-token' };
+            const credsPath = path.join(tempHomeDir, '.qwen', 'oauth_creds.json');
+            await fs.promises.mkdir(path.dirname(credsPath), { recursive: true });
+            await fs.promises.writeFile(credsPath, JSON.stringify(unencryptedCreds));
+            const mockClient = {
+                setCredentials: vi.fn(),
+                getAccessToken: vi.fn().mockResolvedValue({ token: 'test-token' }),
+                getTokenInfo: vi.fn().mockResolvedValue({}),
+                on: vi.fn(),
+            };
+            OAuth2Client.mockImplementation(() => mockClient);
+            await getOauthClient(AuthType.LOGIN_WITH_GOOGLE, mockConfig);
+            expect(OAuthCredentialStorage.loadCredentials).toHaveBeenCalled();
+            expect(mockClient.setCredentials).toHaveBeenCalledWith(cachedCreds);
+            expect(mockClient.setCredentials).not.toHaveBeenCalledWith(unencryptedCreds);
+        });
+        it('should clear credentials using OAuthCredentialStorage', async () => {
+            const { OAuthCredentialStorage } = await import('./oauth-credential-storage.js');
+            // Create a dummy unencrypted credential file. It should be deleted as part of cleanup.
+            const credsPath = path.join(tempHomeDir, '.qwen', 'oauth_creds.json');
+            await fs.promises.mkdir(path.dirname(credsPath), { recursive: true });
+            await fs.promises.writeFile(credsPath, '{}');
+            await clearCachedCredentialFile();
+            expect(OAuthCredentialStorage.clearCredentials).toHaveBeenCalled();
+            expect(fs.existsSync(credsPath)).toBe(false); // The unencrypted file should be cleaned up
+        });
+    });
+});
+//# sourceMappingURL=oauth2.test.js.map