#!/usr/bin/env node
// Test script for delete_connection function
// This script tests deleting connections between memories
import { spawn } from 'child_process';
import dotenv from 'dotenv';
// Load environment variables
dotenv.config({ path: '../.env' });
const testDeleteConnection = () => {
console.log('🔗🗑️ Testing delete_connection function...');
const mcp = spawn('node', ['../build/index.js'], {
env: {
...process.env,
NEO4J_PASSWORD: process.env.NEO4J_PASSWORD,
NEO4J_DATABASE: process.env.NEO4J_DATABASE,
NEO4J_URI: process.env.NEO4J_URI,
NEO4J_USERNAME: process.env.NEO4J_USERNAME
},
stdio: ['pipe', 'pipe', 'pipe']
});
let messageId = 1;
let outputBuffer = '';
let memory1Id = null;
let memory2Id = null;
const sendMessage = (name, args) => {
const message = {
jsonrpc: '2.0',
id: messageId++,
method: 'tools/call',
params: { name, arguments: args }
};
mcp.stdin.write(JSON.stringify(message) + '\n');
};
mcp.stdout.on('data', (data) => {
outputBuffer += data.toString();
const lines = outputBuffer.split('\n');
outputBuffer = lines[lines.length - 1]; // Keep incomplete line
for (const line of lines) {
if (line.trim()) {
try {
const response = JSON.parse(line);
if (response.id === 1) {
// Created first memory
const content = response.result?.content?.[0]?.text;
if (content) {
const result = JSON.parse(content);
memory1Id = result.n._id;
console.log(`✅ Created first memory with ID: ${memory1Id}`);
}
} else if (response.id === 2) {
// Created second memory
const content = response.result?.content?.[0]?.text;
if (content) {
const result = JSON.parse(content);
memory2Id = result.n._id;
console.log(`✅ Created second memory with ID: ${memory2Id}`);
// Create connection
console.log('Creating connection...');
sendMessage('create_connection', {
fromMemoryId: memory1Id,
toMemoryId: memory2Id,
type: 'LIVES_IN',
properties: {
created_at: new Date().toISOString()
}
});
}
} else if (response.id === 3) {
// Created connection
console.log('✅ Created connection');
// Now delete it
console.log('Deleting connection...');
sendMessage('delete_connection', {
fromMemoryId: memory1Id,
toMemoryId: memory2Id,
type: 'LIVES_IN'
});
} else if (response.id === 4) {
// Deleted connection
console.log('✅ delete_connection test passed');
console.log('Response:', response.result?.content?.[0]?.text);
// Clean up - delete the test memories
sendMessage('delete_memory', { nodeId: memory1Id });
sendMessage('delete_memory', { nodeId: memory2Id });
setTimeout(() => mcp.kill(), 1000);
}
} catch (e) {
// Ignore non-JSON lines
}
}
}
});
mcp.stderr.on('data', (data) => {
console.error('❌ Error:', data.toString());
});
// Create two test memories
console.log('Creating test memories...');
sendMessage('create_memory', {
label: 'person',
properties: {
name: 'TestPerson1',
created_at: new Date().toISOString()
}
});
setTimeout(() => {
sendMessage('create_memory', {
label: 'place',
properties: {
name: 'TestLocation1',
created_at: new Date().toISOString()
}
});
}, 500);
setTimeout(() => {
console.log('⏰ Test timeout');
mcp.kill();
}, 15000);
};
testDeleteConnection();