Regenerate code & fix 1 failing test

Author: mehaase

cdp/background_service.py

@@ -27,6 +27,7 @@ class ServiceName(enum.Enum):
     BACKGROUND_SYNC = "backgroundSync"
     PUSH_MESSAGING = "pushMessaging"
     NOTIFICATIONS = "notifications"
+    PAYMENT_HANDLER = "paymentHandler"
 
     def to_json(self) -> str:
         return self.value

cdp/browser.py

@@ -109,6 +109,8 @@ class PermissionType(enum.Enum):
     SENSORS = "sensors"
     VIDEO_CAPTURE = "videoCapture"
     IDLE_DETECTION = "idleDetection"
+    WAKE_LOCK_SCREEN = "wakeLockScreen"
+    WAKE_LOCK_SYSTEM = "wakeLockSystem"
 
     def to_json(self) -> str:
         return self.value

cdp/dom_snapshot.py

@@ -642,13 +642,28 @@ class LayoutTreeSnapshot:
     #: Stacking context information.
     stacking_contexts: 'RareBooleanData'
 
+    #: The offset rect of nodes. Only available when includeDOMRects is set to true
+    offset_rects: typing.Optional[typing.List['Rectangle']] = None
+
+    #: The scroll rect of nodes. Only available when includeDOMRects is set to true
+    scroll_rects: typing.Optional[typing.List['Rectangle']] = None
+
+    #: The client rect of nodes. Only available when includeDOMRects is set to true
+    client_rects: typing.Optional[typing.List['Rectangle']] = None
+
     def to_json(self) -> T_JSON_DICT:
         json: T_JSON_DICT = dict()
         json['nodeIndex'] = [i for i in self.node_index]
         json['styles'] = [i.to_json() for i in self.styles]
         json['bounds'] = [i.to_json() for i in self.bounds]
         json['text'] = [i.to_json() for i in self.text]
         json['stackingContexts'] = self.stacking_contexts.to_json()
+        if self.offset_rects is not None:
+            json['offsetRects'] = [i.to_json() for i in self.offset_rects]
+        if self.scroll_rects is not None:
+            json['scrollRects'] = [i.to_json() for i in self.scroll_rects]
+        if self.client_rects is not None:
+            json['clientRects'] = [i.to_json() for i in self.client_rects]
         return json
 
     @classmethod
@@ -659,6 +674,9 @@ def from_json(cls, json: T_JSON_DICT) -> 'LayoutTreeSnapshot':
             bounds=[Rectangle.from_json(i) for i in json['bounds']],
             text=[StringIndex.from_json(i) for i in json['text']],
             stacking_contexts=RareBooleanData.from_json(json['stackingContexts']),
+            offset_rects=[Rectangle.from_json(i) for i in json['offsetRects']] if 'offsetRects' in json else None,
+            scroll_rects=[Rectangle.from_json(i) for i in json['scrollRects']] if 'scrollRects' in json else None,
+            client_rects=[Rectangle.from_json(i) for i in json['clientRects']] if 'clientRects' in json else None,
         )
 
 
