From c4a50f136ac2292037768c56ee2ce2d144237909 Mon Sep 17 00:00:00 2001
From: "google-labs-jules[bot]"
 <161369871+google-labs-jules[bot]@users.noreply.github.com>
Date: Fri, 10 Jul 2026 20:38:46 +0000
Subject: [PATCH] test: add test for action resolving to a function

Co-authored-by: johnstrand <11484777+johnstrand@users.noreply.github.com>
---
 src/__tests__/Squawk.action.test.ts | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/src/__tests__/Squawk.action.test.ts b/src/__tests__/Squawk.action.test.ts
index e27f966..3e3228e 100644
--- a/src/__tests__/Squawk.action.test.ts
+++ b/src/__tests__/Squawk.action.test.ts
@@ -28,4 +28,19 @@ describe("Squawk action", () => {
 
     pendingSpy.mockRestore();
   });
+
+  it("resolves functional updates correctly", async () => {
+    const store = createStore({
+      testProp: "initial"
+    });
+
+    const updateAction = store.action(() => {
+      // Returning a function that returns the updated state
+      return (() => ({ testProp: "updated" })) as any;
+    });
+
+    await updateAction();
+
+    expect(store.get().testProp).toBe("updated");
+  });
 });