Refactoring CancellationToken tests