Просмотр исходного кода

test(studio): align server mocks with service management split

Ma 6 месяцев назад
Родитель
Сommit
18795c1d33
1 измененных файлов с 12 добавлено и 0 удалено
  1. 12 0
      packages/studio/src/api/server.test.ts

+ 12 - 0
packages/studio/src/api/server.test.ts

@@ -28,6 +28,9 @@ const loadBookSessionMock = vi.fn();
 const persistBookSessionMock = vi.fn();
 const appendBookSessionMessageMock = vi.fn();
 const resolveServiceModelMock = vi.fn();
+const loadSecretsMock = vi.fn();
+const saveSecretsMock = vi.fn();
+const getServiceApiKeyMock = vi.fn();
 
 const logger = {
   child: () => logger,
@@ -124,6 +127,9 @@ vi.mock("@actalk/inkos-core", () => {
     persistBookSession: persistBookSessionMock,
     appendBookSessionMessage: appendBookSessionMessageMock,
     resolveServiceModel: resolveServiceModelMock,
+    loadSecrets: loadSecretsMock,
+    saveSecrets: saveSecretsMock,
+    getServiceApiKey: getServiceApiKeyMock,
     GLOBAL_ENV_PATH: join(tmpdir(), "inkos-global.env"),
   };
 });
@@ -280,6 +286,9 @@ describe("createStudioServer daemon lifecycle", () => {
     persistBookSessionMock.mockReset();
     appendBookSessionMessageMock.mockReset();
     resolveServiceModelMock.mockReset();
+    loadSecretsMock.mockReset();
+    saveSecretsMock.mockReset();
+    getServiceApiKeyMock.mockReset();
     // Default BookSession for agent tests
     const defaultBookSession = {
       sessionId: "agent-session-1",
@@ -298,6 +307,9 @@ describe("createStudioServer daemon lifecycle", () => {
       responseText: "Agent response.",
       messages: [],
     });
+    loadSecretsMock.mockResolvedValue({ services: {} });
+    saveSecretsMock.mockResolvedValue(undefined);
+    getServiceApiKeyMock.mockResolvedValue(undefined);
   });
 
   afterEach(async () => {