Fix issues with tests

Author: mheadd

api/__tests__/api.test.js

@@ -214,8 +214,8 @@ describe('API Integration Tests', () => {
       nock.cleanAll();
       nock('http://ollama:11434')
         .post('/api/chat')
-        .delay(125000) // Longer than our 120 second timeout
-        .reply(200, { message: { content: 'Response' } });
+        .delay(2000) // Short delay that won't timeout but will test timeout handling
+        .replyWithError({ code: 'ECONNABORTED', message: 'timeout of 120000ms exceeded' });
 
       const response = await request(app)
         .post('/chat/default')

api/__tests__/e2e.test.js

@@ -3,6 +3,39 @@ const nock = require('nock');
 const path = require('path');
 const fs = require('fs');
 
+// Set up test helpers that were previously in setup.js
+global.testHelpers = {
+  validPersonas: ['unemployment-benefits', 'parks-recreation', 'business-licensing', 'default'],
+  
+  validateApiResponse: (response) => {
+    // Handle both direct response objects and response bodies with different structures
+    const body = response.body || response;
+    if (body.message && body.message.content) {
+      expect(typeof body.message.content).toBe('string');
+      expect(body.message.content.length).toBeGreaterThan(0);
+    } else if (body.response) {
+      expect(typeof body.response).toBe('string');
+      expect(body.response.length).toBeGreaterThan(0);
+    } else if (typeof body === 'string') {
+      expect(body.length).toBeGreaterThan(0);
+    } else {
+      // Flexible validation for various response formats
+      expect(body).toHaveProperty('message');
+    }
+  }
+};
+
+// Set up test data
+global.testData = {
+  sampleMessages: {
+    unemploymentbenefits: 'I need help applying for unemployment benefits. What documents do I need?',
+    parksrecreation: 'What are the operating hours for the community center?',
+    businesslicensing: 'I need to apply for a restaurant business license. What is the process?',
+    default: 'I need help with city services. Can you direct me to the right department?',
+    general: ['What services do you provide?', 'How can I contact customer support?', 'What are your hours?']
+  }
+};
+
 // Import our server for testing
 const createTestApp = () => {
   const express = require('express');

api/__tests__/setup.js

@@ -61,8 +61,7 @@
       "/node_modules/",
       "/coverage/"
     ],
-    "setupFilesAfterEnv": ["<rootDir>/__tests__/setup.js"],
-    "testTimeout": 10000,
+    "testTimeout": 30000,
     "detectOpenHandles": true,
     "forceExit": false
   },