Enables cancellation token support in async query.

If it is because these are unit tests and are not hitting the actual query pipeline then it is fine, but I think in a "real" if any of these operations are requested with a cancelled cancellation token we should throw from somewhere.