public async Task ExternalCancellation_MutationOperationSent_WaitsForInternalCancellation()
{
// Arrange
var operation = new Set <dynamic>("fake", "fakeKey")
{
IsSent = true
};
var externalCts = new CancellationTokenSource(100);
var internalCts = new CancellationTokenSource();
using var tokenPair = new CancellationTokenPairSource(externalCts.Token, internalCts.Token);
// Act
using var registration = new OperationCancellationRegistration(operation, tokenPair.TokenPair);
tokenPair.Register(() => internalCts.CancelAfter(100));
// Assert
await Assert.ThrowsAnyAsync <OperationCanceledException>(() => operation.Completed.AsTask());
Assert.True(internalCts.IsCancellationRequested);
}
public async Task ExternalCancellation_MutationOperationNotSent_Cancels()
{
// Arrange
var operation = new Set <dynamic>("fake", "fakeKey")
{
IsSent = false
};
var cts = new CancellationTokenSource(100);
var tokenPair = CancellationTokenPair.FromExternalToken(cts.Token);
// Act
using var registration = new OperationCancellationRegistration(operation, tokenPair);
// Assert
await Assert.ThrowsAnyAsync <OperationCanceledException>(() => operation.Completed.AsTask());
}
public async Task ExternalCancellation_ReadOnlyOperation_Cancels(bool isSent)
{
// Arrange
var operation = new Get <dynamic>
{
IsSent = isSent
};
var cts = new CancellationTokenSource(100);
var tokenPair = CancellationTokenPair.FromExternalToken(cts.Token);
// Act
using var registration = new OperationCancellationRegistration(operation, tokenPair);
// Assert
await Assert.ThrowsAnyAsync <OperationCanceledException>(() => operation.Completed.AsTask());
}
public void Dispose_CancelsRegistration()
{
// Arrange
var operation = new Get <dynamic>();
var cts = new CancellationTokenSource();
var tokenPair = CancellationTokenPair.FromInternalToken(cts.Token);
// Act
var registration = new OperationCancellationRegistration(operation, tokenPair);
registration.Dispose();
// Assert
cts.Cancel();
Assert.False(operation.Completed.IsCompleted);
}
