mirror of
https://github.com/czlonkowski/n8n-mcp.git
synced 2026-03-19 17:03:08 +00:00
feat: Add session persistence API for zero-downtime deployments (v2.24.1)
Implements export/restore functionality for MCP sessions to support container restarts without losing user sessions. This enables zero-downtime deployments for multi-tenant platforms and Kubernetes/Docker environments. New Features: - exportSessionState() - Export active sessions to JSON - restoreSessionState() - Restore sessions from exported data - SessionState type - Serializable session structure - Comprehensive test suite (22 tests, 100% passing) Implementation Details: - Only exports sessions with valid n8nApiUrl and n8nApiKey - Automatically filters expired sessions (respects sessionTimeout) - Validates context structure using existing validation - Handles null/invalid sessions gracefully with warnings - Enforces MAX_SESSIONS limit during restore (100 sessions) - Dormant sessions recreate transport/server on first request Files Modified: - src/http-server-single-session.ts: Core export/restore logic - src/mcp-engine.ts: Public API wrapper methods - src/types/session-state.ts: Type definitions - tests/: Comprehensive unit tests Security Note: Session data contains plaintext n8n API keys. Downstream applications MUST encrypt session data before persisting to disk. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com> Conceived by Romuald Członkowski - https://www.aiadvisors.pl/en
This commit is contained in:
255
tests/unit/mcp-engine/session-persistence.test.ts
Normal file
255
tests/unit/mcp-engine/session-persistence.test.ts
Normal file
@@ -0,0 +1,255 @@
|
||||
/**
|
||||
* Unit tests for N8NMCPEngine session persistence wrapper methods
|
||||
*/
|
||||
|
||||
import { describe, it, expect, beforeEach } from 'vitest';
|
||||
import { N8NMCPEngine } from '../../../src/mcp-engine';
|
||||
import { SessionState } from '../../../src/types/session-state';
|
||||
|
||||
describe('N8NMCPEngine - Session Persistence', () => {
|
||||
let engine: N8NMCPEngine;
|
||||
|
||||
beforeEach(() => {
|
||||
engine = new N8NMCPEngine({
|
||||
sessionTimeout: 30 * 60 * 1000,
|
||||
logLevel: 'error' // Quiet during tests
|
||||
});
|
||||
});
|
||||
|
||||
describe('exportSessionState()', () => {
|
||||
it('should return empty array when no sessions exist', () => {
|
||||
const exported = engine.exportSessionState();
|
||||
expect(exported).toEqual([]);
|
||||
});
|
||||
|
||||
it('should delegate to underlying server', () => {
|
||||
// Access private server to create test sessions
|
||||
const engineAny = engine as any;
|
||||
const server = engineAny.server;
|
||||
const serverAny = server as any;
|
||||
|
||||
// Create a mock session
|
||||
serverAny.sessionMetadata['test-session'] = {
|
||||
createdAt: new Date(),
|
||||
lastAccess: new Date()
|
||||
};
|
||||
serverAny.sessionContexts['test-session'] = {
|
||||
n8nApiUrl: 'https://test.example.com',
|
||||
n8nApiKey: 'test-key',
|
||||
instanceId: 'test-instance'
|
||||
};
|
||||
|
||||
const exported = engine.exportSessionState();
|
||||
|
||||
expect(exported).toHaveLength(1);
|
||||
expect(exported[0].sessionId).toBe('test-session');
|
||||
expect(exported[0].context.n8nApiUrl).toBe('https://test.example.com');
|
||||
});
|
||||
|
||||
it('should handle server not initialized', () => {
|
||||
// Create engine without server
|
||||
const engineAny = {} as N8NMCPEngine;
|
||||
const exportMethod = N8NMCPEngine.prototype.exportSessionState.bind(engineAny);
|
||||
|
||||
// Should not throw, should return empty array
|
||||
expect(() => exportMethod()).not.toThrow();
|
||||
const result = exportMethod();
|
||||
expect(result).toEqual([]);
|
||||
});
|
||||
});
|
||||
|
||||
describe('restoreSessionState()', () => {
|
||||
it('should restore sessions via underlying server', () => {
|
||||
const sessions: SessionState[] = [
|
||||
{
|
||||
sessionId: 'restored-session',
|
||||
metadata: {
|
||||
createdAt: new Date().toISOString(),
|
||||
lastAccess: new Date().toISOString()
|
||||
},
|
||||
context: {
|
||||
n8nApiUrl: 'https://restored.example.com',
|
||||
n8nApiKey: 'restored-key',
|
||||
instanceId: 'restored-instance'
|
||||
}
|
||||
}
|
||||
];
|
||||
|
||||
const count = engine.restoreSessionState(sessions);
|
||||
|
||||
expect(count).toBe(1);
|
||||
|
||||
// Verify session was restored
|
||||
const engineAny = engine as any;
|
||||
const server = engineAny.server;
|
||||
const serverAny = server as any;
|
||||
|
||||
expect(serverAny.sessionMetadata['restored-session']).toBeDefined();
|
||||
expect(serverAny.sessionContexts['restored-session']).toMatchObject({
|
||||
n8nApiUrl: 'https://restored.example.com',
|
||||
n8nApiKey: 'restored-key',
|
||||
instanceId: 'restored-instance'
|
||||
});
|
||||
});
|
||||
|
||||
it('should return 0 when restoring empty array', () => {
|
||||
const count = engine.restoreSessionState([]);
|
||||
expect(count).toBe(0);
|
||||
});
|
||||
|
||||
it('should handle server not initialized', () => {
|
||||
const engineAny = {} as N8NMCPEngine;
|
||||
const restoreMethod = N8NMCPEngine.prototype.restoreSessionState.bind(engineAny);
|
||||
|
||||
const sessions: SessionState[] = [
|
||||
{
|
||||
sessionId: 'test',
|
||||
metadata: {
|
||||
createdAt: new Date().toISOString(),
|
||||
lastAccess: new Date().toISOString()
|
||||
},
|
||||
context: {
|
||||
n8nApiUrl: 'https://test.example.com',
|
||||
n8nApiKey: 'test-key',
|
||||
instanceId: 'test-instance'
|
||||
}
|
||||
}
|
||||
];
|
||||
|
||||
// Should not throw, should return 0
|
||||
expect(() => restoreMethod(sessions)).not.toThrow();
|
||||
const result = restoreMethod(sessions);
|
||||
expect(result).toBe(0);
|
||||
});
|
||||
|
||||
it('should return count of successfully restored sessions', () => {
|
||||
const now = Date.now();
|
||||
const sessions: SessionState[] = [
|
||||
{
|
||||
sessionId: 'valid-1',
|
||||
metadata: {
|
||||
createdAt: new Date(now - 10 * 60 * 1000).toISOString(),
|
||||
lastAccess: new Date(now - 5 * 60 * 1000).toISOString()
|
||||
},
|
||||
context: {
|
||||
n8nApiUrl: 'https://valid1.example.com',
|
||||
n8nApiKey: 'key1',
|
||||
instanceId: 'instance1'
|
||||
}
|
||||
},
|
||||
{
|
||||
sessionId: 'valid-2',
|
||||
metadata: {
|
||||
createdAt: new Date(now - 10 * 60 * 1000).toISOString(),
|
||||
lastAccess: new Date(now - 5 * 60 * 1000).toISOString()
|
||||
},
|
||||
context: {
|
||||
n8nApiUrl: 'https://valid2.example.com',
|
||||
n8nApiKey: 'key2',
|
||||
instanceId: 'instance2'
|
||||
}
|
||||
},
|
||||
{
|
||||
sessionId: 'expired',
|
||||
metadata: {
|
||||
createdAt: new Date(now - 60 * 60 * 1000).toISOString(),
|
||||
lastAccess: new Date(now - 45 * 60 * 1000).toISOString() // Expired
|
||||
},
|
||||
context: {
|
||||
n8nApiUrl: 'https://expired.example.com',
|
||||
n8nApiKey: 'expired-key',
|
||||
instanceId: 'expired-instance'
|
||||
}
|
||||
}
|
||||
];
|
||||
|
||||
const count = engine.restoreSessionState(sessions);
|
||||
|
||||
expect(count).toBe(2); // Only 2 valid sessions
|
||||
});
|
||||
});
|
||||
|
||||
describe('Round-trip through engine', () => {
|
||||
it('should preserve sessions through export → restore cycle', () => {
|
||||
// Create mock sessions with current timestamps
|
||||
const engineAny = engine as any;
|
||||
const server = engineAny.server;
|
||||
const serverAny = server as any;
|
||||
|
||||
const now = new Date();
|
||||
const createdAt = new Date(now.getTime() - 10 * 60 * 1000); // 10 minutes ago
|
||||
const lastAccess = new Date(now.getTime() - 5 * 60 * 1000); // 5 minutes ago
|
||||
|
||||
serverAny.sessionMetadata['engine-session'] = {
|
||||
createdAt,
|
||||
lastAccess
|
||||
};
|
||||
serverAny.sessionContexts['engine-session'] = {
|
||||
n8nApiUrl: 'https://engine-test.example.com',
|
||||
n8nApiKey: 'engine-key',
|
||||
instanceId: 'engine-instance',
|
||||
metadata: { env: 'production' }
|
||||
};
|
||||
|
||||
// Export via engine
|
||||
const exported = engine.exportSessionState();
|
||||
expect(exported).toHaveLength(1);
|
||||
|
||||
// Clear sessions
|
||||
delete serverAny.sessionMetadata['engine-session'];
|
||||
delete serverAny.sessionContexts['engine-session'];
|
||||
|
||||
// Restore via engine
|
||||
const count = engine.restoreSessionState(exported);
|
||||
expect(count).toBe(1);
|
||||
|
||||
// Verify data
|
||||
expect(serverAny.sessionMetadata['engine-session']).toBeDefined();
|
||||
expect(serverAny.sessionContexts['engine-session']).toMatchObject({
|
||||
n8nApiUrl: 'https://engine-test.example.com',
|
||||
n8nApiKey: 'engine-key',
|
||||
instanceId: 'engine-instance',
|
||||
metadata: { env: 'production' }
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
describe('Integration with getSessionInfo()', () => {
|
||||
it('should reflect restored sessions in session info', () => {
|
||||
const sessions: SessionState[] = [
|
||||
{
|
||||
sessionId: 'info-session-1',
|
||||
metadata: {
|
||||
createdAt: new Date().toISOString(),
|
||||
lastAccess: new Date().toISOString()
|
||||
},
|
||||
context: {
|
||||
n8nApiUrl: 'https://info1.example.com',
|
||||
n8nApiKey: 'info-key-1',
|
||||
instanceId: 'info-instance-1'
|
||||
}
|
||||
},
|
||||
{
|
||||
sessionId: 'info-session-2',
|
||||
metadata: {
|
||||
createdAt: new Date().toISOString(),
|
||||
lastAccess: new Date().toISOString()
|
||||
},
|
||||
context: {
|
||||
n8nApiUrl: 'https://info2.example.com',
|
||||
n8nApiKey: 'info-key-2',
|
||||
instanceId: 'info-instance-2'
|
||||
}
|
||||
}
|
||||
];
|
||||
|
||||
engine.restoreSessionState(sessions);
|
||||
|
||||
const info = engine.getSessionInfo();
|
||||
|
||||
// Note: getSessionInfo() reflects metadata, not transports
|
||||
// Restored sessions won't have transports until first request
|
||||
expect(info).toBeDefined();
|
||||
});
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user