fix: missing updates for reconnecting stores

src/lib/calculatedValue.ts

@@ -78,14 +78,16 @@ export function calculatedValue<T>(store: Store<T>, notify: () => void): Calcula
 
   async function connect(createConnection: Connection<T>) {
     if (!active) {
+      connection = { active: false };
       return;
     }
 
     const actions: AsyncConnectionActions<any> = {
-      set(value) {
+      set(_value) {
         connection?.active &&
           q(() => {
-            store.set(value);
+            value = _value;
+            notify();
           });
       },
       updateValue(update) {
@@ -147,6 +149,8 @@ export function calculatedValue<T>(store: Store<T>, notify: () => void): Calcula
 
   if (value instanceof Promise) {
     value.finally(() => whenExecuted.resolve()).catch(() => undefined);
+  } else {
+    whenExecuted.resolve();
   }
 
   function check() {
@@ -164,17 +168,13 @@ export function calculatedValue<T>(store: Store<T>, notify: () => void): Calcula
 
   function stop() {
     cancelEffect();
+    whenExecuted.resolve();
+    whenConnected.resolve();
 
     if (connection) {
       connection.active = false;
       connection.cancel?.();
       q.clear();
-
-      whenConnected.reject();
-      whenExecuted.reject();
-    } else {
-      whenConnected.resolve();
-      whenExecuted.resolve();
     }
   }
 

test/core/store.test.ts

@@ -1,6 +1,7 @@
 import { afterEach, beforeEach, describe, expect, test, vi } from 'vitest';
-import { createStore } from '../../src';
+import { createStore, set } from '../../src';
 import { defaultEqual, shallowEqual } from '../../src/lib/equals';
+import { flushPromises } from '../testHelpers';
 
 beforeEach(() => {
   vi.useFakeTimers();
@@ -331,4 +332,22 @@ describe('static store', () => {
     expect(reaction1).toHaveBeenCalledTimes(2);
     expect(reaction2).toHaveBeenCalledTimes(2);
   });
+
+  test('connect', async () => {
+    const state = createStore(({ connect }) => {
+      connect(({ set }) => {
+        set(2);
+        return () => undefined;
+      });
+
+      return 1;
+    });
+
+    expect(state.get()).toBe(1);
+
+    state.subscribe(() => undefined);
+    await flushPromises();
+
+    expect(state.get()).toBe(2);
+  });
 });