@@ -762,7 +780,8 @@ def get_snapshot(
 
 
 def capture_snapshot(
-        computed_styles: typing.List[str]
+        computed_styles: typing.List[str],
+        include_dom_rects: typing.Optional[bool] = None
     ) -> typing.Generator[T_JSON_DICT,T_JSON_DICT,typing.Tuple[typing.List['DocumentSnapshot'], typing.List[str]]]:
     '''
     Returns a document snapshot, including the full DOM tree of the root node (including iframes,
@@ -771,12 +790,15 @@ def capture_snapshot(
     flattened.
 
     :param computed_styles: Whitelist of computed styles to return.
+    :param include_dom_rects: Whether to include DOM rectangles (offsetRects, clientRects, scrollRects) into the snapshot
     :returns: a tuple with the following items:
         0. documents: The nodes in the DOM tree. The DOMNode at index 0 corresponds to the root document.
         1. strings: Shared string table that all string properties refer to with indexes.
     '''
     params: T_JSON_DICT = dict()
     params['computedStyles'] = [i for i in computed_styles]
+    if include_dom_rects is not None:
+        params['includeDOMRects'] = include_dom_rects
     cmd_dict: T_JSON_DICT = {
         'method': 'DOMSnapshot.captureSnapshot',
         'params': params,

cdp/network.py

@@ -931,6 +931,114 @@ def from_json(cls, json: T_JSON_DICT) -> 'Cookie':
         )
 
 
+class SetCookieBlockedReason(enum.Enum):
+    '''
+    Types of reasons why a cookie may not be stored from a response.
+    '''
+    SECURE_ONLY = "SecureOnly"
+    SAME_SITE_STRICT = "SameSiteStrict"
+    SAME_SITE_LAX = "SameSiteLax"
+    SAME_SITE_EXTENDED = "SameSiteExtended"
+    SAME_SITE_UNSPECIFIED_TREATED_AS_LAX = "SameSiteUnspecifiedTreatedAsLax"
+    SAME_SITE_NONE_INSECURE = "SameSiteNoneInsecure"
+    USER_PREFERENCES = "UserPreferences"
+    SYNTAX_ERROR = "SyntaxError"
+    SCHEME_NOT_SUPPORTED = "SchemeNotSupported"
+    OVERWRITE_SECURE = "OverwriteSecure"
+    INVALID_DOMAIN = "InvalidDomain"
+    INVALID_PREFIX = "InvalidPrefix"
+    UNKNOWN_ERROR = "UnknownError"
+
+    def to_json(self) -> str:
+        return self.value
+
+    @classmethod
+    def from_json(cls, json: str) -> 'SetCookieBlockedReason':
+        return cls(json)
+
+
+class CookieBlockedReason(enum.Enum):
+    '''
+    Types of reasons why a cookie may not be sent with a request.
+    '''
+    SECURE_ONLY = "SecureOnly"
+    NOT_ON_PATH = "NotOnPath"
+    DOMAIN_MISMATCH = "DomainMismatch"
+    SAME_SITE_STRICT = "SameSiteStrict"
+    SAME_SITE_LAX = "SameSiteLax"
+    SAME_SITE_EXTENDED = "SameSiteExtended"
+    SAME_SITE_UNSPECIFIED_TREATED_AS_LAX = "SameSiteUnspecifiedTreatedAsLax"
+    SAME_SITE_NONE_INSECURE = "SameSiteNoneInsecure"
+    USER_PREFERENCES = "UserPreferences"
+    UNKNOWN_ERROR = "UnknownError"
+
+    def to_json(self) -> str:
+        return self.value
+
+    @classmethod
+    def from_json(cls, json: str) -> 'CookieBlockedReason':
+        return cls(json)
+
+
+@dataclass
+class BlockedSetCookieWithReason:
+    '''
+    A cookie which was not stored from a response with the corresponding reason.
+    '''
+    #: The reason this cookie was blocked.
+    blocked_reason: 'SetCookieBlockedReason'
+
+    #: The string representing this individual cookie as it would appear in the header.
+    #: This is not the entire "cookie" or "set-cookie" header which could have multiple cookies.
+    cookie_line: str
+
+    #: The cookie object which represents the cookie which was not stored. It is optional because
+    #: sometimes complete cookie information is not available, such as in the case of parsing
+    #: errors.
+    cookie: typing.Optional['Cookie'] = None
+
+    def to_json(self) -> T_JSON_DICT:
+        json: T_JSON_DICT = dict()
+        json['blockedReason'] = self.blocked_reason.to_json()
+        json['cookieLine'] = self.cookie_line
+        if self.cookie is not None:
+            json['cookie'] = self.cookie.to_json()
+        return json
+
+    @classmethod
+    def from_json(cls, json: T_JSON_DICT) -> 'BlockedSetCookieWithReason':
+        return cls(
+            blocked_reason=SetCookieBlockedReason.from_json(json['blockedReason']),
+            cookie_line=str(json['cookieLine']),
+            cookie=Cookie.from_json(json['cookie']) if 'cookie' in json else None,
+        )
+
+
+@dataclass
+class BlockedCookieWithReason:
+    '''
+    A cookie with was not sent with a request with the corresponding reason.
+    '''
+    #: The reason the cookie was blocked.
+    blocked_reason: 'CookieBlockedReason'
+
+    #: The cookie object representing the cookie which was not sent.
+    cookie: 'Cookie'
+
+    def to_json(self) -> T_JSON_DICT:
+        json: T_JSON_DICT = dict()
+        json['blockedReason'] = self.blocked_reason.to_json()
+        json['cookie'] = self.cookie.to_json()
+        return json
+
+    @classmethod
+    def from_json(cls, json: T_JSON_DICT) -> 'BlockedCookieWithReason':
+        return cls(
+            blocked_reason=CookieBlockedReason.from_json(json['blockedReason']),
+            cookie=Cookie.from_json(json['cookie']),
+        )
+
+
 @dataclass
 class CookieParam:
     '''
@@ -1203,12 +1311,16 @@ class SignedExchangeHeader:
     #: Signed exchange response signature.
     signatures: typing.List['SignedExchangeSignature']
 
+    #: Signed exchange header integrity hash in the form of "sha256-<base64-hash-value>".
+    header_integrity: str
+
     def to_json(self) -> T_JSON_DICT:
         json: T_JSON_DICT = dict()
         json['requestUrl'] = self.request_url
         json['responseCode'] = self.response_code
         json['responseHeaders'] = self.response_headers.to_json()
         json['signatures'] = [i.to_json() for i in self.signatures]
+        json['headerIntegrity'] = self.header_integrity
         return json
 
     @classmethod
@@ -1218,6 +1330,7 @@ def from_json(cls, json: T_JSON_DICT) -> 'SignedExchangeHeader':
             response_code=int(json['responseCode']),
             response_headers=Headers.from_json(json['responseHeaders']),
             signatures=[SignedExchangeSignature.from_json(i) for i in json['signatures']],
+            header_integrity=str(json['headerIntegrity']),
         )
 
 
@@ -1384,6 +1497,7 @@ def continue_intercepted_request(
     modifications, or blocks it, or completes it with the provided response bytes. If a network
     fetch occurs as a result which encounters a redirect an additional Network.requestIntercepted
     event will be sent with the same InterceptionId.
+    Deprecated, use Fetch.continueRequest, Fetch.fulfillRequest and Fetch.failRequest instead.
 
     :param interception_id:
     :param error_reason: If set this causes the request to fail with the given reason. Passing ``Aborted`` for requests
@@ -1872,6 +1986,7 @@ def set_request_interception(
     ) -> typing.Generator[T_JSON_DICT,T_JSON_DICT,None]:
     '''
     Sets the requests to intercept that match the provided patterns and optionally resource types.
+    Deprecated, please use Fetch.enable instead.
 
     :param patterns: Requests matching any of these patterns will be forwarded and wait for the corresponding
     continueInterceptedRequest call.
@@ -2026,6 +2141,7 @@ class RequestIntercepted:
     '''
     Details of an intercepted HTTP request, which must be either allowed, blocked, modified or
     mocked.
+    Deprecated, use Fetch.requestPaused instead.
     '''
     #: Each request the page makes will have a unique id, however if any redirects are encountered
     #: while processing that fetch, they will be reported with the same id as the original fetch.
@@ -2363,3 +2479,59 @@ def from_json(cls, json: T_JSON_DICT) -> 'WebSocketWillSendHandshakeRequest':
             wall_time=TimeSinceEpoch.from_json(json['wallTime']),
             request=WebSocketRequest.from_json(json['request'])
         )
