Flow CancellationToken through durable jobs tests

[Copilot]

Durable jobs tests had no per-test timeout mechanism. Operations used CancellationToken.None or local short-lived CancellationTokenSource instances, meaning a hung test would block indefinitely.

Changes

• DurableJobTestsRunner — Accept CancellationToken on all public methods. Replace .WithTimeout() with .WaitAsync(cancellationToken) on grain calls so cancellation is observed at every await point.
• JobShardManagerTestsRunner — Accept CancellationToken on all public methods. Replace all CancellationToken.None with the passed token. Remove local timeout-only CancellationTokenSource instances (the test-level token now serves this role). Use CreateLinkedTokenSource where a shorter sub-timeout is still needed (e.g., verifying an empty shard).
• Test wrappers (InMemoryDurableJobsTests, InMemoryJobShardManagerTests, AzureStorageBlobDurableJobsTests, AzureStorageJobShardManagerTests, AzureStorageJobShardBatchingTests) — Each test method creates a 2-minute CancellationTokenSource and passes the token to the runner.

Example

Before:

[Fact, TestCategory("BVT"), TestCategory("DurableJobs")]
public Task DurableJobGrain()
    => _runner.DurableJobGrain();

After:

[Fact, TestCategory("BVT"), TestCategory("DurableJobs")]
public async Task DurableJobGrain()
{
    using var cts = new CancellationTokenSource(TimeSpan.FromMinutes(2));
    await _runner.DurableJobGrain(cts.Token);
}

Runner methods flow the token to all async operations:

var job = await grain.ScheduleJobAsync("TestJob", dueTime).WaitAsync(cancellationToken);
await grain.WaitForJobToRun(job.Id).WaitAsync(cancellationToken);

Note: xUnit v2 does not support CancellationToken injection on [Fact] methods, so the token is created inside each test method rather than accepted as a parameter.

---

✨ Let Copilot coding agent set things up for you — coding agent works faster and does higher quality work when set up for your repo.

Microsoft Reviewers: Open in CodeFlow

[Copilot]

Pull request overview

This PR adds per-test timeout mechanisms to durable jobs tests by flowing CancellationToken throughout the test infrastructure. Previously, tests used CancellationToken.None or local short-lived CancellationTokenSource instances, meaning hung tests could block indefinitely. The changes ensure every async operation in durable jobs tests can be cancelled via a test-level timeout.

Changes:

• Test runners (DurableJobTestsRunner and JobShardManagerTestsRunner) now accept CancellationToken on all public methods and flow it through all async operations using .WaitAsync(cancellationToken) instead of .WithTimeout()
• All test wrappers create a 2-minute CancellationTokenSource per test method and pass the token to runners
• Local timeout-only CancellationTokenSource instances removed in favor of the test-level token, with CreateLinkedTokenSource used where shorter sub-timeouts are needed

Reviewed changes

Copilot reviewed 7 out of 7 changed files in this pull request and generated no comments.

Show a summary per file

File	Description
test/Tester/DurableJobs/JobShardManagerTestsRunner.cs	Added CancellationToken parameters to all test methods, replaced CancellationToken.None with passed token, removed local timeout-only CTS instances, used CreateLinkedTokenSource for sub-timeouts
test/Tester/DurableJobs/InMemoryJobShardManagerTests.cs	Wrapped each test to create 2-minute CTS and pass token to runner
test/Tester/DurableJobs/DurableJobTestsRunner.cs	Added CancellationToken parameters, replaced .WithTimeout() with .WaitAsync(cancellationToken), removed unused Orleans.Internal import
test/Extensions/TesterAzureUtils/DurableJobs/AzureStorageJobShardManagerTests.cs	Wrapped each test to create 2-minute CTS and pass token to runner
test/Extensions/TesterAzureUtils/DurableJobs/AzureStorageJobShardBatchingTests.cs	Added 2-minute CTS creation and flowed token through all operations
test/Extensions/TesterAzureUtils/DurableJobs/AzureStorageBlobDurableJobsTests.cs	Wrapped each test to create 2-minute CTS and pass token to runner
test/DefaultCluster.Tests/InMemoryDurableJobsTests.cs	Wrapped each test to create 2-minute CTS and pass token to runner