ruvnet--ruflo/v3/plugins/prime-radiant/__tests__/plugin.test.ts

/**
 * Prime Radiant Plugin Lifecycle Tests
 *
 * Tests for the prime-radiant plugin initialization, configuration,
 * WASM loading, and health check functionality.
 */

import { describe, it, expect, beforeEach, afterEach, vi } from 'vitest';

// ============================================================================
// Mock Types
// ============================================================================

interface PluginMetadata {
  name: string;
  version: string;
  description?: string;
  author?: string;
  license?: string;
  wasmSize?: string;
  dependencies?: string[];
  tags?: string[];
}

interface PluginContext {
  config: {
    enabled: boolean;
    priority: number;
    settings: Record<string, unknown>;
  };
  eventBus: {
    emit: (event: string, data?: unknown) => void;
    on: (event: string, handler: (data: unknown) => void) => () => void;
    off: (event: string, handler: (data: unknown) => void) => void;
  };
  logger: {
    debug: (msg: string, ...args: unknown[]) => void;
    info: (msg: string, ...args: unknown[]) => void;
    warn: (msg: string, ...args: unknown[]) => void;
    error: (msg: string, ...args: unknown[]) => void;
  };
  services: Map<string, unknown>;
  coreVersion: string;
  dataDir: string;
}

interface MCPToolDefinition {
  name: string;
  description: string;
  inputSchema: Record<string, unknown>;
  handler: (input: unknown) => Promise<{ content: Array<{ type: string; text: string }> }>;
}

interface HookDefinition {
  name: string;
  event: string;
  priority: 'low' | 'normal' | 'high' | 'critical';
  handler: (context: unknown) => Promise<{ success: boolean }>;
}

type PluginState = 'uninitialized' | 'initializing' | 'initialized' | 'error' | 'shutdown';

// ============================================================================
// Mock Engine Interface
// ============================================================================

interface Engine {
  name: string;
  ready: boolean;
  compute: (input: unknown) => Promise<unknown>;
}

// ============================================================================
// Mock Plugin Implementation
// ============================================================================

class MockPrimeRadiantPlugin {
  public metadata: PluginMetadata;
  public state: PluginState = 'uninitialized';
  private context?: PluginContext;
  private tools: MCPToolDefinition[] = [];
  private hooks: HookDefinition[] = [];
  private engines: Map<string, Engine> = new Map();
  private wasmLoaded = false;

  constructor(metadata: Partial<PluginMetadata> = {}) {
    this.metadata = Object.freeze({
      name: 'prime-radiant',
      version: '0.1.3',
      description: 'Mathematical AI interpretability plugin',
      author: 'rUv',
      license: 'MIT',
      wasmSize: '92KB',
      dependencies: ['@claude-flow/memory', '@claude-flow/security', '@claude-flow/coordination'],
      tags: [
        'sheaf-cohomology',
        'causal-inference',
        'quantum-topology',
        'ai-interpretability',
      ],
      ...metadata,
    });
  }

  async initialize(context: PluginContext): Promise<void> {
    if (this.state === 'initialized') {
      throw new Error('Plugin already initialized');
    }

    this.state = 'initializing';
    this.context = context;

    try {
      // Load WASM
      await this.loadWasm();

      // Initialize engines
      await this.initializeEngines();

      // Register tools
      this.tools = this.createTools();

      // Register hooks
      this.hooks = this.createHooks();

      this.state = 'initialized';
      context.logger.info(`${this.metadata.name} initialized successfully`);
    } catch (error) {
      this.state = 'error';
      throw error;
    }
  }

  async shutdown(): Promise<void> {
    if (this.state !== 'initialized') {
      return;
    }

    // Cleanup engines
    for (const engine of this.engines.values()) {
      // Would call engine.dispose() in real implementation
    }
    this.engines.clear();

    // Clear tools and hooks
    this.tools = [];
    this.hooks = [];
    this.wasmLoaded = false;

    this.state = 'shutdown';
    this.context?.logger.info(`${this.metadata.name} shut down`);
  }

