From 67493e33d492c2f0c3c83f50af1fd0904bcc4251 Mon Sep 17 00:00:00 2001
From: Steve Sanderson <1101362+SteveSandersonMS@users.noreply.github.com>
Date: Tue, 28 Jul 2026 13:09:13 +0000
Subject: [PATCH] Fix flaky .NET session resume E2E test

Subscribe for completion events before sending the initial prompt so a fast response cannot be missed between SendAsync and the post-send test helper subscription.

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
---
 dotnet/test/E2E/SessionE2ETests.cs | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/dotnet/test/E2E/SessionE2ETests.cs b/dotnet/test/E2E/SessionE2ETests.cs
index 17658d78f..e47b6b21b 100644
--- a/dotnet/test/E2E/SessionE2ETests.cs
+++ b/dotnet/test/E2E/SessionE2ETests.cs
@@ -245,8 +245,7 @@ public async Task Should_Resume_A_Session_Using_A_New_Client()
         var session1 = await CreateSessionAsync();
         var sessionId = session1.SessionId;
 
-        await session1.SendAsync(new MessageOptions { Prompt = "What is 1+1?" });
-        var answer = await TestHelper.GetFinalAssistantMessageAsync(session1);
+        var answer = await session1.SendAndWaitAsync(new MessageOptions { Prompt = "What is 1+1?" });
         Assert.NotNull(answer);
         Assert.Contains("2", answer!.Data.Content ?? string.Empty);