[dotnet][bidi] Use subscription id for events to unsubscribe (SeleniumHQ#15251)

nvborisenko · gryznar · commit 853933e50166 · 2025-05-17T14:27:45.000+02:00
diff --git a/dotnet/src/webdriver/BiDi/Communication/Broker.cs b/dotnet/src/webdriver/BiDi/Communication/Broker.cs
@@ -83,6 +83,7 @@ internal Broker(BiDi bidi, ITransport transport)
                 new DateTimeOffsetConverter(),
                 new PrintPageRangeConverter(),
                 new InputOriginConverter(),
+                new SubscriptionConverter(),
                 new JsonStringEnumConverter(JsonNamingPolicy.CamelCase),
 
                 // https://github.com/dotnet/runtime/issues/72604
@@ -223,23 +224,23 @@ public async Task<Subscription> SubscribeAsync<TEventArgs>(string eventName, Act
 
         if (options is BrowsingContextsSubscriptionOptions browsingContextsOptions)
         {
-            await _bidi.SessionModule.SubscribeAsync([eventName], new() { Contexts = browsingContextsOptions.Contexts }).ConfigureAwait(false);
+            var subscribeResult = await _bidi.SessionModule.SubscribeAsync([eventName], new() { Contexts = browsingContextsOptions.Contexts }).ConfigureAwait(false);
 
             var eventHandler = new SyncEventHandler<TEventArgs>(eventName, action, browsingContextsOptions?.Contexts);
 
             handlers.Add(eventHandler);
 
-            return new Subscription(this, eventHandler);
+            return new Subscription(subscribeResult.Subscription, this, eventHandler);
         }
         else
         {
-            await _bidi.SessionModule.SubscribeAsync([eventName]).ConfigureAwait(false);
+            var subscribeResult = await _bidi.SessionModule.SubscribeAsync([eventName]).ConfigureAwait(false);
 
             var eventHandler = new SyncEventHandler<TEventArgs>(eventName, action);
 
             handlers.Add(eventHandler);
 
-            return new Subscription(this, eventHandler);
+            return new Subscription(subscribeResult.Subscription, this, eventHandler);
         }
     }
 
@@ -250,44 +251,51 @@ public async Task<Subscription> SubscribeAsync<TEventArgs>(string eventName, Fun
 
         if (options is BrowsingContextsSubscriptionOptions browsingContextsOptions)
         {
-            await _bidi.SessionModule.SubscribeAsync([eventName], new() { Contexts = browsingContextsOptions.Contexts }).ConfigureAwait(false);
+            var subscribeResult = await _bidi.SessionModule.SubscribeAsync([eventName], new() { Contexts = browsingContextsOptions.Contexts }).ConfigureAwait(false);
 
             var eventHandler = new AsyncEventHandler<TEventArgs>(eventName, func, browsingContextsOptions.Contexts);
 
             handlers.Add(eventHandler);
 
-            return new Subscription(this, eventHandler);
+            return new Subscription(subscribeResult.Subscription, this, eventHandler);
         }
         else
         {
-            await _bidi.SessionModule.SubscribeAsync([eventName]).ConfigureAwait(false);
+            var subscribeResult = await _bidi.SessionModule.SubscribeAsync([eventName]).ConfigureAwait(false);
 
             var eventHandler = new AsyncEventHandler<TEventArgs>(eventName, func);
 
             handlers.Add(eventHandler);
 
-            return new Subscription(this, eventHandler);
+            return new Subscription(subscribeResult.Subscription, this, eventHandler);
         }
     }
 
-    public async Task UnsubscribeAsync(EventHandler eventHandler)
+    public async Task UnsubscribeAsync(Modules.Session.Subscription subscription, EventHandler eventHandler)
     {
         var eventHandlers = _eventHandlers[eventHandler.EventName];
 
         eventHandlers.Remove(eventHandler);
 
-        if (eventHandler.Contexts is not null)
+        if (subscription is not null)
         {
-            if (!eventHandlers.Any(h => eventHandler.Contexts.Equals(h.Contexts)) && !eventHandlers.Any(h => h.Contexts is null))
-            {
-                await _bidi.SessionModule.UnsubscribeAsync([eventHandler.EventName], new() { Contexts = eventHandler.Contexts }).ConfigureAwait(false);
-            }
+            await _bidi.SessionModule.UnsubscribeAsync([subscription]).ConfigureAwait(false);
         }
         else
         {
-            if (!eventHandlers.Any(h => h.Contexts is not null) && !eventHandlers.Any(h => h.Contexts is null))
+            if (eventHandler.Contexts is not null)
+            {
+                if (!eventHandlers.Any(h => eventHandler.Contexts.Equals(h.Contexts)) && !eventHandlers.Any(h => h.Contexts is null))
+                {
+                    await _bidi.SessionModule.UnsubscribeAsync([eventHandler.EventName], new() { Contexts = eventHandler.Contexts }).ConfigureAwait(false);
+                }
+            }
+            else
             {
-                await _bidi.SessionModule.UnsubscribeAsync([eventHandler.EventName]).ConfigureAwait(false);
+                if (!eventHandlers.Any(h => h.Contexts is not null) && !eventHandlers.Any(h => h.Contexts is null))
+                {
+                    await _bidi.SessionModule.UnsubscribeAsync([eventHandler.EventName]).ConfigureAwait(false);
+                }
             }
         }
     }
diff --git a/dotnet/src/webdriver/BiDi/Communication/Json/BiDiJsonSerializerContext.cs b/dotnet/src/webdriver/BiDi/Communication/Json/BiDiJsonSerializerContext.cs
@@ -79,7 +79,9 @@ namespace OpenQA.Selenium.BiDi.Communication.Json;
 [JsonSerializable(typeof(Modules.Session.NewResult))]
 [JsonSerializable(typeof(Modules.Session.EndCommand))]
 [JsonSerializable(typeof(Modules.Session.SubscribeCommand))]
-[JsonSerializable(typeof(Modules.Session.UnsubscribeCommand))]
+[JsonSerializable(typeof(Modules.Session.SubscribeResult))]
+[JsonSerializable(typeof(Modules.Session.UnsubscribeByIdCommand))]
+[JsonSerializable(typeof(Modules.Session.UnsubscribeByAttributesCommand))]
 
 [JsonSerializable(typeof(Modules.Browser.CloseCommand), TypeInfoPropertyName = "Browser_CloseCommand")]
 [JsonSerializable(typeof(Modules.Browser.CreateUserContextCommand))]
diff --git a/dotnet/src/webdriver/BiDi/Communication/Json/Converters/SubscriptionConverter.cs b/dotnet/src/webdriver/BiDi/Communication/Json/Converters/SubscriptionConverter.cs
@@ -0,0 +1,41 @@
+// <copyright file="SubscriptionConverter.cs" company="Selenium Committers">
+// Licensed to the Software Freedom Conservancy (SFC) under one
+// or more contributor license agreements.  See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership.  The SFC licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License.  You may obtain a copy of the License at
+//
+//   http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied.  See the License for the
+// specific language governing permissions and limitations
+// under the License.
+// </copyright>
+
+using System;
+using System.Text.Json;
+using System.Text.Json.Serialization;
+
+#nullable enable
+
+namespace OpenQA.Selenium.BiDi.Communication.Json.Converters;
+
+internal class SubscriptionConverter : JsonConverter<Modules.Session.Subscription>
+{
+    public override Modules.Session.Subscription? Read(ref Utf8JsonReader reader, Type typeToConvert, JsonSerializerOptions options)
+    {
+        var id = reader.GetString();
+
+        return new Modules.Session.Subscription(id!);
+    }
+
+    public override void Write(Utf8JsonWriter writer, Modules.Session.Subscription value, JsonSerializerOptions options)
+    {
+        writer.WriteStringValue(value.Id);
+    }
+}
diff --git a/dotnet/src/webdriver/BiDi/Modules/Session/SessionModule.cs b/dotnet/src/webdriver/BiDi/Modules/Session/SessionModule.cs
@@ -32,7 +32,7 @@ public async Task<StatusResult> StatusAsync(StatusOptions? options = null)
         return await Broker.ExecuteCommandAsync<StatusCommand, StatusResult>(new StatusCommand(), options).ConfigureAwait(false);
     }
 