  async healthCheck(): Promise<{
    healthy: boolean;
    status: string;
    checks: Record<string, { healthy: boolean; message?: string }>;
  }> {
    const checks: Record<string, { healthy: boolean; message?: string }> = {
      state: {
        healthy: this.state === 'initialized',
        message: `Plugin state: ${this.state}`,
      },
      wasm: {
        healthy: this.wasmLoaded,
        message: this.wasmLoaded ? 'WASM loaded' : 'WASM not loaded',
      },
      engines: {
        healthy: this.engines.size === 6,
        message: `${this.engines.size}/6 engines ready`,
      },
      tools: {
        healthy: this.tools.length > 0,
        message: `${this.tools.length} tools registered`,
      },
    };

    const healthy = Object.values(checks).every((c) => c.healthy);

    return {
      healthy,
      status: healthy ? 'healthy' : 'unhealthy',
      checks,
    };
  }

  registerMCPTools(): MCPToolDefinition[] {
    return this.tools;
  }

  registerHooks(): HookDefinition[] {
    return this.hooks;
  }

  getEngine(name: string): Engine | undefined {
    return this.engines.get(name);
  }

  isWasmLoaded(): boolean {
    return this.wasmLoaded;
  }

  getConfig(): Record<string, unknown> {
    return this.context?.config.settings ?? {};
  }

  private async loadWasm(): Promise<void> {
    // Mock WASM loading - in real implementation would load prime-radiant-advanced-wasm
    await new Promise((resolve) => setTimeout(resolve, 10)); // Simulate load time
    this.wasmLoaded = true;
  }

  private async initializeEngines(): Promise<void> {
    const engineNames = [
      'cohomology',
      'spectral',
      'causal',
      'quantum',
      'category',
      'hott',
    ];

    for (const name of engineNames) {
      this.engines.set(name, {
        name,
        ready: true,
        compute: async (input) => ({ result: `${name} computation result` }),
      });
    }
  }

  private createTools(): MCPToolDefinition[] {
    return [
      {
        name: 'pr_coherence_check',
        description: 'Check coherence of vectors using Sheaf Laplacian energy',
        inputSchema: {
          type: 'object',
          properties: {
            vectors: { type: 'array', items: { type: 'array' } },
            threshold: { type: 'number', default: 0.3 },
          },
          required: ['vectors'],
        },
        handler: async (input) => ({
          content: [{ type: 'text', text: JSON.stringify({ coherent: true, energy: 0.1 }) }],
        }),
      },
      {
        name: 'pr_spectral_analyze',
        description: 'Analyze stability using spectral graph theory',
        inputSchema: {
          type: 'object',
          properties: {
            adjacencyMatrix: { type: 'array' },
            analyzeType: { type: 'string', enum: ['stability', 'clustering', 'connectivity'] },
          },
          required: ['adjacencyMatrix'],
        },
        handler: async (input) => ({
          content: [{ type: 'text', text: JSON.stringify({ stable: true, spectralGap: 0.25 }) }],
        }),
      },
      {
        name: 'pr_causal_infer',
        description: 'Perform causal inference using do-calculus',
        inputSchema: {
          type: 'object',
          properties: {
            treatment: { type: 'string' },
            outcome: { type: 'string' },
            graph: { type: 'object' },
          },
          required: ['treatment', 'outcome', 'graph'],
        },
        handler: async (input) => ({
          content: [{ type: 'text', text: JSON.stringify({ causalEffect: -0.35 }) }],
        }),
      },
      {
        name: 'pr_consensus_verify',
        description: 'Verify multi-agent consensus mathematically',
        inputSchema: {
          type: 'object',
          properties: {
            agentStates: { type: 'array' },
            consensusThreshold: { type: 'number', default: 0.8 },
          },
          required: ['agentStates'],
        },
        handler: async (input) => ({
          content: [{ type: 'text', text: JSON.stringify({ consensusAchieved: true }) }],
        }),
      },
      {
        name: 'pr_quantum_topology',
        description: 'Compute quantum topology features',
        inputSchema: {
          type: 'object',
          properties: {
            points: { type: 'array' },
            maxDimension: { type: 'number', default: 2 },
          },
          required: ['points'],
        },
        handler: async (input) => ({
          content: [{ type: 'text', text: JSON.stringify({ bettiNumbers: [1, 2, 0] }) }],
        }),
      },
      {
        name: 'pr_memory_gate',
        description: 'Pre-storage coherence gate for memory entries',
        inputSchema: {
          type: 'object',
          properties: {
            entry: { type: 'object' },
            contextEmbeddings: { type: 'array' },
            thresholds: { type: 'object' },
          },
          required: ['entry'],
        },
        handler: async (input) => ({
          content: [{ type: 'text', text: JSON.stringify({ action: 'allow' }) }],
        }),
      },
    ];
  }

  private createHooks(): HookDefinition[] {
    return [
      {
        name: 'pr/pre-memory-store',
        event: 'pre-memory-store',
        priority: 'high',
        handler: async () => ({ success: true }),
      },
      {
        name: 'pr/pre-consensus',
        event: 'pre-consensus',
        priority: 'high',
        handler: async () => ({ success: true }),
      },
      {
        name: 'pr/post-swarm-task',
        event: 'post-task',
        priority: 'normal',
        handler: async () => ({ success: true }),
      },
      {
        name: 'pr/pre-rag-retrieval',
        event: 'pre-rag-retrieval',
        priority: 'high',
        handler: async () => ({ success: true }),
      },
    ];
  }
}

// ============================================================================
// Test Helpers
// ============================================================================

function createMockContext(overrides: Partial<PluginContext> = {}): PluginContext {
  return {
    config: {
      enabled: true,
      priority: 50,
      settings: {
        coherence: {
          warnThreshold: 0.3,
          rejectThreshold: 0.7,
          cacheEnabled: true,
        },
        spectral: {
          stabilityThreshold: 0.1,
          maxMatrixSize: 1000,
        },
        causal: {
          maxBackdoorPaths: 10,
          confidenceThreshold: 0.8,
        },
      },
    },
    eventBus: {
      emit: vi.fn(),
      on: vi.fn(() => vi.fn()),
      off: vi.fn(),
    },
    logger: {
      debug: vi.fn(),
      info: vi.fn(),
      warn: vi.fn(),
      error: vi.fn(),
    },
    services: new Map(),
    coreVersion: '3.0.0',
    dataDir: '/tmp/prime-radiant',
    ...overrides,
  };
}

// ============================================================================
// Tests
// ============================================================================

