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>
This commit is contained in:
parent
f9dec13660
commit
283caf9810
1 changed files with 2 additions and 2 deletions
|
|
@ -119,7 +119,7 @@ public class AutomaticInputRequiredStatusTests : LoggedTest
|
||||||
finalStatus = await taskStore.GetTaskAsync(mcpTask.TaskId, cancellationToken: TestContext.Current.CancellationToken);
|
finalStatus = await taskStore.GetTaskAsync(mcpTask.TaskId, cancellationToken: TestContext.Current.CancellationToken);
|
||||||
maxAttempts--;
|
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 - Status should have been InputRequired during sampling
|
||||||
Assert.Contains(McpTaskStatus.InputRequired, statusesDuringSampling);
|
Assert.Contains(McpTaskStatus.InputRequired, statusesDuringSampling);
|
||||||
|
|
@ -218,7 +218,7 @@ public class AutomaticInputRequiredStatusTests : LoggedTest
|
||||||
finalStatus = await taskStore.GetTaskAsync(mcpTask.TaskId, cancellationToken: TestContext.Current.CancellationToken);
|
finalStatus = await taskStore.GetTaskAsync(mcpTask.TaskId, cancellationToken: TestContext.Current.CancellationToken);
|
||||||
maxAttempts--;
|
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 - Status should have been InputRequired during elicitation
|
||||||
Assert.Contains(McpTaskStatus.InputRequired, statusesDuringElicitation);
|
Assert.Contains(McpTaskStatus.InputRequired, statusesDuringElicitation);
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue