def get_important_received_items( self, transcript: Transcript ) -> Iterable[Mapping[str, Any]]: for entry in transcript.values(): received = entry.received or None if received is None: continue method = received.get("method") or "" if method == "window/progress" or method == "window/actionRequired": continue yield received
def throw_on_skip(self, transcript: Transcript): failure_messages = [ "Server busy", "timed out", ] for entry in transcript.values(): received = entry.received if received is None: continue if received.get("error"): message = received["error"]["message"] for failure_message in failure_messages: if failure_message in message: raise unittest.SkipTest(message)
def _flag_unhandled_notifications( self, handled_entries: AbstractSet[str], variables: VariableMap, transcript: Transcript, lsp_id_map: _LspIdMap, ) -> Iterable["_ErrorDescription"]: for transcript_id, entry in transcript.items(): if transcript_id in handled_entries: continue received = entry.received if received is None: continue if entry.sent is not None: # We received a request and responded to it. continue method = received["method"] params = received["params"] payload = self._pretty_print_snippet(received) if "id" in received: description = f"""\ An unexpected request of type {method!r} was sent by the language server. Here is the request payload: {payload} """ at_nocommit = "@" + "nocommit" remediation = f"""\ 1) If this was unexpected, then the language server is buggy and should be fixed. 2) To handle this request, add this directive to your test to wait for it and respond to it before proceeding: .{self.wait_for_server_request.__name__}( method={method!r}, params={params!r}, result={{ "{at_nocommit}": "fill in request data here", }}, ) """ else: if any( isinstance(message, _WaitForNotificationSpec) and message.method == method and interpolate_variables( payload=message.params, variables=variables ) == params for message in self._messages ): # This was a notification we we explicitly waiting for, so skip # it. continue uninterpolated_params = uninterpolate_variables( payload=params, variables=variables ) description = f"""\ An unexpected notification of type {method!r} was sent by the language server. Here is the notification payload: {payload} """ remediation = f"""\ 1) If this was unexpected, then the language server is buggy and should be fixed. 2) If all notifications of type {method!r} should be ignored, add this directive anywhere in your test: .{self.ignore_notifications.__name__}(method={method!r}) 3) If this single instance of the notification was expected, add this directive to your test to wait for it before proceeding: .{self.wait_for_notification.__name__}( method={method!r}, params={uninterpolated_params!r}, ) """ previous_request = self._find_previous_request( transcript, lsp_id_map, current_id=transcript_id ) if previous_request is not None: request_context = self._get_context_for_traceback( previous_request.traceback ) else: request_context = "<no previous request was found>" context = f"""\ This was the most recent request issued from the language client before it received the notification: {request_context}""" yield _ErrorDescription( description=description, context=context, remediation=remediation )
def _find_ignored_transcript_ids(self, transcript: Transcript) -> Iterable[str]: for transcript_id, entry in transcript.items(): if (entry.received is not None and entry.received.get("method") in self._ignored_notification_methods): yield transcript_id