expose await

StackExchange.Redis.Tests/PubSub.cs

@@ -396,7 +396,7 @@ async Task RunLoop()
                     }
                 }
 
-                Assert.True(subChannel.IsCompleted);
+                await subChannel.Completion;
                 await Assert.ThrowsAsync<ChannelClosedException>(async delegate
                 {
                     var final = await subChannel.ReadAsync().ForAwait();

StackExchange.Redis/StackExchange/Redis/ChannelMessageQueue.cs

@@ -81,6 +81,12 @@ internal ChannelMessageQueue(RedisChannel redisChannel, RedisSubscriber parent)
             _queue.Reader.Completion.ContinueWith(
                 (_, state) => ((ChannelMessageQueue)state).IsCompleted = true, this, TaskContinuationOptions.ExecuteSynchronously);
         }
+
+        /// <summary>
+        /// An awaitable task the indicates completion of the queue (including drain of data)
+        /// </summary>
+        public Task Completion => _queue.Reader.Completion;
+
         private static readonly UnboundedChannelOptions s_ChannelOptions = new UnboundedChannelOptions
         {
             SingleWriter = true,