@ersbeth/picoflow 1.0.1 → 1.1.1

package/test/nodes/await/flowStateAwait.test.ts

@@ -0,0 +1,999 @@
+import { describe, expect, it, vi } from "vitest";
+import { effect, state, stateAwait } from "#package";
+
+describe("flowStateAwait", () => {
+	describe("unit", () => {
+		describe("initialization", () => {
+			it("should initialize with direct Promise", async () => {
+				const $state = stateAwait(Promise.resolve(1));
+				expect(await $state.pick()).toBe(1);
+			});
+
+			it("should initialize lazily with function", async () => {
+				const initFn = vi.fn(() => Promise.resolve(42));
+				const $state = stateAwait(initFn);
+
+				// Function should not be called yet
+				expect(initFn).not.toHaveBeenCalled();
+
+				// First access triggers initialization
+				const value = await $state.pick();
+				expect(value).toBe(42);
+				expect(initFn).toHaveBeenCalledTimes(1);
+			});
+
+			it("should not call lazy function on creation", () => {
+				const initFn = vi.fn(() => Promise.resolve(100));
+				stateAwait(initFn);
+
+				expect(initFn).not.toHaveBeenCalled();
+			});
+
+			it("should call lazy function on first access", async () => {
+				const initFn = vi.fn(() => Promise.resolve(200));
+				const $state = stateAwait(initFn);
+
+				expect(initFn).not.toHaveBeenCalled();
+				await $state.pick();
+				expect(initFn).toHaveBeenCalledTimes(1);
+			});
+
+			it("should call lazy function only once", async () => {
+				const initFn = vi.fn(() => Promise.resolve(300));
+				const $state = stateAwait(initFn);
+
+				// Multiple accesses should only call init once
+				await $state.pick();
+				await $state.pick();
+				await $state.pick();
+
+				expect(initFn).toHaveBeenCalledTimes(1);
+			});
+
+			it("should cache value after lazy initialization", async () => {
+				const initFn = vi.fn(() => Promise.resolve("cached"));
+				const $state = stateAwait(initFn);
+
+				const value1 = await $state.pick();
+				const value2 = await $state.pick();
+				const value3 = await $state.pick();
+
+				expect(value1).toBe("cached");
+				expect(value2).toBe("cached");
+				expect(value3).toBe("cached");
+				expect(initFn).toHaveBeenCalledTimes(1);
+			});
+
+			it("should handle number values", async () => {
+				const $state = stateAwait(Promise.resolve(42));
+				expect(await $state.pick()).toBe(42);
+			});
+
+			it("should handle string values", async () => {
+				const $state = stateAwait(Promise.resolve("hello"));
+				expect(await $state.pick()).toBe("hello");
+			});
+
+			it("should handle object values", async () => {
+				const obj = { a: 1, b: 2 };
+				const $state = stateAwait(Promise.resolve(obj));
+				const value = await $state.pick();
+				expect(value).toBe(obj);
+				expect(value).toEqual({ a: 1, b: 2 });
+			});
+
+			it("should handle array values", async () => {
+				const arr = [1, 2, 3];
+				const $state = stateAwait(Promise.resolve(arr));
+				const value = await $state.pick();
+				expect(value).toBe(arr);
+				expect(value).toEqual([1, 2, 3]);
+			});
+
+			it("should handle null values", async () => {
+				const $state = stateAwait(Promise.resolve(null));
+				const value = await $state.pick();
+				expect(value).toBeNull();
+			});
+
+			it("should handle undefined values", async () => {
+				const $state = stateAwait(Promise.resolve(undefined));
+				const value = await $state.pick();
+				expect(value).toBeUndefined();
+			});
+		});
+
+		describe("get", () => {
+			it("should return resolved value with tracking context", async () => {
+				const $state = stateAwait(Promise.resolve(5));
+				const $tracker = state(0); // Use state as FlowTracker
+
+				// Wait for Promise to resolve first
+				await $state.pick();
+
+				const value = $state.get($tracker);
+				expect(value).toBe(5);
+			});
+
+			it("should return undefined if Promise not yet resolved", () => {
+				// Create a Promise that resolves after a delay
+				const $state = stateAwait(
+					new Promise<number>((resolve) => setTimeout(() => resolve(10), 100)),
+				);
+				const $tracker = state(0);
+
+				// Before Promise resolves, get() should return undefined
+				const value = $state.get($tracker);
+				expect(value).toBeUndefined();
+			});
+
+			it("should compute lazy value on first get call", async () => {
+				const initFn = vi.fn(() => Promise.resolve(10));
+				const $state = stateAwait(initFn);
+				const $tracker = state(0);
+
+				expect(initFn).not.toHaveBeenCalled();
+
+				// First get() may return undefined, but triggers computation
+				$state.get($tracker);
+				// Wait for Promise to resolve
+				await $state.pick();
+
+				const value2 = $state.get($tracker);
+				expect(value2).toBe(10);
+				expect(initFn).toHaveBeenCalledTimes(1);
+			});
+
+			it("should return cached value on subsequent get calls", async () => {
+				const initFn = vi.fn(() => Promise.resolve(20));
+				const $state = stateAwait(initFn);
+				const $tracker = state(0);
+
+				// Wait for initial resolution
+				await $state.pick();
+
+				const value1 = $state.get($tracker);
+				const value2 = $state.get($tracker);
+				const value3 = $state.get($tracker);
+
+				expect(value1).toBe(20);
+				expect(value2).toBe(20);
+				expect(value3).toBe(20);
+				expect(initFn).toHaveBeenCalledTimes(1);
+			});
+
+			it("should throw error when get is called after disposal", async () => {
+				const $state = stateAwait(Promise.resolve(1));
+				const $tracker = state(0);
+
+				await $state.pick();
+				$state.get($tracker);
+				$state.dispose();
+
+				expect(() => $state.get($tracker)).toThrow(
+					"[PicoFlow] Primitive is disposed",
+				);
+			});
+		});
+
+		describe("pick", () => {
+			it("should return Promise without tracking", async () => {
+				const $state = stateAwait(Promise.resolve(15));
+				const promise = $state.pick();
+
+				expect(promise).toBeInstanceOf(Promise);
+				const value = await promise;
+				expect(value).toBe(15);
+			});
+
+			it("should compute lazy value on first pick call", async () => {
+				const initFn = vi.fn(() => Promise.resolve(25));
+				const $state = stateAwait(initFn);
+
+				expect(initFn).not.toHaveBeenCalled();
+				const promise = $state.pick();
+				expect(promise).toBeInstanceOf(Promise);
+				const value = await promise;
+				expect(value).toBe(25);
+				expect(initFn).toHaveBeenCalledTimes(1);
+			});
+
+			it("should return cached Promise on subsequent pick calls", async () => {
+				const initFn = vi.fn(() => Promise.resolve(30));
+				const $state = stateAwait(initFn);
+
+				const promise1 = $state.pick();
+				const promise2 = $state.pick();
+				const promise3 = $state.pick();
+
+				const value1 = await promise1;
+				const value2 = await promise2;
+				const value3 = await promise3;
+
+				expect(value1).toBe(30);
+				expect(value2).toBe(30);
+				expect(value3).toBe(30);
+				expect(initFn).toHaveBeenCalledTimes(1);
+			});
+
+			it("should await Promise to get value", async () => {
+				const $state = stateAwait(Promise.resolve(35));
+				const value = await $state.pick();
+				expect(value).toBe(35);
+			});
+
+			it("should throw error when pick is called after disposal", async () => {
+				const $state = stateAwait(Promise.resolve(1));
+
+				expect(await $state.pick()).toBe(1);
+
+				$state.dispose();
+
+				await expect($state.pick()).rejects.toThrow(
+					"[PicoFlow] Primitive is disposed",
+				);
+			});
+		});
+
+		describe("set", () => {
+			it("should update value with Promise", async () => {
+				const $state = stateAwait(Promise.resolve(1));
+				expect(await $state.pick()).toBe(1);
+
+				await $state.set(Promise.resolve(2));
+				expect(await $state.pick()).toBe(2);
+			});
+
+			it("should update value with updater function", async () => {
+				const $state = stateAwait(Promise.resolve(1));
+				expect(await $state.pick()).toBe(1);
+
+				await $state.set(async (current) => Promise.resolve(current + 1));
+				expect(await $state.pick()).toBe(2);
+			});
+
+			it("should return Promise<void>", async () => {
+				const $state = stateAwait(Promise.resolve(1));
+				const result = await $state.set(Promise.resolve(2));
+				expect(result).toBeUndefined();
+			});
+
+			it("should not notify when set with same value", async () => {
+				const $state = stateAwait(Promise.resolve(1));
+				const listener = vi.fn();
+				$state.subscribe(listener);
+
+				await vi.waitFor(() => expect(listener).toHaveBeenCalledTimes(1));
+
+				await $state.set(Promise.resolve(1));
+
+				// Wait to ensure no additional calls
+				await new Promise((resolve) => setTimeout(resolve, 50));
+				expect(listener).toHaveBeenCalledTimes(1);
+			});
+
+			it("should notify when set with different value", async () => {
+				const $state = stateAwait(Promise.resolve(1));
+				const listener = vi.fn();
+				$state.subscribe(listener);
+
+				await vi.waitFor(() => expect(listener).toHaveBeenCalledTimes(1));
+
+				await $state.set(Promise.resolve(2));
+				await vi.waitFor(() => expect(listener).toHaveBeenCalledTimes(2));
+				expect(listener).toHaveBeenLastCalledWith(2);
+			});
+
+			it("should not notify when updater returns same value", async () => {
+				const $state = stateAwait(Promise.resolve(5));
+				const listener = vi.fn();
+				$state.subscribe(listener);
+
+				await vi.waitFor(() => expect(listener).toHaveBeenCalledTimes(1));
+
+				await $state.set(async (current) => Promise.resolve(current));
+
+				// Wait to ensure no additional calls
+				await new Promise((resolve) => setTimeout(resolve, 50));
+				expect(listener).toHaveBeenCalledTimes(1);
+			});
+
+			it("should notify when updater returns different value", async () => {
+				const $state = stateAwait(Promise.resolve(5));
+				const listener = vi.fn();
+				$state.subscribe(listener);
+
+				await vi.waitFor(() => expect(listener).toHaveBeenCalledTimes(1));
+
+				await $state.set(async (current) => Promise.resolve(current + 1));
+				await vi.waitFor(() => expect(listener).toHaveBeenCalledTimes(2));
+				expect(listener).toHaveBeenLastCalledWith(6);
+			});
+
+			it("should handle async updater function", async () => {
+				const $state = stateAwait(Promise.resolve(1));
+
+				await $state.set(async (current) => {
+					await new Promise((resolve) => setTimeout(resolve, 10));
+					return Promise.resolve(current + 10);
+				});
+
+				expect(await $state.pick()).toBe(11);
+			});
+
+			it("should throw error when set is called after disposal", async () => {
+				const $state = stateAwait(Promise.resolve(1));
+
+				expect(await $state.pick()).toBe(1);
+
+				await $state.set(Promise.resolve(2));
+				expect(await $state.pick()).toBe(2);
+
+				$state.dispose();
+
+				await expect($state.set(Promise.resolve(3))).rejects.toThrow(
+					"[PicoFlow] Primitive is disposed",
+				);
+			});
+
+			// TODO: Fix this test
+			it.skip("should handle Promise rejection in set", async () => {
+				const $state = stateAwait(Promise.resolve(1));
+
+				await expect(
+					$state.set(Promise.reject(new Error("Set error"))),
+				).rejects.toThrow("Set error");
+			});
+		});
+
+		describe("watch", () => {
+			it("should register dependency when watch is called", () => {
+				const $state = stateAwait(Promise.resolve(35));
+				const $tracker = state(0);
+
+				expect(() => $state.watch($tracker)).not.toThrow();
+			});
+
+			it("should compute lazy value when watch is called", async () => {
+				const initFn = vi.fn(() => Promise.resolve(40));
+				const $state = stateAwait(initFn);
+				const $tracker = state(0);
+
+				expect(initFn).not.toHaveBeenCalled();
+				$state.watch($tracker);
+				// Wait a bit for the computation to complete
+				await new Promise((resolve) => setTimeout(resolve, 10));
+				expect(initFn).toHaveBeenCalledTimes(1);
+			});
+
+			it("should throw error when watch is called after disposal", () => {
+				const $state = stateAwait(Promise.resolve(1));
+				const $tracker = state(0);
+
+				$state.watch($tracker);
+				$state.dispose();
+
+				expect(() => $state.watch($tracker)).toThrow(
+					"[PicoFlow] Primitive is disposed",
+				);
+			});
+		});
+
+		describe("subscribe", () => {
+			it("should return a disposer function", () => {
+				const $state = stateAwait(Promise.resolve(50));
+				const listener = vi.fn();
+				const disposer = $state.subscribe(listener);
+
+				expect(typeof disposer).toBe("function");
+			});
+
+			it("should call listener immediately with resolved value (T | undefined)", async () => {
+				const $state = stateAwait(Promise.resolve(55));
+				const listener = vi.fn();
+
+				$state.subscribe(listener);
+
+				await vi.waitFor(() => expect(listener).toHaveBeenCalledTimes(1));
+				expect(listener).toHaveBeenLastCalledWith(55);
+			});
+
+			it("should call listener with lazy init value", async () => {
+				const $state = stateAwait(() => Promise.resolve(60));
+				const listener = vi.fn();
+
+				$state.subscribe(listener);
+
+				await vi.waitFor(() => expect(listener).toHaveBeenCalledTimes(1));
+				expect(listener).toHaveBeenLastCalledWith(60);
+			});
+
+			it("should call listener when value is updated", async () => {
+				const $state = stateAwait(Promise.resolve(65));
+				const listener = vi.fn();
+
+				$state.subscribe(listener);
+				await vi.waitFor(() => expect(listener).toHaveBeenCalledTimes(1));
+
+				await $state.set(Promise.resolve(70));
+				await vi.waitFor(() => expect(listener).toHaveBeenCalledTimes(2));
+				expect(listener).toHaveBeenLastCalledWith(70);
+			});
+
+			it("should not call listener when updated with same value", async () => {
+				const $state = stateAwait(Promise.resolve(75));
+				const listener = vi.fn();
+
+				$state.subscribe(listener);
+				await vi.waitFor(() => expect(listener).toHaveBeenCalledTimes(1));
+
+				await $state.set(Promise.resolve(80));
+				await vi.waitFor(() => expect(listener).toHaveBeenCalledTimes(2));
+
+				await $state.set(Promise.resolve(80));
+				await new Promise((resolve) => setTimeout(resolve, 50));
+				expect(listener).toHaveBeenCalledTimes(2);
+			});
+
+			it("should not call listener after disposer is called", async () => {
+				const $state = stateAwait(Promise.resolve(85));
+				const listener = vi.fn();
+
+				const unsubscribe = $state.subscribe(listener);
+				await vi.waitFor(() => expect(listener).toHaveBeenCalledTimes(1));
+
+				unsubscribe();
+
+				await $state.set(Promise.resolve(90));
+				await new Promise((resolve) => setTimeout(resolve, 50));
+				expect(listener).toHaveBeenCalledTimes(1);
+			});
+
+			it("should support multiple subscriptions", async () => {
+				const $state = stateAwait(Promise.resolve(95));
+				const listener1 = vi.fn();
+				const listener2 = vi.fn();
+				const listener3 = vi.fn();
+
+				$state.subscribe(listener1);
+				$state.subscribe(listener2);
+				$state.subscribe(listener3);
+
+				await vi.waitFor(() => {
+					expect(listener1).toHaveBeenCalledTimes(1);
+					expect(listener2).toHaveBeenCalledTimes(1);
+					expect(listener3).toHaveBeenCalledTimes(1);
+				});
+
+				expect(listener1).toHaveBeenLastCalledWith(95);
+				expect(listener2).toHaveBeenLastCalledWith(95);
+				expect(listener3).toHaveBeenLastCalledWith(95);
+			});
+
+			it("should throw error when subscribe is called after disposal", () => {
+				const $state = stateAwait(Promise.resolve(1));
+				const listener = vi.fn();
+
+				$state.dispose();
+
+				expect(() => $state.subscribe(listener)).toThrow(
+					"[PicoFlow] Primitive is disposed",
+				);
+			});
+		});
+
+		describe("trigger", () => {
+			it("should return a Promise", () => {
+				const $state = stateAwait(Promise.resolve(1));
+				const result = $state.trigger();
+				expect(result).toBeInstanceOf(Promise);
+			});
+
+			it("should throw error when trigger is called after disposal", async () => {
+				const $state = stateAwait(Promise.resolve(1));
+				$state.dispose();
+				await expect($state.trigger()).rejects.toThrow(
+					"[PicoFlow] Primitive is disposed",
+				);
+			});
+
+			it("should allow multiple triggers", async () => {
+				const $state = stateAwait(Promise.resolve(1));
+				const promise1 = $state.trigger();
+				const promise2 = $state.trigger();
+				const promise3 = $state.trigger();
+
+				expect(promise1).toBeInstanceOf(Promise);
+				expect(promise2).toBeInstanceOf(Promise);
+				expect(promise3).toBeInstanceOf(Promise);
+
+				await Promise.all([promise1, promise2, promise3]);
+			});
+		});
+
+		describe("disposal", () => {
+			it("should have disposed property set to false initially", () => {
+				const $state = stateAwait(Promise.resolve(1));
+				expect($state.disposed).toBe(false);
+			});
+
+			it("should have disposed property set to true after disposal", () => {
+				const $state = stateAwait(Promise.resolve(1));
+				$state.dispose();
+				expect($state.disposed).toBe(true);
+			});
+
+			it("should throw error when disposed twice", () => {
+				const $state = stateAwait(Promise.resolve(1));
+				$state.dispose();
+				expect(() => $state.dispose()).toThrow(
+					"[PicoFlow] Primitive is disposed",
+				);
+			});
+
+			it("should accept self option without throwing", () => {
+				const $state = stateAwait(Promise.resolve(1));
+				expect(() => $state.dispose({ self: true })).not.toThrow();
+				expect($state.disposed).toBe(true);
+			});
+		});
+
+		describe("special cases", () => {
+			it("should handle lazy initialization with function returning different values", async () => {
+				let counter = 0;
+				const $state = stateAwait(() => {
+					counter++;
+					return Promise.resolve(counter);
+				});
+
+				// First access should return 1
+				await $state.pick();
+				expect(await $state.pick()).toBe(1);
+
+				// Subsequent accesses should return cached value (1), not 2
+				expect(await $state.pick()).toBe(1);
+				expect(await $state.pick()).toBe(1);
+				expect(counter).toBe(1);
+			});
+
+			// TODO: Fix this test
+			it.skip("should handle Promise that rejects", async () => {
+				const error = new Error("Promise rejected");
+				const $state = stateAwait(Promise.reject(error));
+
+				await expect($state.pick()).rejects.toThrow("Promise rejected");
+			});
+
+			// TODO: Fix this test
+			it.skip("should handle lazy function that rejects", async () => {
+				const error = new Error("Lazy function rejected");
+				const $state = stateAwait(() => Promise.reject(error));
+
+				await expect($state.pick()).rejects.toThrow("Lazy function rejected");
+			});
+
+			it("should handle Promise that resolves to null", async () => {
+				const $state = stateAwait(Promise.resolve(null));
+				const value = await $state.pick();
+				expect(value).toBeNull();
+			});
+
+			it("should handle Promise that resolves to undefined", async () => {
+				const $state = stateAwait(Promise.resolve(undefined));
+				const value = await $state.pick();
+				expect(value).toBeUndefined();
+			});
+
+			it("should handle complex object values in Promise", async () => {
+				const complexObj = {
+					nested: { deep: { value: 42 } },
+					array: [1, 2, 3],
+				};
+				const $state = stateAwait(Promise.resolve(complexObj));
+				const value = await $state.pick();
+				expect(value).toBe(complexObj);
+				expect(value.nested.deep.value).toBe(42);
+				expect(value.array).toEqual([1, 2, 3]);
+			});
+		});
+	});
+
+	describe("integration", () => {
+		describe("with effects - get", () => {
+			it("should create reactive dependency when get is used in effect", async () => {
+				const $state = stateAwait(Promise.resolve(100));
+				const effectFn = vi.fn();
+
+				effect((t) => {
+					$state.get(t);
+					effectFn();
+				});
+
+				await vi.waitFor(() => expect(effectFn).toHaveBeenCalledTimes(1));
+			});
+
+			it("should call effect with initial value", async () => {
+				const $state = stateAwait(Promise.resolve(1));
+				const effectFn = vi.fn();
+
+				effect((t) => {
+					const value = $state.get(t);
+					effectFn(value);
+				});
+
+				await vi.waitFor(() => expect(effectFn).toHaveBeenCalledTimes(1));
+				// Value might be undefined on first call if Promise hasn't resolved
+				// Wait a bit more for Promise to resolve
+				await new Promise((resolve) => setTimeout(resolve, 10));
+				// Check that eventually the value is correct
+				const lastCall = effectFn.mock.calls[effectFn.mock.calls.length - 1];
+				expect(lastCall[0]).toBe(1);
+			});
+
+			it("should call effect with lazy init value", async () => {
+				const $state = stateAwait(() => Promise.resolve(1));
+				const effectFn = vi.fn();
+
+				effect((t) => {
+					effectFn($state.get(t));
+				});
+
+				await vi.waitFor(() => expect(effectFn).toHaveBeenCalledTimes(1));
+				expect(effectFn).toHaveBeenLastCalledWith(1);
+			});
+
+			it("should call effect when value is updated", async () => {
+				const $state = stateAwait(Promise.resolve(1));
+				const effectFn = vi.fn();
+
+				effect((t) => {
+					effectFn($state.get(t));
+				});
+
+				await vi.waitFor(() => expect(effectFn).toHaveBeenCalledTimes(1));
+				expect(effectFn).toHaveBeenLastCalledWith(1);
+
+				await $state.set(Promise.resolve(2));
+				await vi.waitFor(() => expect(effectFn).toHaveBeenCalledTimes(2));
+				expect(effectFn).toHaveBeenLastCalledWith(2);
+			});
+
+			it("should not call effect when updated with same value", async () => {
+				const $state = stateAwait(Promise.resolve(1));
+				const effectFn = vi.fn();
+
+				effect((t) => {
+					effectFn($state.get(t));
+				});
+
+				await vi.waitFor(() => expect(effectFn).toHaveBeenCalledTimes(1));
+
+				await $state.set(Promise.resolve(2));
+				await vi.waitFor(() => expect(effectFn).toHaveBeenCalledTimes(2));
+				expect(effectFn).toHaveBeenLastCalledWith(2);
+
+				await $state.set(Promise.resolve(2));
+				await new Promise((resolve) => setTimeout(resolve, 50));
+				expect(effectFn).toHaveBeenCalledTimes(2);
+			});
+
+			it("should not call effect after effect is disposed", async () => {
+				const $state = stateAwait(Promise.resolve(1));
+				const effectFn = vi.fn();
+				const $effect = effect((t) => {
+					effectFn($state.get(t));
+				});
+
+				await vi.waitFor(() => expect(effectFn).toHaveBeenCalledTimes(1));
+				expect(effectFn).toHaveBeenLastCalledWith(1);
+
+				await $state.set(Promise.resolve(2));
+				await vi.waitFor(() => expect(effectFn).toHaveBeenCalledTimes(2));
+				expect(effectFn).toHaveBeenLastCalledWith(2);
+
+				$effect.dispose();
+
+				await $state.set(Promise.resolve(3));
+				await new Promise((resolve) => setTimeout(resolve, 50));
+				expect(effectFn).toHaveBeenCalledTimes(2);
+			});
+
+			it("should support multiple effects depending on same state", async () => {
+				const $state = stateAwait(Promise.resolve(200));
+				const effectFn1 = vi.fn();
+				const effectFn2 = vi.fn();
+				const effectFn3 = vi.fn();
+
+				effect((t) => {
+					$state.get(t);
+					effectFn1();
+				});
+
+				effect((t) => {
+					$state.get(t);
+					effectFn2();
+				});
+
+				effect((t) => {
+					$state.get(t);
+					effectFn3();
+				});
+
+				await vi.waitFor(() => {
+					expect(effectFn1).toHaveBeenCalledTimes(1);
+					expect(effectFn2).toHaveBeenCalledTimes(1);
+					expect(effectFn3).toHaveBeenCalledTimes(1);
+				});
+
+				await $state.set(Promise.resolve(300));
+				await vi.waitFor(() => {
+					expect(effectFn1).toHaveBeenCalledTimes(2);
+					expect(effectFn2).toHaveBeenCalledTimes(2);
+					expect(effectFn3).toHaveBeenCalledTimes(2);
+				});
+			});
+
+			it("should support effect depending on multiple states", async () => {
+				const $stateA = stateAwait(Promise.resolve(5));
+				const $stateB = stateAwait(Promise.resolve(10));
+				const effectFn = vi.fn();
+
+				effect((t) => {
+					const a = $stateA.get(t);
+					const b = $stateB.get(t);
+					effectFn(a, b);
+				});
+
+				await vi.waitFor(() => expect(effectFn).toHaveBeenCalledTimes(1));
+				// Wait for Promises to resolve
+				await new Promise((resolve) => setTimeout(resolve, 10));
+				const lastCall = effectFn.mock.calls[effectFn.mock.calls.length - 1];
+				expect(lastCall[0]).toBe(5);
+				expect(lastCall[1]).toBe(10);
+
+				await $stateA.set(Promise.resolve(6));
+				await vi.waitFor(() => expect(effectFn).toHaveBeenCalledTimes(2));
+				expect(effectFn).toHaveBeenLastCalledWith(6, 10);
+
+				await $stateB.set(Promise.resolve(11));
+				await vi.waitFor(() => expect(effectFn).toHaveBeenCalledTimes(3));
+				expect(effectFn).toHaveBeenLastCalledWith(6, 11);
+			});
+		});
+
+		describe("with effects - watch", () => {
+			it("should register dependency when watch is used in effect", async () => {
+				const $state = stateAwait(Promise.resolve(400));
+				const effectFn = vi.fn();
+
+				effect((t) => {
+					$state.watch(t);
+					effectFn();
+				});
+
+				await vi.waitFor(() => expect(effectFn).toHaveBeenCalledTimes(1));
+			});
+
+			it("should trigger re-runs when state changes after watch", async () => {
+				const $state = stateAwait(Promise.resolve(500));
+				const effectFn = vi.fn();
+
+				effect((t) => {
+					$state.watch(t);
+					effectFn();
+				});
+
+				await vi.waitFor(() => expect(effectFn).toHaveBeenCalledTimes(1));
+
+				await $state.set(Promise.resolve(600));
+				await vi.waitFor(() => expect(effectFn).toHaveBeenCalledTimes(2));
+			});
+		});
+
+		describe("with effects - subscribe", () => {
+			it("should call listener immediately when subscribe is used", async () => {
+				const $state = stateAwait(Promise.resolve(700));
+				const listener = vi.fn();
+
+				$state.subscribe(listener);
+
+				await vi.waitFor(() => expect(listener).toHaveBeenCalledTimes(1));
+				expect(listener).toHaveBeenLastCalledWith(700);
+			});
+
+			it("should support multiple subscriptions with effects", async () => {
+				const $state = stateAwait(Promise.resolve(800));
+				const listener1 = vi.fn();
+				const listener2 = vi.fn();
+
+				$state.subscribe(listener1);
+				$state.subscribe(listener2);
+
+				await vi.waitFor(() => {
+					expect(listener1).toHaveBeenCalledTimes(1);
+					expect(listener2).toHaveBeenCalledTimes(1);
+				});
+
+				await $state.set(Promise.resolve(900));
+				await vi.waitFor(() => {
+					expect(listener1).toHaveBeenCalledTimes(2);
+					expect(listener2).toHaveBeenCalledTimes(2);
+				});
+			});
+
+			it("should dispose effect when disposer is called", async () => {
+				const $state = stateAwait(Promise.resolve(1000));
+				const listener = vi.fn();
+
+				const unsubscribe = $state.subscribe(listener);
+				await vi.waitFor(() => expect(listener).toHaveBeenCalledTimes(1));
+
+				await $state.set(Promise.resolve(1100));
+				await vi.waitFor(() => expect(listener).toHaveBeenCalledTimes(2));
+
+				unsubscribe();
+
+				await $state.set(Promise.resolve(1200));
+				await new Promise((resolve) => setTimeout(resolve, 50));
+				expect(listener).toHaveBeenCalledTimes(2);
+			});
+		});
+
+		describe("with effects - disposal", () => {
+			it("should dispose effects when state is disposed", async () => {
+				const $state = stateAwait(Promise.resolve(1300));
+				const effectFn = vi.fn();
+				const $effect = effect((t) => {
+					$state.get(t);
+					effectFn();
+				});
+
+				await vi.waitFor(() => expect(effectFn).toHaveBeenCalledTimes(1));
+				expect($effect.disposed).toBe(false);
+
+				$state.dispose();
+
+				expect($effect.disposed).toBe(true);
+			});
+
+			it("should not dispose effects when state is disposed with self option", async () => {
+				const $state = stateAwait(Promise.resolve(1400));
+				const effectFn = vi.fn();
+				const $effect = effect((t) => {
+					$state.get(t);
+					effectFn();
+				});
+
+				await vi.waitFor(() => expect(effectFn).toHaveBeenCalledTimes(1));
+				expect($effect.disposed).toBe(false);
+
+				$state.dispose({ self: true });
+
+				expect($effect.disposed).toBe(false);
+				expect($state.disposed).toBe(true);
+			});
+
+			it("should unregister effects when state is disposed with self option", async () => {
+				const $state = stateAwait(Promise.resolve(1500));
+				const effectFn = vi.fn();
+				const $effect = effect((t) => {
+					$state.get(t);
+					effectFn();
+				});
+
+				await vi.waitFor(() => expect(effectFn).toHaveBeenCalledTimes(1));
+
+				$state.dispose({ self: true });
+
+				// Effect should still be active but not notified
+				expect($effect.disposed).toBe(false);
+
+				// But state is disposed so operations should fail
+				const $tracker = state(0);
+				expect(() => $state.get($tracker)).toThrow(
+					"[PicoFlow] Primitive is disposed",
+				);
+			});
+		});
+
+		describe("complex cases", () => {
+			it("should handle nested states in effects", async () => {
+				const $stateA = stateAwait(Promise.resolve(1));
+				const $stateB = stateAwait(Promise.resolve(2));
+				const $stateC = stateAwait(
+					Promise.resolve({
+						a: $stateA,
+						b: $stateB,
+					}),
+				);
+
+				const effectFn = vi.fn();
+				effect((t) => {
+					const c = $stateC.get(t);
+					if (c) {
+						const a = c.a.get(t);
+						if (a !== undefined) {
+							effectFn(a);
+						}
+					}
+				});
+
+				await vi.waitFor(() => expect(effectFn).toHaveBeenCalledTimes(1));
+				// Wait for Promises to resolve
+				await new Promise((resolve) => setTimeout(resolve, 10));
+				const lastCall = effectFn.mock.calls[effectFn.mock.calls.length - 1];
+				expect(lastCall[0]).toBe(1);
+
+				await $stateA.set(Promise.resolve(10));
+				await vi.waitFor(() => expect(effectFn).toHaveBeenCalledTimes(2));
+				expect(effectFn).toHaveBeenLastCalledWith(10);
+			});
+
+			it("should handle effect depending on multiple states", async () => {
+				const $stateA = stateAwait(Promise.resolve(10));
+				const $stateB = stateAwait(Promise.resolve(20));
+				const $stateC = stateAwait(Promise.resolve(30));
+				const effectFn = vi.fn();
+
+				effect((t) => {
+					const a = $stateA.get(t);
+					const b = $stateB.get(t);
+					const c = $stateC.get(t);
+					if (a !== undefined && b !== undefined && c !== undefined) {
+						effectFn(a, b, c);
+					}
+				});
+
+				await vi.waitFor(() => expect(effectFn).toHaveBeenCalledTimes(1));
+				// Wait for Promises to resolve
+				await new Promise((resolve) => setTimeout(resolve, 10));
+				const lastCall = effectFn.mock.calls[effectFn.mock.calls.length - 1];
+				expect(lastCall[0]).toBe(10);
+				expect(lastCall[1]).toBe(20);
+				expect(lastCall[2]).toBe(30);
+
+				await $stateA.set(Promise.resolve(11));
+				await vi.waitFor(() => expect(effectFn).toHaveBeenCalledTimes(2));
+				expect(effectFn).toHaveBeenLastCalledWith(11, 20, 30);
+
+				await $stateB.set(Promise.resolve(21));
+				await vi.waitFor(() => expect(effectFn).toHaveBeenCalledTimes(3));
+				expect(effectFn).toHaveBeenLastCalledWith(11, 21, 30);
+			});
+
+			it("should handle state chaining (state A -> state B -> state C)", async () => {
+				const $stateA = stateAwait(Promise.resolve(100));
+				const $stateB = stateAwait(Promise.resolve(200));
+				const $stateC = stateAwait(Promise.resolve(300));
+				const effectFn = vi.fn();
+
+				// Effect watches A, which triggers B, which triggers C
+				effect((t) => {
+					const a = $stateA.get(t);
+					if (a !== undefined && a > 150) {
+						$stateB.set(Promise.resolve(250));
+					}
+				});
+
+				effect((t) => {
+					const b = $stateB.get(t);
+					if (b !== undefined && b > 220) {
+						$stateC.set(Promise.resolve(350));
+					}
+				});
+
+				effect((t) => {
+					const c = $stateC.get(t);
+					if (c !== undefined) {
+						effectFn(c);
+					}
+				});
+
+				await vi.waitFor(() => expect(effectFn).toHaveBeenCalledTimes(1));
+
+				await $stateA.set(Promise.resolve(200));
+				await vi.waitFor(() => expect(effectFn).toHaveBeenCalledTimes(2));
+			});
+		});
+	});
+});