react: schedule transport close as microtask to avoid strict mode issues

Changed `TransportProvider` to schedule the transport close as a microtask on unmount, ensuring React's Strict Mode remount cycle cancels the close during development. Updated corresponding test to verify behavior.

Author: ttypic

src/react/contexts/transport-provider.tsx

@@ -7,9 +7,10 @@
  * to get the stable channel reference and creates the transport once on
  * first render (via useRef).
  *
- * The transport is closed synchronously when the provider unmounts (via
- * useLayoutEffect) so that any in-progress operations are aborted before
- * the channel is detached by ChannelProvider.
+ * The transport is closed when the provider truly unmounts. The close is
+ * scheduled as a microtask so that React Strict Mode's synchronous
+ * remount cycle (mount → fake-unmount → remount) can cancel it before it
+ * fires, avoiding unnecessary transport teardown in development.
  *
  * Multiple TransportProviders can be nested using distinct channelNames.
  * Each provider merges its transport into the parent record, so descendants
@@ -45,6 +46,8 @@ const TransportProviderInner = <TEvent, TMessage>({
   const transportRef = useRef<ClientTransport<TEvent, TMessage> | undefined>(undefined);
   const transportChannelRef = useRef<string>(channelName);
   const transportsToDisposeRef = useRef<ClientTransport<unknown, unknown>[]>([]);
+  const pendingCloseRef = useRef(false);
+
 
   if (!transportRef.current || transportChannelRef.current !== channelName) {
     transportChannelRef.current = channelName;
@@ -70,15 +73,24 @@ const TransportProviderInner = <TEvent, TMessage>({
     },
     [channelName],
   );
-
-  // Synchronously clear the ref on unmount so stale consumers can't call the closed transport.
-  useLayoutEffect(
-    () => () => {
-      void transportRef.current?.close();
-      transportRef.current = undefined;
-    },
-    [],
-  );
+  
+  // Close the transport when the component truly unmounts. The close is
+  // scheduled as a microtask: in React Strict Mode (dev) the component
+  // remounts synchronously before any microtask can drain, so the remount's
+  // effect setup resets pendingCloseRef.current = false and cancels the
+  // close. On a real unmount no remount follows, the microtask fires, and
+  // the transport is closed.
+  useEffect(() => {
+    pendingCloseRef.current = false;
+    return () => {
+      pendingCloseRef.current = true;
+      void Promise.resolve().then(() => {
+        if (pendingCloseRef.current) {
+          void transportRef.current?.close();
+        }
+      });
+    }
+  }, []);
 
   return (
     <TransportContext.Provider value={contextValue}>

test/react/providers/transport-provider.test.ts

@@ -4,6 +4,16 @@ import { act, renderHook } from '@testing-library/react';
 import { createElement, type ReactNode } from 'react';
 import { beforeEach, describe, expect, it, vi } from 'vitest';
 
+/** Flush microtasks (but NOT macrotasks) so deferred promises resolve. */
+const flushMicrotasks = async (): Promise<void> => {
+  await new Promise<void>((resolve) => {
+    queueMicrotask(resolve);
+  });
+  await new Promise<void>((resolve) => {
+    queueMicrotask(resolve);
+  });
+};
+
 import type { ClientTransport } from '../../../src/core/transport/types.js';
 import { TransportProvider } from '../../../src/react/contexts/transport-provider.js';
 import { useClientTransport } from '../../../src/react/use-client-transport.js';
@@ -113,6 +123,21 @@ describe('TransportProvider', () => {
     expect(outerResult.current).not.toBe(innerResult.current);
   });
 
+  it('closes the transport when the provider unmounts', async () => {
+    const created: ReturnType<typeof createMockTransport>[] = [];
+    createClientTransportMock.mockImplementation(() => {
+      const mock = createMockTransport();
+      created.push(mock);
+      return mock.transport;
+    });
+
+    const { unmount } = renderHook(() => useClientTransport({ channelName: 'ai:test' }), { wrapper: wrapDefault });
+    unmount();
+    await flushMicrotasks();
+
+    expect(created[0]?.close).toHaveBeenCalledOnce();
+  });
+
   it('forwards transport options to createClientTransport', () => {
     const logger = {
       trace: vi.fn(),