test: stabilize follow-up integration suites

Author: PatrickSys

tests/impact-2hop.test.ts

@@ -1,4 +1,4 @@
-import { describe, it, expect, beforeEach, afterEach } from 'vitest';
+import { describe, it, expect, beforeEach, afterEach, vi } from 'vitest';
 import { promises as fs } from 'fs';
 import os from 'os';
 import path from 'path';
@@ -15,6 +15,11 @@ import {
   RELATIONSHIPS_FILENAME
 } from '../src/constants/codebase-context.js';
 
+vi.mock('../src/core/reranker.js', () => ({
+  rerank: vi.fn(async (_query: string, results: unknown) => results),
+  getRerankerStatus: vi.fn(() => 'fallback')
+}));
+
 describe('Impact candidates (2-hop)', () => {
   let tempRoot: string | null = null;
   const token = 'UNIQUETOKEN123';

tests/index-migration-atomic-swap.test.ts

@@ -214,7 +214,7 @@ export function greet(user: User): string {
     // Verify staging directory is cleaned up
     const hasStaging = await stagingDirExists(contextDir);
     expect(hasStaging).toBe(false);
-  });
+  }, 30000);
 
   it('should fail closed when meta points to missing artifacts', async () => {
     // Create an index with meta pointing to non-existent files

tests/search-decision-card.test.ts

@@ -5,6 +5,11 @@ import path from 'path';
 import { CodebaseIndexer } from '../src/core/indexer.js';
 import { rmWithRetries } from './test-helpers.js';
 
+vi.mock('../src/core/reranker.js', () => ({
+  rerank: vi.fn(async (_query: string, results: unknown) => results),
+  getRerankerStatus: vi.fn(() => 'fallback')
+}));
+
 type ToolCallRequest = {
   jsonrpc: '2.0';
   id: number;

tests/search-hints.test.ts

@@ -145,7 +145,7 @@ describe('Search Hints', () => {
       // Should be capped at 3
       expect(utilResult.hints.callers.length).toBeLessThanOrEqual(3);
     }
-  });
+  }, 30000);
 
   it('hints include tests when test files are detected', async () => {
     if (!tempRoot) throw new Error('tempRoot not initialized');

tests/search-snippets.test.ts

@@ -5,6 +5,11 @@ import path from 'path';
 import { CodebaseIndexer } from '../src/core/indexer.js';
 import { rmWithRetries } from './test-helpers.js';
 
+vi.mock('../src/core/reranker.js', () => ({
+  rerank: vi.fn(async (_query: string, results: unknown) => results),
+  getRerankerStatus: vi.fn(() => 'fallback')
+}));
+
 describe('Search Snippets with Scope Headers', () => {
   let tempRoot: string | null = null;
 

tests/zombie-guard.test.ts

@@ -117,6 +117,6 @@ describe('zombie process prevention', () => {
     expect(result.code).toBe(1);
     // Should still honor a short timeout (allow CI/Windows process jitter).
     expect(elapsed).toBeGreaterThan(800);
-    expect(elapsed).toBeLessThan(7_000);
-  }, 10_000);
+    expect(elapsed).toBeLessThan(8_000);
+  }, 12_000);
 });