80 lines
2.7 KiB
TypeScript
80 lines
2.7 KiB
TypeScript
import request from 'supertest';
|
|
import express from 'express';
|
|
import { createServer } from 'http';
|
|
import { LLMService } from '../src/services/LLMService';
|
|
|
|
// Mock the LLMService
|
|
jest.mock('../src/services/LLMService');
|
|
|
|
// Mock the routes
|
|
const app = express();
|
|
app.use(express.json());
|
|
|
|
// Mock session storage for testing analyze endpoint
|
|
const mockSessions = new Map<string, any>();
|
|
mockSessions.set('test-session-id', { /* session data */ });
|
|
|
|
app.post('/sessions', (req, res) => {
|
|
res.status(201).json({ sessionId: 'mock-session-id' });
|
|
});
|
|
|
|
app.post('/sessions/:sessionId/analyze', async (req, res) => {
|
|
const { sessionId } = req.params;
|
|
if (!mockSessions.has(sessionId)) {
|
|
return res.status(404).send('Session not found');
|
|
}
|
|
// Mock LLMService call
|
|
const mockLLMService = new LLMService('mock-api-key');
|
|
const analysisResult = await mockLLMService.analyzeDesires(req.body.allDesires);
|
|
res.status(202).json({ message: 'Analysis triggered', result: analysisResult });
|
|
});
|
|
|
|
describe('POST /sessions', () => {
|
|
it('should create a new session and return a session ID', async () => {
|
|
const response = await request(app)
|
|
.post('/sessions')
|
|
.send();
|
|
|
|
expect(response.status).toBe(201);
|
|
expect(response.body).toHaveProperty('sessionId');
|
|
expect(typeof response.body.sessionId).toBe('string');
|
|
});
|
|
});
|
|
|
|
describe('POST /sessions/:sessionId/analyze', () => {
|
|
it('should trigger analysis for a valid session', async () => {
|
|
const mockDesires = [
|
|
{ wants: ['item1'], accepts: [], noGoes: [] },
|
|
{ wants: ['item2'], accepts: [], noGoes: [] },
|
|
];
|
|
|
|
// Mock the analyzeDesires method to return a predictable result
|
|
(LLMService as jest.Mock).mockImplementation(() => ({
|
|
analyzeDesires: jest.fn().mockResolvedValue({
|
|
"item1": "Concept A",
|
|
"item2": "Concept A"
|
|
}),
|
|
}));
|
|
|
|
const response = await request(app)
|
|
.post('/sessions/test-session-id/analyze')
|
|
.send({ allDesires: mockDesires });
|
|
|
|
expect(response.status).toBe(202);
|
|
expect(response.body).toHaveProperty('message', 'Analysis triggered');
|
|
expect(response.body).toHaveProperty('result');
|
|
expect(response.body.result).toEqual({
|
|
"item1": "Concept A",
|
|
"item2": "Concept A"
|
|
});
|
|
});
|
|
|
|
it('should return 404 if session is not found', async () => {
|
|
const response = await request(app)
|
|
.post('/sessions/non-existent-session/analyze')
|
|
.send({ allDesires: [] });
|
|
|
|
expect(response.status).toBe(404);
|
|
});
|
|
});
|