Merge branch 'feat/extend' of https://github.com/udecode/jotai-x into…

… feat/extend
udecode · Dec 21, 2023 · d278211 · d278211
2 parents 56dcfc2 + 301380c
commit d278211
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/packages/jotai-x/src/createAtomStore.spec.tsx b/packages/jotai-x/src/createAtomStore.spec.tsx
@@ -342,6 +342,16 @@ describe('createAtomStore', () => {
       expect(result.current).toBe('Jane is 98 years old');
     });
 
+    it('includes extended atom in set hooks', () => {
+      const { result } = renderHook(() => Object.keys(useUserStore().set));
+      expect(result.current).toContain('bio');
+    });
+
+    it('includes extended atom in use hooks', () => {
+      const { result } = renderHook(() => Object.keys(useUserStore().use));
+      expect(result.current).toContain('bio');
+    });
+
     it('computes extended atom based on current state', () => {
       const { getByText } = render(
         <UserProvider name="John" age={42}>