@ersbeth/picoflow 0.2.4 → 1.0.0

package/test/state.test.ts

@@ -2,193 +2,211 @@ import { describe, expect, test, vi } from "vitest";
 import { effect, state } from "#package";
 
 describe("state", () => {
-    test("is initialized", () => {
-        const $state = state(1);
-        expect($state.get()).toBe(1);
-    });
+	test("is initialized", () => {
+		const $state = state(1);
+		expect($state.pick()).toBe(1);
+	});
 
-    test("is lazy initialized", () => {
-        const $state = state(() => 1);
-        expect($state.get()).toBe(1);
-    });
+	test("is lazy initialized", () => {
+		const $state = state(() => 1);
+		expect($state.pick()).toBe(1);
+	});
 
-    test("is updated", () => {
-        const $state = state(1);
-        expect($state.get()).toBe(1);
+	test("is updated", () => {
+		const $state = state(1);
+		expect($state.pick()).toBe(1);
 
-        $state.set(2);
-        expect($state.get()).toBe(2);
-    });
+		$state.set(2);
+		expect($state.pick()).toBe(2);
+	});
 
-    test("is updated from current", () => {
-        const $state = state(1);
-        expect($state.get()).toBe(1);
+	test("is updated from current", () => {
+		const $state = state(1);
+		expect($state.pick()).toBe(1);
 
-        $state.set((state) => state + 1);
-        expect($state.get()).toBe(2);
-    });
+		$state.set((state) => state + 1);
+		expect($state.pick()).toBe(2);
+	});
 
-    test("set throws when disposed", () => {
-        const $state = state(1);
+	test("set throws when disposed", () => {
+		const $state = state(1);
 
-        expect($state.get()).toBe(1);
+		expect($state.pick()).toBe(1);
 
-        $state.set(2);
-        expect($state.get()).toBe(2);
+		$state.set(2);
+		expect($state.pick()).toBe(2);
 
-        $state.dispose();
+		$state.dispose();
 
-        expect(() => $state.set(3)).toThrow("[PicoFlow] Primitive is disposed");
-    });
+		expect(() => $state.set(3)).toThrow("[PicoFlow] Primitive is disposed");
+	});
 
-    test("get throws when disposed", () => {
-        const $state = state(1);
+	test("get throws when disposed", () => {
+		const $state = state(1);
 
-        expect($state.get()).toBe(1);
+		expect($state.pick()).toBe(1);
 
-        $state.set(2);
-        expect($state.get()).toBe(2);
+		$state.set(2);
+		expect($state.pick()).toBe(2);
 
-        $state.dispose();
+		$state.dispose();
 
-        expect(() => $state.get()).toThrow("[PicoFlow] Primitive is disposed");
-    });
+		expect(() => $state.pick()).toThrow("[PicoFlow] Primitive is disposed");
+	});
 });
 
 describe("effect", () => {
-    test("called with init value", () => {
-        const $state = state(1);
-        const effectFn = vi.fn();
-        effect((get) => effectFn(get($state)));
-
-        expect(effectFn).toHaveBeenCalledTimes(1);
-        expect(effectFn).toHaveBeenLastCalledWith(1);
-    });
-
-    test("called with lazy init value", () => {
-        const $state = state(() => 1);
-        const effectFn = vi.fn();
-        effect((get) => effectFn(get($state)));
-
-        expect(effectFn).toHaveBeenCalledTimes(1);
-        expect(effectFn).toHaveBeenLastCalledWith(1);
-    });
-
-    test("called when updated", () => {
-        const $state = state(1);
-        const effectFn = vi.fn();
-        effect((get) => effectFn(get($state)));
-
-        expect(effectFn).toHaveBeenCalledTimes(1);
-        expect(effectFn).toHaveBeenLastCalledWith(1);
-
-        $state.set(2);
-        expect(effectFn).toHaveBeenCalledTimes(2);
-        expect(effectFn).toHaveBeenLastCalledWith(2);
-    });
-
-    test("NOT called when updated with the same value", () => {
-        const $state = state(1);
-        const effectFn = vi.fn();
-        effect((get) => effectFn(get($state)));
-
-        expect(effectFn).toHaveBeenCalledTimes(1);
-        expect(effectFn).toHaveBeenLastCalledWith(1);
-
-        $state.set(2);
-        expect(effectFn).toHaveBeenCalledTimes(2);
-        expect(effectFn).toHaveBeenLastCalledWith(2);
-
-        $state.set(2);
-        expect(effectFn).toHaveBeenCalledTimes(2);
-        expect(effectFn).toHaveBeenLastCalledWith(2);
-    });
-
-    test("NOT called when disposed", () => {
-        const $state = state(1);
-        const effectFn = vi.fn();
-        const $effect = effect((get) => effectFn(get($state)));
-
-        expect(effectFn).toHaveBeenCalledTimes(1);
-        expect(effectFn).toHaveBeenLastCalledWith(1);
-
-        $state.set(2);
-        expect(effectFn).toHaveBeenCalledTimes(2);
-        expect(effectFn).toHaveBeenLastCalledWith(2);
-
-        $effect.dispose();
-
-        $state.set(3);
-        expect(effectFn).toHaveBeenCalledTimes(2);
-        expect(effectFn).toHaveBeenLastCalledWith(2);
-    });
+	test("called with init value", () => {
+		const $state = state(1);
+		const effectFn = vi.fn();
+		effect((t) => effectFn($state.get(t)));
+
+		expect(effectFn).toHaveBeenCalledTimes(1);
+		expect(effectFn).toHaveBeenLastCalledWith(1);
+	});
+
+	test("called with lazy init value", () => {
+		const $state = state(() => 1);
+		const effectFn = vi.fn();
+		effect((t) => effectFn($state.get(t)));
+
+		expect(effectFn).toHaveBeenCalledTimes(1);
+		expect(effectFn).toHaveBeenLastCalledWith(1);
+	});
+
+	test("called when updated", () => {
+		const $state = state(1);
+		const effectFn = vi.fn();
+		effect((t) => effectFn($state.get(t)));
+
+		expect(effectFn).toHaveBeenCalledTimes(1);
+		expect(effectFn).toHaveBeenLastCalledWith(1);
+
+		$state.set(2);
+		expect(effectFn).toHaveBeenCalledTimes(2);
+		expect(effectFn).toHaveBeenLastCalledWith(2);
+	});
+
+	test("NOT called when updated with the same value", () => {
+		const $state = state(1);
+		const effectFn = vi.fn();
+		effect((t) => effectFn($state.get(t)));
+
+		expect(effectFn).toHaveBeenCalledTimes(1);
+		expect(effectFn).toHaveBeenLastCalledWith(1);
+
+		$state.set(2);
+		expect(effectFn).toHaveBeenCalledTimes(2);
+		expect(effectFn).toHaveBeenLastCalledWith(2);
+
+		$state.set(2);
+		expect(effectFn).toHaveBeenCalledTimes(2);
+		expect(effectFn).toHaveBeenLastCalledWith(2);
+	});
+
+	test("NOT called when disposed", () => {
+		const $state = state(1);
+		const effectFn = vi.fn();
+		const $effect = effect((t) => effectFn($state.get(t)));
+
+		expect(effectFn).toHaveBeenCalledTimes(1);
+		expect(effectFn).toHaveBeenLastCalledWith(1);
+
+		$state.set(2);
+		expect(effectFn).toHaveBeenCalledTimes(2);
+		expect(effectFn).toHaveBeenLastCalledWith(2);
+
+		$effect.dispose();
+
+		$state.set(3);
+		expect(effectFn).toHaveBeenCalledTimes(2);
+		expect(effectFn).toHaveBeenLastCalledWith(2);
+	});
+
+	test("nested states", () => {
+		const $stateA = state(1);
+		const $stateB = state(2);
+		const $stateC = state({
+			a: $stateA,
+			b: $stateB,
+		});
+
+		const effectFn = vi.fn();
+		effect((t) => {
+			const a = $stateC.get(t).a.get(t);
+			effectFn(a);
+		});
+
+		expect(effectFn).toHaveBeenCalledTimes(1);
+		expect(effectFn).toHaveBeenLastCalledWith(1);
+	});
 });
 
 describe("subscribe", () => {
-    test("called with init value", () => {
-        const $state = state(1);
-        const effectFn = vi.fn();
-        $state.subscribe((value) => effectFn(value));
-
-        expect(effectFn).toHaveBeenCalledTimes(1);
-        expect(effectFn).toHaveBeenLastCalledWith(1);
-    });
-
-    test("called with lazy init value", () => {
-        const $state = state(() => 1);
-        const effectFn = vi.fn();
-        $state.subscribe((value) => effectFn(value));
-
-        expect(effectFn).toHaveBeenCalledTimes(1);
-        expect(effectFn).toHaveBeenLastCalledWith(1);
-    });
-
-    test("called when updated", () => {
-        const $state = state(1);
-        const effectFn = vi.fn();
-        $state.subscribe((value) => effectFn(value));
-
-        expect(effectFn).toHaveBeenCalledTimes(1);
-        expect(effectFn).toHaveBeenLastCalledWith(1);
-
-        $state.set(2);
-        expect(effectFn).toHaveBeenCalledTimes(2);
-        expect(effectFn).toHaveBeenLastCalledWith(2);
-    });
-
-    test("NOT called when updated with the same value", () => {
-        const $state = state(1);
-        const effectFn = vi.fn();
-        $state.subscribe((value) => effectFn(value));
-
-        expect(effectFn).toHaveBeenCalledTimes(1);
-        expect(effectFn).toHaveBeenLastCalledWith(1);
-
-        $state.set(2);
-        expect(effectFn).toHaveBeenCalledTimes(2);
-        expect(effectFn).toHaveBeenLastCalledWith(2);
-
-        $state.set(2);
-        expect(effectFn).toHaveBeenCalledTimes(2);
-        expect(effectFn).toHaveBeenLastCalledWith(2);
-    });
-
-    test("NOT called when disposed", () => {
-        const $state = state(1);
-        const effectFn = vi.fn();
-        const unsubscribe = $state.subscribe((value) => effectFn(value));
-
-        expect(effectFn).toHaveBeenCalledTimes(1);
-        expect(effectFn).toHaveBeenLastCalledWith(1);
-
-        $state.set(2);
-        expect(effectFn).toHaveBeenCalledTimes(2);
-        expect(effectFn).toHaveBeenLastCalledWith(2);
-
-        unsubscribe();
-
-        $state.set(3);
-        expect(effectFn).toHaveBeenCalledTimes(2);
-        expect(effectFn).toHaveBeenLastCalledWith(2);
-    });
+	test("called with init value", () => {
+		const $state = state(1);
+		const effectFn = vi.fn();
+		$state.subscribe((value) => effectFn(value));
+
+		expect(effectFn).toHaveBeenCalledTimes(1);
+		expect(effectFn).toHaveBeenLastCalledWith(1);
+	});
+
+	test("called with lazy init value", () => {
+		const $state = state(() => 1);
+		const effectFn = vi.fn();
+		$state.subscribe((value) => effectFn(value));
+
+		expect(effectFn).toHaveBeenCalledTimes(1);
+		expect(effectFn).toHaveBeenLastCalledWith(1);
+	});
+
+	test("called when updated", () => {
+		const $state = state(1);
+		const effectFn = vi.fn();
+		$state.subscribe((value) => effectFn(value));
+
+		expect(effectFn).toHaveBeenCalledTimes(1);
+		expect(effectFn).toHaveBeenLastCalledWith(1);
+
+		$state.set(2);
+		expect(effectFn).toHaveBeenCalledTimes(2);
+		expect(effectFn).toHaveBeenLastCalledWith(2);
+	});
+
+	test("NOT called when updated with the same value", () => {
+		const $state = state(1);
+		const effectFn = vi.fn();
+		$state.subscribe((value) => effectFn(value));
+
+		expect(effectFn).toHaveBeenCalledTimes(1);
+		expect(effectFn).toHaveBeenLastCalledWith(1);
+
+		$state.set(2);
+		expect(effectFn).toHaveBeenCalledTimes(2);
+		expect(effectFn).toHaveBeenLastCalledWith(2);
+
+		$state.set(2);
+		expect(effectFn).toHaveBeenCalledTimes(2);
+		expect(effectFn).toHaveBeenLastCalledWith(2);
+	});
+
+	test("NOT called when disposed", () => {
+		const $state = state(1);
+		const effectFn = vi.fn();
+		const unsubscribe = $state.subscribe((value) => effectFn(value));
+
+		expect(effectFn).toHaveBeenCalledTimes(1);
+		expect(effectFn).toHaveBeenLastCalledWith(1);
+
+		$state.set(2);
+		expect(effectFn).toHaveBeenCalledTimes(2);
+		expect(effectFn).toHaveBeenLastCalledWith(2);
+
+		unsubscribe();
+
+		$state.set(3);
+		expect(effectFn).toHaveBeenCalledTimes(2);
+		expect(effectFn).toHaveBeenLastCalledWith(2);
+	});
 });