Made functions async

Author: shbenzer

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

@@ -36,7 +36,7 @@ def __init__(self, conn):
         self.conn = conn
         self.callbacks = {}
 
-    def continue_response(self, request_id, status_code, headers=None, body=None):
+    async def continue_response(self, request_id, status_code, headers=None, body=None):
         params = {
             'requestId': request_id,
             'status': status_code
@@ -45,9 +45,9 @@ def continue_response(self, request_id, status_code, headers=None, body=None):
             params['headers'] = headers
         if body is not None:
             params['body'] = body
-        self.conn.execute('network.continueResponse', params)
+        await self.conn.execute('network.continueResponse', params)
 
-    def continue_request(self, request_id, url=None, method=None, headers=None, postData=None):
+    async def continue_request(self, request_id, url=None, method=None, headers=None, postData=None):
         params = {
             'requestId': request_id
         }
@@ -59,23 +59,23 @@ def continue_request(self, request_id, url=None, method=None, headers=None, post
             params['headers'] = headers
         if postData is not None:
             params['postData'] = postData
-        self.conn.execute('network.continueRequest', params)
+        await self.conn.execute('network.continueRequest', params)
 
-    def add_intercept(self, phases=None, contexts=None, url_patterns=None):
+    async def add_intercept(self, phases=None, contexts=None, url_patterns=None):
         if phases is None:
             phases = []
         params = {
             'phases': phases,
             'contexts': contexts,
             'urlPatterns': url_patterns
         }
-        self.conn.execute('network.addIntercept', params)
+        await self.conn.execute('network.addIntercept', params)
 
-    def remove_intercept(self, intercept):
-        self.conn.execute('network.removeIntercept', {'intercept': intercept})
+    async def remove_intercept(self, intercept):
+        await self.conn.execute('network.removeIntercept', {'intercept': intercept})
 
-    def continue_with_auth(self, request_id, username, password):
-        self.conn.execute(
+    async def continue_with_auth(self, request_id, username, password):
+        await self.conn.execute(
             'network.continueWithAuth',
             {
                 'request': request_id,
@@ -88,7 +88,7 @@ def continue_with_auth(self, request_id, username, password):
             }
         )
 
-    def on(self, event, callback):
+    async def on(self, event, callback):
         event = self.EVENTS.get(event, event)
         self.callbacks[event] = callback
-        session_subscribe(self.conn, event, self.handle_event)
+        await session_subscribe(self.conn, event, self.handle_event)