@ai-setting/roy-agent-core 1.3.10 → 1.3.14

package/src/env/llm/llm.test.ts

@@ -1,233 +0,0 @@
-import { describe, test, expect, beforeEach, afterEach, vi } from "bun:test";
-import { LLMComponent } from "./llm";
-import { ConfigComponent } from "../../config/config-component";
-import type { Environment } from "../interface";
-
-describe("LLMComponent", () => {
-  let component: LLMComponent;
-  let configComponent: ConfigComponent;
-  let mockEnv: Environment;
-
-  beforeEach(() => {
-    component = new LLMComponent();
-    configComponent = new ConfigComponent();
-    
-    // Mock environment
-    mockEnv = {
-      getComponent: vi.fn((name: string) => {
-        if (name === "config") return configComponent;
-        return undefined;
-      }),
-      hasComponent: vi.fn((name: string) => name === "config"),
-      logger: vi.fn(),
-      trace: vi.fn(),
-    } as unknown as Environment;
-  });
-
-  afterEach(() => {
-    vi.restoreAllMocks();
-  });
-
-  describe("constructor", () => {
-    test("should create instance with correct name and version", () => {
-      expect(component.name).toBe("llm");
-      expect(component.version).toBe("2.0.0");
-    });
-
-    test("should have hookManager", () => {
-      expect(component.hookManager).toBeDefined();
-    });
-  });
-
-  describe("init", () => {
-    test("should initialize with ConfigComponent", async () => {
-      await component.init({
-        name: "llm",
-        version: "1.0.0",
-        enabled: true,
-        env: mockEnv,
-        options: { configComponent }
-      });
-      expect(component.getStatus()).toBe("running");
-    });
-
-    test("should throw error without ConfigComponent", async () => {
-      await expect(component.init({
-        name: "llm",
-        version: "1.0.0",
-        enabled: true,
-        env: mockEnv,
-      })).rejects.toThrow("ConfigComponent is required");
-    });
-  });
-
-  describe("hooks", () => {
-    test("should add hook", async () => {
-      await component.init({
-        name: "llm",
-        version: "1.0.0",
-        enabled: true,
-        env: mockEnv,
-        options: { configComponent }
-      });
-
-      const beforeFn = async () => {};
-      const afterFn = async () => {};
-
-      component.addHook("llm.before-invoke", "test-before-hook", beforeFn);
-      component.addHook("llm.after-invoke", "test-after-hook", afterFn);
-
-      // 通过 hookManager 验证 hooks 已注册
-      expect(component.hookManager.count("llm.before-invoke")).toBe(1);
-      expect(component.hookManager.count("llm.after-invoke")).toBe(1);
-    });
-
-    test("should remove hook", async () => {
-      await component.init({
-        name: "llm",
-        version: "1.0.0",
-        enabled: true,
-        env: mockEnv,
-        options: { configComponent }
-      });
-
-      const hookFn = async () => {};
-
-      component.addHook("llm.before-invoke", "test-hook", hookFn);
-      expect(component.hookManager.count("llm.before-invoke")).toBe(1);
-
-      component.removeHook("llm.before-invoke", "test-hook");
-      expect(component.hookManager.count("llm.before-invoke")).toBe(0);
-    });
-  });
-
-  describe("lifecycle", () => {
-    test("should transition through lifecycle states", async () => {
-      expect(component.getStatus()).toBe("created");
-
-      await component.init({
-        name: "llm",
-        version: "1.0.0",
-        enabled: true,
-        env: mockEnv,
-        options: { configComponent }
-      });
-      expect(component.getStatus()).toBe("running");
-
-      await component.onStop();
-      expect(component.getStatus()).toBe("stopped");
-
-      await component.onStart();
-      expect(component.getStatus()).toBe("running");
-    });
-  });
-
-  describe("invoke", () => {
-    test("should throw error when component is not running", async () => {
-      await expect(component.invoke({
-        messages: [{ role: "user", content: "Hello" }]
-      })).rejects.toThrow("LLMComponent is not running");
-    });
-
-    test("should throw error when apiKey is missing", async () => {
-      // ConfigComponent is empty, no API key configured
-      await component.init({
-        name: "llm",
-        version: "1.0.0",
-        enabled: true,
-        env: mockEnv,
-        options: { configComponent }
-      });
-
-      await expect(component.invoke({
-        messages: [{ role: "user", content: "Hello" }]
-      })).rejects.toThrow(/API key not found for provider/);
-    });
-
-    test("should throw error when provider does not exist", async () => {
-      await component.init({
-        name: "llm",
-        version: "1.0.0",
-        enabled: true,
-        env: mockEnv,
-        options: { configComponent }
-      });
-
-      await expect(component.invoke({
-        messages: [{ role: "user", content: "Hello" }],
-        providerId: "non-existent-provider"
-      })).rejects.toThrow(/API key not found for provider/);
-    });
-  });
-
-  describe("config hot reload", () => {
-    test("should register config watcher on init", async () => {
-      const watchSpy = vi.spyOn(configComponent, "watch");
-      
-      await component.init({
-        name: "llm",
-        version: "1.0.0",
-        enabled: true,
-        env: mockEnv,
-        options: { configComponent }
-      });
-
-      // Verify watch was called for llm.* patterns
-      expect(watchSpy).toHaveBeenCalled();
-    });
-
-    test("should cleanup watcher on stop", async () => {
-      const watchSpy = vi.spyOn(configComponent, "watch");
-      
-      await component.init({
-        name: "llm",
-        version: "1.0.0",
-        enabled: true,
-        env: mockEnv,
-        options: { configComponent }
-      });
-
-      // Get the unwatch function returned by watch
-      const watchCall = watchSpy.mock.calls[0];
-      const callback = watchCall?.[1] as Function;
-      
-      // Stop should be successful (watcher cleaned up)
-      await component.onStop();
-      expect(component.getStatus()).toBe("stopped");
-    });
-
-    test("should handle provider config reload", async () => {
-      // Set providers config before init
-      configComponent.set("llm.providers", {
-        openai: {
-          apiKey: "test-key",
-          baseURL: "https://api.openai.com/v1"
-        }
-      });
-      
-      await component.init({
-        name: "llm",
-        version: "1.0.0",
-        enabled: true,
-        env: mockEnv,
-        options: { configComponent }
-      });
-
-      // Verify provider is loaded from config
-      const provider = component.getProvider("openai");
-      expect(provider).toBeDefined();
-      expect(provider?.apiKey).toBe("test-key");
-
-      // Simulate config change event
-      component.onConfigChange({
-        key: "llm.providers",
-        oldValue: { openai: { apiKey: "test-key" } },
-        newValue: { openai: { apiKey: "new-key" } }
-      });
-
-      // After config change event, onConfigChange should be called successfully
-      // (Verification is done by checking no error is thrown)
-      expect(true).toBe(true);
-    });
-  });
-});