[UPDATE PRIMITIVE] Fix failing CI: update tool registration counts for codeql_lsp_document_symbols and codeqlGenerateLogSummaryTool (#125)

* Initial plan

* fix: update test assertions for new codeql_lsp_document_symbols and codeqlGenerateLogSummaryTool tools

Co-authored-by: data-douser <70299490+data-douser@users.noreply.github.com>

---------

Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com>
Co-authored-by: data-douser <70299490+data-douser@users.noreply.github.com>

Author: Copilot

server/test/src/tools/codeql-tools.test.ts

@@ -46,9 +46,9 @@ describe('registerCodeQLTools', () => {
     // rank_sarif_results has been removed in favor of SARIF prompts
     expect(toolNames).not.toContain('rank_sarif_results');
 
-    // Total tools registered: 2 high-level helpers + 12 specialized tools + 23 CLI tools = 37
+    // Total tools registered: 2 high-level helpers + 12 specialized tools + 24 CLI tools = 38
     // (codeql_lsp_diagnostics moved to registerLSPTools in tools/lsp/)
-    expect(mockServer.tool).toHaveBeenCalledTimes(37);
+    expect(mockServer.tool).toHaveBeenCalledTimes(38);
   });
 
   it('should register validate_codeql_query with correct parameters', () => {

server/test/src/tools/lsp/lsp-tools.test.ts

@@ -15,8 +15,10 @@ vi.mock('../../../../src/tools/lsp/lsp-diagnostics', () => ({
 
 // Mock lsp-handlers
 vi.mock('../../../../src/tools/lsp/lsp-handlers', () => ({
+  extractNamesFromDocumentSymbols: vi.fn().mockReturnValue([]),
   lspCompletion: vi.fn().mockResolvedValue([]),
   lspDefinition: vi.fn().mockResolvedValue([]),
+  lspDocumentSymbols: vi.fn().mockResolvedValue([]),
   lspReferences: vi.fn().mockResolvedValue([]),
 }));
 
@@ -36,10 +38,10 @@ describe('registerLSPTools', () => {
     expect(registerLspDiagnosticsTool).toHaveBeenCalledWith(mockServer);
   });
 
-  it('should register 3 file-based LSP tools directly', () => {
+  it('should register 4 file-based LSP tools directly', () => {
     registerLSPTools(mockServer);
-    // 3 tools registered directly via server.tool (diagnostics is registered via delegate)
-    expect(mockServer.tool).toHaveBeenCalledTimes(3);
+    // 4 tools registered directly via server.tool (diagnostics is registered via delegate)
+    expect(mockServer.tool).toHaveBeenCalledTimes(4);
   });
 
   it('should register codeql_lsp_completion', () => {
@@ -74,6 +76,17 @@ describe('registerLSPTools', () => {
     expect(typeof refsCall![3]).toBe('function');
   });
 
+  it('should register codeql_lsp_document_symbols', () => {
+    registerLSPTools(mockServer);
+    const calls = (mockServer.tool as ReturnType<typeof vi.fn>).mock.calls;
+    const docSymsCall = calls.find((c: unknown[]) => c[0] === 'codeql_lsp_document_symbols');
+    expect(docSymsCall).toBeDefined();
+    expect(docSymsCall![1]).toContain('classes, predicates, modules');
+    expect(docSymsCall![2]).toHaveProperty('file_path');
+    expect(docSymsCall![2]).toHaveProperty('names_only');
+    expect(typeof docSymsCall![3]).toBe('function');
+  });
+
   it('should include optional parameters in all file-based schemas', () => {
     registerLSPTools(mockServer);
     const calls = (mockServer.tool as ReturnType<typeof vi.fn>).mock.calls;
@@ -124,6 +137,18 @@ describe('registerLSPTools', () => {
       expect(parsed).toHaveProperty('locations');
     });
 
+    it('codeql_lsp_document_symbols handler should return JSON with symbolCount', async () => {
+      registerLSPTools(mockServer);
+      const calls = (mockServer.tool as ReturnType<typeof vi.fn>).mock.calls;
+      const handler = calls.find((c: unknown[]) => c[0] === 'codeql_lsp_document_symbols')![3];
+
+      const result = await handler({ file_path: '/test.ql' });
+
+      const parsed = JSON.parse(result.content[0].text);
+      expect(parsed).toHaveProperty('symbolCount');
+      expect(parsed).toHaveProperty('symbols');
+    });
+
     it('handler should return isError on failure', async () => {
       // Re-mock to throw
       const { lspCompletion } = await import('../../../../src/tools/lsp/lsp-handlers');