Task completion source for events

SeleniumHQ · Sep 10, 2024 · d23ff62 · d23ff62
1 parent 2b3816b
commit d23ff62
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/...tnet/SeleniumDocs/BiDi/BrowsingContext/BrowsingContextTest.BrowsingContextCreatedEvent.cs b/...tnet/SeleniumDocs/BiDi/BrowsingContext/BrowsingContextTest.BrowsingContextCreatedEvent.cs
@@ -13,12 +13,14 @@ public async Task BrowsingContextCreatedEvent()
     {
         await using var bidi = await driver.AsBidirectionalAsync();
 
-        BrowsingContextInfo info = null;
+        TaskCompletionSource<BrowsingContextInfo> tcs = new();
 
-        await bidi.OnBrowsingContextCreatedAsync(e => info = e);
+        await bidi.OnBrowsingContextCreatedAsync(tcs.SetResult);
 
         driver.SwitchTo().NewWindow(OpenQA.Selenium.WindowType.Window);
 
+        var info = await tcs.Task.WaitAsync(TimeSpan.FromSeconds(5));
+
         Assert.IsNotNull(info);
         Console.WriteLine(info);
     }

diff --git a/...es/dotnet/SeleniumDocs/BiDi/BrowsingContext/BrowsingContextTest.NavigationStartedEvent.cs b/...es/dotnet/SeleniumDocs/BiDi/BrowsingContext/BrowsingContextTest.NavigationStartedEvent.cs
@@ -1,6 +1,7 @@
 using Microsoft.VisualStudio.TestTools.UnitTesting;
 using OpenQA.Selenium.BiDi;
 using OpenQA.Selenium.BiDi.Modules.BrowsingContext;
+using System;
 using System.Threading.Tasks;
 
 namespace SeleniumDocs.BiDi.BrowsingContext;
@@ -12,12 +13,14 @@ public async Task NavigationStartedEvent()
     {
         var browsingContext = await driver.AsBidirectionalContextAsync();
 
-        NavigationInfo info = null;
+        TaskCompletionSource<NavigationInfo> tcs = new();
 
-        await browsingContext.OnNavigationStartedAsync(e => info = e);
+        await browsingContext.OnNavigationStartedAsync(tcs.SetResult);
 
         await browsingContext.NavigateAsync("https://www.selenium.dev/selenium/web/bidi/logEntryAdded.html", new() { Wait = ReadinessState.Complete });
 
+        var info = await tcs.Task.WaitAsync(TimeSpan.FromSeconds(5));
+
         Assert.IsNotNull(info);
     }
 }