Enhance OpenAI Test Script with Advanced Model Selection and Error Handling

- Added comprehensive model configuration and selection mechanism
- Implemented dynamic model handling for OpenAI and DeepSeek providers
- Enhanced error handling and retry logic for API interactions
- Expanded system analysis with more detailed XML-based response parsing
- Updated configuration loading with improved model selection and validation
- Added colored logging and more informative console output
- Integrated new test suite for API endpoints

__tests__/api/index.test.ts

@@ -0,0 +1,163 @@
+import { jest, describe, it, expect, beforeEach, afterEach } from '@jest/globals';
+import express from 'express';
+import request from 'supertest';
+import { config } from 'dotenv';
+import { resolve } from 'path';
+import type { Entity } from '../../src/types/hass';
+import { TokenManager } from '../../src/security/index';
+import { MCP_SCHEMA } from '../../src/mcp/schema';
+
+// Load test environment variables
+config({ path: resolve(process.cwd(), '.env.test') });
+
+// Mock dependencies
+jest.mock('../../src/security/index', () => ({
+    TokenManager: {
+        validateToken: jest.fn().mockImplementation((token) => token === 'valid-test-token'),
+    },
+    rateLimiter: (req: any, res: any, next: any) => next(),
+    securityHeaders: (req: any, res: any, next: any) => next(),
+    validateRequest: (req: any, res: any, next: any) => next(),
+    sanitizeInput: (req: any, res: any, next: any) => next(),
+    errorHandler: (err: any, req: any, res: any, next: any) => {
+        res.status(500).json({ error: err.message });
+    },
+}));
+
+// Create mock entity
+const mockEntity: Entity = {
+    entity_id: 'light.living_room',
+    state: 'off',
+    attributes: {},
+    last_changed: new Date().toISOString(),
+    last_updated: new Date().toISOString(),
+    context: {
+        id: '123',
+        parent_id: null,
+        user_id: null
+    }
+};
+
+// Mock Home Assistant module
+jest.mock('../../src/hass/index');
+
+// Mock LiteMCP
+jest.mock('litemcp', () => ({
+    LiteMCP: jest.fn().mockImplementation(() => ({
+        name: 'home-assistant',
+        version: '0.1.0',
+        tools: []
+    }))
+}));
+
+// Create Express app for testing
+const app = express();
+app.use(express.json());
+
+// Add test routes that mimic our actual routes
+app.get('/mcp', (_req, res) => {
+    res.json(MCP_SCHEMA);
+});
+
+app.get('/state', (req, res) => {
+    const authHeader = req.headers.authorization;
+    if (!authHeader || !authHeader.startsWith('Bearer ') || authHeader.split(' ')[1] !== 'valid-test-token') {
+        return res.status(401).json({ error: 'Unauthorized' });
+    }
+    res.json([mockEntity]);
+});
+
+app.post('/command', (req, res) => {
+    const authHeader = req.headers.authorization;
+    if (!authHeader || !authHeader.startsWith('Bearer ') || authHeader.split(' ')[1] !== 'valid-test-token') {
+        return res.status(401).json({ error: 'Unauthorized' });
+    }
+
+    const { command, entity_id } = req.body;
+    if (!command || !entity_id) {
+        return res.status(400).json({ error: 'Missing required parameters' });
+    }
+
+    if (command === 'invalid_command') {
+        return res.status(400).json({ error: 'Invalid command' });
+    }
+
+    res.json({ success: true });
+});
+
+describe('API Endpoints', () => {
+    describe('GET /mcp', () => {
+        it('should return MCP schema without authentication', async () => {
+            const response = await request(app)
+                .get('/mcp')
+                .expect('Content-Type', /json/)
+                .expect(200);
+
+            expect(response.body).toBeDefined();
+            expect(response.body).toHaveProperty('name');
+            expect(response.body).toHaveProperty('version');
+            expect(response.body).toHaveProperty('tools');
+        });
+    });
+
+    describe('Protected Endpoints', () => {
+        describe('GET /state', () => {
+            it('should return 401 without authentication', async () => {
+                await request(app)
+                    .get('/state')
+                    .expect(401);
+            });
+
+            it('should return state with valid token', async () => {
+                const response = await request(app)
+                    .get('/state')
+                    .set('Authorization', 'Bearer valid-test-token')
+                    .expect('Content-Type', /json/)
+                    .expect(200);
+
+                expect(response.body).toBeDefined();
+                expect(Array.isArray(response.body)).toBeTruthy();
+                expect(response.body[0]).toHaveProperty('entity_id', 'light.living_room');
+                expect(response.body[0]).toHaveProperty('state', 'off');
+            });
+        });
+
+        describe('POST /command', () => {
+            it('should return 401 without authentication', async () => {
+                await request(app)
+                    .post('/command')
+                    .send({
+                        command: 'turn_on',
+                        entity_id: 'light.living_room'
+                    })
+                    .expect(401);
+            });
+
+            it('should process valid command with authentication', async () => {
+                const response = await request(app)
+                    .set('Authorization', 'Bearer valid-test-token')
+                    .post('/command')
+                    .send({
+                        command: 'turn_on',
+                        entity_id: 'light.living_room'
+                    })
+                    .expect('Content-Type', /json/)
+                    .expect(200);
+
+                expect(response.body).toBeDefined();
+                expect(response.body).toHaveProperty('success', true);
+            });
+
+            it('should validate command parameters', async () => {
+                await request(app)
+                    .post('/command')
+                    .set('Authorization', 'Bearer valid-test-token')
+                    .send({
+                        command: 'invalid_command',
+                        entity_id: 'light.living_room'
+                    })
+                    .expect(400);
+            });
+        });
+    });
+});