fix: address widget review feedback

packages/widgets/src/definition.spec.ts

@@ -1,6 +1,8 @@
 import { describe, expect, test } from "vitest";
 
-import { getWidgetQueryKeys } from "./definition";
+import { getWidgetQueryKeys, getWidgetRuntimeQueries, setWidgetRuntimeQueries } from "./definition";
+
+const togglePolling = () => undefined;
 
 describe("getWidgetQueryKeys", () => {
   test("falls back to the widget router namespace", () => {
@@ -12,8 +14,21 @@ describe("getWidgetQueryKeys", () => {
   });
 
   test("preserves every explicit key for multi-root widgets", () => {
-    const queryKeys = [[["widget", "audioStats"]], [["widget", "mediaServer"]]] as const;
+    const queryKeys = [
+      [["widget", "audioStats", "getStats"]],
+      [["widget", "mediaServer", "getCurrentStreams"]],
+    ] as const;
 
     expect(getWidgetQueryKeys({ kind: "audioStats", queryKeys })).toBe(queryKeys);
   });
+
+  test("registers runtime queries without replacing imperative widget actions", () => {
+    const widgetStateRef = { current: { togglePolling } };
+    const runtimeQueries = [{ path: ["widget", "calendar", "findAllEvents"], input: { month: 7 } }];
+
+    setWidgetRuntimeQueries(widgetStateRef, runtimeQueries);
+
+    expect(widgetStateRef.current.togglePolling).toBe(togglePolling);
+    expect(getWidgetRuntimeQueries(widgetStateRef)).toEqual(runtimeQueries);
+  });
 });