-    public async Task SubscribeAsync(IEnumerable<string> events, SubscribeOptions? options = null)
+    public async Task<SubscribeResult> SubscribeAsync(IEnumerable<string> events, SubscribeOptions? options = null)
     {
         var @params = new SubscribeCommandParameters(events);
 
@@ -41,19 +41,23 @@ public async Task SubscribeAsync(IEnumerable<string> events, SubscribeOptions? o
             @params.Contexts = options.Contexts;
         }
 
-        await Broker.ExecuteCommandAsync(new SubscribeCommand(@params), options).ConfigureAwait(false);
+        return await Broker.ExecuteCommandAsync<SubscribeCommand, SubscribeResult>(new(@params), options).ConfigureAwait(false);
     }
 
-    public async Task UnsubscribeAsync(IEnumerable<string> events, UnsubscribeOptions? options = null)
+    public async Task UnsubscribeAsync(IEnumerable<Subscription> subscriptions, UnsubscribeByIdOptions? options = null)
     {
-        var @params = new SubscribeCommandParameters(events);
+        var @params = new UnsubscribeByIdCommandParameters(subscriptions);
 
-        if (options is not null)
-        {
-            @params.Contexts = options.Contexts;
-        }
+        await Broker.ExecuteCommandAsync(new UnsubscribeByIdCommand(@params), options).ConfigureAwait(false);
+    }
+
+    public async Task UnsubscribeAsync(IEnumerable<string> eventNames = null, UnsubscribeByAttributesOptions? options = null)
+    {
+        var @params = new UnsubscribeByAttributesCommandParameters(eventNames);
+
+        @params.Contexts = options?.Contexts;
 
-        await Broker.ExecuteCommandAsync(new UnsubscribeCommand(@params), options).ConfigureAwait(false);
+        await Broker.ExecuteCommandAsync(new UnsubscribeByAttributesCommand(@params), options).ConfigureAwait(false);
     }
 
     public async Task<NewResult> NewAsync(CapabilitiesRequest capabilitiesRequest, NewOptions? options = null)
diff --git a/dotnet/src/webdriver/BiDi/Modules/Session/SubscribeCommand.cs b/dotnet/src/webdriver/BiDi/Modules/Session/SubscribeCommand.cs
@@ -36,3 +36,5 @@ public record SubscribeOptions : CommandOptions
 {
     public IEnumerable<BrowsingContext.BrowsingContext>? Contexts { get; set; }
 }
+
+internal record SubscribeResult(Subscription Subscription);
diff --git a/dotnet/src/webdriver/BiDi/Modules/Session/Subscription.cs b/dotnet/src/webdriver/BiDi/Modules/Session/Subscription.cs
@@ -0,0 +1,32 @@
+// <copyright file="Subscription.cs" company="Selenium Committers">
+// Licensed to the Software Freedom Conservancy (SFC) under one
+// or more contributor license agreements.  See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership.  The SFC licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License.  You may obtain a copy of the License at
+//
+//   http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied.  See the License for the
+// specific language governing permissions and limitations
+// under the License.
+// </copyright>
+
+#nullable enable
+
+namespace OpenQA.Selenium.BiDi.Modules.Session;
+
+public sealed class Subscription
+{
+    internal Subscription(string id)
+    {
+        Id = id;
+    }
+
+    internal string Id { get; }
+}
diff --git a/dotnet/src/webdriver/BiDi/Modules/Session/UnsubscribeCommand.cs b/dotnet/src/webdriver/BiDi/Modules/Session/UnsubscribeCommand.cs
@@ -18,12 +18,31 @@
 // </copyright>
 
 using OpenQA.Selenium.BiDi.Communication;
+using System;
+using System.Collections.Generic;
 
 #nullable enable
 
 namespace OpenQA.Selenium.BiDi.Modules.Session;
 
-internal class UnsubscribeCommand(SubscribeCommandParameters @params)
-    : Command<SubscribeCommandParameters>(@params, "session.unsubscribe");
+internal class UnsubscribeByIdCommand(UnsubscribeByIdCommandParameters @params)
+    : Command<UnsubscribeByIdCommandParameters>(@params, "session.unsubscribe");
 