describe('PrimeRadiantPlugin', () => {
  let plugin: MockPrimeRadiantPlugin;
  let context: PluginContext;

  beforeEach(() => {
    plugin = new MockPrimeRadiantPlugin();
    context = createMockContext();
  });

  afterEach(async () => {
    if (plugin.state === 'initialized') {
      await plugin.shutdown();
    }
  });

  describe('metadata', () => {
    it('should have correct plugin name', () => {
      expect(plugin.metadata.name).toBe('prime-radiant');
    });

    it('should have correct version', () => {
      expect(plugin.metadata.version).toBe('0.1.3');
    });

    it('should specify WASM size', () => {
      expect(plugin.metadata.wasmSize).toBe('92KB');
    });

    it('should have required dependencies', () => {
      expect(plugin.metadata.dependencies).toContain('@claude-flow/memory');
      expect(plugin.metadata.dependencies).toContain('@claude-flow/security');
      expect(plugin.metadata.dependencies).toContain('@claude-flow/coordination');
    });

    it('should have expected tags', () => {
      expect(plugin.metadata.tags).toContain('sheaf-cohomology');
      expect(plugin.metadata.tags).toContain('causal-inference');
      expect(plugin.metadata.tags).toContain('ai-interpretability');
    });

    it('should freeze metadata', () => {
      expect(Object.isFrozen(plugin.metadata)).toBe(true);
    });
  });

  describe('initialization', () => {
    it('should start in uninitialized state', () => {
      expect(plugin.state).toBe('uninitialized');
    });

    it('should initialize successfully with valid context', async () => {
      await plugin.initialize(context);

      expect(plugin.state).toBe('initialized');
    });

    it('should log initialization success', async () => {
      await plugin.initialize(context);

      expect(context.logger.info).toHaveBeenCalledWith(
        expect.stringContaining('initialized successfully')
      );
    });

    it('should reject double initialization', async () => {
      await plugin.initialize(context);

      await expect(plugin.initialize(context)).rejects.toThrow('already initialized');
    });

    it('should load WASM during initialization', async () => {
      expect(plugin.isWasmLoaded()).toBe(false);

      await plugin.initialize(context);

      expect(plugin.isWasmLoaded()).toBe(true);
    });

    it('should initialize all 6 engines', async () => {
      await plugin.initialize(context);

      expect(plugin.getEngine('cohomology')).toBeDefined();
      expect(plugin.getEngine('spectral')).toBeDefined();
      expect(plugin.getEngine('causal')).toBeDefined();
      expect(plugin.getEngine('quantum')).toBeDefined();
      expect(plugin.getEngine('category')).toBeDefined();
      expect(plugin.getEngine('hott')).toBeDefined();
    });

    it('should mark all engines as ready', async () => {
      await plugin.initialize(context);

      for (const engineName of ['cohomology', 'spectral', 'causal', 'quantum', 'category', 'hott']) {
        const engine = plugin.getEngine(engineName);
        expect(engine?.ready).toBe(true);
      }
    });
  });

  describe('shutdown', () => {
    it('should shutdown gracefully', async () => {
      await plugin.initialize(context);
      await plugin.shutdown();

      expect(plugin.state).toBe('shutdown');
    });

    it('should clear engines on shutdown', async () => {
      await plugin.initialize(context);
      await plugin.shutdown();

      expect(plugin.getEngine('cohomology')).toBeUndefined();
    });

    it('should unload WASM on shutdown', async () => {
      await plugin.initialize(context);
      await plugin.shutdown();

      expect(plugin.isWasmLoaded()).toBe(false);
    });

    it('should be safe to call shutdown multiple times', async () => {
      await plugin.initialize(context);
      await plugin.shutdown();
      await plugin.shutdown(); // Should not throw

      expect(plugin.state).toBe('shutdown');
    });

    it('should be safe to shutdown uninitialized plugin', async () => {
      await plugin.shutdown(); // Should not throw

      expect(plugin.state).toBe('uninitialized');
    });
  });

  describe('health check', () => {
    it('should report healthy when initialized', async () => {
      await plugin.initialize(context);

      const health = await plugin.healthCheck();

      expect(health.healthy).toBe(true);
      expect(health.status).toBe('healthy');
    });

    it('should report unhealthy when not initialized', async () => {
      const health = await plugin.healthCheck();

      expect(health.healthy).toBe(false);
      expect(health.status).toBe('unhealthy');
    });

    it('should include WASM check', async () => {
      await plugin.initialize(context);

      const health = await plugin.healthCheck();

      expect(health.checks.wasm.healthy).toBe(true);
      expect(health.checks.wasm.message).toContain('WASM loaded');
    });

    it('should include engines check', async () => {
      await plugin.initialize(context);

      const health = await plugin.healthCheck();

      expect(health.checks.engines.healthy).toBe(true);
      expect(health.checks.engines.message).toContain('6/6');
    });

    it('should include tools check', async () => {
      await plugin.initialize(context);

      const health = await plugin.healthCheck();

      expect(health.checks.tools.healthy).toBe(true);
    });
  });

  describe('MCP tools', () => {
    it('should register MCP tools after initialization', async () => {
      await plugin.initialize(context);

      const tools = plugin.registerMCPTools();

      expect(tools.length).toBeGreaterThan(0);
    });

    it('should include coherence_check tool', async () => {
      await plugin.initialize(context);

      const tools = plugin.registerMCPTools();
      const tool = tools.find((t) => t.name === 'pr_coherence_check');

      expect(tool).toBeDefined();
      expect(tool?.description).toContain('Sheaf Laplacian');
    });

    it('should include spectral_analyze tool', async () => {
      await plugin.initialize(context);

      const tools = plugin.registerMCPTools();
      const tool = tools.find((t) => t.name === 'pr_spectral_analyze');

      expect(tool).toBeDefined();
      expect(tool?.description).toContain('spectral');
    });

    it('should include causal_infer tool', async () => {
      await plugin.initialize(context);

      const tools = plugin.registerMCPTools();
      const tool = tools.find((t) => t.name === 'pr_causal_infer');

      expect(tool).toBeDefined();
      expect(tool?.description).toContain('causal');
    });

    it('should include consensus_verify tool', async () => {
      await plugin.initialize(context);

      const tools = plugin.registerMCPTools();
      const tool = tools.find((t) => t.name === 'pr_consensus_verify');

      expect(tool).toBeDefined();
    });

    it('should include quantum_topology tool', async () => {
      await plugin.initialize(context);

      const tools = plugin.registerMCPTools();
      const tool = tools.find((t) => t.name === 'pr_quantum_topology');

      expect(tool).toBeDefined();
    });

    it('should include memory_gate tool', async () => {
      await plugin.initialize(context);

      const tools = plugin.registerMCPTools();
      const tool = tools.find((t) => t.name === 'pr_memory_gate');

      expect(tool).toBeDefined();
    });

    it('should have valid input schemas', async () => {
      await plugin.initialize(context);

      const tools = plugin.registerMCPTools();

      for (const tool of tools) {
        expect(tool.inputSchema).toBeDefined();
        expect(tool.inputSchema.type).toBe('object');
        expect(tool.inputSchema.properties).toBeDefined();
      }
    });

    it('should have working handlers', async () => {
      await plugin.initialize(context);

      const tools = plugin.registerMCPTools();
      const tool = tools[0];

      const result = await tool.handler({ vectors: [[0.1, 0.2]] });

      expect(result.content).toBeDefined();
      expect(result.content[0].type).toBe('text');
    });
  });

  describe('hooks', () => {
    it('should register hooks after initialization', async () => {
      await plugin.initialize(context);

      const hooks = plugin.registerHooks();

      expect(hooks.length).toBeGreaterThan(0);
    });

    it('should include pre-memory-store hook', async () => {
      await plugin.initialize(context);

      const hooks = plugin.registerHooks();
      const hook = hooks.find((h) => h.name === 'pr/pre-memory-store');

      expect(hook).toBeDefined();
      expect(hook?.event).toBe('pre-memory-store');
      expect(hook?.priority).toBe('high');
    });

    it('should include pre-consensus hook', async () => {
      await plugin.initialize(context);

      const hooks = plugin.registerHooks();
      const hook = hooks.find((h) => h.name === 'pr/pre-consensus');

      expect(hook).toBeDefined();
    });

    it('should include post-swarm-task hook', async () => {
      await plugin.initialize(context);

      const hooks = plugin.registerHooks();
      const hook = hooks.find((h) => h.name === 'pr/post-swarm-task');

      expect(hook).toBeDefined();
      expect(hook?.event).toBe('post-task');
    });

    it('should include pre-rag-retrieval hook', async () => {
      await plugin.initialize(context);

      const hooks = plugin.registerHooks();
      const hook = hooks.find((h) => h.name === 'pr/pre-rag-retrieval');

      expect(hook).toBeDefined();
    });
  });

  describe('configuration', () => {
    it('should access plugin settings', async () => {
      await plugin.initialize(context);

      const config = plugin.getConfig();

      expect(config.coherence).toBeDefined();
      expect((config.coherence as any).warnThreshold).toBe(0.3);
      expect((config.coherence as any).rejectThreshold).toBe(0.7);
    });

    it('should access spectral settings', async () => {
      await plugin.initialize(context);

      const config = plugin.getConfig();

      expect(config.spectral).toBeDefined();
      expect((config.spectral as any).stabilityThreshold).toBe(0.1);
    });

    it('should access causal settings', async () => {
      await plugin.initialize(context);

      const config = plugin.getConfig();

      expect(config.causal).toBeDefined();
      expect((config.causal as any).maxBackdoorPaths).toBe(10);
    });
  });
});

describe('PrimeRadiantPlugin WASM Loading', () => {
  it('should handle WASM load failure gracefully', async () => {
    // This would test error handling for WASM load failure
    // In the mock, WASM always loads successfully
    const plugin = new MockPrimeRadiantPlugin();
    const context = createMockContext();

    // Mock would need to be modified to test failure scenarios
    await plugin.initialize(context);
    expect(plugin.isWasmLoaded()).toBe(true);
  });

  it('should load WASM within performance target', async () => {
    const plugin = new MockPrimeRadiantPlugin();
    const context = createMockContext();

    const startTime = performance.now();
    await plugin.initialize(context);
    const duration = performance.now() - startTime;

    // Target: <50ms for WASM load
    expect(duration).toBeLessThan(100); // Allow some margin for mock
  });
});