memento-mcp

import { describe, test, expect, vi } from 'vitest'; import { handleCreateRelations } from '../createRelations.js'; describe('handleCreateRelations', () => { test('should create relations and return results', async () => { // Arrange const args = { relations: [{ from: 'Entity1', to: 'Entity2', relationType: 'KNOWS' }], }; const mockResult = { success: true }; const mockKnowledgeGraphManager = { createRelations: vi.fn().mockResolvedValue(mockResult), }; // Act const response = await handleCreateRelations(args, mockKnowledgeGraphManager); // Assert expect(mockKnowledgeGraphManager.createRelations).toHaveBeenCalledWith(args.relations); expect(response).toEqual({ content: [ { type: 'text', text: JSON.stringify(mockResult, null, 2), }, ], }); }); });