feat: vendor permission system source

pi-permission-system/test/service-lifecycle.test.ts

@@ -0,0 +1,157 @@
+import { beforeEach, describe, expect, it, vi } from "vitest";
+import type { RegisteredChildDetector } from "#src/authority/subagent-detection";
+import type { PermissionsService } from "#src/service";
+import {
+  PermissionServiceLifecycle,
+  type ServiceLifecycle,
+} from "#src/service-lifecycle";
+
+import { makeCtx } from "#test/helpers/handler-fixtures";
+
+// ── module stubs ───────────────────────────────────────────────────────────
+
+const mockIsRegisteredChild = vi.fn<(ctx: unknown) => boolean>();
+const mockPublishPermissionsService = vi.hoisted(() => vi.fn<() => void>());
+const mockUnpublishPermissionsService = vi.hoisted(() => vi.fn<() => void>());
+const mockEmitReadyEvent = vi.hoisted(() => vi.fn<() => void>());
+
+vi.mock("#src/service", () => ({
+  publishPermissionsService: mockPublishPermissionsService,
+  unpublishPermissionsService: mockUnpublishPermissionsService,
+}));
+vi.mock("#src/permission-events", () => ({
+  emitReadyEvent: mockEmitReadyEvent,
+}));
+
+// ── helpers ────────────────────────────────────────────────────────────────
+
+function makeService(): PermissionsService {
+  return {
+    checkPermission: vi.fn(),
+    getToolPermission: vi.fn(),
+    registerToolInputFormatter: vi.fn(),
+    registerToolAccessExtractor: vi.fn(),
+    registerAuthorizer: vi.fn(),
+  };
+}
+
+function makeDetection(): RegisteredChildDetector {
+  return { isRegisteredChild: mockIsRegisteredChild };
+}
+
+function makeLifecycle(overrides?: { subscriptions?: (() => void)[] }) {
+  const service = makeService();
+  const detection = makeDetection();
+  const events = { emit: vi.fn(), on: vi.fn() };
+  const subscriptions = overrides?.subscriptions ?? [];
+  const lifecycle = new PermissionServiceLifecycle(
+    service,
+    detection,
+    events,
+    subscriptions,
+  );
+  return { lifecycle, service, detection, events, subscriptions };
+}
+
+beforeEach(() => {
+  mockIsRegisteredChild.mockReset();
+  mockIsRegisteredChild.mockReturnValue(false);
+  mockPublishPermissionsService.mockReset();
+  mockUnpublishPermissionsService.mockReset();
+  mockEmitReadyEvent.mockReset();
+});
+
+// ── ServiceLifecycle interface shape ──────────────────────────────────────
+
+it("PermissionServiceLifecycle satisfies ServiceLifecycle", () => {
+  const { lifecycle } = makeLifecycle();
+  const _: ServiceLifecycle = lifecycle;
+  expect(_).toBeDefined();
+});
+
+// ── activate ──────────────────────────────────────────────────────────────
+
+describe("activate", () => {
+  it("publishes the service for a non-child session", () => {
+    const ctx = makeCtx();
+    const { lifecycle, service } = makeLifecycle();
+    mockIsRegisteredChild.mockReturnValue(false);
+    lifecycle.activate(ctx);
+    expect(mockPublishPermissionsService).toHaveBeenCalledWith(service);
+  });
+
+  it("skips publishing for a registered child session", () => {
+    const ctx = makeCtx();
+    const { lifecycle } = makeLifecycle();
+    mockIsRegisteredChild.mockReturnValue(true);
+    lifecycle.activate(ctx);
+    expect(mockPublishPermissionsService).not.toHaveBeenCalled();
+  });
+
+  it("always emits the ready event, even for a child session", () => {
+    const ctx = makeCtx();
+    const { lifecycle, events } = makeLifecycle();
+    mockIsRegisteredChild.mockReturnValue(true);
+    lifecycle.activate(ctx);
+    expect(mockEmitReadyEvent).toHaveBeenCalledWith(events);
+  });
+
+  it("emits ready after publishing the service", () => {
+    const ctx = makeCtx();
+    const order: string[] = [];
+    mockPublishPermissionsService.mockImplementation(() =>
+      order.push("publish"),
+    );
+    mockEmitReadyEvent.mockImplementation(() => order.push("ready"));
+    const { lifecycle } = makeLifecycle();
+    lifecycle.activate(ctx);
+    expect(order).toEqual(["publish", "ready"]);
+  });
+
+  it("consults the detector with ctx", () => {
+    const ctx = makeCtx();
+    const { lifecycle } = makeLifecycle();
+    lifecycle.activate(ctx);
+    expect(mockIsRegisteredChild).toHaveBeenCalledWith(ctx);
+  });
+});
+
+// ── teardown ──────────────────────────────────────────────────────────────
+
+describe("teardown", () => {
+  it("calls each subscription unsubscribe function", () => {
+    const unsub1 = vi.fn();
+    const unsub2 = vi.fn();
+    const unsub3 = vi.fn();
+    const { lifecycle } = makeLifecycle({
+      subscriptions: [unsub1, unsub2, unsub3],
+    });
+    lifecycle.teardown();
+    expect(unsub1).toHaveBeenCalledOnce();
+    expect(unsub2).toHaveBeenCalledOnce();
+    expect(unsub3).toHaveBeenCalledOnce();
+  });
+
+  it("unpublishes the service after running subscriptions", () => {
+    const order: string[] = [];
+    const unsub = vi.fn(() => order.push("unsub"));
+    mockUnpublishPermissionsService.mockImplementation(() =>
+      order.push("unpublish"),
+    );
+    const { lifecycle } = makeLifecycle({ subscriptions: [unsub] });
+    lifecycle.teardown();
+    expect(order).toEqual(["unsub", "unpublish"]);
+  });
+
+  it("passes the service to unpublishPermissionsService", () => {
+    const { lifecycle, service } = makeLifecycle();
+    lifecycle.teardown();
+    expect(mockUnpublishPermissionsService).toHaveBeenCalledWith(service);
+  });
+
+  it("works with no subscriptions", () => {
+    const { lifecycle } = makeLifecycle({ subscriptions: [] });
+    expect(() => lifecycle.teardown()).not.toThrow();
+    expect(mockUnpublishPermissionsService).toHaveBeenCalledOnce();
+  });
+});