session start works

backend/tests/llmService.test.ts

@@ -0,0 +1,54 @@
+import { GoogleGenerativeAI } from '@google/generative-ai';
+import { LLMService } from '../src/services/LLMService';
+
+// Mock the GoogleGenerativeAI class and its methods
+jest.mock('@google/generative-ai', () => ({
+    GoogleGenerativeAI: jest.fn().mockImplementation(() => ({
+        getGenerativeModel: jest.fn().mockReturnValue({
+            generateContent: jest.fn().mockResolvedValue({
+                response: {
+                    text: jest.fn().mockReturnValue(
+                        JSON.stringify({
+                            "item1": "Concept A",
+                            "item2": "Concept A",
+                            "item3": "Concept B"
+                        })
+                    ),
+                },
+            }),
+        }),
+    })),
+}));
+
+describe('LLMService', () => {
+    let llmService: LLMService;
+    const mockApiKey = 'test-api-key';
+
+    beforeEach(() => {
+        llmService = new LLMService(mockApiKey);
+    });
+
+    it('should call the Gemini API with the correct prompt and return parsed content', async () => {
+        const desires = [
+            { wants: ['item1'], accepts: [], noGoes: [] },
+            { wants: ['item2'], accepts: [], noGoes: [] },
+            { wants: [], accepts: ['item3'], noGoes: [] },
+        ];
+
+        const result = await llmService.analyzeDesires(desires);
+
+        expect(GoogleGenerativeAI).toHaveBeenCalledWith(mockApiKey);
+        expect(llmService['model'].generateContent).toHaveBeenCalled();
+        expect(result).toEqual({
+            "item1": "Concept A",
+            "item2": "Concept A",
+            "item3": "Concept B"
+        });
+    });
+
+    it('should handle API errors gracefully', async () => {
+        llmService['model'].generateContent.mockRejectedValueOnce(new Error('API Error'));
+
+        await expect(llmService.analyzeDesires([])).rejects.toThrow('API Error');
+    });
+});