Fix race condition in AutomaticInputRequiredStatusTests polling logic (#1223)

Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com>
Co-authored-by: ericstj <8918108+ericstj@users.noreply.github.com>
Co-authored-by: stephentoub <2642209+stephentoub@users.noreply.github.com>

Author: 3 people

tests/ModelContextProtocol.Tests/Server/AutomaticInputRequiredStatusTests.cs

@@ -119,7 +119,7 @@ public async Task TaskStatus_TransitionsToInputRequired_DuringSampleAsync()
             finalStatus = await taskStore.GetTaskAsync(mcpTask.TaskId, cancellationToken: TestContext.Current.CancellationToken);
             maxAttempts--;
         }
-        while (finalStatus?.Status == McpTaskStatus.Working && maxAttempts > 0);
+        while (finalStatus?.Status is not McpTaskStatus.Completed && maxAttempts > 0);
 
         // Assert - Status should have been InputRequired during sampling
         Assert.Contains(McpTaskStatus.InputRequired, statusesDuringSampling);
@@ -218,7 +218,7 @@ public async Task TaskStatus_TransitionsToInputRequired_DuringElicitAsync()
             finalStatus = await taskStore.GetTaskAsync(mcpTask.TaskId, cancellationToken: TestContext.Current.CancellationToken);
             maxAttempts--;
         }
-        while (finalStatus?.Status == McpTaskStatus.Working && maxAttempts > 0);
+        while (finalStatus?.Status is not McpTaskStatus.Completed && maxAttempts > 0);
 
         // Assert - Status should have been InputRequired during elicitation
         Assert.Contains(McpTaskStatus.InputRequired, statusesDuringElicitation);