Fixed wait until available #61

Author: FlorianRappl

Diff for: CHANGELOG.md

@@ -4,6 +4,7 @@ Released on Tuesday, March 31 2020.
 
 - Fixed deadlock with some versions of jQuery (#63)
 - Enabled the instance creator factory (#11)
+- Return `IDocument` from `WaitUntilAvailable()` (#61)
 
 # 0.13.0
 

Diff for: src/AngleSharp.Js/JsDocumentExtensions.cs

@@ -102,10 +102,11 @@ public static Task<IDocument> WhenStable(this Task<IDocument> documentTask) =>
         /// <param name="documentTask">The document task to await.</param>
         /// <param name="cancellation">The timeout cancellation, if any.</param>
         /// <returns>A task that is finished when the document is available.</returns>
-        public static async Task WaitUntilAvailable(this Task<IDocument> documentTask, CancellationToken cancellation = default)
+        public static async Task<IDocument> WaitUntilAvailable(this Task<IDocument> documentTask, CancellationToken cancellation = default)
         {
             var document = await documentTask.ConfigureAwait(false);
             await document.WaitUntilAvailable(cancellation).ConfigureAwait(false);
+            return document;
         }
 
         /// <summary>
@@ -116,7 +117,7 @@ public static async Task WaitUntilAvailable(this Task<IDocument> documentTask, C
         /// <param name="document">The document to await.</param>
         /// <param name="cancellation">The timeout cancellation, if any.</param>
         /// <returns>A task that is finished when the document is available.</returns>
-        public static async Task WaitUntilAvailable(this IDocument document, CancellationToken cancellation = default)
+        public static async Task<IDocument> WaitUntilAvailable(this IDocument document, CancellationToken cancellation = default)
         {
             if (document.ReadyState != DocumentReadyState.Complete)
             {
@@ -126,6 +127,7 @@ public static async Task WaitUntilAvailable(this IDocument document, Cancellatio
             }
 
             await document.WhenStable().ConfigureAwait(false);
+            return document;
         }
     }
 }