🤖 tests: wait for subscription in forkWorkspace streams

Change-Id: I1a869f9aea822126fdf7098161d5f5198fe12ee2
Signed-off-by: Thomas Kosiewski <tk@coder.com>

Author: ThomasK33

tests/ipc/forkWorkspace.test.ts

@@ -94,6 +94,8 @@ describeIntegration("Workspace fork", () => {
         // User expects: forked workspace is functional - can send messages to it
         const collector = createStreamCollector(env.orpc, forkedWorkspaceId);
         collector.start();
+        await collector.waitForSubscription();
+
         const sendResult = await sendMessageWithModel(
           env,
           forkedWorkspaceId,
@@ -149,6 +151,8 @@ describeIntegration("Workspace fork", () => {
         // Send a message that requires the historical context
         const collector = createStreamCollector(env.orpc, forkedWorkspaceId);
         collector.start();
+        await collector.waitForSubscription();
+
         const sendResult = await sendMessageWithModel(
           env,
           forkedWorkspaceId,
@@ -202,6 +206,10 @@ describeIntegration("Workspace fork", () => {
         const forkedCollector = createStreamCollector(env.orpc, forkedWorkspaceId);
         sourceCollector.start();
         forkedCollector.start();
+        await Promise.all([
+          sourceCollector.waitForSubscription(),
+          forkedCollector.waitForSubscription(),
+        ]);
 
         // Send different messages to both concurrently
         const [sourceResult, forkedResult] = await Promise.all([
@@ -251,6 +259,7 @@ describeIntegration("Workspace fork", () => {
         // Start collector before starting stream
         const sourceCollector = createStreamCollector(env.orpc, sourceWorkspaceId);
         sourceCollector.start();
+        await sourceCollector.waitForSubscription();
 
         // Start a stream in the source workspace (don't await)
         void sendMessageWithModel(
@@ -284,6 +293,8 @@ describeIntegration("Workspace fork", () => {
         // Send a message to the forked workspace
         const forkedCollector = createStreamCollector(env.orpc, forkedWorkspaceId);
         forkedCollector.start();
+        await forkedCollector.waitForSubscription();
+
         const forkedSendResult = await sendMessageWithModel(
           env,
           forkedWorkspaceId,