+
+
+@event_class('Network.requestWillBeSentExtraInfo')
+@dataclass
+class RequestWillBeSentExtraInfo:
+    '''
+    Fired when additional information about a requestWillBeSent event is available from the
+    network stack. Not every requestWillBeSent event will have an additional
+    requestWillBeSentExtraInfo fired for it, and there is no guarantee whether requestWillBeSent
+    or requestWillBeSentExtraInfo will be fired first for the same request.
+    '''
+    #: Request identifier. Used to match this information to an existing requestWillBeSent event.
+    request_id: 'RequestId'
+    #: A list of cookies which will not be sent with this request along with corresponding reasons
+    #: for blocking.
+    blocked_cookies: typing.List['BlockedCookieWithReason']
+    #: Raw request headers as they will be sent over the wire.
+    headers: 'Headers'
+
+    @classmethod
+    def from_json(cls, json: T_JSON_DICT) -> 'RequestWillBeSentExtraInfo':
+        return cls(
+            request_id=RequestId.from_json(json['requestId']),
+            blocked_cookies=[BlockedCookieWithReason.from_json(i) for i in json['blockedCookies']],
+            headers=Headers.from_json(json['headers'])
+        )
+
+
+@event_class('Network.responseReceivedExtraInfo')
+@dataclass
+class ResponseReceivedExtraInfo:
+    '''
+    Fired when additional information about a responseReceived event is available from the network
+    stack. Not every responseReceived event will have an additional responseReceivedExtraInfo for
+    it, and responseReceivedExtraInfo may be fired before or after responseReceived.
+    '''
+    #: Request identifier. Used to match this information to another responseReceived event.
+    request_id: 'RequestId'
+    #: A list of cookies which were not stored from the response along with the corresponding
+    #: reasons for blocking. The cookies here may not be valid due to syntax errors, which
+    #: are represented by the invalid cookie line string instead of a proper cookie.
+    blocked_cookies: typing.List['BlockedSetCookieWithReason']
+    #: Raw response headers as they were received over the wire.
+    headers: 'Headers'
+    #: Raw response header text as it was received over the wire. The raw text may not always be
+    #: available, such as in the case of HTTP/2 or QUIC.
+    headers_text: typing.Optional[str]
+
+    @classmethod
+    def from_json(cls, json: T_JSON_DICT) -> 'ResponseReceivedExtraInfo':
+        return cls(
+            request_id=RequestId.from_json(json['requestId']),
+            blocked_cookies=[BlockedSetCookieWithReason.from_json(i) for i in json['blockedCookies']],
+            headers=Headers.from_json(json['headers']),
+            headers_text=str(json['headersText']) if 'headersText' in json else None
+        )

