schemathesis 3.25.6__py3-none-any.whl → 4.0.0a1__py3-none-any.whl

schemathesis/core/errors.py

@@ -0,0 +1,370 @@
+"""Base error handling that is not tied to any specific API specification or execution context."""
+
+from __future__ import annotations
+
+import enum
+import re
+import traceback
+from types import TracebackType
+from typing import TYPE_CHECKING, Any, Callable, NoReturn
+
+from schemathesis.core.output import truncate_json
+
+if TYPE_CHECKING:
+    from jsonschema import SchemaError as JsonSchemaError
+    from jsonschema import ValidationError
+    from requests import RequestException
+
+    from schemathesis.core.compat import RefResolutionError
+
+
+SCHEMA_ERROR_SUGGESTION = "Ensure that the definition complies with the OpenAPI specification"
+SERIALIZERS_SUGGESTION_MESSAGE = (
+    "You can register your own serializer with `schemathesis.serializer` "
+    "and Schemathesis will be able to make API calls with this media type. \n"
+    "See https://schemathesis.readthedocs.io/en/stable/how.html#payload-serialization for more information."
+)
+
+
+class SchemathesisError(Exception):
+    """Base exception class for all Schemathesis errors."""
+
+
+class InvalidSchema(SchemathesisError):
+    """Indicates errors in API schema validation or processing."""
+
+    def __init__(
+        self,
+        message: str | None = None,
+        path: str | None = None,
+        method: str | None = None,
+        full_path: str | None = None,
+    ) -> None:
+        self.message = message
+        self.path = path
+        self.method = method
+        self.full_path = full_path
+
+    @classmethod
+    def from_jsonschema_error(
+        cls, error: ValidationError, path: str | None, method: str | None, full_path: str | None
+    ) -> InvalidSchema:
+        if error.absolute_path:
+            part = error.absolute_path[-1]
+            if isinstance(part, int) and len(error.absolute_path) > 1:
+                parent = error.absolute_path[-2]
+                message = f"Invalid definition for element at index {part} in `{parent}`"
+            else:
+                message = f"Invalid `{part}` definition"
+        else:
+            message = "Invalid schema definition"
+        error_path = " -> ".join(str(entry) for entry in error.path) or "[root]"
+        message += f"\n\nLocation:\n    {error_path}"
+        instance = truncate_json(error.instance)
+        message += f"\n\nProblematic definition:\n{instance}"
+        message += "\n\nError details:\n    "
+        # This default message contains the instance which we already printed
+        if "is not valid under any of the given schemas" in error.message:
+            message += "The provided definition doesn't match any of the expected formats or types."
+        else:
+            message += error.message
+        message += f"\n\n{SCHEMA_ERROR_SUGGESTION}"
+        return cls(message, path=path, method=method, full_path=full_path)
+
+    @classmethod
+    def from_reference_resolution_error(
+        cls, error: RefResolutionError, path: str | None, method: str | None, full_path: str | None
+    ) -> InvalidSchema:
+        notes = getattr(error, "__notes__", [])
+        # Some exceptions don't have the actual reference in them, hence we add it manually via notes
+        pointer = f"'{notes[0]}'"
+        message = "Unresolvable JSON pointer in the schema"
+        # Get the pointer value from "Unresolvable JSON pointer: 'components/UnknownParameter'"
+        message += f"\n\nError details:\n    JSON pointer: {pointer}"
+        message += "\n    This typically means that the schema is referencing a component that doesn't exist."
+        message += f"\n\n{SCHEMA_ERROR_SUGGESTION}"
+        return cls(message, path=path, method=method, full_path=full_path)
+
+    def as_failing_test_function(self) -> Callable:
+        """Create a test function that will fail.
+
+        This approach allows us to use default pytest reporting style for operation-level schema errors.
+        """
+
+        def actual_test(*args: Any, **kwargs: Any) -> NoReturn:
+            __tracebackhide__ = True
+            raise self
+
+        return actual_test
+
+
+class InvalidRegexType(InvalidSchema):
+    """Raised when an invalid type is used where a regex pattern is expected."""
+
+
+class MalformedMediaType(ValueError):
+    """Raised on parsing of incorrect media type."""
+
+
+class InvalidRegexPattern(InvalidSchema):
+    """Raised when a string pattern is not a valid regular expression."""
+
+    @classmethod
+    def from_hypothesis_jsonschema_message(cls, message: str) -> InvalidRegexPattern:
+        match = re.search(r"pattern='(.*?)'.*?\((.*?)\)", message)
+        if match:
+            message = f"Invalid regular expression. Pattern `{match.group(1)}` is not recognized - `{match.group(2)}`"
+        return cls(message)
+
+    @classmethod
+    def from_schema_error(cls, error: JsonSchemaError, *, from_examples: bool) -> InvalidRegexPattern:
+        if from_examples:
+            message = (
+                "Failed to generate test cases from examples for this API operation because of "
+                f"unsupported regular expression `{error.instance}`"
+            )
+        else:
+            message = (
+                "Failed to generate test cases for this API operation because of "
+                f"unsupported regular expression `{error.instance}`"
+            )
+        return cls(message)
+
+
+class InvalidHeadersExample(InvalidSchema):
+    @classmethod
+    def from_headers(cls, headers: dict[str, str]) -> InvalidHeadersExample:
+        message = (
+            "Failed to generate test cases from examples for this API operation because of "
+            "some header examples are invalid:\n"
+        )
+        for key, value in headers.items():
+            message += f"\n  - {key!r}={value!r}"
+        message += "\n\nEnsure the header examples comply with RFC 7230, Section 3.2"
+        return cls(message)
+
+
+class IncorrectUsage(SchemathesisError):
+    """Indicates incorrect usage of Schemathesis' public API."""
+
+
+class InvalidRateLimit(IncorrectUsage):
+    """Incorrect input for rate limiting."""
+
+    def __init__(self, value: str) -> None:
+        super().__init__(
+            f"Invalid rate limit value: `{value}`. Should be in form `limit/interval`. "
+            "Example: `10/m` for 10 requests per minute."
+        )
+
+
+class InternalError(SchemathesisError):
+    """Internal error in Schemathesis."""
+
+
+class SerializationError(SchemathesisError):
+    """Can't serialize request payload."""
+
+
+NAMESPACE_DEFINITION_URL = "https://github.com/OAI/OpenAPI-Specification/blob/master/versions/3.0.3.md#xmlNamespace"
+UNBOUND_PREFIX_MESSAGE_TEMPLATE = (
+    "Unbound prefix: `{prefix}`. "
+    "You need to define this namespace in your API schema via the `xml.namespace` keyword. "
+    f"See more at {NAMESPACE_DEFINITION_URL}"
+)
+
+
+class UnboundPrefix(SerializationError):
+    """XML serialization error.
+
+    It happens when the schema does not define a namespace that is used by some of its parts.
+    """
+
+    def __init__(self, prefix: str):
+        super().__init__(UNBOUND_PREFIX_MESSAGE_TEMPLATE.format(prefix=prefix))
+
+
+SERIALIZATION_NOT_POSSIBLE_MESSAGE = (
+    f"Schemathesis can't serialize data to any of the defined media types: {{}} \n{SERIALIZERS_SUGGESTION_MESSAGE}"
+)
+SERIALIZATION_FOR_TYPE_IS_NOT_POSSIBLE_MESSAGE = (
+    f"Schemathesis can't serialize data to {{}} \n{SERIALIZERS_SUGGESTION_MESSAGE}"
+)
+RECURSIVE_REFERENCE_ERROR_MESSAGE = (
+    "Currently, Schemathesis can't generate data for this operation due to "
+    "recursive references in the operation definition. See more information in "
+    "this issue - https://github.com/schemathesis/schemathesis/issues/947"
+)
+
+
+class SerializationNotPossible(SerializationError):
+    """Not possible to serialize data to specified media type(s).
+
+    This error occurs in two scenarios:
+    1. When attempting to serialize to a specific media type that isn't supported
+    2. When none of the available media types can be used for serialization
+    """
+
+    def __init__(self, message: str, media_types: list[str]) -> None:
+        self.message = message
+        self.media_types = media_types
+
+    def __str__(self) -> str:
+        return self.message
+
+    @classmethod
+    def from_media_types(cls, *media_types: str) -> SerializationNotPossible:
+        """Create error when no available media type can be used."""
+        return cls(SERIALIZATION_NOT_POSSIBLE_MESSAGE.format(", ".join(media_types)), media_types=list(media_types))
+
+    @classmethod
+    def for_media_type(cls, media_type: str) -> SerializationNotPossible:
+        """Create error when a specific required media type isn't supported."""
+        return cls(SERIALIZATION_FOR_TYPE_IS_NOT_POSSIBLE_MESSAGE.format(media_type), media_types=[media_type])
+
+
+class OperationNotFound(LookupError, SchemathesisError):
+    """Raised when an API operation cannot be found in the schema.
+
+    This error typically occurs during schema access in user code when trying to
+    reference a non-existent operation.
+    """
+
+    def __init__(self, message: str, item: str) -> None:
+        self.message = message
+        self.item = item
+
+    def __str__(self) -> str:
+        return self.message
+
+
+@enum.unique
+class LoaderErrorKind(str, enum.Enum):
+    # Connection related issues
+    CONNECTION_SSL = "connection_ssl"
+    CONNECTION_OTHER = "connection_other"
+    NETWORK_OTHER = "network_other"
+
+    # HTTP error codes
+    HTTP_SERVER_ERROR = "http_server_error"
+    HTTP_CLIENT_ERROR = "http_client_error"
+    HTTP_NOT_FOUND = "http_not_found"
+    HTTP_FORBIDDEN = "http_forbidden"
+
+    # Content decoding issues
+    SYNTAX_ERROR = "syntax_error"
+    UNEXPECTED_CONTENT_TYPE = "unexpected_content_type"
+    YAML_NUMERIC_STATUS_CODES = "yaml_numeric_status_codes"
+    YAML_NON_STRING_KEYS = "yaml_non_string_keys"
+
+    # Open API validation
+    OPEN_API_INVALID_SCHEMA = "open_api_invalid_schema"
+    OPEN_API_UNSPECIFIED_VERSION = "open_api_unspecified_version"
+    OPEN_API_UNSUPPORTED_VERSION = "open_api_unsupported_version"
+
+    # GraphQL validation
+    GRAPHQL_INVALID_SCHEMA = "graphql_invalid_schema"
+
+    # Unclassified
+    UNCLASSIFIED = "unclassified"
+
+
+class LoaderError(SchemathesisError):
+    """Failed to load an API schema."""
+
+    def __init__(
+        self,
+        kind: LoaderErrorKind,
+        message: str,
+        url: str | None = None,
+        extras: list[str] | None = None,
+    ) -> None:
+        self.kind = kind
+        self.message = message
+        self.url = url
+        self.extras = extras or []
+
+    def __str__(self) -> str:
+        return self.message
+
+
+def get_request_error_extras(exc: RequestException) -> list[str]:
+    """Extract additional context from a request exception."""
+    from requests.exceptions import ChunkedEncodingError, ConnectionError, SSLError
+    from urllib3.exceptions import MaxRetryError
+
+    if isinstance(exc, SSLError):
+        reason = str(exc.args[0].reason)
+        return [_remove_ssl_line_number(reason).strip()]
+    if isinstance(exc, ConnectionError):
+        inner = exc.args[0]
+        if isinstance(inner, MaxRetryError) and inner.reason is not None:
+            arg = inner.reason.args[0]
+            if isinstance(arg, str):
+                if ":" not in arg:
+                    reason = arg
+                else:
+                    _, reason = arg.split(":", maxsplit=1)
+            else:
+                reason = f"Max retries exceeded with url: {inner.url}"
+            return [reason.strip()]
+        return [" ".join(map(_clean_inner_request_message, inner.args))]
+    if isinstance(exc, ChunkedEncodingError):
+        return [str(exc.args[0].args[1])]
+    return []
+
+
+def _remove_ssl_line_number(text: str) -> str:
+    return re.sub(r"\(_ssl\.c:\d+\)", "", text)
+
+
+def _clean_inner_request_message(message: object) -> str:
+    if isinstance(message, str) and message.startswith("HTTPConnectionPool"):
+        return re.sub(r"HTTPConnectionPool\(.+?\): ", "", message).rstrip(".")
+    return str(message)
+
+
+def get_request_error_message(exc: RequestException) -> str:
+    """Extract user-facing message from a request exception."""
+    from requests.exceptions import ChunkedEncodingError, ConnectionError, ReadTimeout, SSLError
+
+    if isinstance(exc, ReadTimeout):
+        _, duration = exc.args[0].args[0][:-1].split("read timeout=")
+        return f"Read timed out after {duration} seconds"
+    if isinstance(exc, SSLError):
+        return "SSL verification problem"
+    if isinstance(exc, ConnectionError):
+        return "Connection failed"
+    if isinstance(exc, ChunkedEncodingError):
+        return "Connection broken. The server declared chunked encoding but sent an invalid chunk"
+    return str(exc)
+
+
+def split_traceback(traceback: str) -> list[str]:
+    return [entry for entry in traceback.splitlines() if entry]
+
+
+def format_exception(
+    error: Exception,
+    *,
+    with_traceback: bool = False,
+    skip_frames: int = 0,
+) -> str:
+    """Format exception with optional traceback."""
+    if not with_traceback:
+        lines = traceback.format_exception_only(type(error), error)
+        return "".join(lines).strip()
+
+    trace = error.__traceback__
+    if skip_frames > 0:
+        trace = extract_nth_traceback(trace, skip_frames)
+    lines = traceback.format_exception(type(error), error, trace)
+    return "".join(lines).strip()
+
+
+def extract_nth_traceback(trace: TracebackType | None, n: int) -> TracebackType | None:
+    depth = 0
+    while depth < n and trace is not None:
+        trace = trace.tb_next
+        depth += 1
+    return trace

schemathesis/core/failures.py

@@ -0,0 +1,285 @@
+from __future__ import annotations
+
+import http.client
+import textwrap
+from collections.abc import Sequence
+from dataclasses import dataclass
+from enum import Enum, auto
+from json import JSONDecodeError
+from typing import Callable
+
+from schemathesis.core.compat import BaseExceptionGroup
+from schemathesis.core.output import OutputConfig, prepare_response_payload
+from schemathesis.core.transport import Response
+
+
+class Severity(Enum):
+    # For server errors, security issues like ignored auth
+    CRITICAL = auto()
+    # For schema violations
+    HIGH = auto()
+    # For content type issues, header problems
+    MEDIUM = auto()
+    # For performance issues, minor inconsistencies
+    LOW = auto()
+
+    def __lt__(self, other: Severity) -> bool:
+        # Lower values are more severe
+        return self.value < other.value
+
+
+@dataclass
+class Failure(AssertionError):
+    """API check failure."""
+
+    __slots__ = ("operation", "title", "message", "code", "case_id", "severity")
+
+    def __init__(
+        self,
+        *,
+        operation: str,
+        title: str,
+        message: str,
+        code: str,
+        case_id: str | None = None,
+        severity: Severity = Severity.MEDIUM,
+    ) -> None:
+        self.operation = operation
+        self.title = title
+        self.message = message
+        self.code = code
+        self.case_id = case_id
+        self.severity = severity
+
+    def __str__(self) -> str:
+        if not self.message:
+            return self.title
+        return f"{self.title}\n\n{self.message}"
+
+    def __lt__(self, other: Failure) -> bool:
+        return (
+            self.severity,
+            self.__class__.__name__,
+            self.message,
+        ) < (other.severity, other.__class__.__name__, other.message)
+
+    # Comparison & hashing is done purely on classes to simplify keeping the minimized failure during shrinking
+    def __hash__(self) -> int:
+        return hash(self.__class__)
+
+    def __eq__(self, other: object, /) -> bool:
+        if not isinstance(other, Failure):
+            return NotImplemented
+        return type(self) is type(other) and self.operation == other.operation and self._unique_key == other._unique_key
+
+    @classmethod
+    def from_assertion(cls, *, name: str, operation: str, exc: AssertionError) -> Failure:
+        return Failure(
+            operation=operation,
+            title=f"Custom check failed: `{name}`",
+            message=str(exc),
+            code="custom",
+        )
+
+    @property
+    def _unique_key(self) -> str:
+        return self.message
+
+
+@dataclass
+class MaxResponseTimeConfig:
+    limit: float = 10.0
+
+
+class ResponseTimeExceeded(Failure):
+    """Response took longer than expected."""
+
+    __slots__ = ("operation", "elapsed", "deadline", "title", "message", "code", "case_id", "severity")
+
+    def __init__(
+        self,
+        *,
+        operation: str,
+        elapsed: float,
+        deadline: int,
+        message: str,
+        title: str = "Response time limit exceeded",
+        code: str = "response_time_exceeded",
+        case_id: str | None = None,
+    ) -> None:
+        self.operation = operation
+        self.elapsed = elapsed
+        self.deadline = deadline
+        self.title = title
+        self.message = message
+        self.code = code
+        self.case_id = case_id
+        self.severity = Severity.LOW
+
+    @property
+    def _unique_key(self) -> str:
+        return self.title
+
+
+class ServerError(Failure):
+    """Server responded with an error."""
+
+    __slots__ = ("operation", "status_code", "title", "message", "code", "case_id", "severity")
+
+    def __init__(
+        self,
+        *,
+        operation: str,
+        status_code: int,
+        title: str = "Server error",
+        message: str = "",
+        code: str = "server_error",
+        case_id: str | None = None,
+    ) -> None:
+        self.operation = operation
+        self.status_code = status_code
+        self.title = title
+        self.message = message
+        self.code = code
+        self.case_id = case_id
+        self.severity = Severity.CRITICAL
+
+    @property
+    def _unique_key(self) -> str:
+        return str(self.status_code)
+
+
+class MalformedJson(Failure):
+    """Failed to deserialize JSON."""
+
+    __slots__ = (
+        "operation",
+        "validation_message",
+        "document",
+        "position",
+        "lineno",
+        "colno",
+        "message",
+        "title",
+        "code",
+        "case_id",
+        "severity",
+    )
+
+    def __init__(
+        self,
+        *,
+        operation: str,
+        validation_message: str,
+        document: str,
+        position: int,
+        lineno: int,
+        colno: int,
+        message: str,
+        title: str = "JSON deserialization error",
+        code: str = "malformed_json",
+        case_id: str | None = None,
+    ) -> None:
+        self.operation = operation
+        self.validation_message = validation_message
+        self.document = document
+        self.position = position
+        self.lineno = lineno
+        self.colno = colno
+        self.message = message
+        self.title = title
+        self.code = code
+        self.case_id = case_id
+        self.severity = Severity.MEDIUM
+
+    @property
+    def _unique_key(self) -> str:
+        return self.title
+
+    @classmethod
+    def from_exception(cls, *, operation: str, exc: JSONDecodeError) -> MalformedJson:
+        return cls(
+            operation=operation,
+            message=str(exc),
+            validation_message=exc.msg,
+            document=exc.doc,
+            position=exc.pos,
+            lineno=exc.lineno,
+            colno=exc.colno,
+        )
+
+
+class FailureGroup(BaseExceptionGroup):
+    """Multiple distinct check failures."""
+
+    exceptions: Sequence[Failure]
+
+    def __new__(cls, failures: Sequence[Failure], message: str | None = None) -> FailureGroup:
+        if message is None:
+            message = failure_report_title(failures)
+        return super().__new__(cls, message, list(failures))
+
+
+class MessageBlock(Enum):
+    CASE_ID = "case_id"
+    FAILURE = "failure"
+    STATUS = "status"
+    CURL = "curl"
+
+
+BlockFormatter = Callable[[MessageBlock, str], str]
+
+
+def failure_report_title(failures: Sequence[Failure]) -> str:
+    message = f"Schemathesis found {len(failures)} distinct failure"
+    if len(failures) > 1:
+        message += "s"
+    return message
+
+
+def format_failures(
+    *,
+    case_id: str | None,
+    response: Response | None,
+    failures: Sequence[Failure],
+    curl: str,
+    formatter: BlockFormatter | None = None,
+    config: OutputConfig,
+) -> str:
+    """Format failure information with custom styling."""
+    formatter = formatter or (lambda _, x: x)
+
+    if case_id is not None:
+        output = formatter(MessageBlock.CASE_ID, f"{case_id}\n")
+    else:
+        output = ""
+
+    # Failures
+    for idx, failure in enumerate(failures):
+        output += formatter(MessageBlock.FAILURE, f"\n- {failure.title}")
+        if failure.message:
+            output += "\n\n"
+            output += textwrap.indent(failure.message, "    ")
+        if idx != len(failures):
+            output += "\n"
+
+    # Response status
+    if isinstance(response, Response):
+        reason = http.client.responses.get(response.status_code, "Unknown")
+        output += formatter(MessageBlock.STATUS, f"\n[{response.status_code}] {reason}:\n")
+        # Response payload
+        if response.content is None or not response.content:
+            output += "\n    <EMPTY>"
+        else:
+            try:
+                payload = prepare_response_payload(response.text, config=config)
+                output += textwrap.indent(f"\n`{payload}`", prefix="    ")
+            except UnicodeDecodeError:
+                output += "\n    <BINARY>"
+    else:
+        output += "\n    <NO RESPONSE>"
+
+    # cURL
+    output += "\n" + formatter(MessageBlock.CURL, f"\nReproduce with: \n\n    {curl}")
+
+    return output

schemathesis/core/fs.py

@@ -0,0 +1,19 @@
+import os
+import pathlib
+
+
+def ensure_parent(path: os.PathLike, fail_silently: bool = True) -> None:
+    # Try to create the parent dir
+    try:
+        pathlib.Path(path).parent.mkdir(mode=0o755, parents=True, exist_ok=True)
+    except OSError:
+        if not fail_silently:
+            raise
+
+
+def file_exists(path: str) -> bool:
+    try:
+        return pathlib.Path(path).is_file()
+    except OSError:
+        # For example, path could be too long
+        return False

schemathesis/{_lazy_import.py → core/lazy_import.py}

@@ -1,4 +1,5 @@
 from __future__ import annotations
+
 from typing import Any, Callable