add tests: atoms accessed with write get should have their dependenci…

…es updated

tests/vanilla/dependency.test.tsx

@@ -1,5 +1,5 @@
 import { expect, it, vi } from 'vitest'
-import { atom, createStore } from 'jotai/vanilla'
+import { atom, createStore, type Getter } from 'jotai/vanilla'
 
 it('can propagate updates with async atom chains', async () => {
   const store = createStore()
@@ -334,3 +334,30 @@ it('handles complex dependency chains', async () => {
   resolve()
   expect(await promise2).toBe(10)
 })
+
+it('updates dependencies for atoms accessed with write get', () => {
+  const store = createStore()
+  const a = atom(0)
+  const bRead = vi.fn((get: Getter) => get(a))
+  const b = atom(bRead)
+  const c = atom(null, (get) => get(b))
+  store.sub(a, () => {})
+  store.set(c)
+  bRead.mockClear()
+  store.set(a, 1)
+  // note: this behavior is required for the next test to work
+  expect(bRead).toHaveBeenCalledOnce()
+  expect(store.get(a)).toBe(1)
+})
+
+it('can read sync derived atom in write without initializing', () => {
+  const store = createStore()
+  const a = atom(0)
+  const b = atom((get) => get(a) + 1)
+  const c = atom(null, (get, set) => set(a, get(b)))
+  store.set(c)
+  expect(store.get(a)).toBe(1)
+  store.set(c)
+  // note: this is why write get needs to update deps
+  expect(store.get(a)).toBe(2)
+})