diff --git a/packages/core/src/code_assist/server.test.ts b/packages/core/src/code_assist/server.test.ts
index 6944af47..6246fd4e 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');
 
@@ -115,7 +116,14 @@ describe('CodeAssistServer', () => {
     const client = new OAuth2Client();
     const server = new CodeAssistServer(client, 'test-project');
     const mockResponse = {
-      // TODO: Add mock response
+      currentTier: {
+        id: UserTierId.FREE,
+        name: 'Free',
+        description: 'free tier',
+      },
+      allowedTiers: [],
+      ineligibleTiers: [],
+      cloudaicompanionProject: 'projects/test',
     };
     vi.spyOn(server, 'requestPost').mockResolvedValue(mockResponse);
 
@@ -127,7 +135,7 @@ describe('CodeAssistServer', () => {
       'loadCodeAssist',
       expect.any(Object),
     );
-    expect(response).toBe(mockResponse);
+    expect(response).toEqual(mockResponse);
   });
 
   it('should return 0 for countTokens', async () => {