-public record UnsubscribeOptions : SubscribeOptions;
+internal class UnsubscribeByAttributesCommand(UnsubscribeByAttributesCommandParameters @params)
+    : Command<UnsubscribeByAttributesCommandParameters>(@params, "session.unsubscribe");
+
+internal record UnsubscribeByIdCommandParameters(IEnumerable<Subscription> Subscriptions) : CommandParameters;
+
+public record UnsubscribeByIdOptions : CommandOptions;
+
+internal record UnsubscribeByAttributesCommandParameters(IEnumerable<string> Events) : CommandParameters
+{
+    [Obsolete("Contexts param is deprecated and will be removed in the future versions")]
+    // https://w3c.github.io/webdriver-bidi/#type-session-UnsubscribeByAttributesRequest
+    public IEnumerable<BrowsingContext.BrowsingContext>? Contexts { get; set; }
+}
+
+public record UnsubscribeByAttributesOptions : CommandOptions
+{
+    public IEnumerable<BrowsingContext.BrowsingContext>? Contexts { get; set; }
+}
diff --git a/dotnet/src/webdriver/BiDi/Subscription.cs b/dotnet/src/webdriver/BiDi/Subscription.cs
@@ -28,18 +28,20 @@ namespace OpenQA.Selenium.BiDi;
 
 public class Subscription : IAsyncDisposable
 {
+    private readonly Modules.Session.Subscription _subscription;
     private readonly Broker _broker;
     private readonly Communication.EventHandler _eventHandler;
 
-    internal Subscription(Broker broker, Communication.EventHandler eventHandler)
+    internal Subscription(Modules.Session.Subscription subscription, Broker broker, Communication.EventHandler eventHandler)
     {
+        _subscription = subscription;
         _broker = broker;
         _eventHandler = eventHandler;
     }
 
     public async Task UnsubscribeAsync()
     {
-        await _broker.UnsubscribeAsync(_eventHandler).ConfigureAwait(false);
+        await _broker.UnsubscribeAsync(_subscription, _eventHandler).ConfigureAwait(false);
     }
 
     public async ValueTask DisposeAsync()

Original file line number	Diff line number	Diff line change
`@@ -83,6 +83,7 @@ internal Broker(BiDi bidi, ITransport transport)`
`83`	`83`	`new DateTimeOffsetConverter(),`
`84`	`84`	`new PrintPageRangeConverter(),`
`85`	`85`	`new InputOriginConverter(),`
	`86`	`+ new SubscriptionConverter(),`
`86`	`87`	`new JsonStringEnumConverter(JsonNamingPolicy.CamelCase),`
`87`	`88`
`88`	`89`	`// https://github.com/dotnet/runtime/issues/72604`
`@@ -223,23 +224,23 @@ public async Task<Subscription> SubscribeAsync<TEventArgs>(string eventName, Act`
`223`	`224`
`224`	`225`	`if (options is BrowsingContextsSubscriptionOptions browsingContextsOptions)`
`225`	`226`	`{`
`226`		`- await _bidi.SessionModule.SubscribeAsync([eventName], new() { Contexts = browsingContextsOptions.Contexts }).ConfigureAwait(false);`
	`227`	`+ var subscribeResult = await _bidi.SessionModule.SubscribeAsync([eventName], new() { Contexts = browsingContextsOptions.Contexts }).ConfigureAwait(false);`
`227`	`228`
`228`	`229`	`var eventHandler = new SyncEventHandler<TEventArgs>(eventName, action, browsingContextsOptions?.Contexts);`
`229`	`230`
`230`	`231`	`handlers.Add(eventHandler);`
`231`	`232`
`232`		`- return new Subscription(this, eventHandler);`
	`233`	`+ return new Subscription(subscribeResult.Subscription, this, eventHandler);`
`233`	`234`	`}`
`234`	`235`	`else`
`235`	`236`	`{`
`236`		`- await _bidi.SessionModule.SubscribeAsync([eventName]).ConfigureAwait(false);`
	`237`	`+ var subscribeResult = await _bidi.SessionModule.SubscribeAsync([eventName]).ConfigureAwait(false);`
`237`	`238`
`238`	`239`	`var eventHandler = new SyncEventHandler<TEventArgs>(eventName, action);`
`239`	`240`
`240`	`241`	`handlers.Add(eventHandler);`
`241`	`242`
`242`		`- return new Subscription(this, eventHandler);`
	`243`	`+ return new Subscription(subscribeResult.Subscription, this, eventHandler);`
`243`	`244`	`}`
`244`	`245`	`}`
`245`	`246`
`@@ -250,44 +251,51 @@ public async Task<Subscription> SubscribeAsync<TEventArgs>(string eventName, Fun`
`250`	`251`
`251`	`252`	`if (options is BrowsingContextsSubscriptionOptions browsingContextsOptions)`
`252`	`253`	`{`
`253`		`- await _bidi.SessionModule.SubscribeAsync([eventName], new() { Contexts = browsingContextsOptions.Contexts }).ConfigureAwait(false);`
	`254`	`+ var subscribeResult = await _bidi.SessionModule.SubscribeAsync([eventName], new() { Contexts = browsingContextsOptions.Contexts }).ConfigureAwait(false);`
`254`	`255`
`255`	`256`	`var eventHandler = new AsyncEventHandler<TEventArgs>(eventName, func, browsingContextsOptions.Contexts);`
`256`	`257`
`257`	`258`	`handlers.Add(eventHandler);`
`258`	`259`
`259`		`- return new Subscription(this, eventHandler);`
	`260`	`+ return new Subscription(subscribeResult.Subscription, this, eventHandler);`
`260`	`261`	`}`
`261`	`262`	`else`
`262`	`263`	`{`
`263`		`- await _bidi.SessionModule.SubscribeAsync([eventName]).ConfigureAwait(false);`
	`264`	`+ var subscribeResult = await _bidi.SessionModule.SubscribeAsync([eventName]).ConfigureAwait(false);`
`264`	`265`
`265`	`266`	`var eventHandler = new AsyncEventHandler<TEventArgs>(eventName, func);`
`266`	`267`
`267`	`268`	`handlers.Add(eventHandler);`
`268`	`269`
`269`		`- return new Subscription(this, eventHandler);`
	`270`	`+ return new Subscription(subscribeResult.Subscription, this, eventHandler);`
`270`	`271`	`}`
`271`	`272`	`}`
`272`	`273`
`273`		`- public async Task UnsubscribeAsync(EventHandler eventHandler)`
	`274`	`+ public async Task UnsubscribeAsync(Modules.Session.Subscription subscription, EventHandler eventHandler)`
`274`	`275`	`{`
`275`	`276`	`var eventHandlers = _eventHandlers[eventHandler.EventName];`
`276`	`277`
`277`	`278`	`eventHandlers.Remove(eventHandler);`
`278`	`279`
`279`		`- if (eventHandler.Contexts is not null)`
	`280`	`+ if (subscription is not null)`
`280`	`281`	`{`
`281`		`- if (!eventHandlers.Any(h => eventHandler.Contexts.Equals(h.Contexts)) && !eventHandlers.Any(h => h.Contexts is null))`
`282`		`- {`
`283`		`- await _bidi.SessionModule.UnsubscribeAsync([eventHandler.EventName], new() { Contexts = eventHandler.Contexts }).ConfigureAwait(false);`
`284`		`- }`
	`282`	`+ await _bidi.SessionModule.UnsubscribeAsync([subscription]).ConfigureAwait(false);`
`285`	`283`	`}`
`286`	`284`	`else`
`287`	`285`	`{`
`288`		`- if (!eventHandlers.Any(h => h.Contexts is not null) && !eventHandlers.Any(h => h.Contexts is null))`
	`286`	`+ if (eventHandler.Contexts is not null)`
	`287`	`+ {`
	`288`	`+ if (!eventHandlers.Any(h => eventHandler.Contexts.Equals(h.Contexts)) && !eventHandlers.Any(h => h.Contexts is null))`
	`289`	`+ {`
	`290`	`+ await _bidi.SessionModule.UnsubscribeAsync([eventHandler.EventName], new() { Contexts = eventHandler.Contexts }).ConfigureAwait(false);`
	`291`	`+ }`
	`292`	`+ }`
	`293`	`+ else`
`289`	`294`	`{`
`290`		`- await _bidi.SessionModule.UnsubscribeAsync([eventHandler.EventName]).ConfigureAwait(false);`
	`295`	`+ if (!eventHandlers.Any(h => h.Contexts is not null) && !eventHandlers.Any(h => h.Contexts is null))`
	`296`	`+ {`
	`297`	`+ await _bidi.SessionModule.UnsubscribeAsync([eventHandler.EventName]).ConfigureAwait(false);`
	`298`	`+ }`
`291`	`299`	`}`
`292`	`300`	`}`
`293`	`301`	`}`
Original file line number	Diff line number	Diff line change
`@@ -32,7 +32,7 @@ public async Task<StatusResult> StatusAsync(StatusOptions? options = null)`
`32`	`32`	`return await Broker.ExecuteCommandAsync<StatusCommand, StatusResult>(new StatusCommand(), options).ConfigureAwait(false);`
`33`	`33`	`}`
`34`	`34`
`35`		`- public async Task SubscribeAsync(IEnumerable<string> events, SubscribeOptions? options = null)`
	`35`	`+ public async Task<SubscribeResult> SubscribeAsync(IEnumerable<string> events, SubscribeOptions? options = null)`
`36`	`36`	`{`
`37`	`37`	`var @params = new SubscribeCommandParameters(events);`
`38`	`38`
`@@ -41,19 +41,23 @@ public async Task SubscribeAsync(IEnumerable<string> events, SubscribeOptions? o`
`41`	`41`	`@params.Contexts = options.Contexts;`
`42`	`42`	`}`
`43`	`43`
`44`		`- await Broker.ExecuteCommandAsync(new SubscribeCommand(@params), options).ConfigureAwait(false);`
	`44`	`+ return await Broker.ExecuteCommandAsync<SubscribeCommand, SubscribeResult>(new(@params), options).ConfigureAwait(false);`
`45`	`45`	`}`
`46`	`46`
`47`		`- public async Task UnsubscribeAsync(IEnumerable<string> events, UnsubscribeOptions? options = null)`
	`47`	`+ public async Task UnsubscribeAsync(IEnumerable<Subscription> subscriptions, UnsubscribeByIdOptions? options = null)`
`48`	`48`	`{`
`49`		`- var @params = new SubscribeCommandParameters(events);`
	`49`	`+ var @params = new UnsubscribeByIdCommandParameters(subscriptions);`
`50`	`50`
`51`		`- if (options is not null)`
`52`		`- {`
`53`		`- @params.Contexts = options.Contexts;`
`54`		`- }`
	`51`	`+ await Broker.ExecuteCommandAsync(new UnsubscribeByIdCommand(@params), options).ConfigureAwait(false);`
	`52`	`+ }`
	`53`	`+`
	`54`	`+ public async Task UnsubscribeAsync(IEnumerable<string> eventNames = null, UnsubscribeByAttributesOptions? options = null)`
	`55`	`+ {`
	`56`	`+ var @params = new UnsubscribeByAttributesCommandParameters(eventNames);`
	`57`	`+`
	`58`	`+ @params.Contexts = options?.Contexts;`
`55`	`59`
`56`		`- await Broker.ExecuteCommandAsync(new UnsubscribeCommand(@params), options).ConfigureAwait(false);`
	`60`	`+ await Broker.ExecuteCommandAsync(new UnsubscribeByAttributesCommand(@params), options).ConfigureAwait(false);`
`57`	`61`	`}`
`58`	`62`
`59`	`63`	`public async Task<NewResult> NewAsync(CapabilitiesRequest capabilitiesRequest, NewOptions? options = null)`
Original file line number	Diff line number	Diff line change
`@@ -36,3 +36,5 @@ public record SubscribeOptions : CommandOptions`
`36`	`36`	`{`
`37`	`37`	`public IEnumerable<BrowsingContext.BrowsingContext>? Contexts { get; set; }`
`38`	`38`	`}`
	`39`	`+`
	`40`	`+internal record SubscribeResult(Subscription Subscription);`
Original file line number	Diff line number	Diff line change
`@@ -28,18 +28,20 @@ namespace OpenQA.Selenium.BiDi;`
`28`	`28`
`29`	`29`	`public class Subscription : IAsyncDisposable`
`30`	`30`	`{`
	`31`	`+ private readonly Modules.Session.Subscription _subscription;`
`31`	`32`	`private readonly Broker _broker;`
`32`	`33`	`private readonly Communication.EventHandler _eventHandler;`
`33`	`34`
`34`		`- internal Subscription(Broker broker, Communication.EventHandler eventHandler)`
	`35`	`+ internal Subscription(Modules.Session.Subscription subscription, Broker broker, Communication.EventHandler eventHandler)`
`35`	`36`	`{`
	`37`	`+ _subscription = subscription;`
`36`	`38`	`_broker = broker;`
`37`	`39`	`_eventHandler = eventHandler;`
`38`	`40`	`}`
`39`	`41`
`40`	`42`	`public async Task UnsubscribeAsync()`
`41`	`43`	`{`
`42`		`- await _broker.UnsubscribeAsync(_eventHandler).ConfigureAwait(false);`
	`44`	`+ await _broker.UnsubscribeAsync(_subscription, _eventHandler).ConfigureAwait(false);`
`43`	`45`	`}`
`44`	`46`
`45`	`47`	`public async ValueTask DisposeAsync()`