Add sendAndWait for all languages

Author: SteveSandersonMS

dotnet/test/SessionTests.cs

@@ -329,15 +329,16 @@ public async Task Send_Returns_Immediately_While_Events_Stream_In_Background()
 
         session.On(evt => events.Add(evt.Type));
 
-        await session.SendAsync(new MessageOptions { Prompt = "What is 1+1?" });
+        // Use a slow command so we can verify SendAsync() returns before completion
+        await session.SendAsync(new MessageOptions { Prompt = "Run 'sleep 2 && echo done'" });
 
-        // send() should return before turn completes (no session.idle yet)
+        // SendAsync() should return before turn completes (no session.idle yet)
         Assert.DoesNotContain("session.idle", events);
 
         // Wait for turn to complete
         var message = await TestHelper.GetFinalAssistantMessageAsync(session);
 
-        Assert.Contains("2", message?.Data.Content ?? string.Empty);
+        Assert.Contains("done", message?.Data.Content ?? string.Empty);
         Assert.Contains("session.idle", events);
         Assert.Contains("assistant.message", events);
     }
@@ -364,8 +365,9 @@ public async Task SendAndWait_Throws_On_Timeout()
     {
         var session = await Client.CreateSessionAsync();
 
+        // Use a slow command to ensure timeout triggers before completion
         var ex = await Assert.ThrowsAsync<TimeoutException>(() =>
-            session.SendAndWaitAsync(new MessageOptions { Prompt = "What is 3+3?" }, TimeSpan.FromMilliseconds(1)));
+            session.SendAndWaitAsync(new MessageOptions { Prompt = "Run 'sleep 2 && echo done'" }, TimeSpan.FromMilliseconds(100)));
 
         Assert.Contains("timed out", ex.Message);
     }

go/session.go

@@ -152,17 +152,18 @@ func (s *Session) SendAndWait(options MessageOptions, timeout time.Duration) (*S
 	var mu sync.Mutex
 
 	unsubscribe := s.On(func(event SessionEvent) {
-		if event.Type == generated.AssistantMessage {
+		switch event.Type {
+		case generated.AssistantMessage:
 			mu.Lock()
 			eventCopy := event
 			lastAssistantMessage = &eventCopy
 			mu.Unlock()
-		} else if event.Type == generated.SessionIdle {
+		case generated.SessionIdle:
 			select {
 			case idleCh <- struct{}{}:
 			default:
 			}
-		} else if event.Type == generated.SessionError {
+		case generated.SessionError:
 			errMsg := "session error"
 			if event.Data.Message != nil {
 				errMsg = *event.Data.Message

nodejs/test/e2e/session.test.ts

@@ -334,15 +334,16 @@ describe("Send Blocking Behavior", async () => {
             events.push(event.type);
         });
 
-        await session.send({ prompt: "What is 1+1?" });
+        // Use a slow command so we can verify send() returns before completion
+        await session.send({ prompt: "Run 'sleep 2 && echo done'" });
 
         // send() should return before turn completes (no session.idle yet)
         expect(events).not.toContain("session.idle");
 
         // Wait for turn to complete
         const message = await getFinalAssistantMessage(session);
 
-        expect(message.data.content).toContain("2");
+        expect(message.data.content).toContain("done");
         expect(events).toContain("session.idle");
         expect(events).toContain("assistant.message");
     });
@@ -367,8 +368,9 @@ describe("Send Blocking Behavior", async () => {
     it("sendAndWait throws on timeout", async () => {
         const session = await client.createSession();
 
-        await expect(session.sendAndWait({ prompt: "What is 3+3?" }, 1)).rejects.toThrow(
-            /Timeout after 1ms/
-        );
+        // Use a slow command to ensure timeout triggers before completion
+        await expect(
+            session.sendAndWait({ prompt: "Run 'sleep 2 && echo done'" }, 100)
+        ).rejects.toThrow(/Timeout after 100ms/);
     });
 });

python/copilot/session.py

@@ -14,10 +14,12 @@
 from .types import (
     MessageOptions,
     PermissionHandler,
-    SessionEvent as SessionEventTypeAlias,
     Tool,
     ToolHandler,
 )
+from .types import (
+    SessionEvent as SessionEventTypeAlias,
+)
 
 
 class CopilotSession:

python/e2e/test_permissions.py

@@ -8,7 +8,7 @@
 
 from copilot import PermissionRequest, PermissionRequestResult
 
-from .testharness import E2ETestContext, get_final_assistant_message
+from .testharness import E2ETestContext
 from .testharness.helper import read_file, write_file
 
 pytestmark = pytest.mark.asyncio(loop_scope="module")