fix(core): derived atom is not re-evaluated if it read between multip…

…le set (#2960)

* add failing test

* fix it

src/vanilla/store.ts

@@ -434,10 +434,8 @@ const buildStore = (...storeArgs: StoreArgs): Store => {
     while (stack.length) {
       const aState = stack.pop()!
       for (const [d, s] of getMountedOrPendingDependents(aState)) {
-        if (!invalidatedAtoms.has(d)) {
-          invalidatedAtoms.set(d, s.n)
-          stack.push(s)
-        }
+        invalidatedAtoms.set(d, s.n)
+        stack.push(s)
       }
     }
   }

tests/vanilla/store.test.tsx

@@ -1209,3 +1209,17 @@ it('allows subcribing to atoms during mount', () => {
   store.sub(a, () => {})
   expect(bMounted).toBe(true)
 })
+
+it('updates with reading derived atoms (#2959)', () => {
+  const store = createStore()
+  const countAtom = atom(0)
+  const countDerivedAtom = atom((get) => get(countAtom))
+  const countUpAtom = atom(null, (get, set) => {
+    set(countAtom, 1)
+    get(countDerivedAtom)
+    set(countAtom, 2)
+  })
+  store.sub(countDerivedAtom, () => {})
+  store.set(countUpAtom)
+  expect(store.get(countDerivedAtom)).toBe(2)
+})