feat: implement unsubscribe by id

src/bidiMapper/BidiNoOpParser.ts

@@ -218,6 +218,9 @@ export class BidiNoOpParser implements BidiCommandParameterParser {
   parseSubscribeParams(params: unknown): Session.SubscriptionRequest {
     return params as Session.SubscriptionRequest;
   }
+  parseUnsubscribeParams(params: unknown): Session.UnsubscribeByAttributesRequest|Session.UnsubscribeByIdRequest {
+    return params as Session.UnsubscribeByAttributesRequest|Session.UnsubscribeByIdRequest;
+  }
   // keep-sorted end
 
   // Storage module

src/bidiMapper/BidiParser.ts

@@ -138,6 +138,7 @@ export interface BidiCommandParameterParser {
   // Session module
   // keep-sorted start block=yes
   parseSubscribeParams(params: unknown): Session.SubscriptionRequest;
+  parseUnsubscribeParams(params: unknown): Session.UnsubscribeByAttributesRequest|Session.UnsubscribeByIdRequest;
   // keep-sorted end
 
   // Storage module

src/bidiMapper/modules/session/EventManager.ts

@@ -306,6 +306,11 @@ export class EventManager extends EventEmitter<EventManagerEventsMap> {
     await this.toggleModulesIfNeeded();
   }
 
+  async unsubscribeByIds(subscriptionIds: string[]): Promise<void> {
+    this.#subscriptionManager.unsubscribeById(subscriptionIds);
+    await this.toggleModulesIfNeeded();
+  }
+
   async toggleModulesIfNeeded(): Promise<void> {
     // TODO(1): Only update changed subscribers
     // TODO(2): Enable for Worker Targets

src/bidiMapper/modules/session/SessionProcessor.ts

@@ -157,9 +157,15 @@ export class SessionProcessor {
   }
 
   async unsubscribe(
-    params: Session.SubscriptionRequest,
+    params: Session.UnsubscribeByAttributesRequest|Session.UnsubscribeByIdRequest,
     channel: BidiPlusChannel = {},
   ): Promise<EmptyResult> {
+    if ('subscriptions' in params) {
+      await this.#eventManager.unsubscribeByIds(
+        params.subscriptions,
+      );
+      return {};
+    }
     await this.#eventManager.unsubscribe(
       params.events as ChromiumBidi.EventNames[],
       params.contexts ?? [],

src/bidiMapper/modules/session/SubscriptionManager.ts

@@ -85,6 +85,7 @@ export type Subscription = {
 
 export class SubscriptionManager {
   #subscriptions: Subscription[] = [];
+  #knownSubscriptionIds = new Set<string>();
   #browsingContextStorage: BrowsingContextStorage;
 
   constructor(browsingContextStorage: BrowsingContextStorage) {
@@ -222,6 +223,7 @@ export class SubscriptionManager {
       channel,
     };
     this.#subscriptions.push(subscription);
+    this.#knownSubscriptionIds.add(subscription.id);
     return subscription;
   }
 
@@ -349,8 +351,17 @@ export class SubscriptionManager {
   /**
    * Unsubscribes by subscriptionId.
    */
-  unsubscribeById(_subscription: string) {
-    // TODO: implement.
+  unsubscribeById(subscriptionIds: string[]) {
+    const subscriptionIdsSet = new Set(subscriptionIds);
+    const unknownIds = difference(subscriptionIdsSet, this.#knownSubscriptionIds);
+
+    if (unknownIds.size !== 0) {
+      throw new InvalidArgumentException('No subscription found');
+    }
+    this.#subscriptions = this.#subscriptions.filter(subscription => {
+      return subscriptionIdsSet.has(subscription.id);
+    });
+    this.#knownSubscriptionIds = difference(this.#knownSubscriptionIds, subscriptionIdsSet);
   }
 }
 

src/bidiTab/BidiParser.ts

@@ -217,6 +217,9 @@ export class BidiParser implements BidiCommandParameterParser {
   parseSubscribeParams(params: unknown): Session.SubscriptionRequest {
     return Parser.Session.parseSubscribeParams(params);
   }
+  parseUnsubbscribeParams(params: unknown): Session.UnsubscribeByAttributesRequest|Session.UnsubscribeByIdRequest {
+    return Parser.Session.parseUnsubscribeParams(params);
+  }
   // keep-sorted end
 
   // Storage module

src/protocol-parser/protocol-parser.ts

@@ -278,6 +278,15 @@ export namespace Session {
       WebDriverBidi.Session.SubscriptionRequestSchema,
     ) as Protocol.Session.SubscriptionRequest;
   }
+
+  export function parseUnsubscribeParams(
+    params: unknown,
+  ): Protocol.Session.UnsubscribeByAttributesRequest|Protocol.Session.UnsubscribeByIdRequest {
+    return parseObject(
+      params,
+      WebDriverBidi.Session.UnsubscribeByIdRequestSchema,
+    ) as Protocol.Session.UnsubscribeByAttributesRequest|Protocol.Session.UnsubscribeByIdRequest;
+  }
 }
 
 export namespace Input {

tests/session/test_subscription.py

@@ -525,6 +525,21 @@ async def test_subscribeWithoutContext_bufferedEventsFromNotClosedContextsAreRet
     assert {"type": "success", "id": command_id, 'result': ANY} == resp
 
 
+@pytest.mark.asyncio
+async def test_unsubscribe_by_id(websocket):
+    res = await subscribe(websocket, ["log.entryAdded"])
+    await execute_command(
+        websocket,
+        {
+            "method": "session.unsubscribe",
+            "params": {
+                "subscriptions": [
+                    res["subscription"]
+                ]
+            }
+        })
+
+
 @pytest.mark.asyncio
 @pytest.mark.parametrize("channel_name", ["channel", "goog:channel"])
 async def test_unsubscribeIsAtomic(websocket, context_id, iframe_id,

tests/test_helpers.py

@@ -66,7 +66,7 @@ async def subscribe(websocket,
         command[channel_name
                 if channel_name is not None else "goog:channel"] = channel
 
-    await execute_command(websocket, command)
+    return await execute_command(websocket, command)
 
 
 async def send_JSON_command(websocket, command: dict) -> int: