Skip to content

Commit c9311b6

Browse files
committed
Implement remove_request_handler
1 parent edc5665 commit c9311b6

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

py/selenium/webdriver/common/bidi/network.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
from dataclasses import dataclass, fields, is_dataclass
44

55
from selenium.webdriver.common.bidi.cdp import import_devtools
6-
from selenium.webdriver.common.bidi.session import session_subscribe
6+
from selenium.webdriver.common.bidi.session import session_subscribe, session_unsubscribe
77

88
from . import script
99

@@ -514,3 +514,7 @@ async def add_listener(self, event, callback):
514514
async def continue_request(self, params: ContinueRequestParameters):
515515
result = await self.conn.execute(ContinueRequest(params).cmd())
516516
return result
517+
518+
async def remove_intercept(self, event, params: RemoveInterceptParameters):
519+
await self.conn.execute(session_unsubscribe(event.event_class))
520+
await self.conn.execute(RemoveIntercept(params).cmd())

py/selenium/webdriver/remote/network.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,9 @@ async def get(self, url, conn):
3434
params = NavigateParameters(context=self.driver.current_window_handle, url=url)
3535
await conn.execute(Navigate(params).cmd())
3636

37+
async def remove_request_handler(self):
38+
await self.network.remove_intercept(event=BeforeRequestSent, params=network.RemoveInterceptParameters(self.intercept))
39+
3740
def _callback(self, request_filter, handler):
3841
async def callback(request):
3942
if request_filter(request):

0 commit comments

Comments
 (0)