@holoscript/framework 6.0.3

package/src/agents/__tests__/AgentRegistry.prod.test.ts

@@ -0,0 +1,125 @@
+/**
+ * AgentRegistry — Production Test Suite
+ *
+ * Covers: register, deregister, heartbeat, discover, findBest,
+ * get, has, getAllManifests, size, getStatusCounts, clear, export/import.
+ */
+import { describe, it, expect, beforeEach, afterEach } from 'vitest';
+import { AgentRegistry } from '../AgentRegistry';
+import type { AgentManifest } from '../AgentManifest';
+
+function mkManifest(id: string, extra: Partial<AgentManifest> = {}): AgentManifest {
+  return {
+    id,
+    name: `Agent ${id}`,
+    version: '1.0.0',
+    capabilities: [{ type: 'compute', domain: 'physics' }],
+    endpoints: [{ protocol: 'http', address: 'localhost', port: 3000 }],
+    trustLevel: 'local',
+    status: 'online',
+    ...extra,
+  } as AgentManifest;
+}
+
+describe('AgentRegistry — Production', () => {
+  let reg: AgentRegistry;
+
+  beforeEach(() => {
+    reg = new AgentRegistry({ autoCleanup: false });
+  });
+  afterEach(() => {
+    reg.stop();
+  });
+
+  // ─── Registration ────────────────────────────────────────────────
+  it('register adds an agent', async () => {
+    await reg.register(mkManifest('a1'));
+    expect(reg.has('a1')).toBe(true);
+    expect(reg.size).toBe(1); // size is a getter
+  });
+
+  it('register updates existing agent (no reject)', async () => {
+    await reg.register(mkManifest('a1'));
+    await reg.register(mkManifest('a1', { version: '2.0.0' }));
+    // Duplicate register updates, not rejects
+    expect(reg.size).toBe(1);
+    expect(reg.get('a1')!.version).toBe('2.0.0');
+  });
+
+  it('deregister removes an agent', async () => {
+    await reg.register(mkManifest('a1'));
+    await reg.deregister('a1');
+    expect(reg.has('a1')).toBe(false);
+    expect(reg.size).toBe(0);
+  });
+
+  // ─── Heartbeat ───────────────────────────────────────────────────
+  it('heartbeat updates agent entry', async () => {
+    await reg.register(mkManifest('a1'));
+    await reg.heartbeat('a1');
+    expect(reg.has('a1')).toBe(true);
+  });
+
+  it('heartbeat on unknown agent throws', async () => {
+    await expect(reg.heartbeat('unknown')).rejects.toThrow();
+  });
+
+  // ─── Discovery ───────────────────────────────────────────────────
+  it('discover returns matching agents', async () => {
+    await reg.register(
+      mkManifest('a1', { capabilities: [{ type: 'compute', domain: 'physics' }] } as any)
+    );
+    await reg.register(
+      mkManifest('a2', { capabilities: [{ type: 'render', domain: 'graphics' }] } as any)
+    );
+    const results = await reg.discover({ type: 'compute' });
+    expect(results.length).toBeGreaterThanOrEqual(1);
+    expect(results.some((r) => r.id === 'a1')).toBe(true);
+  });
+
+  it('findBest returns best matching agent', async () => {
+    await reg.register(mkManifest('a1'));
+    const best = await reg.findBest({ type: 'compute' });
+    expect(best).not.toBeNull();
+    expect(best!.id).toBe('a1');
+  });
+
+  // ─── Getters ─────────────────────────────────────────────────────
+  it('get returns manifest by ID', async () => {
+    await reg.register(mkManifest('a1'));
+    const m = reg.get('a1');
+    expect(m).toBeDefined();
+    expect(m!.name).toBe('Agent a1');
+  });
+
+  it('getAllManifests returns all', async () => {
+    await reg.register(mkManifest('a1'));
+    await reg.register(mkManifest('a2'));
+    expect(reg.getAllManifests().length).toBe(2);
+  });
+
+  it('getStatusCounts counts agents by status', async () => {
+    await reg.register(mkManifest('a1', { status: 'online' }));
+    const counts = reg.getStatusCounts();
+    expect(counts['online']).toBeGreaterThanOrEqual(1);
+  });
+
+  // ─── Clear ───────────────────────────────────────────────────────
+  it('clear removes all agents', async () => {
+    await reg.register(mkManifest('a1'));
+    await reg.register(mkManifest('a2'));
+    reg.clear();
+    expect(reg.size).toBe(0);
+  });
+
+  // ─── Export/Import ───────────────────────────────────────────────
+  it('export and import round-trip', async () => {
+    await reg.register(mkManifest('a1'));
+    const data = reg.export();
+    expect(data.agents.length).toBe(1);
+    const reg2 = new AgentRegistry({ autoCleanup: false });
+    await reg2.import(data);
+    expect(reg2.has('a1')).toBe(true);
+    reg2.stop();
+  });
+});

package/src/agents/__tests__/AgentRegistry.test.ts