cdp/overlay.py

@@ -142,19 +142,23 @@ def enable() -> typing.Generator[T_JSON_DICT,T_JSON_DICT,None]:
 
 def get_highlight_object_for_test(
         node_id: 'dom.NodeId',
-        include_distance: typing.Optional[bool] = None
+        include_distance: typing.Optional[bool] = None,
+        include_style: typing.Optional[bool] = None
     ) -> typing.Generator[T_JSON_DICT,T_JSON_DICT,dict]:
     '''
     For testing.
 
     :param node_id: Id of the node to get highlight object for.
     :param include_distance: Whether to include distance info.
+    :param include_style: Whether to include style info.
     :returns: Highlight data for the node.
     '''
     params: T_JSON_DICT = dict()
     params['nodeId'] = node_id.to_json()
     if include_distance is not None:
         params['includeDistance'] = include_distance
+    if include_style is not None:
+        params['includeStyle'] = include_style
     cmd_dict: T_JSON_DICT = {
         'method': 'Overlay.getHighlightObjectForTest',
         'params': params,
@@ -398,6 +402,23 @@ def set_show_paint_rects(
     json = yield cmd_dict
 
 
+def set_show_layout_shift_regions(
+        result: bool
+    ) -> typing.Generator[T_JSON_DICT,T_JSON_DICT,None]:
+    '''
+    Requests that backend shows layout shift regions
+
+    :param result: True for showing layout shift regions
+    '''
+    params: T_JSON_DICT = dict()
+    params['result'] = result
+    cmd_dict: T_JSON_DICT = {
+        'method': 'Overlay.setShowLayoutShiftRegions',
+        'params': params,
+    }
+    json = yield cmd_dict
+
+
 def set_show_scroll_bottleneck_rects(
         show: bool
     ) -> typing.Generator[T_JSON_DICT,T_JSON_DICT,None]: