skybridge 0.0.0-dev.fc2cd30 → 0.0.0-dev.fc2ea93

package/dist/web/components/modal-provider.js.map

@@ -1 +1 @@
-{"version":3,"file":"modal-provider.js","sourceRoot":"","sources":["../../../src/web/components/modal-provider.tsx"],"names":[],"mappings":";AAAA,OAAO,EAAkB,SAAS,EAAE,oBAAoB,EAAE,MAAM,OAAO,CAAC;AACxE,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAEpD,MAAM,WAAW,GAAG;;;;;;;;;;;;;;;;;CAiBnB,CAAC;AAEF,MAAM,UAAU,aAAa,CAAC,EAAE,QAAQ,EAA2B;IACjE,MAAM,OAAO,GAAG,aAAa,CAAC,WAAW,EAAE,CAAC;IAE5C,MAAM,EAAE,IAAI,EAAE,GAAG,oBAAoB,CACnC,OAAO,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC,SAAS,EAChD,OAAO,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC,WAAW,CACnD,CAAC;IACF,MAAM,MAAM,GAAG,IAAI,KAAK,OAAO,CAAC;IAEhC,MAAM,mBAAmB,GAAG,CAAC,CAAmB,EAAE,EAAE;QAClD,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;YACjC,OAAO,CAAC,UAAU,EAAE,CAAC;QACvB,CAAC;IACH,CAAC,CAAC;IAEF,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO;QACT,CAAC;QACD,MAAM,OAAO,GAAG,CAAC,CAAgB,EAAE,EAAE;YACnC,IAAI,CAAC,CAAC,GAAG,KAAK,QAAQ,EAAE,CAAC;gBACvB,OAAO,CAAC,UAAU,EAAE,CAAC;YACvB,CAAC;QACH,CAAC,CAAC;QACF,QAAQ,CAAC,gBAAgB,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QAC9C,OAAO,GAAG,EAAE,CAAC,QAAQ,CAAC,mBAAmB,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAChE,CAAC,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;IAEtB,OAAO,CACL,8BACE,0BAAQ,WAAW,GAAS,EAC3B,MAAM,IAAI,CAET,cACE,IAAI,EAAC,QAAQ,EACb,SAAS,EAAC,mBAAmB,EAC7B,OAAO,EAAE,mBAAmB,GAC5B,CACH,EACD,cAAK,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,SAAS,YACtD,QAAQ,GACL,IACL,CACJ,CAAC;AACJ,CAAC","sourcesContent":["import { type ReactNode, useEffect, useSyncExternalStore } from \"react\";\nimport { McpAppAdaptor } from \"../bridges/index.js\";\n\nconst modalStyles = `\n.sb-modal-backdrop {\n  position: fixed;\n  inset: 0;\n  background: rgba(0, 0, 0, 0.5);\n  z-index: 9998;\n}\n.sb-modal-container {\n  border-radius: 12px;\n  position: fixed;\n  inset: 0;\n  margin: auto;\n  width: fit-content;\n  height: fit-content;\n  background: white;\n  z-index: 9999;\n}\n`;\n\nexport function ModalProvider({ children }: { children: ReactNode }) {\n  const adaptor = McpAppAdaptor.getInstance();\n\n  const { mode } = useSyncExternalStore(\n    adaptor.getHostContextStore(\"display\").subscribe,\n    adaptor.getHostContextStore(\"display\").getSnapshot,\n  );\n  const isOpen = mode === \"modal\";\n\n  const handleBackdropClick = (e: React.MouseEvent) => {\n    if (e.target === e.currentTarget) {\n      adaptor.closeModal();\n    }\n  };\n\n  useEffect(() => {\n    if (!isOpen) {\n      return;\n    }\n    const handler = (e: KeyboardEvent) => {\n      if (e.key === \"Escape\") {\n        adaptor.closeModal();\n      }\n    };\n    document.addEventListener(\"keydown\", handler);\n    return () => document.removeEventListener(\"keydown\", handler);\n  }, [isOpen, adaptor]);\n\n  return (\n    <>\n      <style>{modalStyles}</style>\n      {isOpen && (\n        // biome-ignore lint/a11y/useKeyWithClickEvents: backdrop isn't focusable\n        <div\n          role=\"dialog\"\n          className=\"sb-modal-backdrop\"\n          onClick={handleBackdropClick}\n        />\n      )}\n      <div className={isOpen ? \"sb-modal-container\" : undefined}>\n        {children}\n      </div>\n    </>\n  );\n}\n"]}
+{"version":3,"file":"modal-provider.js","sourceRoot":"","sources":["../../../src/web/components/modal-provider.tsx"],"names":[],"mappings":";AAAA,OAAO,EAAkB,SAAS,EAAE,oBAAoB,EAAE,MAAM,OAAO,CAAC;AACxE,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAEjD,MAAM,WAAW,GAAG;;;;;;;;;;;;;;;;;CAiBnB,CAAC;AAEF,MAAM,UAAU,aAAa,CAAC,EAAE,QAAQ,EAA2B;IACjE,MAAM,OAAO,GAAG,UAAU,EAAE,CAAC;IAC7B,MAAM,YAAY,GAAG,OAAO,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC;IAC5D,MAAM,EAAE,IAAI,EAAE,GAAG,oBAAoB,CACnC,YAAY,CAAC,SAAS,EACtB,YAAY,CAAC,WAAW,CACzB,CAAC;IACF,MAAM,MAAM,GAAG,IAAI,KAAK,OAAO,CAAC;IAEhC,MAAM,mBAAmB,GAAG,CAAC,CAAmB,EAAE,EAAE;QAClD,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;YACjC,OAAO,CAAC,UAAU,EAAE,CAAC;QACvB,CAAC;IACH,CAAC,CAAC;IAEF,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO;QACT,CAAC;QACD,MAAM,OAAO,GAAG,CAAC,CAAgB,EAAE,EAAE;YACnC,IAAI,CAAC,CAAC,GAAG,KAAK,QAAQ,EAAE,CAAC;gBACvB,OAAO,CAAC,UAAU,EAAE,CAAC;YACvB,CAAC;QACH,CAAC,CAAC;QACF,QAAQ,CAAC,gBAAgB,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QAC9C,OAAO,GAAG,EAAE,CAAC,QAAQ,CAAC,mBAAmB,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAChE,CAAC,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;IAEtB,OAAO,CACL,8BACE,0BAAQ,WAAW,GAAS,EAC3B,MAAM,IAAI,CAET,cACE,IAAI,EAAC,QAAQ,EACb,SAAS,EAAC,mBAAmB,EAC7B,OAAO,EAAE,mBAAmB,GAC5B,CACH,EACD,cAAK,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,SAAS,YACtD,QAAQ,GACL,IACL,CACJ,CAAC;AACJ,CAAC","sourcesContent":["import { type ReactNode, useEffect, useSyncExternalStore } from \"react\";\nimport { getAdaptor } from \"../bridges/index.js\";\n\nconst modalStyles = `\n.sb-modal-backdrop {\n  position: fixed;\n  inset: 0;\n  background: rgba(0, 0, 0, 0.5);\n  z-index: 9998;\n}\n.sb-modal-container {\n  border-radius: 12px;\n  position: fixed;\n  inset: 0;\n  margin: auto;\n  width: fit-content;\n  height: fit-content;\n  background: white;\n  z-index: 9999;\n}\n`;\n\nexport function ModalProvider({ children }: { children: ReactNode }) {\n  const adaptor = getAdaptor();\n  const displayStore = adaptor.getHostContextStore(\"display\");\n  const { mode } = useSyncExternalStore(\n    displayStore.subscribe,\n    displayStore.getSnapshot,\n  );\n  const isOpen = mode === \"modal\";\n\n  const handleBackdropClick = (e: React.MouseEvent) => {\n    if (e.target === e.currentTarget) {\n      adaptor.closeModal();\n    }\n  };\n\n  useEffect(() => {\n    if (!isOpen) {\n      return;\n    }\n    const handler = (e: KeyboardEvent) => {\n      if (e.key === \"Escape\") {\n        adaptor.closeModal();\n      }\n    };\n    document.addEventListener(\"keydown\", handler);\n    return () => document.removeEventListener(\"keydown\", handler);\n  }, [isOpen, adaptor]);\n\n  return (\n    <>\n      <style>{modalStyles}</style>\n      {isOpen && (\n        // biome-ignore lint/a11y/useKeyWithClickEvents: backdrop isn't focusable\n        <div\n          role=\"dialog\"\n          className=\"sb-modal-backdrop\"\n          onClick={handleBackdropClick}\n        />\n      )}\n      <div className={isOpen ? \"sb-modal-container\" : undefined}>\n        {children}\n      </div>\n    </>\n  );\n}\n"]}

package/dist/web/create-store.test.js

@@ -1,12 +1,14 @@
 import { cleanup } from "@testing-library/react";
 import { afterEach, beforeEach, describe, expect, it, vi, } from "vitest";
-import { McpAppAdaptor } from "./bridges/mcp-app/adaptor.js";
+import { HostAdaptor } from "./bridges/adaptor.js";
+import { getAdaptor } from "./bridges/get-adaptor.js";
 import { McpAppBridge } from "./bridges/mcp-app/bridge.js";
 import { createStore } from "./create-store.js";
 import { VIEW_CONTEXT_KEY } from "./data-llm.js";
 import { getMcpAppHostPostMessageMock, MockResizeObserver, } from "./hooks/test/utils.js";
 describe("createStore", () => {
     afterEach(() => {
+        HostAdaptor.resetInstance();
         vi.unstubAllGlobals();
         vi.resetAllMocks();
     });
@@ -78,16 +80,17 @@ describe("createStore", () => {
     });
     describe("mcp-app mode", () => {
         beforeEach(() => {
+            vi.stubGlobal("openai", undefined);
             vi.stubGlobal("skybridge", { hostType: "mcp-app" });
             vi.stubGlobal("ResizeObserver", MockResizeObserver);
         });
         afterEach(async () => {
             cleanup();
+            HostAdaptor.resetInstance();
             McpAppBridge.resetInstance();
-            McpAppAdaptor.resetInstance();
         });
         it("should initialize with null viewState", () => {
-            const adaptor = McpAppAdaptor.getInstance();
+            const adaptor = getAdaptor();
             const viewState = adaptor.getHostContextStore("viewState").getSnapshot();
             expect(viewState).toBeNull();
         });
@@ -99,7 +102,7 @@ describe("createStore", () => {
             expect(store.getState()).toEqual({ count: 0 });
         });
         it("should update in-memory state via setViewState", async () => {
-            const adaptor = McpAppAdaptor.getInstance();
+            const adaptor = getAdaptor();
             vi.spyOn(adaptor, "setViewState").mockResolvedValue(undefined);
             const storeCreator = (set) => ({
                 count: 0,
@@ -114,7 +117,7 @@ describe("createStore", () => {
         it("should notify listeners when view state changes", async () => {
             const postMessageMock = getMcpAppHostPostMessageMock();
             vi.stubGlobal("parent", { postMessage: postMessageMock });
-            const adaptor = McpAppAdaptor.getInstance();
+            const adaptor = getAdaptor();
             const listener = vi.fn();
             adaptor.getHostContextStore("viewState").subscribe(listener);
             await adaptor.setViewState({ count: 42 });

package/dist/web/create-store.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"create-store.test.js","sourceRoot":"","sources":["../../src/web/create-store.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,wBAAwB,CAAC;AACjD,OAAO,EACL,SAAS,EACT,UAAU,EACV,QAAQ,EACR,MAAM,EACN,EAAE,EAEF,EAAE,GACH,MAAM,QAAQ,CAAC;AAEhB,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAC3D,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AACjD,OAAO,EACL,4BAA4B,EAC5B,kBAAkB,GACnB,MAAM,uBAAuB,CAAC;AAE/B,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,gBAAgB,EAAE,CAAC;QACtB,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,IAAI,UAGH,CAAC;QAEF,UAAU,CAAC,GAAG,EAAE;YACd,UAAU,GAAG;gBACX,WAAW,EAAE,IAAI;gBACjB,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;aACrD,CAAC;YACF,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;YACpC,EAAE,CAAC,UAAU,CAAC,WAAW,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YAErD,MAAM,YAAY,GAA+C,GAAG,EAAE,CAAC,CAAC;gBACtE,KAAK,EAAE,CAAC;aACT,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,CAAC,CAAC;YAExC,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAElD,MAAM,YAAY,GAA+C,GAAG,EAAE,CAAC,CAAC;gBACtE,KAAK,EAAE,CAAC;gBACR,IAAI,EAAE,SAAS;aAChB,CAAC,CAAC;YACH,MAAM,YAAY,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;YAEnD,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;YAEtD,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;YAEzE,MAAM,YAAY,GAA+C,GAAG,EAAE,CAAC,CAAC;gBACtE,KAAK,EAAE,CAAC;gBACR,IAAI,EAAE,SAAS;aAChB,CAAC,CAAC;YACH,MAAM,WAAW,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;YAClD,UAAU,CAAC,WAAW,GAAG,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC;YAEvD,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,CAAC,CAAC;YAExC,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAE5E,MAAM,YAAY,GAA+C,CAC/D,GAAG,EACH,EAAE,CAAC,CAAC;gBACJ,KAAK,EAAE,CAAC;gBACR,SAAS,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,GAAG,CAAC,EAAE,CAAC,CAAC;aAC9D,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,CAAC,CAAC;YACxC,KAAK,CAAC,QAAQ,EAAE,CAAC,SAAS,EAAE,CAAC;YAE7B,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;gBACpB,MAAM,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC,gBAAgB,EAAE,CAAC;YACvD,CAAC,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YAC9D,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;gBACvB,YAAY,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE;gBAC1B,cAAc,EAAE,EAAE;aACnB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YAEvD,MAAM,YAAY,GAA+C,GAAG,EAAE,CAAC,CAAC;gBACtE,KAAK,EAAE,CAAC;aACT,CAAC,CAAC;YACH,MAAM,WAAW,GAAG;gBAClB,KAAK,EAAE,CAAC;gBACR,CAAC,gBAAgB,CAAC,EAAE,eAAe;aACpC,CAAC;YACF,UAAU,CAAC,WAAW,GAAG,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC;YAEvD,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,CAAC,CAAC;YAExC,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;YAC/C,MAAM,CACH,KAAK,CAAC,QAAQ,EAA8B,CAAC,gBAAgB,CAAC,CAChE,CAAC,aAAa,EAAE,CAAC;QACpB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,UAAU,CAAC,GAAG,EAAE;YACd,EAAE,CAAC,UAAU,CAAC,WAAW,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;YACpD,EAAE,CAAC,UAAU,CAAC,gBAAgB,EAAE,kBAAkB,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,SAAS,CAAC,KAAK,IAAI,EAAE;YACnB,OAAO,EAAE,CAAC;YACV,YAAY,CAAC,aAAa,EAAE,CAAC;YAC7B,aAAa,CAAC,aAAa,EAAE,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,OAAO,GAAG,aAAa,CAAC,WAAW,EAAE,CAAC;YAC5C,MAAM,SAAS,GAAG,OAAO,CAAC,mBAAmB,CAAC,WAAW,CAAC,CAAC,WAAW,EAAE,CAAC;YAEzE,MAAM,CAAC,SAAS,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yEAAyE,EAAE,GAAG,EAAE;YAEjF,MAAM,YAAY,GAA+C,GAAG,EAAE,CAAC,CAAC;gBACtE,KAAK,EAAE,CAAC;aACT,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,CAAC,CAAC;YAExC,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,OAAO,GAAG,aAAa,CAAC,WAAW,EAAE,CAAC;YAC5C,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAG/D,MAAM,YAAY,GAA+C,CAC/D,GAAG,EACH,EAAE,CAAC,CAAC;gBACJ,KAAK,EAAE,CAAC;gBACR,SAAS,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,GAAG,CAAC,EAAE,CAAC,CAAC;aAC9D,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,CAAC,CAAC;YACxC,KAAK,CAAC,QAAQ,EAAE,CAAC,SAAS,EAAE,CAAC;YAE7B,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;gBACpB,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;YAClE,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,eAAe,GAAG,4BAA4B,EAAE,CAAC;YACvD,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,EAAE,WAAW,EAAE,eAAe,EAAE,CAAC,CAAC;YAE1D,MAAM,OAAO,GAAG,aAAa,CAAC,WAAW,EAAE,CAAC;YAC5C,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAEzB,OAAO,CAAC,mBAAmB,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YAC7D,MAAM,OAAO,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;YAE1C,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,MAAM,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,yBAAyB,EAAE,CAAC,EAC9D,GAAG,CACJ,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,CAAC;YACpC,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,WAAW,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,OAAO,CAAC;gBACrE,KAAK,EAAE,EAAE;aACV,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { cleanup } from \"@testing-library/react\";\nimport {\n  afterEach,\n  beforeEach,\n  describe,\n  expect,\n  it,\n  type Mock,\n  vi,\n} from \"vitest\";\nimport type { StateCreator } from \"zustand\";\nimport { McpAppAdaptor } from \"./bridges/mcp-app/adaptor.js\";\nimport { McpAppBridge } from \"./bridges/mcp-app/bridge.js\";\nimport { createStore } from \"./create-store.js\";\nimport { VIEW_CONTEXT_KEY } from \"./data-llm.js\";\nimport {\n  getMcpAppHostPostMessageMock,\n  MockResizeObserver,\n} from \"./hooks/test/utils.js\";\n\ndescribe(\"createStore\", () => {\n  afterEach(() => {\n    vi.unstubAllGlobals();\n    vi.resetAllMocks();\n  });\n\n  describe(\"apps-sdk mode\", () => {\n    let OpenaiMock: {\n      widgetState: unknown;\n      setWidgetState: Mock;\n    };\n\n    beforeEach(() => {\n      OpenaiMock = {\n        widgetState: null,\n        setWidgetState: vi.fn().mockResolvedValue(undefined),\n      };\n      vi.stubGlobal(\"openai\", OpenaiMock);\n      vi.stubGlobal(\"skybridge\", { hostType: \"apps-sdk\" });\n    });\n\n    it(\"should create a store without default state\", () => {\n      type TestState = { count: number };\n      const storeCreator: StateCreator<TestState, [], [], TestState> = () => ({\n        count: 0,\n      });\n\n      const store = createStore(storeCreator);\n\n      expect(store.getState()).toEqual({ count: 0 });\n    });\n\n    it(\"should create a store with default state\", () => {\n      type TestState = { count: number; name: string };\n      const storeCreator: StateCreator<TestState, [], [], TestState> = () => ({\n        count: 0,\n        name: \"initial\",\n      });\n      const defaultState = { count: 5, name: \"default\" };\n\n      const store = createStore(storeCreator, defaultState);\n\n      expect(store.getState()).toEqual({ count: 5, name: \"default\" });\n    });\n\n    it(\"should initialize from window.openai.widgetState when available\", () => {\n      type TestState = { count: number; name: string };\n      const storeCreator: StateCreator<TestState, [], [], TestState> = () => ({\n        count: 0,\n        name: \"initial\",\n      });\n      const windowState = { count: 20, name: \"window\" };\n      OpenaiMock.widgetState = { modelContent: windowState };\n\n      const store = createStore(storeCreator);\n\n      expect(store.getState()).toEqual({ count: 20, name: \"window\" });\n    });\n\n    it(\"should persist state changes to window.openai.setWidgetState\", async () => {\n      type TestState = { count: number; increment: () => void };\n      const storeCreator: StateCreator<TestState, [], [], TestState> = (\n        set,\n      ) => ({\n        count: 0,\n        increment: () => set((state) => ({ count: state.count + 1 })),\n      });\n\n      const store = createStore(storeCreator);\n      store.getState().increment();\n\n      await vi.waitFor(() => {\n        expect(OpenaiMock.setWidgetState).toHaveBeenCalled();\n      });\n\n      const callArgs = OpenaiMock.setWidgetState.mock.calls[0]?.[0];\n      expect(callArgs).toEqual({\n        modelContent: { count: 1 },\n        privateContent: {},\n      });\n    });\n\n    it(\"should filter view context from initial state\", () => {\n      type TestState = { count: number };\n      const storeCreator: StateCreator<TestState, [], [], TestState> = () => ({\n        count: 0,\n      });\n      const windowState = {\n        count: 5,\n        [VIEW_CONTEXT_KEY]: \"context-value\",\n      };\n      OpenaiMock.widgetState = { modelContent: windowState };\n\n      const store = createStore(storeCreator);\n\n      expect(store.getState()).toEqual({ count: 5 });\n      expect(\n        (store.getState() as Record<string, unknown>)[VIEW_CONTEXT_KEY],\n      ).toBeUndefined();\n    });\n  });\n\n  describe(\"mcp-app mode\", () => {\n    beforeEach(() => {\n      vi.stubGlobal(\"skybridge\", { hostType: \"mcp-app\" });\n      vi.stubGlobal(\"ResizeObserver\", MockResizeObserver);\n    });\n\n    afterEach(async () => {\n      cleanup();\n      McpAppBridge.resetInstance();\n      McpAppAdaptor.resetInstance();\n    });\n\n    it(\"should initialize with null viewState\", () => {\n      const adaptor = McpAppAdaptor.getInstance();\n      const viewState = adaptor.getHostContextStore(\"viewState\").getSnapshot();\n\n      expect(viewState).toBeNull();\n    });\n\n    it(\"should create a store with default state when no persisted state exists\", () => {\n      type TestState = { count: number };\n      const storeCreator: StateCreator<TestState, [], [], TestState> = () => ({\n        count: 0,\n      });\n\n      const store = createStore(storeCreator);\n\n      expect(store.getState()).toEqual({ count: 0 });\n    });\n\n    it(\"should update in-memory state via setViewState\", async () => {\n      const adaptor = McpAppAdaptor.getInstance();\n      vi.spyOn(adaptor, \"setViewState\").mockResolvedValue(undefined);\n\n      type TestState = { count: number; increment: () => void };\n      const storeCreator: StateCreator<TestState, [], [], TestState> = (\n        set,\n      ) => ({\n        count: 0,\n        increment: () => set((state) => ({ count: state.count + 1 })),\n      });\n\n      const store = createStore(storeCreator);\n      store.getState().increment();\n\n      await vi.waitFor(() => {\n        expect(adaptor.setViewState).toHaveBeenCalledWith({ count: 1 });\n      });\n    });\n\n    it(\"should notify listeners when view state changes\", async () => {\n      const postMessageMock = getMcpAppHostPostMessageMock();\n      vi.stubGlobal(\"parent\", { postMessage: postMessageMock });\n\n      const adaptor = McpAppAdaptor.getInstance();\n      const listener = vi.fn();\n\n      adaptor.getHostContextStore(\"viewState\").subscribe(listener);\n      await adaptor.setViewState({ count: 42 });\n\n      expect(postMessageMock).toHaveBeenCalledWith(\n        expect.objectContaining({ method: \"ui/update-model-context\" }),\n        \"*\",\n      );\n      expect(listener).toHaveBeenCalled();\n      expect(adaptor.getHostContextStore(\"viewState\").getSnapshot()).toEqual({\n        count: 42,\n      });\n    });\n  });\n});\n"]}
+{"version":3,"file":"create-store.test.js","sourceRoot":"","sources":["../../src/web/create-store.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,wBAAwB,CAAC;AACjD,OAAO,EACL,SAAS,EACT,UAAU,EACV,QAAQ,EACR,MAAM,EACN,EAAE,EAEF,EAAE,GACH,MAAM,QAAQ,CAAC;AAEhB,OAAO,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AACnD,OAAO,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AACtD,OAAO,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAC3D,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AACjD,OAAO,EACL,4BAA4B,EAC5B,kBAAkB,GACnB,MAAM,uBAAuB,CAAC;AAE/B,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,SAAS,CAAC,GAAG,EAAE;QACb,WAAW,CAAC,aAAa,EAAE,CAAC;QAC5B,EAAE,CAAC,gBAAgB,EAAE,CAAC;QACtB,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,IAAI,UAGH,CAAC;QAEF,UAAU,CAAC,GAAG,EAAE;YACd,UAAU,GAAG;gBACX,WAAW,EAAE,IAAI;gBACjB,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;aACrD,CAAC;YACF,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;YACpC,EAAE,CAAC,UAAU,CAAC,WAAW,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YAErD,MAAM,YAAY,GAA+C,GAAG,EAAE,CAAC,CAAC;gBACtE,KAAK,EAAE,CAAC;aACT,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,CAAC,CAAC;YAExC,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAElD,MAAM,YAAY,GAA+C,GAAG,EAAE,CAAC,CAAC;gBACtE,KAAK,EAAE,CAAC;gBACR,IAAI,EAAE,SAAS;aAChB,CAAC,CAAC;YACH,MAAM,YAAY,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;YAEnD,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;YAEtD,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;YAEzE,MAAM,YAAY,GAA+C,GAAG,EAAE,CAAC,CAAC;gBACtE,KAAK,EAAE,CAAC;gBACR,IAAI,EAAE,SAAS;aAChB,CAAC,CAAC;YACH,MAAM,WAAW,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;YAClD,UAAU,CAAC,WAAW,GAAG,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC;YAEvD,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,CAAC,CAAC;YAExC,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAE5E,MAAM,YAAY,GAA+C,CAC/D,GAAG,EACH,EAAE,CAAC,CAAC;gBACJ,KAAK,EAAE,CAAC;gBACR,SAAS,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,GAAG,CAAC,EAAE,CAAC,CAAC;aAC9D,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,CAAC,CAAC;YACxC,KAAK,CAAC,QAAQ,EAAE,CAAC,SAAS,EAAE,CAAC;YAE7B,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;gBACpB,MAAM,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC,gBAAgB,EAAE,CAAC;YACvD,CAAC,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YAC9D,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;gBACvB,YAAY,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE;gBAC1B,cAAc,EAAE,EAAE;aACnB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YAEvD,MAAM,YAAY,GAA+C,GAAG,EAAE,CAAC,CAAC;gBACtE,KAAK,EAAE,CAAC;aACT,CAAC,CAAC;YACH,MAAM,WAAW,GAAG;gBAClB,KAAK,EAAE,CAAC;gBACR,CAAC,gBAAgB,CAAC,EAAE,eAAe;aACpC,CAAC;YACF,UAAU,CAAC,WAAW,GAAG,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC;YAEvD,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,CAAC,CAAC;YAExC,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;YAC/C,MAAM,CACH,KAAK,CAAC,QAAQ,EAA8B,CAAC,gBAAgB,CAAC,CAChE,CAAC,aAAa,EAAE,CAAC;QACpB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,UAAU,CAAC,GAAG,EAAE;YACd,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;YACnC,EAAE,CAAC,UAAU,CAAC,WAAW,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;YACpD,EAAE,CAAC,UAAU,CAAC,gBAAgB,EAAE,kBAAkB,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,SAAS,CAAC,KAAK,IAAI,EAAE;YACnB,OAAO,EAAE,CAAC;YACV,WAAW,CAAC,aAAa,EAAE,CAAC;YAC5B,YAAY,CAAC,aAAa,EAAE,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,OAAO,GAAG,UAAU,EAAE,CAAC;YAC7B,MAAM,SAAS,GAAG,OAAO,CAAC,mBAAmB,CAAC,WAAW,CAAC,CAAC,WAAW,EAAE,CAAC;YAEzE,MAAM,CAAC,SAAS,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yEAAyE,EAAE,GAAG,EAAE;YAEjF,MAAM,YAAY,GAA+C,GAAG,EAAE,CAAC,CAAC;gBACtE,KAAK,EAAE,CAAC;aACT,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,CAAC,CAAC;YAExC,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,OAAO,GAAG,UAAU,EAAE,CAAC;YAC7B,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAG/D,MAAM,YAAY,GAA+C,CAC/D,GAAG,EACH,EAAE,CAAC,CAAC;gBACJ,KAAK,EAAE,CAAC;gBACR,SAAS,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,GAAG,CAAC,EAAE,CAAC,CAAC;aAC9D,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,WAAW,CAAC,YAAY,CAAC,CAAC;YACxC,KAAK,CAAC,QAAQ,EAAE,CAAC,SAAS,EAAE,CAAC;YAE7B,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;gBACpB,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;YAClE,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,eAAe,GAAG,4BAA4B,EAAE,CAAC;YACvD,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,EAAE,WAAW,EAAE,eAAe,EAAE,CAAC,CAAC;YAE1D,MAAM,OAAO,GAAG,UAAU,EAAE,CAAC;YAC7B,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAEzB,OAAO,CAAC,mBAAmB,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YAC7D,MAAM,OAAO,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;YAE1C,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,MAAM,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,yBAAyB,EAAE,CAAC,EAC9D,GAAG,CACJ,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,CAAC;YACpC,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,WAAW,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,OAAO,CAAC;gBACrE,KAAK,EAAE,EAAE;aACV,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { cleanup } from \"@testing-library/react\";\nimport {\n  afterEach,\n  beforeEach,\n  describe,\n  expect,\n  it,\n  type Mock,\n  vi,\n} from \"vitest\";\nimport type { StateCreator } from \"zustand\";\nimport { HostAdaptor } from \"./bridges/adaptor.js\";\nimport { getAdaptor } from \"./bridges/get-adaptor.js\";\nimport { McpAppBridge } from \"./bridges/mcp-app/bridge.js\";\nimport { createStore } from \"./create-store.js\";\nimport { VIEW_CONTEXT_KEY } from \"./data-llm.js\";\nimport {\n  getMcpAppHostPostMessageMock,\n  MockResizeObserver,\n} from \"./hooks/test/utils.js\";\n\ndescribe(\"createStore\", () => {\n  afterEach(() => {\n    HostAdaptor.resetInstance();\n    vi.unstubAllGlobals();\n    vi.resetAllMocks();\n  });\n\n  describe(\"apps-sdk mode\", () => {\n    let OpenaiMock: {\n      widgetState: unknown;\n      setWidgetState: Mock;\n    };\n\n    beforeEach(() => {\n      OpenaiMock = {\n        widgetState: null,\n        setWidgetState: vi.fn().mockResolvedValue(undefined),\n      };\n      vi.stubGlobal(\"openai\", OpenaiMock);\n      vi.stubGlobal(\"skybridge\", { hostType: \"apps-sdk\" });\n    });\n\n    it(\"should create a store without default state\", () => {\n      type TestState = { count: number };\n      const storeCreator: StateCreator<TestState, [], [], TestState> = () => ({\n        count: 0,\n      });\n\n      const store = createStore(storeCreator);\n\n      expect(store.getState()).toEqual({ count: 0 });\n    });\n\n    it(\"should create a store with default state\", () => {\n      type TestState = { count: number; name: string };\n      const storeCreator: StateCreator<TestState, [], [], TestState> = () => ({\n        count: 0,\n        name: \"initial\",\n      });\n      const defaultState = { count: 5, name: \"default\" };\n\n      const store = createStore(storeCreator, defaultState);\n\n      expect(store.getState()).toEqual({ count: 5, name: \"default\" });\n    });\n\n    it(\"should initialize from window.openai.widgetState when available\", () => {\n      type TestState = { count: number; name: string };\n      const storeCreator: StateCreator<TestState, [], [], TestState> = () => ({\n        count: 0,\n        name: \"initial\",\n      });\n      const windowState = { count: 20, name: \"window\" };\n      OpenaiMock.widgetState = { modelContent: windowState };\n\n      const store = createStore(storeCreator);\n\n      expect(store.getState()).toEqual({ count: 20, name: \"window\" });\n    });\n\n    it(\"should persist state changes to window.openai.setWidgetState\", async () => {\n      type TestState = { count: number; increment: () => void };\n      const storeCreator: StateCreator<TestState, [], [], TestState> = (\n        set,\n      ) => ({\n        count: 0,\n        increment: () => set((state) => ({ count: state.count + 1 })),\n      });\n\n      const store = createStore(storeCreator);\n      store.getState().increment();\n\n      await vi.waitFor(() => {\n        expect(OpenaiMock.setWidgetState).toHaveBeenCalled();\n      });\n\n      const callArgs = OpenaiMock.setWidgetState.mock.calls[0]?.[0];\n      expect(callArgs).toEqual({\n        modelContent: { count: 1 },\n        privateContent: {},\n      });\n    });\n\n    it(\"should filter view context from initial state\", () => {\n      type TestState = { count: number };\n      const storeCreator: StateCreator<TestState, [], [], TestState> = () => ({\n        count: 0,\n      });\n      const windowState = {\n        count: 5,\n        [VIEW_CONTEXT_KEY]: \"context-value\",\n      };\n      OpenaiMock.widgetState = { modelContent: windowState };\n\n      const store = createStore(storeCreator);\n\n      expect(store.getState()).toEqual({ count: 5 });\n      expect(\n        (store.getState() as Record<string, unknown>)[VIEW_CONTEXT_KEY],\n      ).toBeUndefined();\n    });\n  });\n\n  describe(\"mcp-app mode\", () => {\n    beforeEach(() => {\n      vi.stubGlobal(\"openai\", undefined);\n      vi.stubGlobal(\"skybridge\", { hostType: \"mcp-app\" });\n      vi.stubGlobal(\"ResizeObserver\", MockResizeObserver);\n    });\n\n    afterEach(async () => {\n      cleanup();\n      HostAdaptor.resetInstance();\n      McpAppBridge.resetInstance();\n    });\n\n    it(\"should initialize with null viewState\", () => {\n      const adaptor = getAdaptor();\n      const viewState = adaptor.getHostContextStore(\"viewState\").getSnapshot();\n\n      expect(viewState).toBeNull();\n    });\n\n    it(\"should create a store with default state when no persisted state exists\", () => {\n      type TestState = { count: number };\n      const storeCreator: StateCreator<TestState, [], [], TestState> = () => ({\n        count: 0,\n      });\n\n      const store = createStore(storeCreator);\n\n      expect(store.getState()).toEqual({ count: 0 });\n    });\n\n    it(\"should update in-memory state via setViewState\", async () => {\n      const adaptor = getAdaptor();\n      vi.spyOn(adaptor, \"setViewState\").mockResolvedValue(undefined);\n\n      type TestState = { count: number; increment: () => void };\n      const storeCreator: StateCreator<TestState, [], [], TestState> = (\n        set,\n      ) => ({\n        count: 0,\n        increment: () => set((state) => ({ count: state.count + 1 })),\n      });\n\n      const store = createStore(storeCreator);\n      store.getState().increment();\n\n      await vi.waitFor(() => {\n        expect(adaptor.setViewState).toHaveBeenCalledWith({ count: 1 });\n      });\n    });\n\n    it(\"should notify listeners when view state changes\", async () => {\n      const postMessageMock = getMcpAppHostPostMessageMock();\n      vi.stubGlobal(\"parent\", { postMessage: postMessageMock });\n\n      const adaptor = getAdaptor();\n      const listener = vi.fn();\n\n      adaptor.getHostContextStore(\"viewState\").subscribe(listener);\n      await adaptor.setViewState({ count: 42 });\n\n      expect(postMessageMock).toHaveBeenCalledWith(\n        expect.objectContaining({ method: \"ui/update-model-context\" }),\n        \"*\",\n      );\n      expect(listener).toHaveBeenCalled();\n      expect(adaptor.getHostContextStore(\"viewState\").getSnapshot()).toEqual({\n        count: 42,\n      });\n    });\n  });\n});\n"]}

package/dist/web/data-llm.test.js

@@ -1,13 +1,16 @@
 import { jsx as _jsx, jsxs as _jsxs, Fragment as _Fragment } from "react/jsx-runtime";
 import { cleanup, render } from "@testing-library/react";
 import { afterEach, beforeEach, describe, expect, it, vi, } from "vitest";
-import { McpAppAdaptor, McpAppBridge } from "./bridges/mcp-app/index.js";
+import { HostAdaptor } from "./bridges/adaptor.js";
+import { getAdaptor } from "./bridges/get-adaptor.js";
+import { McpAppBridge } from "./bridges/mcp-app/index.js";
 import { DataLLM } from "./data-llm.js";
 import { getMcpAppHostPostMessageMock, MockResizeObserver, } from "./hooks/test/utils.js";
 describe("DataLLM", () => {
     afterEach(() => {
         // Clean up React components BEFORE unstubbing globals
         cleanup();
+        HostAdaptor.resetInstance();
         vi.unstubAllGlobals();
         vi.resetAllMocks();
     });
@@ -88,6 +91,7 @@ describe("DataLLM", () => {
         let postMessageMock;
         beforeEach(() => {
             vi.stubGlobal("skybridge", { hostType: "mcp-app" });
+            vi.stubGlobal("openai", undefined);
             vi.stubGlobal("ResizeObserver", MockResizeObserver);
             postMessageMock = getMcpAppHostPostMessageMock();
             vi.stubGlobal("parent", { postMessage: postMessageMock });
@@ -95,10 +99,10 @@ describe("DataLLM", () => {
         afterEach(() => {
             cleanup();
             McpAppBridge.resetInstance();
-            McpAppAdaptor.resetInstance();
+            HostAdaptor.resetInstance();
         });
         it("should register a node and update view state via adaptor", async () => {
-            const adaptor = McpAppAdaptor.getInstance();
+            const adaptor = getAdaptor();
             render(_jsx(DataLLM, { content: "Test content", children: _jsx("div", { children: "Child" }) }));
             await vi.waitFor(() => {
                 const viewState = adaptor
@@ -108,7 +112,7 @@ describe("DataLLM", () => {
             });
         });
         it("should preserve existing view state when updating context", async () => {
-            const adaptor = McpAppAdaptor.getInstance();
+            const adaptor = getAdaptor();
             await adaptor.setViewState({ existingKey: "existingValue" });
             render(_jsx(DataLLM, { content: "Test content", children: _jsx("div", { children: "Child" }) }));
             await vi.waitFor(() => {
@@ -120,7 +124,7 @@ describe("DataLLM", () => {
             });
         });
         it("should handle multiple DataLLM components sharing state through adaptor", async () => {
-            const adaptor = McpAppAdaptor.getInstance();
+            const adaptor = getAdaptor();
             render(_jsxs(_Fragment, { children: [_jsx(DataLLM, { content: "First component", children: _jsx("div", { children: "First" }) }), _jsx(DataLLM, { content: "Second component", children: _jsx("div", { children: "Second" }) })] }));
             await vi.waitFor(() => {
                 const viewState = adaptor

package/dist/web/data-llm.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"data-llm.test.js","sourceRoot":"","sources":["../../src/web/data-llm.test.tsx"],"names":[],"mappings":";AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EACL,SAAS,EACT,UAAU,EACV,QAAQ,EACR,MAAM,EACN,EAAE,EAEF,EAAE,GACH,MAAM,QAAQ,CAAC;AAChB,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AACzE,OAAO,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AACxC,OAAO,EACL,4BAA4B,EAC5B,kBAAkB,GACnB,MAAM,uBAAuB,CAAC;AAE/B,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;IACvB,SAAS,CAAC,GAAG,EAAE;QACb,sDAAsD;QACtD,OAAO,EAAE,CAAC;QACV,EAAE,CAAC,gBAAgB,EAAE,CAAC;QACtB,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,IAAI,UAA0D,CAAC;QAE/D,UAAU,CAAC,GAAG,EAAE;YACd,UAAU,GAAG;gBACX,WAAW,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE;gBACjC,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;aACxB,CAAC;YACF,kDAAkD;YAClD,MAAM,CAAC,cAAc,CAAC,UAAU,EAAE,QAAQ,EAAE;gBAC1C,KAAK,EAAE,UAAU;gBACjB,QAAQ,EAAE,IAAI;gBACd,YAAY,EAAE,IAAI;aACnB,CAAC,CAAC;YACH,kDAAkD;YAClD,IAAI,OAAO,MAAM,KAAK,WAAW,EAAE,CAAC;gBAClC,MAAM,CAAC,cAAc,CAAC,MAAM,EAAE,QAAQ,EAAE;oBACtC,KAAK,EAAE,UAAU;oBACjB,QAAQ,EAAE,IAAI;oBACd,YAAY,EAAE,IAAI;iBACnB,CAAC,CAAC;YACL,CAAC;YACD,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;YACpC,EAAE,CAAC,UAAU,CAAC,WAAW,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,SAAS,CAAC,GAAG,EAAE;YACb,0DAA0D;YAC1D,IAAI,OAAO,MAAM,KAAK,WAAW,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;gBACnD,MAAM,CAAC,MAAM,CAAC,cAAc,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;gBACvC,MAAM,CAAC,MAAM,CAAC,WAAW,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC;YACvE,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;YACnE,MAAM,CACJ,KAAC,OAAO,IAAC,OAAO,EAAC,cAAc,YAC7B,kCAAgB,GACR,CACX,CAAC;YAEF,MAAM,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC,gBAAgB,EAAE,CAAC;YACrD,MAAM,QAAQ,GACZ,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,CAAC;YAC7D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;YAClD,MAAM,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;YAClE,UAAU,CAAC,WAAW,GAAG;gBACvB,YAAY,EAAE,EAAE,WAAW,EAAE,eAAe,EAAE;aAC/C,CAAC;YAEF,MAAM,CACJ,KAAC,OAAO,IAAC,OAAO,EAAC,cAAc,YAC7B,kCAAgB,GACR,CACX,CAAC;YAEF,MAAM,QAAQ,GACZ,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,CAAC;YAC7D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,aAAa,EAAE,eAAe,CAAC,CAAC;YAChE,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,CACJ,MAAC,OAAO,IAAC,OAAO,EAAC,SAAS,aACxB,KAAC,OAAO,IAAC,OAAO,EAAC,UAAU,GAAG,EAC9B,KAAC,OAAO,IAAC,OAAO,EAAC,UAAU,YACzB,KAAC,OAAO,IAAC,OAAO,EAAC,SAAS,YACxB,oCAAkB,GACV,GACF,IACF,CACX,CAAC;YAEF,MAAM,QAAQ,GACZ,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAClC,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAChD,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,CAAC;YACvB,MAAM,OAAO,GAAG,QAAQ,EAAE,cAAwB,CAAC;YACnD,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;YACvC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;YAC1C,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;YAC1C,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,CACzB,KAAC,OAAO,IAAC,OAAO,EAAC,iBAAiB,YAChC,kCAAgB,GACR,CACX,CAAC;YAEF,MAAM,YAAY,GAAG,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;YAEjE,QAAQ,CACN,KAAC,OAAO,IAAC,OAAO,EAAC,iBAAiB,YAChC,kCAAgB,GACR,CACX,CAAC;YAEF,MAAM,CAAC,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,eAAe,CACjE,YAAY,CACb,CAAC;YACF,MAAM,YAAY,GAChB,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAClC,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAChD,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,CAAC;YACvB,MAAM,CAAC,YAAY,EAAE,cAAc,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;YACvE,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,CACxB,KAAC,OAAO,IAAC,OAAO,EAAC,mBAAmB,YAClC,kCAAgB,GACR,CACX,CAAC;YAEF,MAAM,kBAAkB,GAAG,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;YAEvE,OAAO,EAAE,CAAC;YAEV,MAAM,CAAC,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,eAAe,CACjE,kBAAkB,CACnB,CAAC;YACF,MAAM,YAAY,GAChB,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAClC,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAChD,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,CAAC;YACvB,MAAM,CAAC,YAAY,EAAE,cAAc,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QAC1E,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,IAAI,eAAgE,CAAC;QAErE,UAAU,CAAC,GAAG,EAAE;YACd,EAAE,CAAC,UAAU,CAAC,WAAW,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;YACpD,EAAE,CAAC,UAAU,CAAC,gBAAgB,EAAE,kBAAkB,CAAC,CAAC;YACpD,eAAe,GAAG,4BAA4B,EAAE,CAAC;YACjD,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,EAAE,WAAW,EAAE,eAAe,EAAE,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,SAAS,CAAC,GAAG,EAAE;YACb,OAAO,EAAE,CAAC;YACV,YAAY,CAAC,aAAa,EAAE,CAAC;YAC7B,aAAa,CAAC,aAAa,EAAE,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,MAAM,OAAO,GAAG,aAAa,CAAC,WAAW,EAAE,CAAC;YAE5C,MAAM,CACJ,KAAC,OAAO,IAAC,OAAO,EAAC,cAAc,YAC7B,kCAAgB,GACR,CACX,CAAC;YAEF,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;gBACpB,MAAM,SAAS,GAAG,OAAO;qBACtB,mBAAmB,CAAC,WAAW,CAAC;qBAChC,WAAW,EAAE,CAAC;gBACjB,MAAM,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAChE,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,OAAO,GAAG,aAAa,CAAC,WAAW,EAAE,CAAC;YAC5C,MAAM,OAAO,CAAC,YAAY,CAAC,EAAE,WAAW,EAAE,eAAe,EAAE,CAAC,CAAC;YAE7D,MAAM,CACJ,KAAC,OAAO,IAAC,OAAO,EAAC,cAAc,YAC7B,kCAAgB,GACR,CACX,CAAC;YAEF,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;gBACpB,MAAM,SAAS,GAAG,OAAO;qBACtB,mBAAmB,CAAC,WAAW,CAAC;qBAChC,WAAW,EAAE,CAAC;gBACjB,MAAM,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;gBACrD,MAAM,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAChE,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;YACvF,MAAM,OAAO,GAAG,aAAa,CAAC,WAAW,EAAE,CAAC;YAE5C,MAAM,CACJ,8BACE,KAAC,OAAO,IAAC,OAAO,EAAC,iBAAiB,YAChC,kCAAgB,GACR,EACV,KAAC,OAAO,IAAC,OAAO,EAAC,kBAAkB,YACjC,mCAAiB,GACT,IACT,CACJ,CAAC;YAEF,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;gBACpB,MAAM,SAAS,GAAG,OAAO;qBACtB,mBAAmB,CAAC,WAAW,CAAC;qBAChC,WAAW,EAAE,CAAC;gBACjB,MAAM,OAAO,GAAG,SAAS,EAAE,cAAwB,CAAC;gBACpD,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;gBAC/C,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;YAClD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;YAC3E,MAAM,CACJ,KAAC,OAAO,IAAC,OAAO,EAAC,cAAc,YAC7B,kCAAgB,GACR,CACX,CAAC;YAEF,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;gBACpB,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,MAAM,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,yBAAyB,EAAE,CAAC,EAC9D,GAAG,CACJ,CAAC;YACJ,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { cleanup, render } from \"@testing-library/react\";\nimport {\n  afterEach,\n  beforeEach,\n  describe,\n  expect,\n  it,\n  type Mock,\n  vi,\n} from \"vitest\";\nimport { McpAppAdaptor, McpAppBridge } from \"./bridges/mcp-app/index.js\";\nimport { DataLLM } from \"./data-llm.js\";\nimport {\n  getMcpAppHostPostMessageMock,\n  MockResizeObserver,\n} from \"./hooks/test/utils.js\";\n\ndescribe(\"DataLLM\", () => {\n  afterEach(() => {\n    // Clean up React components BEFORE unstubbing globals\n    cleanup();\n    vi.unstubAllGlobals();\n    vi.resetAllMocks();\n  });\n\n  describe(\"apps-sdk mode\", () => {\n    let OpenaiMock: { widgetState: unknown; setWidgetState: Mock };\n\n    beforeEach(() => {\n      OpenaiMock = {\n        widgetState: { modelContent: {} },\n        setWidgetState: vi.fn(),\n      };\n      // Use Object.defineProperty to ensure it persists\n      Object.defineProperty(globalThis, \"openai\", {\n        value: OpenaiMock,\n        writable: true,\n        configurable: true,\n      });\n      // Also set on window for browser-like environment\n      if (typeof window !== \"undefined\") {\n        Object.defineProperty(window, \"openai\", {\n          value: OpenaiMock,\n          writable: true,\n          configurable: true,\n        });\n      }\n      vi.stubGlobal(\"openai\", OpenaiMock);\n      vi.stubGlobal(\"skybridge\", { hostType: \"apps-sdk\" });\n    });\n\n    afterEach(() => {\n      // Keep the mock available for React cleanup, but reset it\n      if (typeof window !== \"undefined\" && window.openai) {\n        window.openai.setWidgetState = vi.fn();\n        window.openai.widgetState = { modelContent: {}, privateContent: {} };\n      }\n    });\n\n    it(\"should register a node with content and call setViewState\", () => {\n      render(\n        <DataLLM content=\"Test content\">\n          <div>Child</div>\n        </DataLLM>,\n      );\n\n      expect(OpenaiMock.setWidgetState).toHaveBeenCalled();\n      const callArgs =\n        OpenaiMock.setWidgetState.mock.calls[0]?.[0]?.modelContent;\n      expect(callArgs).toHaveProperty(\"__view_context\");\n      expect(callArgs?.__view_context).toContain(\"- Test content\");\n    });\n    it(\"should preserve existing viewState when updating context\", () => {\n      OpenaiMock.widgetState = {\n        modelContent: { existingKey: \"existingValue\" },\n      };\n\n      render(\n        <DataLLM content=\"Test content\">\n          <div>Child</div>\n        </DataLLM>,\n      );\n\n      const callArgs =\n        OpenaiMock.setWidgetState.mock.calls[0]?.[0]?.modelContent;\n      expect(callArgs).toHaveProperty(\"existingKey\", \"existingValue\");\n      expect(callArgs).toHaveProperty(\"__view_context\");\n    });\n\n    it(\"should handle deeply nested DataLLM components\", () => {\n      render(\n        <DataLLM content=\"Level 1\">\n          <DataLLM content=\"Level 2A\" />\n          <DataLLM content=\"Level 2B\">\n            <DataLLM content=\"Level 3\">\n              <div>Content</div>\n            </DataLLM>\n          </DataLLM>\n        </DataLLM>,\n      );\n\n      const callArgs =\n        OpenaiMock.setWidgetState.mock.calls[\n          OpenaiMock.setWidgetState.mock.calls.length - 1\n        ]?.[0]?.modelContent;\n      const context = callArgs?.__view_context as string;\n      expect(context).toContain(\"- Level 1\");\n      expect(context).toContain(\"  - Level 2A\");\n      expect(context).toContain(\"  - Level 2B\");\n      expect(context).toContain(\"    - Level 3\");\n    });\n\n    it(\"should update context when content changes\", () => {\n      const { rerender } = render(\n        <DataLLM content=\"Initial content\">\n          <div>Child</div>\n        </DataLLM>,\n      );\n\n      const initialCalls = OpenaiMock.setWidgetState.mock.calls.length;\n\n      rerender(\n        <DataLLM content=\"Updated content\">\n          <div>Child</div>\n        </DataLLM>,\n      );\n\n      expect(OpenaiMock.setWidgetState.mock.calls.length).toBeGreaterThan(\n        initialCalls,\n      );\n      const lastCallArgs =\n        OpenaiMock.setWidgetState.mock.calls[\n          OpenaiMock.setWidgetState.mock.calls.length - 1\n        ]?.[0]?.modelContent;\n      expect(lastCallArgs?.__view_context).toContain(\"- Updated content\");\n    });\n\n    it(\"should remove node and update context when component unmounts\", () => {\n      const { unmount } = render(\n        <DataLLM content=\"Content to remove\">\n          <div>Child</div>\n        </DataLLM>,\n      );\n\n      const callsBeforeUnmount = OpenaiMock.setWidgetState.mock.calls.length;\n\n      unmount();\n\n      expect(OpenaiMock.setWidgetState.mock.calls.length).toBeGreaterThan(\n        callsBeforeUnmount,\n      );\n      const lastCallArgs =\n        OpenaiMock.setWidgetState.mock.calls[\n          OpenaiMock.setWidgetState.mock.calls.length - 1\n        ]?.[0]?.modelContent;\n      expect(lastCallArgs?.__view_context).not.toContain(\"Content to remove\");\n    });\n  });\n\n  describe(\"mcp-app mode\", () => {\n    let postMessageMock: ReturnType<typeof getMcpAppHostPostMessageMock>;\n\n    beforeEach(() => {\n      vi.stubGlobal(\"skybridge\", { hostType: \"mcp-app\" });\n      vi.stubGlobal(\"ResizeObserver\", MockResizeObserver);\n      postMessageMock = getMcpAppHostPostMessageMock();\n      vi.stubGlobal(\"parent\", { postMessage: postMessageMock });\n    });\n\n    afterEach(() => {\n      cleanup();\n      McpAppBridge.resetInstance();\n      McpAppAdaptor.resetInstance();\n    });\n\n    it(\"should register a node and update view state via adaptor\", async () => {\n      const adaptor = McpAppAdaptor.getInstance();\n\n      render(\n        <DataLLM content=\"Test content\">\n          <div>Child</div>\n        </DataLLM>,\n      );\n\n      await vi.waitFor(() => {\n        const viewState = adaptor\n          .getHostContextStore(\"viewState\")\n          .getSnapshot();\n        expect(viewState?.__view_context).toContain(\"- Test content\");\n      });\n    });\n\n    it(\"should preserve existing view state when updating context\", async () => {\n      const adaptor = McpAppAdaptor.getInstance();\n      await adaptor.setViewState({ existingKey: \"existingValue\" });\n\n      render(\n        <DataLLM content=\"Test content\">\n          <div>Child</div>\n        </DataLLM>,\n      );\n\n      await vi.waitFor(() => {\n        const viewState = adaptor\n          .getHostContextStore(\"viewState\")\n          .getSnapshot();\n        expect(viewState?.existingKey).toBe(\"existingValue\");\n        expect(viewState?.__view_context).toContain(\"- Test content\");\n      });\n    });\n\n    it(\"should handle multiple DataLLM components sharing state through adaptor\", async () => {\n      const adaptor = McpAppAdaptor.getInstance();\n\n      render(\n        <>\n          <DataLLM content=\"First component\">\n            <div>First</div>\n          </DataLLM>\n          <DataLLM content=\"Second component\">\n            <div>Second</div>\n          </DataLLM>\n        </>,\n      );\n\n      await vi.waitFor(() => {\n        const viewState = adaptor\n          .getHostContextStore(\"viewState\")\n          .getSnapshot();\n        const context = viewState?.__view_context as string;\n        expect(context).toContain(\"- First component\");\n        expect(context).toContain(\"- Second component\");\n      });\n    });\n\n    it(\"should call ui/update-model-context when view state changes\", async () => {\n      render(\n        <DataLLM content=\"Test content\">\n          <div>Child</div>\n        </DataLLM>,\n      );\n\n      await vi.waitFor(() => {\n        expect(postMessageMock).toHaveBeenCalledWith(\n          expect.objectContaining({ method: \"ui/update-model-context\" }),\n          \"*\",\n        );\n      });\n    });\n  });\n});\n"]}
+{"version":3,"file":"data-llm.test.js","sourceRoot":"","sources":["../../src/web/data-llm.test.tsx"],"names":[],"mappings":";AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EACL,SAAS,EACT,UAAU,EACV,QAAQ,EACR,MAAM,EACN,EAAE,EAEF,EAAE,GACH,MAAM,QAAQ,CAAC;AAChB,OAAO,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AACnD,OAAO,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AACtD,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAC1D,OAAO,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AACxC,OAAO,EACL,4BAA4B,EAC5B,kBAAkB,GACnB,MAAM,uBAAuB,CAAC;AAE/B,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;IACvB,SAAS,CAAC,GAAG,EAAE;QACb,sDAAsD;QACtD,OAAO,EAAE,CAAC;QACV,WAAW,CAAC,aAAa,EAAE,CAAC;QAC5B,EAAE,CAAC,gBAAgB,EAAE,CAAC;QACtB,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,IAAI,UAA0D,CAAC;QAE/D,UAAU,CAAC,GAAG,EAAE;YACd,UAAU,GAAG;gBACX,WAAW,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE;gBACjC,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;aACxB,CAAC;YACF,kDAAkD;YAClD,MAAM,CAAC,cAAc,CAAC,UAAU,EAAE,QAAQ,EAAE;gBAC1C,KAAK,EAAE,UAAU;gBACjB,QAAQ,EAAE,IAAI;gBACd,YAAY,EAAE,IAAI;aACnB,CAAC,CAAC;YACH,kDAAkD;YAClD,IAAI,OAAO,MAAM,KAAK,WAAW,EAAE,CAAC;gBAClC,MAAM,CAAC,cAAc,CAAC,MAAM,EAAE,QAAQ,EAAE;oBACtC,KAAK,EAAE,UAAU;oBACjB,QAAQ,EAAE,IAAI;oBACd,YAAY,EAAE,IAAI;iBACnB,CAAC,CAAC;YACL,CAAC;YACD,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;YACpC,EAAE,CAAC,UAAU,CAAC,WAAW,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,SAAS,CAAC,GAAG,EAAE;YACb,0DAA0D;YAC1D,IAAI,OAAO,MAAM,KAAK,WAAW,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;gBACnD,MAAM,CAAC,MAAM,CAAC,cAAc,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;gBACvC,MAAM,CAAC,MAAM,CAAC,WAAW,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC;YACvE,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;YACnE,MAAM,CACJ,KAAC,OAAO,IAAC,OAAO,EAAC,cAAc,YAC7B,kCAAgB,GACR,CACX,CAAC;YAEF,MAAM,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC,gBAAgB,EAAE,CAAC;YACrD,MAAM,QAAQ,GACZ,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,CAAC;YAC7D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;YAClD,MAAM,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;YAClE,UAAU,CAAC,WAAW,GAAG;gBACvB,YAAY,EAAE,EAAE,WAAW,EAAE,eAAe,EAAE;aAC/C,CAAC;YAEF,MAAM,CACJ,KAAC,OAAO,IAAC,OAAO,EAAC,cAAc,YAC7B,kCAAgB,GACR,CACX,CAAC;YAEF,MAAM,QAAQ,GACZ,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,CAAC;YAC7D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,aAAa,EAAE,eAAe,CAAC,CAAC;YAChE,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,CACJ,MAAC,OAAO,IAAC,OAAO,EAAC,SAAS,aACxB,KAAC,OAAO,IAAC,OAAO,EAAC,UAAU,GAAG,EAC9B,KAAC,OAAO,IAAC,OAAO,EAAC,UAAU,YACzB,KAAC,OAAO,IAAC,OAAO,EAAC,SAAS,YACxB,oCAAkB,GACV,GACF,IACF,CACX,CAAC;YAEF,MAAM,QAAQ,GACZ,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAClC,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAChD,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,CAAC;YACvB,MAAM,OAAO,GAAG,QAAQ,EAAE,cAAwB,CAAC;YACnD,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;YACvC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;YAC1C,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;YAC1C,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,CACzB,KAAC,OAAO,IAAC,OAAO,EAAC,iBAAiB,YAChC,kCAAgB,GACR,CACX,CAAC;YAEF,MAAM,YAAY,GAAG,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;YAEjE,QAAQ,CACN,KAAC,OAAO,IAAC,OAAO,EAAC,iBAAiB,YAChC,kCAAgB,GACR,CACX,CAAC;YAEF,MAAM,CAAC,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,eAAe,CACjE,YAAY,CACb,CAAC;YACF,MAAM,YAAY,GAChB,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAClC,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAChD,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,CAAC;YACvB,MAAM,CAAC,YAAY,EAAE,cAAc,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;YACvE,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,CACxB,KAAC,OAAO,IAAC,OAAO,EAAC,mBAAmB,YAClC,kCAAgB,GACR,CACX,CAAC;YAEF,MAAM,kBAAkB,GAAG,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;YAEvE,OAAO,EAAE,CAAC;YAEV,MAAM,CAAC,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,eAAe,CACjE,kBAAkB,CACnB,CAAC;YACF,MAAM,YAAY,GAChB,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAClC,UAAU,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAChD,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,CAAC;YACvB,MAAM,CAAC,YAAY,EAAE,cAAc,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QAC1E,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,IAAI,eAAgE,CAAC;QAErE,UAAU,CAAC,GAAG,EAAE;YACd,EAAE,CAAC,UAAU,CAAC,WAAW,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;YACpD,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;YACnC,EAAE,CAAC,UAAU,CAAC,gBAAgB,EAAE,kBAAkB,CAAC,CAAC;YACpD,eAAe,GAAG,4BAA4B,EAAE,CAAC;YACjD,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,EAAE,WAAW,EAAE,eAAe,EAAE,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,SAAS,CAAC,GAAG,EAAE;YACb,OAAO,EAAE,CAAC;YACV,YAAY,CAAC,aAAa,EAAE,CAAC;YAC7B,WAAW,CAAC,aAAa,EAAE,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,MAAM,OAAO,GAAG,UAAU,EAAE,CAAC;YAE7B,MAAM,CACJ,KAAC,OAAO,IAAC,OAAO,EAAC,cAAc,YAC7B,kCAAgB,GACR,CACX,CAAC;YAEF,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;gBACpB,MAAM,SAAS,GAAG,OAAO;qBACtB,mBAAmB,CAAC,WAAW,CAAC;qBAChC,WAAW,EAAE,CAAC;gBACjB,MAAM,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAChE,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,OAAO,GAAG,UAAU,EAAE,CAAC;YAC7B,MAAM,OAAO,CAAC,YAAY,CAAC,EAAE,WAAW,EAAE,eAAe,EAAE,CAAC,CAAC;YAE7D,MAAM,CACJ,KAAC,OAAO,IAAC,OAAO,EAAC,cAAc,YAC7B,kCAAgB,GACR,CACX,CAAC;YAEF,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;gBACpB,MAAM,SAAS,GAAG,OAAO;qBACtB,mBAAmB,CAAC,WAAW,CAAC;qBAChC,WAAW,EAAE,CAAC;gBACjB,MAAM,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;gBACrD,MAAM,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAChE,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;YACvF,MAAM,OAAO,GAAG,UAAU,EAAE,CAAC;YAE7B,MAAM,CACJ,8BACE,KAAC,OAAO,IAAC,OAAO,EAAC,iBAAiB,YAChC,kCAAgB,GACR,EACV,KAAC,OAAO,IAAC,OAAO,EAAC,kBAAkB,YACjC,mCAAiB,GACT,IACT,CACJ,CAAC;YAEF,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;gBACpB,MAAM,SAAS,GAAG,OAAO;qBACtB,mBAAmB,CAAC,WAAW,CAAC;qBAChC,WAAW,EAAE,CAAC;gBACjB,MAAM,OAAO,GAAG,SAAS,EAAE,cAAwB,CAAC;gBACpD,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;gBAC/C,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;YAClD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;YAC3E,MAAM,CACJ,KAAC,OAAO,IAAC,OAAO,EAAC,cAAc,YAC7B,kCAAgB,GACR,CACX,CAAC;YAEF,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;gBACpB,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,MAAM,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,yBAAyB,EAAE,CAAC,EAC9D,GAAG,CACJ,CAAC;YACJ,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { cleanup, render } from \"@testing-library/react\";\nimport {\n  afterEach,\n  beforeEach,\n  describe,\n  expect,\n  it,\n  type Mock,\n  vi,\n} from \"vitest\";\nimport { HostAdaptor } from \"./bridges/adaptor.js\";\nimport { getAdaptor } from \"./bridges/get-adaptor.js\";\nimport { McpAppBridge } from \"./bridges/mcp-app/index.js\";\nimport { DataLLM } from \"./data-llm.js\";\nimport {\n  getMcpAppHostPostMessageMock,\n  MockResizeObserver,\n} from \"./hooks/test/utils.js\";\n\ndescribe(\"DataLLM\", () => {\n  afterEach(() => {\n    // Clean up React components BEFORE unstubbing globals\n    cleanup();\n    HostAdaptor.resetInstance();\n    vi.unstubAllGlobals();\n    vi.resetAllMocks();\n  });\n\n  describe(\"apps-sdk mode\", () => {\n    let OpenaiMock: { widgetState: unknown; setWidgetState: Mock };\n\n    beforeEach(() => {\n      OpenaiMock = {\n        widgetState: { modelContent: {} },\n        setWidgetState: vi.fn(),\n      };\n      // Use Object.defineProperty to ensure it persists\n      Object.defineProperty(globalThis, \"openai\", {\n        value: OpenaiMock,\n        writable: true,\n        configurable: true,\n      });\n      // Also set on window for browser-like environment\n      if (typeof window !== \"undefined\") {\n        Object.defineProperty(window, \"openai\", {\n          value: OpenaiMock,\n          writable: true,\n          configurable: true,\n        });\n      }\n      vi.stubGlobal(\"openai\", OpenaiMock);\n      vi.stubGlobal(\"skybridge\", { hostType: \"apps-sdk\" });\n    });\n\n    afterEach(() => {\n      // Keep the mock available for React cleanup, but reset it\n      if (typeof window !== \"undefined\" && window.openai) {\n        window.openai.setWidgetState = vi.fn();\n        window.openai.widgetState = { modelContent: {}, privateContent: {} };\n      }\n    });\n\n    it(\"should register a node with content and call setViewState\", () => {\n      render(\n        <DataLLM content=\"Test content\">\n          <div>Child</div>\n        </DataLLM>,\n      );\n\n      expect(OpenaiMock.setWidgetState).toHaveBeenCalled();\n      const callArgs =\n        OpenaiMock.setWidgetState.mock.calls[0]?.[0]?.modelContent;\n      expect(callArgs).toHaveProperty(\"__view_context\");\n      expect(callArgs?.__view_context).toContain(\"- Test content\");\n    });\n    it(\"should preserve existing viewState when updating context\", () => {\n      OpenaiMock.widgetState = {\n        modelContent: { existingKey: \"existingValue\" },\n      };\n\n      render(\n        <DataLLM content=\"Test content\">\n          <div>Child</div>\n        </DataLLM>,\n      );\n\n      const callArgs =\n        OpenaiMock.setWidgetState.mock.calls[0]?.[0]?.modelContent;\n      expect(callArgs).toHaveProperty(\"existingKey\", \"existingValue\");\n      expect(callArgs).toHaveProperty(\"__view_context\");\n    });\n\n    it(\"should handle deeply nested DataLLM components\", () => {\n      render(\n        <DataLLM content=\"Level 1\">\n          <DataLLM content=\"Level 2A\" />\n          <DataLLM content=\"Level 2B\">\n            <DataLLM content=\"Level 3\">\n              <div>Content</div>\n            </DataLLM>\n          </DataLLM>\n        </DataLLM>,\n      );\n\n      const callArgs =\n        OpenaiMock.setWidgetState.mock.calls[\n          OpenaiMock.setWidgetState.mock.calls.length - 1\n        ]?.[0]?.modelContent;\n      const context = callArgs?.__view_context as string;\n      expect(context).toContain(\"- Level 1\");\n      expect(context).toContain(\"  - Level 2A\");\n      expect(context).toContain(\"  - Level 2B\");\n      expect(context).toContain(\"    - Level 3\");\n    });\n\n    it(\"should update context when content changes\", () => {\n      const { rerender } = render(\n        <DataLLM content=\"Initial content\">\n          <div>Child</div>\n        </DataLLM>,\n      );\n\n      const initialCalls = OpenaiMock.setWidgetState.mock.calls.length;\n\n      rerender(\n        <DataLLM content=\"Updated content\">\n          <div>Child</div>\n        </DataLLM>,\n      );\n\n      expect(OpenaiMock.setWidgetState.mock.calls.length).toBeGreaterThan(\n        initialCalls,\n      );\n      const lastCallArgs =\n        OpenaiMock.setWidgetState.mock.calls[\n          OpenaiMock.setWidgetState.mock.calls.length - 1\n        ]?.[0]?.modelContent;\n      expect(lastCallArgs?.__view_context).toContain(\"- Updated content\");\n    });\n\n    it(\"should remove node and update context when component unmounts\", () => {\n      const { unmount } = render(\n        <DataLLM content=\"Content to remove\">\n          <div>Child</div>\n        </DataLLM>,\n      );\n\n      const callsBeforeUnmount = OpenaiMock.setWidgetState.mock.calls.length;\n\n      unmount();\n\n      expect(OpenaiMock.setWidgetState.mock.calls.length).toBeGreaterThan(\n        callsBeforeUnmount,\n      );\n      const lastCallArgs =\n        OpenaiMock.setWidgetState.mock.calls[\n          OpenaiMock.setWidgetState.mock.calls.length - 1\n        ]?.[0]?.modelContent;\n      expect(lastCallArgs?.__view_context).not.toContain(\"Content to remove\");\n    });\n  });\n\n  describe(\"mcp-app mode\", () => {\n    let postMessageMock: ReturnType<typeof getMcpAppHostPostMessageMock>;\n\n    beforeEach(() => {\n      vi.stubGlobal(\"skybridge\", { hostType: \"mcp-app\" });\n      vi.stubGlobal(\"openai\", undefined);\n      vi.stubGlobal(\"ResizeObserver\", MockResizeObserver);\n      postMessageMock = getMcpAppHostPostMessageMock();\n      vi.stubGlobal(\"parent\", { postMessage: postMessageMock });\n    });\n\n    afterEach(() => {\n      cleanup();\n      McpAppBridge.resetInstance();\n      HostAdaptor.resetInstance();\n    });\n\n    it(\"should register a node and update view state via adaptor\", async () => {\n      const adaptor = getAdaptor();\n\n      render(\n        <DataLLM content=\"Test content\">\n          <div>Child</div>\n        </DataLLM>,\n      );\n\n      await vi.waitFor(() => {\n        const viewState = adaptor\n          .getHostContextStore(\"viewState\")\n          .getSnapshot();\n        expect(viewState?.__view_context).toContain(\"- Test content\");\n      });\n    });\n\n    it(\"should preserve existing view state when updating context\", async () => {\n      const adaptor = getAdaptor();\n      await adaptor.setViewState({ existingKey: \"existingValue\" });\n\n      render(\n        <DataLLM content=\"Test content\">\n          <div>Child</div>\n        </DataLLM>,\n      );\n\n      await vi.waitFor(() => {\n        const viewState = adaptor\n          .getHostContextStore(\"viewState\")\n          .getSnapshot();\n        expect(viewState?.existingKey).toBe(\"existingValue\");\n        expect(viewState?.__view_context).toContain(\"- Test content\");\n      });\n    });\n\n    it(\"should handle multiple DataLLM components sharing state through adaptor\", async () => {\n      const adaptor = getAdaptor();\n\n      render(\n        <>\n          <DataLLM content=\"First component\">\n            <div>First</div>\n          </DataLLM>\n          <DataLLM content=\"Second component\">\n            <div>Second</div>\n          </DataLLM>\n        </>,\n      );\n\n      await vi.waitFor(() => {\n        const viewState = adaptor\n          .getHostContextStore(\"viewState\")\n          .getSnapshot();\n        const context = viewState?.__view_context as string;\n        expect(context).toContain(\"- First component\");\n        expect(context).toContain(\"- Second component\");\n      });\n    });\n\n    it(\"should call ui/update-model-context when view state changes\", async () => {\n      render(\n        <DataLLM content=\"Test content\">\n          <div>Child</div>\n        </DataLLM>,\n      );\n\n      await vi.waitFor(() => {\n        expect(postMessageMock).toHaveBeenCalledWith(\n          expect.objectContaining({ method: \"ui/update-model-context\" }),\n          \"*\",\n        );\n      });\n    });\n  });\n});\n"]}

package/dist/web/hooks/use-call-tool.test.js

@@ -1,18 +1,27 @@
 import { act, renderHook, waitFor } from "@testing-library/react";
 import { afterEach, beforeEach, describe, expect, expectTypeOf, it, vi, } from "vitest";
+import { HostAdaptor } from "../bridges/adaptor.js";
+import { AppsSdkBridge } from "../bridges/apps-sdk/bridge.js";
+import { getAdaptor } from "../bridges/get-adaptor.js";
+import { McpAppBridge } from "../bridges/mcp-app/bridge.js";
 import { useCallTool } from "./use-call-tool.js";
 describe("useCallTool - onSuccess callback", () => {
-    let OpenaiMock;
+    let callToolMock;
     beforeEach(() => {
-        OpenaiMock = {
-            callTool: vi.fn(),
-        };
-        vi.stubGlobal("openai", OpenaiMock);
+        HostAdaptor.resetInstance();
+        McpAppBridge.resetInstance();
+        AppsSdkBridge.resetInstance();
+        vi.stubGlobal("parent", { postMessage: vi.fn() });
         vi.stubGlobal("skybridge", { hostType: "apps-sdk" });
+        vi.stubGlobal("openai", { callTool: vi.fn() });
+        callToolMock = vi.spyOn(getAdaptor(), "callTool");
     });
     afterEach(() => {
         vi.unstubAllGlobals();
         vi.resetAllMocks();
+        HostAdaptor.resetInstance();
+        McpAppBridge.resetInstance();
+        AppsSdkBridge.resetInstance();
     });
     const toolName = "test-tool";
     const args = { input: "test input" };
@@ -28,9 +37,9 @@ describe("useCallTool - onSuccess callback", () => {
             content: [{ type: "text", text: "result" }],
             structuredContent: { value: 1 },
             isError: false,
-            _meta: { secret: "only visible to widget" },
+            meta: { secret: "only visible to widget" },
         };
-        OpenaiMock.callTool.mockResolvedValueOnce(rawSdkResponse);
+        callToolMock.mockResolvedValueOnce(rawSdkResponse);
         const { result } = renderHook(() => useCallTool(toolName));
         await act(async () => {
             result.current.callTool(args);
@@ -41,17 +50,17 @@ describe("useCallTool - onSuccess callback", () => {
             });
         });
     });
-    it("should call window.openai.callTool with correct arguments", async () => {
+    it("should call adaptor.callTool with correct arguments", async () => {
         const { result } = renderHook(() => useCallTool(toolName));
         await act(async () => {
             result.current.callTool(args);
         });
-        expect(OpenaiMock.callTool).toHaveBeenCalledWith(toolName, args);
+        expect(callToolMock).toHaveBeenCalledWith(toolName, args);
     });
     it("should call onSuccess callback with correct data and toolArgs on successful execution", async () => {
         const onSuccess = vi.fn();
         const onError = vi.fn();
-        OpenaiMock.callTool.mockResolvedValueOnce(data);
+        callToolMock.mockResolvedValueOnce(data);
         const { result } = renderHook(() => useCallTool(toolName));
         act(() => {
             result.current.callTool(args, {
@@ -67,7 +76,7 @@ describe("useCallTool - onSuccess callback", () => {
     it("should call onError callback with error and toolArgs on failed execution", async () => {
         const onSuccess = vi.fn();
         const onError = vi.fn();
-        OpenaiMock.callTool.mockRejectedValueOnce(error);
+        callToolMock.mockRejectedValueOnce(error);
         const { result } = renderHook(() => useCallTool(toolName));
         act(() => {
             result.current.callTool(args, {
@@ -84,7 +93,7 @@ describe("useCallTool - onSuccess callback", () => {
         const onSuccess = vi.fn();
         const onError = vi.fn();
         const onSettled = vi.fn();
-        OpenaiMock.callTool.mockResolvedValueOnce(data);
+        callToolMock.mockResolvedValueOnce(data);
         const { result } = renderHook(() => useCallTool(toolName));
         act(() => {
             result.current.callTool(args, {
@@ -103,7 +112,7 @@ describe("useCallTool - onSuccess callback", () => {
         const onSuccess = vi.fn();
         const onError = vi.fn();
         const onSettled = vi.fn();
-        OpenaiMock.callTool.mockRejectedValueOnce(error);
+        callToolMock.mockRejectedValueOnce(error);
         const { result } = renderHook(() => useCallTool(toolName));
         act(() => {
             result.current.callTool(args, {
@@ -130,7 +139,7 @@ describe("useCallTool - onSuccess callback", () => {
         };
         const { promise: firstCallToolPromise, resolve: resolveFirstCallTool } = Promise.withResolvers();
         const { promise: secondCallToolPromise, resolve: resolveSecondCallTool } = Promise.withResolvers();
-        OpenaiMock.callTool
+        callToolMock
             .mockImplementationOnce(() => firstCallToolPromise)
             .mockImplementationOnce(() => secondCallToolPromise);
         await act(() => {
@@ -149,17 +158,22 @@ describe("useCallTool - onSuccess callback", () => {
     });
 });
 describe("useCallTool - TypeScript typing", () => {
-    let OpenaiMock;
+    let callToolMock;
     beforeEach(() => {
-        OpenaiMock = {
-            callTool: vi.fn(),
-        };
-        vi.stubGlobal("openai", OpenaiMock);
+        HostAdaptor.resetInstance();
+        McpAppBridge.resetInstance();
+        AppsSdkBridge.resetInstance();
+        vi.stubGlobal("parent", { postMessage: vi.fn() });
         vi.stubGlobal("skybridge", { hostType: "apps-sdk" });
+        vi.stubGlobal("openai", { callTool: vi.fn() });
+        callToolMock = vi.spyOn(getAdaptor(), "callTool");
     });
     afterEach(() => {
         vi.unstubAllGlobals();
         vi.resetAllMocks();
+        HostAdaptor.resetInstance();
+        McpAppBridge.resetInstance();
+        AppsSdkBridge.resetInstance();
     });
     it("should have correct return types when ToolArgs is null and ToolResponse is specified", async () => {
         const { result } = renderHook(() => useCallTool("test-tool"));
@@ -169,11 +183,11 @@ describe("useCallTool - TypeScript typing", () => {
             isError: false,
             meta: { id: 123 },
         };
-        OpenaiMock.callTool.mockResolvedValueOnce(data);
+        callToolMock.mockResolvedValueOnce(data);
         await act(async () => {
             result.current.callTool();
         });
-        expect(OpenaiMock.callTool).toHaveBeenCalledWith("test-tool", null);
+        expect(callToolMock).toHaveBeenCalledWith("test-tool", null);
         expectTypeOf(result.current.data);
     });
     it("should correctly type callToolAsync return value", async () => {
@@ -185,7 +199,7 @@ describe("useCallTool - TypeScript typing", () => {
             isError: false,
             meta: {},
         };
-        OpenaiMock.callTool.mockResolvedValueOnce(mockResponse);
+        callToolMock.mockResolvedValueOnce(mockResponse);
         const returnedValue = await act(async () => {
             return result.current.callToolAsync(testArgs);
         });
@@ -200,7 +214,7 @@ describe("useCallTool - TypeScript typing", () => {
             isError: false,
             meta: {},
         };
-        OpenaiMock.callTool.mockResolvedValueOnce(mockResponse);
+        callToolMock.mockResolvedValueOnce(mockResponse);
         const returnedValue = await act(async () => {
             return result.current.callToolAsync();
         });

package/dist/web/hooks/use-call-tool.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"use-call-tool.test.js","sourceRoot":"","sources":["../../../src/web/hooks/use-call-tool.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,UAAU,EAAE,OAAO,EAAE,MAAM,wBAAwB,CAAC;AAClE,OAAO,EACL,SAAS,EACT,UAAU,EACV,QAAQ,EACR,MAAM,EACN,YAAY,EACZ,EAAE,EAEF,EAAE,GACH,MAAM,QAAQ,CAAC;AAGhB,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAEjD,QAAQ,CAAC,kCAAkC,EAAE,GAAG,EAAE;IAChD,IAAI,UAA8B,CAAC;IAEnC,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,GAAG;YACX,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;SAClB,CAAC;QACF,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QACpC,EAAE,CAAC,UAAU,CAAC,WAAW,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,gBAAgB,EAAE,CAAC;QACtB,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,MAAM,QAAQ,GAAG,WAAW,CAAC;IAC7B,MAAM,IAAI,GAAG,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC;IACrC,MAAM,IAAI,GAAqB;QAC7B,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;QACzD,iBAAiB,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE;QACrC,OAAO,EAAE,KAAK;QACd,IAAI,EAAE,EAAE;KACT,CAAC;IACF,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;IAEtC,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,cAAc,GAAG;YACrB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;YACpD,iBAAiB,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE;YAC/B,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,EAAE,MAAM,EAAE,wBAAwB,EAAE;SAC5C,CAAC;QACF,UAAU,CAAC,QAAQ,CAAC,qBAAqB,CAAC,cAAc,CAAC,CAAC;QAC1D,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,WAAW,CAA2B,QAAQ,CAAC,CAChD,CAAC;QAEF,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,aAAa,CAAC;gBACxC,IAAI,EAAE,EAAE,MAAM,EAAE,wBAAwB,EAAE;aAC3C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,WAAW,CAA2B,QAAQ,CAAC,CAChD,CAAC;QACF,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uFAAuF,EAAE,KAAK,IAAI,EAAE;QACrG,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC1B,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACxB,UAAU,CAAC,QAAQ,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC;QAChD,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,WAAW,CAA2B,QAAQ,CAAC,CAChD,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE;gBAC5B,SAAS;gBACT,OAAO;aACR,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YACnD,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;QACxF,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC1B,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACxB,UAAU,CAAC,QAAQ,CAAC,qBAAqB,CAAC,KAAK,CAAC,CAAC;QACjD,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,WAAW,CAA2B,QAAQ,CAAC,CAChD,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE;gBAC5B,SAAS;gBACT,OAAO;aACR,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YACzC,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sFAAsF,EAAE,KAAK,IAAI,EAAE;QACpG,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC1B,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACxB,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC1B,UAAU,CAAC,QAAQ,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC;QAChD,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,WAAW,CAA2B,QAAQ,CAAC,CAChD,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE;gBAC5B,SAAS;gBACT,OAAO;gBACP,SAAS;aACV,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YACnD,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC;YAC9D,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kFAAkF,EAAE,KAAK,IAAI,EAAE;QAChG,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC1B,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACxB,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC1B,UAAU,CAAC,QAAQ,CAAC,qBAAqB,CAAC,KAAK,CAAC,CAAC;QACjD,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,WAAW,CAA2B,QAAQ,CAAC,CAChD,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE;gBAC5B,SAAS;gBACT,OAAO;gBACP,SAAS;aACV,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;YAClD,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,SAAS,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;YAC/D,MAAM,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC3C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,WAAW,CAA2B,QAAQ,CAAC,CAChD,CAAC;QAEF,MAAM,aAAa,GAAG;YACpB,GAAG,IAAI;YACP,iBAAiB,EAAE,EAAE,MAAM,EAAE,mBAAmB,EAAE;SACnD,CAAC;QACF,MAAM,cAAc,GAAG;YACrB,GAAG,IAAI;YACP,iBAAiB,EAAE,EAAE,MAAM,EAAE,oBAAoB,EAAE;SACpD,CAAC;QACF,MAAM,EAAE,OAAO,EAAE,oBAAoB,EAAE,OAAO,EAAE,oBAAoB,EAAE,GACpE,OAAO,CAAC,aAAa,EAAE,CAAC;QAC1B,MAAM,EAAE,OAAO,EAAE,qBAAqB,EAAE,OAAO,EAAE,qBAAqB,EAAE,GACtE,OAAO,CAAC,aAAa,EAAE,CAAC;QAC1B,UAAU,CAAC,QAAQ;aAChB,sBAAsB,CAAC,GAAG,EAAE,CAAC,oBAAoB,CAAC;aAClD,sBAAsB,CAAC,GAAG,EAAE,CAAC,qBAAqB,CAAC,CAAC;QAEvD,MAAM,GAAG,CAAC,GAAG,EAAE;YACb,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YAC9B,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YAC9B,oBAAoB,CAAC,aAAa,CAAC,CAAC;YACpC,OAAO,oBAAoB,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACjD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAC/C,qBAAqB,CAAC,cAAc,CAAC,CAAC;QAEtC,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YACjD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;IAC/C,IAAI,UAA8B,CAAC;IAEnC,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,GAAG;YACX,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;SAClB,CAAC;QACF,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QACpC,EAAE,CAAC,UAAU,CAAC,WAAW,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,gBAAgB,EAAE,CAAC;QACtB,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sFAAsF,EAAE,KAAK,IAAI,EAAE;QAMpG,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,WAAW,CAAqB,WAAW,CAAC,CAC7C,CAAC;QACF,MAAM,IAAI,GAAiB;YACzB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;YAClD,iBAAiB,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE;YACrC,OAAO,EAAE,KAAK;YACd,IAAI,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE;SAClB,CAAC;QAEF,UAAU,CAAC,QAAQ,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC;QAEhD,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;QACpE,YAAY,CAA0B,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAMhE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,WAAW,CAAyB,WAAW,CAAC,CACjD,CAAC;QAEF,MAAM,QAAQ,GAAa,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;QAC7C,MAAM,YAAY,GAAiB;YACjC,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;YACpD,iBAAiB,EAAE,EAAE,MAAM,EAAE,aAAa,EAAE;YAC5C,OAAO,EAAE,KAAK;YACd,IAAI,EAAE,EAAE;SACT,CAAC;QAEF,UAAU,CAAC,QAAQ,CAAC,qBAAqB,CAAC,YAAY,CAAC,CAAC;QAExD,MAAM,aAAa,GAAG,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACzC,OAAO,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QACH,YAAY,CAAsB,aAAa,CAAC,CAAC;QACjD,MAAM,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QAKzE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,WAAW,CAAqB,WAAW,CAAC,CAC7C,CAAC;QAEF,MAAM,YAAY,GAAiB;YACjC,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;YAClD,iBAAiB,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE;YACxC,OAAO,EAAE,KAAK;YACd,IAAI,EAAE,EAAE;SACT,CAAC;QAEF,UAAU,CAAC,QAAQ,CAAC,qBAAqB,CAAC,YAAY,CAAC,CAAC;QAExD,MAAM,aAAa,GAAG,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACzC,OAAO,MAAM,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;QACxC,CAAC,CAAC,CAAC;QACH,YAAY,CAAsB,aAAa,CAAC,CAAC;QACjD,MAAM,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { act, renderHook, waitFor } from \"@testing-library/react\";\nimport {\n  afterEach,\n  beforeEach,\n  describe,\n  expect,\n  expectTypeOf,\n  it,\n  type Mock,\n  vi,\n} from \"vitest\";\n\nimport type { CallToolResponse } from \"../bridges/types.js\";\nimport { useCallTool } from \"./use-call-tool.js\";\n\ndescribe(\"useCallTool - onSuccess callback\", () => {\n  let OpenaiMock: { callTool: Mock };\n\n  beforeEach(() => {\n    OpenaiMock = {\n      callTool: vi.fn(),\n    };\n    vi.stubGlobal(\"openai\", OpenaiMock);\n    vi.stubGlobal(\"skybridge\", { hostType: \"apps-sdk\" });\n  });\n\n  afterEach(() => {\n    vi.unstubAllGlobals();\n    vi.resetAllMocks();\n  });\n\n  const toolName = \"test-tool\";\n  const args = { input: \"test input\" };\n  const data: CallToolResponse = {\n    content: [{ type: \"text\" as const, text: \"test result\" }],\n    structuredContent: { result: \"test\" },\n    isError: false,\n    meta: {},\n  };\n  const error = new Error(\"test error\");\n\n  it(\"should normalize _meta to meta when SDK returns _meta instead of meta\", async () => {\n    const rawSdkResponse = {\n      content: [{ type: \"text\" as const, text: \"result\" }],\n      structuredContent: { value: 1 },\n      isError: false,\n      _meta: { secret: \"only visible to widget\" },\n    };\n    OpenaiMock.callTool.mockResolvedValueOnce(rawSdkResponse);\n    const { result } = renderHook(() =>\n      useCallTool<typeof args, typeof data>(toolName),\n    );\n\n    await act(async () => {\n      result.current.callTool(args);\n    });\n\n    await waitFor(() => {\n      expect(result.current.data).toMatchObject({\n        meta: { secret: \"only visible to widget\" },\n      });\n    });\n  });\n\n  it(\"should call window.openai.callTool with correct arguments\", async () => {\n    const { result } = renderHook(() =>\n      useCallTool<typeof args, typeof data>(toolName),\n    );\n    await act(async () => {\n      result.current.callTool(args);\n    });\n    expect(OpenaiMock.callTool).toHaveBeenCalledWith(toolName, args);\n  });\n\n  it(\"should call onSuccess callback with correct data and toolArgs on successful execution\", async () => {\n    const onSuccess = vi.fn();\n    const onError = vi.fn();\n    OpenaiMock.callTool.mockResolvedValueOnce(data);\n    const { result } = renderHook(() =>\n      useCallTool<typeof args, typeof data>(toolName),\n    );\n\n    act(() => {\n      result.current.callTool(args, {\n        onSuccess,\n        onError,\n      });\n    });\n\n    await waitFor(() => {\n      expect(onSuccess).toHaveBeenCalledWith(data, args);\n      expect(onError).not.toHaveBeenCalled();\n    });\n  });\n\n  it(\"should call onError callback with error and toolArgs on failed execution\", async () => {\n    const onSuccess = vi.fn();\n    const onError = vi.fn();\n    OpenaiMock.callTool.mockRejectedValueOnce(error);\n    const { result } = renderHook(() =>\n      useCallTool<typeof args, typeof data>(toolName),\n    );\n\n    act(() => {\n      result.current.callTool(args, {\n        onSuccess,\n        onError,\n      });\n    });\n\n    await waitFor(() => {\n      expect(onSuccess).not.toHaveBeenCalled();\n      expect(onError).toHaveBeenCalledWith(error, args);\n    });\n  });\n\n  it(\"should call onSettled callback with data and undefined error on successful execution\", async () => {\n    const onSuccess = vi.fn();\n    const onError = vi.fn();\n    const onSettled = vi.fn();\n    OpenaiMock.callTool.mockResolvedValueOnce(data);\n    const { result } = renderHook(() =>\n      useCallTool<typeof args, typeof data>(toolName),\n    );\n\n    act(() => {\n      result.current.callTool(args, {\n        onSuccess,\n        onError,\n        onSettled,\n      });\n    });\n\n    await waitFor(() => {\n      expect(onSuccess).toHaveBeenCalledWith(data, args);\n      expect(onSettled).toHaveBeenCalledWith(data, undefined, args);\n      expect(onError).not.toHaveBeenCalled();\n    });\n  });\n\n  it(\"should call onSettled callback with undefined data and error on failed execution\", async () => {\n    const onSuccess = vi.fn();\n    const onError = vi.fn();\n    const onSettled = vi.fn();\n    OpenaiMock.callTool.mockRejectedValueOnce(error);\n    const { result } = renderHook(() =>\n      useCallTool<typeof args, typeof data>(toolName),\n    );\n\n    act(() => {\n      result.current.callTool(args, {\n        onSuccess,\n        onError,\n        onSettled,\n      });\n    });\n\n    await waitFor(() => {\n      expect(onError).toHaveBeenCalledWith(error, args);\n      expect(onSettled).toHaveBeenCalledWith(undefined, error, args);\n      expect(onSuccess).not.toHaveBeenCalled();\n    });\n  });\n\n  it(\"should always return last call started state\", async () => {\n    const { result } = renderHook(() =>\n      useCallTool<typeof args, typeof data>(toolName),\n    );\n\n    const firstCallData = {\n      ...data,\n      structuredContent: { result: \"first call result\" },\n    };\n    const secondCallData = {\n      ...data,\n      structuredContent: { result: \"second call result\" },\n    };\n    const { promise: firstCallToolPromise, resolve: resolveFirstCallTool } =\n      Promise.withResolvers();\n    const { promise: secondCallToolPromise, resolve: resolveSecondCallTool } =\n      Promise.withResolvers();\n    OpenaiMock.callTool\n      .mockImplementationOnce(() => firstCallToolPromise)\n      .mockImplementationOnce(() => secondCallToolPromise);\n\n    await act(() => {\n      result.current.callTool(args);\n      result.current.callTool(args);\n      resolveFirstCallTool(firstCallData);\n      return firstCallToolPromise;\n    });\n\n    expect(result.current.status).toEqual(\"pending\");\n    expect(result.current.data).toEqual(undefined);\n    resolveSecondCallTool(secondCallData);\n\n    await waitFor(() => {\n      expect(result.current.status).toEqual(\"success\");\n      expect(result.current.data).toEqual(secondCallData);\n    });\n  });\n});\n\ndescribe(\"useCallTool - TypeScript typing\", () => {\n  let OpenaiMock: { callTool: Mock };\n\n  beforeEach(() => {\n    OpenaiMock = {\n      callTool: vi.fn(),\n    };\n    vi.stubGlobal(\"openai\", OpenaiMock);\n    vi.stubGlobal(\"skybridge\", { hostType: \"apps-sdk\" });\n  });\n\n  afterEach(() => {\n    vi.unstubAllGlobals();\n    vi.resetAllMocks();\n  });\n\n  it(\"should have correct return types when ToolArgs is null and ToolResponse is specified\", async () => {\n    type TestResponse = CallToolResponse & {\n      structuredContent: { result: string };\n      meta: { id: number };\n    };\n\n    const { result } = renderHook(() =>\n      useCallTool<null, TestResponse>(\"test-tool\"),\n    );\n    const data: TestResponse = {\n      content: [{ type: \"text\" as const, text: \"test\" }],\n      structuredContent: { result: \"test\" },\n      isError: false,\n      meta: { id: 123 },\n    };\n\n    OpenaiMock.callTool.mockResolvedValueOnce(data);\n\n    await act(async () => {\n      result.current.callTool();\n    });\n\n    expect(OpenaiMock.callTool).toHaveBeenCalledWith(\"test-tool\", null);\n    expectTypeOf<typeof data | undefined>(result.current.data);\n  });\n\n  it(\"should correctly type callToolAsync return value\", async () => {\n    type TestArgs = { query: string };\n    type TestResponse = CallToolResponse & {\n      structuredContent: { answer: string };\n    };\n\n    const { result } = renderHook(() =>\n      useCallTool<TestArgs, TestResponse>(\"test-tool\"),\n    );\n\n    const testArgs: TestArgs = { query: \"test\" };\n    const mockResponse: TestResponse = {\n      content: [{ type: \"text\" as const, text: \"answer\" }],\n      structuredContent: { answer: \"test answer\" },\n      isError: false,\n      meta: {},\n    };\n\n    OpenaiMock.callTool.mockResolvedValueOnce(mockResponse);\n\n    const returnedValue = await act(async () => {\n      return result.current.callToolAsync(testArgs);\n    });\n    expectTypeOf<typeof mockResponse>(returnedValue);\n    expect(returnedValue).toEqual(mockResponse);\n  });\n\n  it(\"should correctly type callToolAsync when ToolArgs is null\", async () => {\n    type TestResponse = CallToolResponse & {\n      structuredContent: { data: string };\n    };\n\n    const { result } = renderHook(() =>\n      useCallTool<null, TestResponse>(\"test-tool\"),\n    );\n\n    const mockResponse: TestResponse = {\n      content: [{ type: \"text\" as const, text: \"data\" }],\n      structuredContent: { data: \"test data\" },\n      isError: false,\n      meta: {},\n    };\n\n    OpenaiMock.callTool.mockResolvedValueOnce(mockResponse);\n\n    const returnedValue = await act(async () => {\n      return result.current.callToolAsync();\n    });\n    expectTypeOf<typeof mockResponse>(returnedValue);\n    expect(returnedValue).toEqual(mockResponse);\n  });\n});\n"]}
+{"version":3,"file":"use-call-tool.test.js","sourceRoot":"","sources":["../../../src/web/hooks/use-call-tool.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,UAAU,EAAE,OAAO,EAAE,MAAM,wBAAwB,CAAC;AAClE,OAAO,EACL,SAAS,EACT,UAAU,EACV,QAAQ,EACR,MAAM,EACN,YAAY,EACZ,EAAE,EAEF,EAAE,GACH,MAAM,QAAQ,CAAC;AAChB,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AACpD,OAAO,EAAE,aAAa,EAAE,MAAM,+BAA+B,CAAC;AAC9D,OAAO,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AACvD,OAAO,EAAE,YAAY,EAAE,MAAM,8BAA8B,CAAC;AAE5D,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAEjD,QAAQ,CAAC,kCAAkC,EAAE,GAAG,EAAE;IAChD,IAAI,YAAkB,CAAC;IAEvB,UAAU,CAAC,GAAG,EAAE;QACd,WAAW,CAAC,aAAa,EAAE,CAAC;QAC5B,YAAY,CAAC,aAAa,EAAE,CAAC;QAC7B,aAAa,CAAC,aAAa,EAAE,CAAC;QAC9B,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,EAAE,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QAClD,EAAE,CAAC,UAAU,CAAC,WAAW,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC,CAAC;QACrD,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QAC/C,YAAY,GAAG,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,EAAE,UAAU,CAAoB,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,gBAAgB,EAAE,CAAC;QACtB,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,WAAW,CAAC,aAAa,EAAE,CAAC;QAC5B,YAAY,CAAC,aAAa,EAAE,CAAC;QAC7B,aAAa,CAAC,aAAa,EAAE,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,MAAM,QAAQ,GAAG,WAAW,CAAC;IAC7B,MAAM,IAAI,GAAG,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC;IACrC,MAAM,IAAI,GAAqB;QAC7B,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;QACzD,iBAAiB,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE;QACrC,OAAO,EAAE,KAAK;QACd,IAAI,EAAE,EAAE;KACT,CAAC;IACF,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;IAEtC,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,cAAc,GAAG;YACrB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;YACpD,iBAAiB,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE;YAC/B,OAAO,EAAE,KAAK;YACd,IAAI,EAAE,EAAE,MAAM,EAAE,wBAAwB,EAAE;SAC3C,CAAC;QACF,YAAY,CAAC,qBAAqB,CAAC,cAAc,CAAC,CAAC;QACnD,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,WAAW,CAA2B,QAAQ,CAAC,CAChD,CAAC;QAEF,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,aAAa,CAAC;gBACxC,IAAI,EAAE,EAAE,MAAM,EAAE,wBAAwB,EAAE;aAC3C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,WAAW,CAA2B,QAAQ,CAAC,CAChD,CAAC;QACF,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uFAAuF,EAAE,KAAK,IAAI,EAAE;QACrG,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC1B,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACxB,YAAY,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC;QACzC,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,WAAW,CAA2B,QAAQ,CAAC,CAChD,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE;gBAC5B,SAAS;gBACT,OAAO;aACR,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YACnD,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;QACxF,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC1B,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACxB,YAAY,CAAC,qBAAqB,CAAC,KAAK,CAAC,CAAC;QAC1C,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,WAAW,CAA2B,QAAQ,CAAC,CAChD,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE;gBAC5B,SAAS;gBACT,OAAO;aACR,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YACzC,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sFAAsF,EAAE,KAAK,IAAI,EAAE;QACpG,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC1B,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACxB,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC1B,YAAY,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC;QACzC,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,WAAW,CAA2B,QAAQ,CAAC,CAChD,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE;gBAC5B,SAAS;gBACT,OAAO;gBACP,SAAS;aACV,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YACnD,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC;YAC9D,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kFAAkF,EAAE,KAAK,IAAI,EAAE;QAChG,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC1B,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACxB,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC1B,YAAY,CAAC,qBAAqB,CAAC,KAAK,CAAC,CAAC;QAC1C,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,WAAW,CAA2B,QAAQ,CAAC,CAChD,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE;gBAC5B,SAAS;gBACT,OAAO;gBACP,SAAS;aACV,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;YAClD,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,SAAS,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;YAC/D,MAAM,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC3C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,WAAW,CAA2B,QAAQ,CAAC,CAChD,CAAC;QAEF,MAAM,aAAa,GAAG;YACpB,GAAG,IAAI;YACP,iBAAiB,EAAE,EAAE,MAAM,EAAE,mBAAmB,EAAE;SACnD,CAAC;QACF,MAAM,cAAc,GAAG;YACrB,GAAG,IAAI;YACP,iBAAiB,EAAE,EAAE,MAAM,EAAE,oBAAoB,EAAE;SACpD,CAAC;QACF,MAAM,EAAE,OAAO,EAAE,oBAAoB,EAAE,OAAO,EAAE,oBAAoB,EAAE,GACpE,OAAO,CAAC,aAAa,EAAE,CAAC;QAC1B,MAAM,EAAE,OAAO,EAAE,qBAAqB,EAAE,OAAO,EAAE,qBAAqB,EAAE,GACtE,OAAO,CAAC,aAAa,EAAE,CAAC;QAC1B,YAAY;aACT,sBAAsB,CAAC,GAAG,EAAE,CAAC,oBAAoB,CAAC;aAClD,sBAAsB,CAAC,GAAG,EAAE,CAAC,qBAAqB,CAAC,CAAC;QAEvD,MAAM,GAAG,CAAC,GAAG,EAAE;YACb,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YAC9B,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YAC9B,oBAAoB,CAAC,aAAa,CAAC,CAAC;YACpC,OAAO,oBAAoB,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACjD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAC/C,qBAAqB,CAAC,cAAc,CAAC,CAAC;QAEtC,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YACjD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;IAC/C,IAAI,YAAkB,CAAC;IAEvB,UAAU,CAAC,GAAG,EAAE;QACd,WAAW,CAAC,aAAa,EAAE,CAAC;QAC5B,YAAY,CAAC,aAAa,EAAE,CAAC;QAC7B,aAAa,CAAC,aAAa,EAAE,CAAC;QAC9B,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,EAAE,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QAClD,EAAE,CAAC,UAAU,CAAC,WAAW,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC,CAAC;QACrD,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QAC/C,YAAY,GAAG,EAAE,CAAC,KAAK,CAAC,UAAU,EAAE,EAAE,UAAU,CAAoB,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,gBAAgB,EAAE,CAAC;QACtB,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,WAAW,CAAC,aAAa,EAAE,CAAC;QAC5B,YAAY,CAAC,aAAa,EAAE,CAAC;QAC7B,aAAa,CAAC,aAAa,EAAE,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sFAAsF,EAAE,KAAK,IAAI,EAAE;QAMpG,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,WAAW,CAAqB,WAAW,CAAC,CAC7C,CAAC;QACF,MAAM,IAAI,GAAiB;YACzB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;YAClD,iBAAiB,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE;YACrC,OAAO,EAAE,KAAK;YACd,IAAI,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE;SAClB,CAAC;QAEF,YAAY,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC;QAEzC,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;QAC7D,YAAY,CAA0B,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAMhE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,WAAW,CAAyB,WAAW,CAAC,CACjD,CAAC;QAEF,MAAM,QAAQ,GAAa,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;QAC7C,MAAM,YAAY,GAAiB;YACjC,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;YACpD,iBAAiB,EAAE,EAAE,MAAM,EAAE,aAAa,EAAE;YAC5C,OAAO,EAAE,KAAK;YACd,IAAI,EAAE,EAAE;SACT,CAAC;QAEF,YAAY,CAAC,qBAAqB,CAAC,YAAY,CAAC,CAAC;QAEjD,MAAM,aAAa,GAAG,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACzC,OAAO,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QACH,YAAY,CAAsB,aAAa,CAAC,CAAC;QACjD,MAAM,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QAKzE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,WAAW,CAAqB,WAAW,CAAC,CAC7C,CAAC;QAEF,MAAM,YAAY,GAAiB;YACjC,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;YAClD,iBAAiB,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE;YACxC,OAAO,EAAE,KAAK;YACd,IAAI,EAAE,EAAE;SACT,CAAC;QAEF,YAAY,CAAC,qBAAqB,CAAC,YAAY,CAAC,CAAC;QAEjD,MAAM,aAAa,GAAG,MAAM,GAAG,CAAC,KAAK,IAAI,EAAE;YACzC,OAAO,MAAM,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;QACxC,CAAC,CAAC,CAAC;QACH,YAAY,CAAsB,aAAa,CAAC,CAAC;QACjD,MAAM,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { act, renderHook, waitFor } from \"@testing-library/react\";\nimport {\n  afterEach,\n  beforeEach,\n  describe,\n  expect,\n  expectTypeOf,\n  it,\n  type Mock,\n  vi,\n} from \"vitest\";\nimport { HostAdaptor } from \"../bridges/adaptor.js\";\nimport { AppsSdkBridge } from \"../bridges/apps-sdk/bridge.js\";\nimport { getAdaptor } from \"../bridges/get-adaptor.js\";\nimport { McpAppBridge } from \"../bridges/mcp-app/bridge.js\";\nimport type { CallToolResponse } from \"../bridges/types.js\";\nimport { useCallTool } from \"./use-call-tool.js\";\n\ndescribe(\"useCallTool - onSuccess callback\", () => {\n  let callToolMock: Mock;\n\n  beforeEach(() => {\n    HostAdaptor.resetInstance();\n    McpAppBridge.resetInstance();\n    AppsSdkBridge.resetInstance();\n    vi.stubGlobal(\"parent\", { postMessage: vi.fn() });\n    vi.stubGlobal(\"skybridge\", { hostType: \"apps-sdk\" });\n    vi.stubGlobal(\"openai\", { callTool: vi.fn() });\n    callToolMock = vi.spyOn(getAdaptor(), \"callTool\") as unknown as Mock;\n  });\n\n  afterEach(() => {\n    vi.unstubAllGlobals();\n    vi.resetAllMocks();\n    HostAdaptor.resetInstance();\n    McpAppBridge.resetInstance();\n    AppsSdkBridge.resetInstance();\n  });\n\n  const toolName = \"test-tool\";\n  const args = { input: \"test input\" };\n  const data: CallToolResponse = {\n    content: [{ type: \"text\" as const, text: \"test result\" }],\n    structuredContent: { result: \"test\" },\n    isError: false,\n    meta: {},\n  };\n  const error = new Error(\"test error\");\n\n  it(\"should normalize _meta to meta when SDK returns _meta instead of meta\", async () => {\n    const rawSdkResponse = {\n      content: [{ type: \"text\" as const, text: \"result\" }],\n      structuredContent: { value: 1 },\n      isError: false,\n      meta: { secret: \"only visible to widget\" },\n    };\n    callToolMock.mockResolvedValueOnce(rawSdkResponse);\n    const { result } = renderHook(() =>\n      useCallTool<typeof args, typeof data>(toolName),\n    );\n\n    await act(async () => {\n      result.current.callTool(args);\n    });\n\n    await waitFor(() => {\n      expect(result.current.data).toMatchObject({\n        meta: { secret: \"only visible to widget\" },\n      });\n    });\n  });\n\n  it(\"should call adaptor.callTool with correct arguments\", async () => {\n    const { result } = renderHook(() =>\n      useCallTool<typeof args, typeof data>(toolName),\n    );\n    await act(async () => {\n      result.current.callTool(args);\n    });\n    expect(callToolMock).toHaveBeenCalledWith(toolName, args);\n  });\n\n  it(\"should call onSuccess callback with correct data and toolArgs on successful execution\", async () => {\n    const onSuccess = vi.fn();\n    const onError = vi.fn();\n    callToolMock.mockResolvedValueOnce(data);\n    const { result } = renderHook(() =>\n      useCallTool<typeof args, typeof data>(toolName),\n    );\n\n    act(() => {\n      result.current.callTool(args, {\n        onSuccess,\n        onError,\n      });\n    });\n\n    await waitFor(() => {\n      expect(onSuccess).toHaveBeenCalledWith(data, args);\n      expect(onError).not.toHaveBeenCalled();\n    });\n  });\n\n  it(\"should call onError callback with error and toolArgs on failed execution\", async () => {\n    const onSuccess = vi.fn();\n    const onError = vi.fn();\n    callToolMock.mockRejectedValueOnce(error);\n    const { result } = renderHook(() =>\n      useCallTool<typeof args, typeof data>(toolName),\n    );\n\n    act(() => {\n      result.current.callTool(args, {\n        onSuccess,\n        onError,\n      });\n    });\n\n    await waitFor(() => {\n      expect(onSuccess).not.toHaveBeenCalled();\n      expect(onError).toHaveBeenCalledWith(error, args);\n    });\n  });\n\n  it(\"should call onSettled callback with data and undefined error on successful execution\", async () => {\n    const onSuccess = vi.fn();\n    const onError = vi.fn();\n    const onSettled = vi.fn();\n    callToolMock.mockResolvedValueOnce(data);\n    const { result } = renderHook(() =>\n      useCallTool<typeof args, typeof data>(toolName),\n    );\n\n    act(() => {\n      result.current.callTool(args, {\n        onSuccess,\n        onError,\n        onSettled,\n      });\n    });\n\n    await waitFor(() => {\n      expect(onSuccess).toHaveBeenCalledWith(data, args);\n      expect(onSettled).toHaveBeenCalledWith(data, undefined, args);\n      expect(onError).not.toHaveBeenCalled();\n    });\n  });\n\n  it(\"should call onSettled callback with undefined data and error on failed execution\", async () => {\n    const onSuccess = vi.fn();\n    const onError = vi.fn();\n    const onSettled = vi.fn();\n    callToolMock.mockRejectedValueOnce(error);\n    const { result } = renderHook(() =>\n      useCallTool<typeof args, typeof data>(toolName),\n    );\n\n    act(() => {\n      result.current.callTool(args, {\n        onSuccess,\n        onError,\n        onSettled,\n      });\n    });\n\n    await waitFor(() => {\n      expect(onError).toHaveBeenCalledWith(error, args);\n      expect(onSettled).toHaveBeenCalledWith(undefined, error, args);\n      expect(onSuccess).not.toHaveBeenCalled();\n    });\n  });\n\n  it(\"should always return last call started state\", async () => {\n    const { result } = renderHook(() =>\n      useCallTool<typeof args, typeof data>(toolName),\n    );\n\n    const firstCallData = {\n      ...data,\n      structuredContent: { result: \"first call result\" },\n    };\n    const secondCallData = {\n      ...data,\n      structuredContent: { result: \"second call result\" },\n    };\n    const { promise: firstCallToolPromise, resolve: resolveFirstCallTool } =\n      Promise.withResolvers();\n    const { promise: secondCallToolPromise, resolve: resolveSecondCallTool } =\n      Promise.withResolvers();\n    callToolMock\n      .mockImplementationOnce(() => firstCallToolPromise)\n      .mockImplementationOnce(() => secondCallToolPromise);\n\n    await act(() => {\n      result.current.callTool(args);\n      result.current.callTool(args);\n      resolveFirstCallTool(firstCallData);\n      return firstCallToolPromise;\n    });\n\n    expect(result.current.status).toEqual(\"pending\");\n    expect(result.current.data).toEqual(undefined);\n    resolveSecondCallTool(secondCallData);\n\n    await waitFor(() => {\n      expect(result.current.status).toEqual(\"success\");\n      expect(result.current.data).toEqual(secondCallData);\n    });\n  });\n});\n\ndescribe(\"useCallTool - TypeScript typing\", () => {\n  let callToolMock: Mock;\n\n  beforeEach(() => {\n    HostAdaptor.resetInstance();\n    McpAppBridge.resetInstance();\n    AppsSdkBridge.resetInstance();\n    vi.stubGlobal(\"parent\", { postMessage: vi.fn() });\n    vi.stubGlobal(\"skybridge\", { hostType: \"apps-sdk\" });\n    vi.stubGlobal(\"openai\", { callTool: vi.fn() });\n    callToolMock = vi.spyOn(getAdaptor(), \"callTool\") as unknown as Mock;\n  });\n\n  afterEach(() => {\n    vi.unstubAllGlobals();\n    vi.resetAllMocks();\n    HostAdaptor.resetInstance();\n    McpAppBridge.resetInstance();\n    AppsSdkBridge.resetInstance();\n  });\n\n  it(\"should have correct return types when ToolArgs is null and ToolResponse is specified\", async () => {\n    type TestResponse = CallToolResponse & {\n      structuredContent: { result: string };\n      meta: { id: number };\n    };\n\n    const { result } = renderHook(() =>\n      useCallTool<null, TestResponse>(\"test-tool\"),\n    );\n    const data: TestResponse = {\n      content: [{ type: \"text\" as const, text: \"test\" }],\n      structuredContent: { result: \"test\" },\n      isError: false,\n      meta: { id: 123 },\n    };\n\n    callToolMock.mockResolvedValueOnce(data);\n\n    await act(async () => {\n      result.current.callTool();\n    });\n\n    expect(callToolMock).toHaveBeenCalledWith(\"test-tool\", null);\n    expectTypeOf<typeof data | undefined>(result.current.data);\n  });\n\n  it(\"should correctly type callToolAsync return value\", async () => {\n    type TestArgs = { query: string };\n    type TestResponse = CallToolResponse & {\n      structuredContent: { answer: string };\n    };\n\n    const { result } = renderHook(() =>\n      useCallTool<TestArgs, TestResponse>(\"test-tool\"),\n    );\n\n    const testArgs: TestArgs = { query: \"test\" };\n    const mockResponse: TestResponse = {\n      content: [{ type: \"text\" as const, text: \"answer\" }],\n      structuredContent: { answer: \"test answer\" },\n      isError: false,\n      meta: {},\n    };\n\n    callToolMock.mockResolvedValueOnce(mockResponse);\n\n    const returnedValue = await act(async () => {\n      return result.current.callToolAsync(testArgs);\n    });\n    expectTypeOf<typeof mockResponse>(returnedValue);\n    expect(returnedValue).toEqual(mockResponse);\n  });\n\n  it(\"should correctly type callToolAsync when ToolArgs is null\", async () => {\n    type TestResponse = CallToolResponse & {\n      structuredContent: { data: string };\n    };\n\n    const { result } = renderHook(() =>\n      useCallTool<null, TestResponse>(\"test-tool\"),\n    );\n\n    const mockResponse: TestResponse = {\n      content: [{ type: \"text\" as const, text: \"data\" }],\n      structuredContent: { data: \"test data\" },\n      isError: false,\n      meta: {},\n    };\n\n    callToolMock.mockResolvedValueOnce(mockResponse);\n\n    const returnedValue = await act(async () => {\n      return result.current.callToolAsync();\n    });\n    expectTypeOf<typeof mockResponse>(returnedValue);\n    expect(returnedValue).toEqual(mockResponse);\n  });\n});\n"]}

package/dist/web/hooks/use-display-mode.test.js

@@ -1,39 +1,75 @@
-import { act, renderHook } from "@testing-library/react";
-import { afterEach, beforeEach, describe, expect, it, vi, } from "vitest";
+import { act, fireEvent, renderHook, waitFor } from "@testing-library/react";
+import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
+import { HostAdaptor } from "../bridges/adaptor.js";
+import { AppsSdkBridge } from "../bridges/apps-sdk/bridge.js";
+import { getAdaptor } from "../bridges/get-adaptor.js";
+import { McpAppBridge } from "../bridges/mcp-app/bridge.js";
+import { getMcpAppHostPostMessageMock, MockResizeObserver, } from "./test/utils.js";
 import { useDisplayMode } from "./use-display-mode.js";
 describe("useDisplayMode", () => {
-    let OpenaiMock;
     beforeEach(() => {
-        OpenaiMock = {
-            displayMode: "inline",
-            requestDisplayMode: vi.fn().mockResolvedValue({ mode: "inline" }),
-        };
-        vi.stubGlobal("openai", OpenaiMock);
+        HostAdaptor.resetInstance();
+        McpAppBridge.resetInstance();
+        AppsSdkBridge.resetInstance();
         vi.stubGlobal("skybridge", { hostType: "apps-sdk" });
+        vi.stubGlobal("openai", { view: { mode: "inline" } });
+        vi.stubGlobal("ResizeObserver", MockResizeObserver);
     });
     afterEach(() => {
         vi.unstubAllGlobals();
         vi.resetAllMocks();
+        HostAdaptor.resetInstance();
+        McpAppBridge.resetInstance();
+        AppsSdkBridge.resetInstance();
     });
-    it("should return the current display mode from window.openai.displayMode", () => {
-        OpenaiMock.displayMode = "inline";
+    it("should return the current display mode from MCP host context", async () => {
+        vi.stubGlobal("parent", {
+            postMessage: getMcpAppHostPostMessageMock({ displayMode: "inline" }),
+        });
         const { result } = renderHook(() => useDisplayMode());
-        expect(result.current[0]).toBe("inline");
+        await waitFor(() => {
+            expect(result.current[0]).toBe("inline");
+        });
     });
-    it("should return different display modes when window.openai.displayMode changes", () => {
-        OpenaiMock.displayMode = "inline";
-        const { result, rerender } = renderHook(() => useDisplayMode());
-        expect(result.current[0]).toBe("inline");
-        OpenaiMock.displayMode = "fullscreen";
-        rerender();
-        expect(result.current[0]).toBe("fullscreen");
+    it("should update display mode on host-context-changed notification", async () => {
+        vi.stubGlobal("parent", {
+            postMessage: getMcpAppHostPostMessageMock({ displayMode: "inline" }),
+        });
+        const { result } = renderHook(() => useDisplayMode());
+        await waitFor(() => {
+            expect(result.current[0]).toBe("inline");
+        });
+        act(() => {
+            fireEvent(window, new MessageEvent("message", {
+                source: window.parent,
+                data: {
+                    jsonrpc: "2.0",
+                    method: "ui/notifications/host-context-changed",
+                    params: { displayMode: "fullscreen" },
+                },
+            }));
+        });
+        await waitFor(() => {
+            expect(result.current[0]).toBe("fullscreen");
+        });
     });
-    it("should call window.openai.requestDisplayMode with correct mode when setDisplayMode is called", async () => {
+    it("should call requestDisplayMode on the adaptor when setDisplayMode is called", async () => {
+        vi.stubGlobal("parent", {
+            postMessage: getMcpAppHostPostMessageMock({ displayMode: "inline" }),
+        });
         const { result } = renderHook(() => useDisplayMode());
+        await waitFor(() => expect(result.current[0]).toBe("inline"));
+        // Patch mcp.getApp to avoid real protocol request
+        const adaptor = getAdaptor();
+        const fakeApp = {
+            requestDisplayMode: vi.fn().mockResolvedValue({ mode: "fullscreen" }),
+        };
+        // biome-ignore lint/suspicious/noExplicitAny: test seam
+        adaptor.mcp.getApp = vi.fn().mockResolvedValue(fakeApp);
         await act(async () => {
             await result.current[1]("fullscreen");
         });
-        expect(OpenaiMock.requestDisplayMode).toHaveBeenCalledWith({
+        expect(fakeApp.requestDisplayMode).toHaveBeenCalledWith({
             mode: "fullscreen",
         });
     });