chore: Add unit tests to claudeAi service

jest.config.js

@@ -1,6 +1,6 @@
 module.exports = {
     testEnvironment: 'node',
-    roots: ['<rootDir>/tests'],
+    roots: ['<rootDir>'],
     moduleFileExtensions: ['js', 'json', 'jsx', 'ts', 'tsx', 'node'],
     moduleNameMapper: {
         '^@/(.*)$': '<rootDir>/$1',

services/claudeAi.js

@@ -1,24 +1,20 @@
-const Anthropic = require("@anthropic-ai/sdk");
+const Anthropic = require('@anthropic-ai/sdk');
 require('dotenv').config();
 
 const anthropic = new Anthropic({
     apiKey: process.env['ANTHROPIC_API_KEY'], // This is the default and can be omitted
 });
 
-
-async function generateClaudeContent(content, model_type = "claude-3-5-sonnet-20240620") {
+async function generateClaudeContent(content, model_type = 'claude-3-opus-20240229') {
     try {
         const result = await anthropic.messages.create({
             max_tokens: 4096, // This is the maximum number of tokens that can be generated
             messages: content,
-            model: 'claude-3-opus-20240229',
+            model: model_type,
         });
-
         return result.content[0].text;
-
     } catch (error) {
-        console.error("Error in generateClaudeContent:", error);
+        console.error('Error in generateClaudeContent:', error);
     }
-
 }
-module.exports = { generateClaudeContent };
+module.exports = {generateClaudeContent};

services/claudeAi.test.js

@@ -0,0 +1,61 @@
+const createSpy = jest.fn().mockResolvedValue({
+    content: [
+        {
+            text: 'mock-result-text',
+        },
+    ],
+});
+jest.doMock('@anthropic-ai/sdk', () => {
+    return jest.fn().mockImplementation(() => {
+        return {
+            messages: {
+                create: createSpy,
+            },
+        };
+    });
+});
+
+const {generateClaudeContent} = require('./claudeAi');
+
+describe('claudeAi service', () => {
+    afterEach(() => {
+        createSpy.mockClear();
+    });
+    it('should call Anthropic.messages.create with the given args', async () => {
+        const mockContent = 'this is a mock content';
+
+        await generateClaudeContent(mockContent);
+
+        expect(createSpy).toHaveBeenCalledWith({
+            max_tokens: 4096,
+            messages: 'this is a mock content',
+            model: 'claude-3-opus-20240229',
+        });
+    });
+
+    it('should call Anthropic.messages.create with a given model', async () => {
+        const mockContent = 'this is a mock content';
+        const mockModel = 'claude-3-5-sonnet-20240620';
+
+        await generateClaudeContent(mockContent, mockModel);
+
+        expect(createSpy).toHaveBeenCalledWith({
+            max_tokens: 4096,
+            messages: 'this is a mock content',
+            model: 'claude-3-5-sonnet-20240620',
+        });
+    });
+
+    it('should console error if something went wrong', async () => {
+        const mockContent = 'this is a mock content';
+        const mockError = new Error('Something went wrong');
+        jest.spyOn(console, 'error');
+        createSpy.mockImplementationOnce(() => {
+            console.log('throwing :>> ');
+            throw mockError;
+        });
+
+        await generateClaudeContent(mockContent);
+        expect(console.error).toHaveBeenCalledWith('Error in generateClaudeContent:', mockError);
+    });
+});