From d13fc96224f757c70eae7e066d69b2bb249cdc62 Mon Sep 17 00:00:00 2001 From: kamja Date: Mon, 7 Jul 2025 16:38:57 +0900 Subject: [PATCH 1/4] test(core): add mock response for loadCodeAssist --- packages/core/src/code_assist/server.test.ts | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/packages/core/src/code_assist/server.test.ts b/packages/core/src/code_assist/server.test.ts index 6944af47dbc..5f63e3662dc 100644 --- a/packages/core/src/code_assist/server.test.ts +++ b/packages/core/src/code_assist/server.test.ts @@ -115,7 +115,14 @@ describe('CodeAssistServer', () => { const client = new OAuth2Client(); const server = new CodeAssistServer(client, 'test-project'); const mockResponse = { - // TODO: Add mock response + currentTier: { + id: 'free-tier', + name: 'Free', + description: 'free tier', + }, + allowedTiers: [], + ineligibleTiers: [], + cloudaicompanionProject: 'projects/test', }; vi.spyOn(server, 'requestPost').mockResolvedValue(mockResponse); @@ -127,7 +134,8 @@ describe('CodeAssistServer', () => { 'loadCodeAssist', expect.any(Object), ); - expect(response).toBe(mockResponse); + expect(response.cloudaicompanionProject).toBe('projects/test'); + expect(response.currentTier?.id).toBe('free-tier'); }); it('should return 0 for countTokens', async () => { From f680c2af1ce1c627369357c32cc4e31202bfb976 Mon Sep 17 00:00:00 2001 From: HyeongHo Jun <88872409+kamja44@users.noreply.github.com> Date: Tue, 8 Jul 2025 08:24:27 +0900 Subject: [PATCH 2/4] Update packages/core/src/code_assist/server.test.ts Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> --- packages/core/src/code_assist/server.test.ts | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/packages/core/src/code_assist/server.test.ts b/packages/core/src/code_assist/server.test.ts index 5f63e3662dc..e7e5ca6abcb 100644 --- a/packages/core/src/code_assist/server.test.ts +++ b/packages/core/src/code_assist/server.test.ts @@ -134,8 +134,7 @@ describe('CodeAssistServer', () => { 'loadCodeAssist', expect.any(Object), ); - expect(response.cloudaicompanionProject).toBe('projects/test'); - expect(response.currentTier?.id).toBe('free-tier'); + expect(response).toEqual(mockResponse); }); it('should return 0 for countTokens', async () => { From 27c9ef914696d332ccd33c96cac2c93b35f978c7 Mon Sep 17 00:00:00 2001 From: HyeongHo Jun <88872409+kamja44@users.noreply.github.com> Date: Wed, 16 Jul 2025 16:19:24 +0900 Subject: [PATCH 3/4] Update packages/core/src/code_assist/server.test.ts Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> --- packages/core/src/code_assist/server.test.ts | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/packages/core/src/code_assist/server.test.ts b/packages/core/src/code_assist/server.test.ts index e7e5ca6abcb..e84d68deaf5 100644 --- a/packages/core/src/code_assist/server.test.ts +++ b/packages/core/src/code_assist/server.test.ts @@ -116,7 +116,7 @@ describe('CodeAssistServer', () => { const server = new CodeAssistServer(client, 'test-project'); const mockResponse = { currentTier: { - id: 'free-tier', + id: UserTierId.FREE, name: 'Free', description: 'free tier', }, From f8cc67377989ff96d38efe640bebe8230168d5f2 Mon Sep 17 00:00:00 2001 From: kamja Date: Wed, 16 Jul 2025 16:20:44 +0900 Subject: [PATCH 4/4] import userTierId --- packages/core/src/code_assist/server.test.ts | 1 + 1 file changed, 1 insertion(+) diff --git a/packages/core/src/code_assist/server.test.ts b/packages/core/src/code_assist/server.test.ts index e84d68deaf5..6246fd4e11d 100644 --- a/packages/core/src/code_assist/server.test.ts +++ b/packages/core/src/code_assist/server.test.ts @@ -7,6 +7,7 @@ import { describe, it, expect, vi } from 'vitest'; import { CodeAssistServer } from './server.js'; import { OAuth2Client } from 'google-auth-library'; +import { UserTierId } from './types.js'; vi.mock('google-auth-library');