@@ -0,0 +1,148 @@
+import { describe, it, expect, beforeEach, afterEach } from 'vitest';
+import { AgentRegistry } from '../AgentRegistry';
+import type { AgentManifest } from '../AgentManifest';
+
+function makeManifest(id: string, overrides: Partial<AgentManifest> = {}): AgentManifest {
+  return {
+    id,
+    name: id,
+    version: '1.0.0',
+    capabilities: [{ type: 'compute', domain: 'general' }],
+    status: 'online',
+    trustLevel: 'local',
+    endpoints: [{ protocol: 'http', address: 'localhost' }],
+    ...overrides,
+  } as AgentManifest;
+}
+
+describe('AgentRegistry', () => {
+  let registry: AgentRegistry;
+
+  beforeEach(() => {
+    registry = new AgentRegistry({ autoCleanup: false });
+  });
+
+  afterEach(() => {
+    registry.stop();
+  });
+
+  // ---------------------------------------------------------------------------
+  // Registration
+  // ---------------------------------------------------------------------------
+
+  it('register adds an agent', async () => {
+    await registry.register(makeManifest('agent-1'));
+    expect(registry.size).toBe(1); // size is a getter
+  });
+
+  it('has returns true for registered agent', async () => {
+    await registry.register(makeManifest('agent-1'));
+    expect(registry.has('agent-1')).toBe(true);
+    expect(registry.has('nope')).toBe(false);
+  });
+
+  it('get returns manifest', async () => {
+    await registry.register(makeManifest('agent-1'));
+    const m = registry.get('agent-1');
+    expect(m).toBeDefined();
+    expect(m!.id).toBe('agent-1');
+  });
+
+  it('register allows updating same ID', async () => {
+    await registry.register(makeManifest('agent-1'));
+    // Registering the same ID again is an update, not an error
+    await registry.register(makeManifest('agent-1', { version: '2.0.0' }));
+    expect(registry.size).toBe(1);
+  });
+
+  // ---------------------------------------------------------------------------
+  // Deregistration
+  // ---------------------------------------------------------------------------
+
+  it('deregister removes an agent', async () => {
+    await registry.register(makeManifest('agent-1'));
+    await registry.deregister('agent-1');
+    expect(registry.size).toBe(0);
+  });
+
+  // ---------------------------------------------------------------------------
+  // Heartbeat
+  // ---------------------------------------------------------------------------
+
+  it('heartbeat updates agent entry', async () => {
+    await registry.register(makeManifest('agent-1'));
+    await registry.heartbeat('agent-1');
+    expect(registry.has('agent-1')).toBe(true);
+  });
+
+  // ---------------------------------------------------------------------------
+  // Discovery
+  // ---------------------------------------------------------------------------
+
+  it('discover finds matching agents', async () => {
+    await registry.register(
+      makeManifest('a1', { capabilities: [{ type: 'compute', domain: 'general' }] })
+    );
+    await registry.register(
+      makeManifest('a2', { capabilities: [{ type: 'render', domain: 'graphics' }] })
+    );
+    const results = await registry.discover({ type: 'compute' });
+    expect(results.length).toBeGreaterThanOrEqual(1);
+  });
+
+  it('findBest returns the best match', async () => {
+    await registry.register(
+      makeManifest('a1', { capabilities: [{ type: 'compute', domain: 'general' }] })
+    );
+    const best = await registry.findBest({ type: 'compute' });
+    expect(best).not.toBeNull();
+    expect(best!.id).toBe('a1');
+  });
+
+  it('findBest returns null when no match', async () => {
+    const best = await registry.findBest({ type: 'compute' });
+    expect(best).toBeNull();
+  });
+
+  // ---------------------------------------------------------------------------
+  // getAllManifests
+  // ---------------------------------------------------------------------------
+
+  it('getAllManifests returns all agents', async () => {
+    await registry.register(makeManifest('a1'));
+    await registry.register(makeManifest('a2'));
+    expect(registry.getAllManifests()).toHaveLength(2);
+  });
+
+  // ---------------------------------------------------------------------------
+  // Status Counts
+  // ---------------------------------------------------------------------------
+
+  it('getStatusCounts reports by status', async () => {
+    await registry.register(makeManifest('a1'));
+    const counts = registry.getStatusCounts();
+    expect(counts['online']).toBe(1);
+  });
+
+  // ---------------------------------------------------------------------------
+  // Clear
+  // ---------------------------------------------------------------------------
+
+  it('clear removes all agents', async () => {
+    await registry.register(makeManifest('a1'));
+    await registry.register(makeManifest('a2'));
+    registry.clear();
+    expect(registry.size).toBe(0);
+  });
+
+  // ---------------------------------------------------------------------------
+  // Export
+  // ---------------------------------------------------------------------------
+
+  it('export returns registry snapshot', async () => {
+    await registry.register(makeManifest('a1'));
+    const data = registry.export();
+    expect(data.agents).toHaveLength(1);
+    expect(data.timestamp).toBeGreaterThan(0);
+  });
+});