fix(tests): update test output

Author: JoshWheeler08

proxy/src/adapters/bedrock.spec.ts

@@ -8,6 +8,11 @@ describe('BedrockAdapter', () => {
   beforeEach(() => {
     // Arrange
     adapter = new BedrockAdapter();
+    jest.spyOn(console, 'warn').mockImplementation();
+  });
+
+  afterEach(() => {
+    jest.restoreAllMocks();
   });
 
   describe('canHandleRequest', () => {
@@ -493,7 +498,6 @@ describe('BedrockAdapter', () => {
   describe('handler selection', () => {
     it('should log warning when no handler found for model', () => {
       // Arrange
-      const consoleSpy = jest.spyOn(console, 'warn').mockImplementation();
       const request: ProviderRequest = {
         url: '/model/unknown.model-v1/invoke',
         method: 'POST',
@@ -507,11 +511,9 @@ describe('BedrockAdapter', () => {
       adapter.transformRequest(request as any);
 
       // Assert
-      expect(consoleSpy).toHaveBeenCalledWith(
+      expect(console.warn).toHaveBeenCalledWith(
         expect.stringContaining('No handler found for Bedrock model: unknown.model-v1')
       );
-
-      consoleSpy.mockRestore();
     });
 
     it('should fall back to Anthropic handler when no handler matches', () => {

proxy/src/config.spec.ts

@@ -11,9 +11,15 @@ describe('config', () => {
 
   beforeEach(() => {
     jest.clearAllMocks();
+    jest.spyOn(console, 'log').mockImplementation();
+    jest.spyOn(console, 'error').mockImplementation();
     delete process.env.CONFIG_PATH;
   });
 
+  afterEach(() => {
+    jest.restoreAllMocks();
+  });
+
   describe('loadConfig', () => {
     it('should load configuration from default path when CONFIG_PATH is not set', () => {
       // Arrange

proxy/src/runners/runner-orchestrator.spec.ts

@@ -11,6 +11,7 @@ describe('RunnerOrchestrator', () => {
 
   beforeEach(() => {
     jest.clearAllMocks();
+    jest.spyOn(console, 'log').mockImplementation();
 
     // Arrange - default config
     mockConfig = {
@@ -43,37 +44,31 @@ describe('RunnerOrchestrator', () => {
     };
   });
 
+  afterEach(() => {
+    jest.restoreAllMocks();
+  });
+
   describe('constructor', () => {
     it('should initialize vLLM runner', () => {
-      // Arrange
-      const consoleSpy = jest.spyOn(console, 'log').mockImplementation();
-
-      // Act
+      // Arrange & Act
       new RunnerOrchestrator(mockConfig);
 
       // Assert
       expect(VLLMRunner).toHaveBeenCalledWith('http://localhost:8000');
-      expect(consoleSpy).toHaveBeenCalledWith(
+      expect(console.log).toHaveBeenCalledWith(
         expect.stringContaining('Registered runner: vllm-runner-1')
       );
-
-      consoleSpy.mockRestore();
     });
 
     it('should initialize Ollama runner', () => {
-      // Arrange
-      const consoleSpy = jest.spyOn(console, 'log').mockImplementation();
-
-      // Act
+      // Arrange & Act
       new RunnerOrchestrator(mockConfig);
 
       // Assert
       expect(OllamaRunner).toHaveBeenCalledWith('http://localhost:11434');
-      expect(consoleSpy).toHaveBeenCalledWith(
+      expect(console.log).toHaveBeenCalledWith(
         expect.stringContaining('Registered runner: ollama-runner-1')
       );
-
-      consoleSpy.mockRestore();
     });
 
     it('should initialize TGI runner as vLLM runner', () => {