schemathesis 3.13.0__py3-none-any.whl → 4.4.2__py3-none-any.whl

schemathesis/models.py

@@ -1,1041 +0,0 @@
-# pylint: disable=too-many-lines
-import base64
-import datetime
-import http
-from collections import Counter
-from contextlib import contextmanager
-from copy import deepcopy
-from enum import Enum
-from io import StringIO
-from itertools import chain, zip_longest
-from logging import LogRecord
-from typing import (
-    TYPE_CHECKING,
-    Any,
-    Callable,
-    Dict,
-    Generator,
-    Generic,
-    Iterator,
-    List,
-    NoReturn,
-    Optional,
-    Sequence,
-    Tuple,
-    Type,
-    TypeVar,
-    Union,
-    cast,
-)
-from urllib.parse import quote, unquote, urljoin, urlsplit, urlunsplit
-
-import attr
-import curlify
-import requests
-import werkzeug
-from hypothesis import event, note, reject
-from hypothesis import strategies as st
-from requests.structures import CaseInsensitiveDict
-from starlette.testclient import TestClient as ASGIClient
-
-from . import failures, serializers
-from .constants import (
-    DEFAULT_RESPONSE_TIMEOUT,
-    SERIALIZERS_SUGGESTION_MESSAGE,
-    USER_AGENT,
-    CodeSampleStyle,
-    DataGenerationMethod,
-)
-from .exceptions import (
-    CheckFailed,
-    FailureContext,
-    InvalidSchema,
-    SerializationNotPossible,
-    get_grouped_exception,
-    get_timeout_error,
-)
-from .hooks import GLOBAL_HOOK_DISPATCHER, HookContext, HookDispatcher
-from .parameters import Parameter, ParameterSet, PayloadAlternatives
-from .serializers import Serializer, SerializerContext
-from .types import Body, Cookies, FormData, Headers, NotSet, PathParameters, Query
-from .utils import (
-    NOT_SET,
-    GenericResponse,
-    WSGIResponse,
-    deprecated_property,
-    get_response_payload,
-    maybe_set_assertion_message,
-)
-
-if TYPE_CHECKING:
-    from .schemas import BaseSchema
-    from .stateful import Stateful, StatefulTest
-
-
-@attr.s(slots=True)  # pragma: no mutate
-class CaseSource:
-    """Data sources, used to generate a test case."""
-
-    case: "Case" = attr.ib()  # pragma: no mutate
-    response: GenericResponse = attr.ib()  # pragma: no mutate
-
-
-def cant_serialize(media_type: str) -> NoReturn:  # type: ignore
-    """Reject the current example if we don't know how to send this data to the application."""
-    event_text = f"Can't serialize data to `{media_type}`."
-    note(f"{event_text} {SERIALIZERS_SUGGESTION_MESSAGE}")
-    event(event_text)
-    reject()  # type: ignore
-
-
-@attr.s(slots=True, repr=False)  # pragma: no mutate
-class Case:  # pylint: disable=too-many-public-methods
-    """A single test case parameters."""
-
-    operation: "APIOperation" = attr.ib()  # pragma: no mutate
-    path_parameters: Optional[PathParameters] = attr.ib(default=None)  # pragma: no mutate
-    headers: Optional[CaseInsensitiveDict] = attr.ib(default=None)  # pragma: no mutate
-    cookies: Optional[Cookies] = attr.ib(default=None)  # pragma: no mutate
-    query: Optional[Query] = attr.ib(default=None)  # pragma: no mutate
-    # By default, there is no body, but we can't use `None` as the default value because it clashes with `null`
-    # which is a valid payload.
-    body: Union[Body, NotSet] = attr.ib(default=NOT_SET)  # pragma: no mutate
-
-    source: Optional[CaseSource] = attr.ib(default=None)  # pragma: no mutate
-    # The media type for cases with a payload. For example, "application/json"
-    media_type: Optional[str] = attr.ib(default=None)  # pragma: no mutate
-    # The way the case was generated (None for manually crafted ones)
-    data_generation_method: Optional[DataGenerationMethod] = attr.ib(default=None)  # pragma: no mutate
-
-    def __repr__(self) -> str:
-        parts = [f"{self.__class__.__name__}("]
-        first = True
-        for name in ("path_parameters", "headers", "cookies", "query", "body"):
-            value = getattr(self, name)
-            if value not in (None, NOT_SET):
-                if first:
-                    first = False
-                else:
-                    parts.append(", ")
-                parts.extend((name, "=", repr(value)))
-        return "".join(parts) + ")"
-
-    @deprecated_property(removed_in="4.0", replacement="operation")
-    def endpoint(self) -> "APIOperation":
-        return self.operation
-
-    @property
-    def path(self) -> str:
-        return self.operation.path
-
-    @property
-    def full_path(self) -> str:
-        return self.operation.full_path
-
-    @property
-    def method(self) -> str:
-        return self.operation.method.upper()
-
-    @property
-    def base_url(self) -> Optional[str]:
-        return self.operation.base_url
-
-    @property
-    def app(self) -> Any:
-        return self.operation.app
-
-    def set_source(self, response: GenericResponse, case: "Case") -> None:
-        self.source = CaseSource(case=case, response=response)
-
-    @property
-    def formatted_path(self) -> str:
-        # pylint: disable=not-a-mapping
-        try:
-            return self.path.format(**self.path_parameters or {})
-        except KeyError as exc:
-            # This may happen when a path template has a placeholder for variable "X", but parameter "X" is not defined
-            # in the parameters list.
-            # When `exc` is formatted, it is the missing key name in quotes. E.g. 'id'
-            raise InvalidSchema(f"Path parameter {exc} is not defined") from exc
-
-    def get_full_base_url(self) -> Optional[str]:
-        """Create a full base url, adding "localhost" for WSGI apps."""
-        parts = urlsplit(self.base_url)
-        if not parts.hostname:
-            path = cast(str, parts.path or "")
-            return urlunsplit(("http", "localhost", path or "", "", ""))
-        return self.base_url
-
-    def as_text_lines(self, headers: Optional[Dict[str, Any]] = None) -> List[str]:
-        """Textual representation.
-
-        Each component is a separate line.
-        """
-        output = {
-            "Path parameters": self.path_parameters,
-            "Headers": self.headers,
-            "Cookies": self.cookies,
-            "Query": self.query,
-            "Body": self.body,
-        }
-        if headers:
-            final_headers = output["Headers"] or {}
-            final_headers = cast(Dict[str, Any], final_headers)
-            final_headers.update(headers)
-            output["Headers"] = final_headers
-        max_length = max(map(len, output))
-        template = f"{{:<{max_length}}} : {{}}"
-
-        def should_display(key: str, value: Any) -> bool:
-            if key == "Body":
-                return value is not NOT_SET
-            return value is not None
-
-        return [template.format(key, value) for key, value in output.items() if should_display(key, value)]
-
-    def get_code_to_reproduce(
-        self, headers: Optional[Dict[str, Any]] = None, request: Optional[requests.PreparedRequest] = None
-    ) -> str:
-        """Construct a Python code to reproduce this case with `requests`."""
-        if request is not None:
-            kwargs: Dict[str, Any] = {
-                "method": request.method,
-                "url": request.url,
-                "headers": request.headers,
-                "data": request.body,
-            }
-        else:
-            base_url = self.get_full_base_url()
-            kwargs = self.as_requests_kwargs(base_url)
-        if headers:
-            final_headers = kwargs["headers"] or {}
-            final_headers.update(headers)
-            kwargs["headers"] = final_headers
-        method = kwargs["method"].lower()
-
-        def should_display(key: str, value: Any) -> bool:
-            if key in ("method", "url"):
-                return False
-            # Parameters are either absent because they are not defined or are optional
-            return value not in (None, {})
-
-        printed_kwargs = ", ".join(
-            f"{key}={repr(value)}" for key, value in kwargs.items() if should_display(key, value)
-        )
-        url = _escape_single_quotes(kwargs["url"])
-        args_repr = f"'{url}'"
-        if printed_kwargs:
-            args_repr += f", {printed_kwargs}"
-        return f"requests.{method}({args_repr})"
-
-    def as_curl_command(self, headers: Optional[Dict[str, Any]] = None) -> str:
-        """Construct a curl command for a given case."""
-        base_url = self.get_full_base_url()
-        kwargs = self.as_requests_kwargs(base_url)
-        if headers:
-            final_headers = kwargs["headers"] or {}
-            final_headers.update(headers)
-            kwargs["headers"] = final_headers
-        request = requests.Request(**kwargs)
-        prepared = request.prepare()
-        if isinstance(prepared.body, bytes):
-            # Note, it may be not sufficient to reproduce the error :(
-            prepared.body = prepared.body.decode("utf-8", errors="replace")
-        return curlify.to_curl(prepared)
-
-    def _get_base_url(self, base_url: Optional[str] = None) -> str:
-        if base_url is None:
-            if self.base_url is not None:
-                base_url = self.base_url
-            else:
-                raise ValueError(
-                    "Base URL is required as `base_url` argument in `call` or should be specified "
-                    "in the schema constructor as a part of Schema URL."
-                )
-        return base_url
-
-    def _get_headers(self, headers: Optional[Dict[str, str]] = None) -> CaseInsensitiveDict:
-        final_headers = self.headers.copy() if self.headers is not None else CaseInsensitiveDict()
-        if headers:
-            final_headers.update(headers)
-        final_headers.setdefault("User-Agent", USER_AGENT)
-        return final_headers
-
-    def _get_serializer(self) -> Optional[Serializer]:
-        """Get a serializer for the payload, if there is any."""
-        if self.media_type is not None:
-            cls = serializers.get(self.media_type)
-            if cls is None:
-                all_media_types = self.operation.get_request_payload_content_types()
-                if all(serializers.get(media_type) is None for media_type in all_media_types):
-                    raise SerializationNotPossible.from_media_types(*all_media_types)
-                cant_serialize(self.media_type)
-            return cls()
-        return None
-
-    def as_requests_kwargs(
-        self, base_url: Optional[str] = None, headers: Optional[Dict[str, str]] = None
-    ) -> Dict[str, Any]:
-        """Convert the case into a dictionary acceptable by requests."""
-        final_headers = self._get_headers(headers)
-        if self.media_type and self.media_type != "multipart/form-data" and self.body is not NOT_SET:
-            # `requests` will handle multipart form headers with the proper `boundary` value.
-            if "content-type" not in {header.lower() for header in final_headers}:
-                final_headers["Content-Type"] = self.media_type
-        base_url = self._get_base_url(base_url)
-        formatted_path = self.formatted_path.lstrip("/")  # pragma: no mutate
-        url = unquote(urljoin(base_url + "/", quote(formatted_path)))
-        extra: Dict[str, Any]
-        serializer = self._get_serializer()
-        if serializer is not None and self.body is not NOT_SET:
-            context = SerializerContext(case=self)
-            extra = serializer.as_requests(context, self.body)
-        else:
-            extra = {}
-        additional_headers = extra.pop("headers", None)
-        if additional_headers:
-            # Additional headers, needed for the serializer
-            for key, value in additional_headers.items():
-                final_headers.setdefault(key, value)
-        return {
-            "method": self.method,
-            "url": url,
-            "cookies": self.cookies,
-            "headers": final_headers,
-            "params": self.query,
-            **extra,
-        }
-
-    def call(
-        self,
-        base_url: Optional[str] = None,
-        session: Optional[requests.Session] = None,
-        headers: Optional[Dict[str, Any]] = None,
-        **kwargs: Any,
-    ) -> requests.Response:
-        """Make a network call with `requests`."""
-        if session is None:
-            session = requests.Session()
-            close_session = True
-        else:
-            close_session = False
-        data = self.as_requests_kwargs(base_url, headers)
-        data.update(kwargs)
-        data.setdefault("timeout", DEFAULT_RESPONSE_TIMEOUT / 1000)
-        try:
-            response = session.request(**data)  # type: ignore
-        except requests.Timeout as exc:
-            timeout = 1000 * data["timeout"]  # It is defined and not empty, since the exception happened
-            code_message = self._get_code_message(self.operation.schema.code_sample_style, exc.request)
-            raise get_timeout_error(timeout)(
-                f"\n\n1. Request timed out after {timeout:.2f}ms\n\n----------\n\n{code_message}",
-                context=failures.RequestTimeout(timeout=timeout),
-            ) from None
-        if close_session:
-            session.close()
-        return response
-
-    def as_werkzeug_kwargs(self, headers: Optional[Dict[str, str]] = None) -> Dict[str, Any]:
-        """Convert the case into a dictionary acceptable by werkzeug.Client."""
-        final_headers = self._get_headers(headers)
-        if self.media_type and self.body is not NOT_SET:
-            # If we need to send a payload, then the Content-Type header should be set
-            final_headers["Content-Type"] = self.media_type
-        extra: Dict[str, Any]
-        serializer = self._get_serializer()
-        if serializer is not None and self.body is not NOT_SET:
-            context = SerializerContext(case=self)
-            extra = serializer.as_werkzeug(context, self.body)
-        else:
-            extra = {}
-        return {
-            "method": self.method,
-            "path": self.operation.schema.get_full_path(self.formatted_path),
-            # Convert to a regular dictionary, as we use `CaseInsensitiveDict` which is not supported by Werkzeug
-            "headers": dict(final_headers),
-            "query_string": self.query,
-            **extra,
-        }
-
-    def call_wsgi(self, app: Any = None, headers: Optional[Dict[str, str]] = None, **kwargs: Any) -> WSGIResponse:
-        application = app or self.app
-        if application is None:
-            raise RuntimeError(
-                "WSGI application instance is required. "
-                "Please, set `app` argument in the schema constructor or pass it to `call_wsgi`"
-            )
-        data = self.as_werkzeug_kwargs(headers)
-        client = werkzeug.Client(application, WSGIResponse)
-        with cookie_handler(client, self.cookies):
-            response = client.open(**data, **kwargs)
-        requests_kwargs = self.as_requests_kwargs(base_url=self.get_full_base_url(), headers=headers)
-        response.request = requests.Request(**requests_kwargs).prepare()
-        return response
-
-    def call_asgi(
-        self,
-        app: Any = None,
-        base_url: Optional[str] = None,
-        headers: Optional[Dict[str, str]] = None,
-        **kwargs: Any,
-    ) -> requests.Response:
-        application = app or self.app
-        if application is None:
-            raise RuntimeError(
-                "ASGI application instance is required. "
-                "Please, set `app` argument in the schema constructor or pass it to `call_asgi`"
-            )
-        if base_url is None:
-            base_url = self.get_full_base_url()
-        client = ASGIClient(application)
-
-        return self.call(base_url=base_url, session=client, headers=headers, **kwargs)
-
-    def validate_response(
-        self,
-        response: GenericResponse,
-        checks: Tuple["CheckFunction", ...] = (),
-        additional_checks: Tuple["CheckFunction", ...] = (),
-        code_sample_style: Optional[str] = None,
-    ) -> None:
-        """Validate application response.
-
-        By default, all available checks will be applied.
-
-        :param response: Application response.
-        :param checks: A tuple of check functions that accept ``response`` and ``case``.
-        :param additional_checks: A tuple of additional checks that will be executed after ones from the ``checks``
-            argument.
-        :param code_sample_style: Controls the style of code samples for failure reproduction.
-        """
-        from .checks import ALL_CHECKS  # pylint: disable=import-outside-toplevel
-
-        checks = checks or ALL_CHECKS
-        errors = []
-        for check in chain(checks, additional_checks):
-            try:
-                check(response, self)
-            except AssertionError as exc:
-                maybe_set_assertion_message(exc, check.__name__)
-                errors.append(exc)
-        if errors:
-            exception_cls = get_grouped_exception(self.operation.verbose_name, *errors)
-            formatted_errors = "\n\n".join(f"{idx}. {error.args[0]}" for idx, error in enumerate(errors, 1))
-            code_sample_style = (
-                CodeSampleStyle.from_str(code_sample_style)
-                if code_sample_style is not None
-                else self.operation.schema.code_sample_style
-            )
-            code_message = self._get_code_message(code_sample_style, response.request)
-            payload = get_response_payload(response)
-            raise exception_cls(
-                f"\n\n{formatted_errors}\n\n----------\n\nResponse payload: `{payload}`\n\n{code_message}"
-            )
-
-    def _get_code_message(self, code_sample_style: CodeSampleStyle, request: requests.PreparedRequest) -> str:
-        if code_sample_style == CodeSampleStyle.python:
-            code = self.get_code_to_reproduce(request=request)
-            return f"Run this Python code to reproduce this response: \n\n    {code}\n"
-        if code_sample_style == CodeSampleStyle.curl:
-            code = self.as_curl_command(headers=dict(request.headers))
-            return f"Run this cURL command to reproduce this response: \n\n    {code}\n"
-        raise ValueError(f"Unknown code sample style: {code_sample_style.name}")
-
-    def call_and_validate(
-        self,
-        base_url: Optional[str] = None,
-        session: Optional[requests.Session] = None,
-        headers: Optional[Dict[str, Any]] = None,
-        checks: Tuple["CheckFunction", ...] = (),
-        code_sample_style: Optional[str] = None,
-        **kwargs: Any,
-    ) -> requests.Response:
-        response = self.call(base_url, session, headers, **kwargs)
-        self.validate_response(response, checks, code_sample_style=code_sample_style)
-        return response
-
-    def get_full_url(self) -> str:
-        """Make a full URL to the current API operation, including query parameters."""
-        base_url = self.base_url or "http://localhost"
-        kwargs = self.as_requests_kwargs(base_url)
-        request = requests.Request(**kwargs)
-        prepared = requests.Session().prepare_request(request)  # type: ignore
-        return cast(str, prepared.url)
-
-    def partial_deepcopy(self) -> "Case":
-        return self.__class__(
-            operation=self.operation.partial_deepcopy(),
-            data_generation_method=self.data_generation_method,
-            path_parameters=deepcopy(self.path_parameters),
-            headers=deepcopy(self.headers),
-            cookies=deepcopy(self.cookies),
-            query=deepcopy(self.query),
-            body=deepcopy(self.body),
-        )
-
-
-BACKSLASH = "\\"
-SINGLE_QUOTE = "'"
-ESCAPED_SINGLE_QUOTE = "\\'"
-
-
-def _escape_single_quotes(url: str) -> str:
-    """Escape single quotes in a string, so it is usable as in generated Python code.
-
-    The usual ``str.replace`` is not suitable as it may convert already escaped quotes to not-escaped.
-    """
-    output = StringIO()
-    url_chars = zip_longest(url, url[1:])
-    for pair in url_chars:
-        if pair == (BACKSLASH, BACKSLASH):
-            # Escaped backslash
-            output.write(BACKSLASH)
-            output.write(BACKSLASH)
-            next(url_chars)
-        elif pair == (BACKSLASH, SINGLE_QUOTE):
-            # Already escaped quote - write it and skip one char to avoid double-escaping
-            output.write(ESCAPED_SINGLE_QUOTE)
-            next(url_chars)
-        elif pair[0] == SINGLE_QUOTE:
-            # Escape single quote on first occurrence. It can't be already escaped one as it is handled above
-            output.write(ESCAPED_SINGLE_QUOTE)
-        else:
-            # Default case
-            output.write(pair[0])
-    return output.getvalue()
-
-
-@contextmanager
-def cookie_handler(client: werkzeug.Client, cookies: Optional[Cookies]) -> Generator[None, None, None]:
-    """Set cookies required for a call."""
-    if not cookies:
-        yield
-    else:
-        for key, value in cookies.items():
-            client.set_cookie("localhost", key, value)
-        yield
-        for key in cookies:
-            client.delete_cookie("localhost", key)
-
-
-P = TypeVar("P", bound=Parameter)
-D = TypeVar("D")
-
-
-@attr.s  # pragma: no mutate
-class OperationDefinition(Generic[P, D]):
-    """A wrapper to store not resolved API operation definitions.
-
-    To prevent recursion errors we need to store definitions without resolving references. But operation definitions
-    itself can be behind a reference (when there is a ``$ref`` in ``paths`` values), therefore we need to store this
-    scope change to have a proper reference resolving later.
-    """
-
-    raw: D = attr.ib()  # pragma: no mutate
-    resolved: D = attr.ib()  # pragma: no mutate
-    scope: str = attr.ib()  # pragma: no mutate
-    parameters: Sequence[P] = attr.ib()  # pragma: no mutate
-
-
-C = TypeVar("C", bound=Case)
-
-
-@attr.s(eq=False)  # pragma: no mutate
-class APIOperation(Generic[P, C]):
-    """A single operation defined in an API.
-
-    You can get one via a ``schema`` instance.
-
-    .. code-block:: python
-
-        # Get the POST /items operation
-        operation = schema["/items"]["POST"]
-
-    """
-
-    # `path` does not contain `basePath`
-    # Example <scheme>://<host>/<basePath>/users - "/users" is path
-    # https://swagger.io/docs/specification/2-0/api-host-and-base-path/
-    path: str = attr.ib()  # pragma: no mutate
-    method: str = attr.ib()  # pragma: no mutate
-    definition: OperationDefinition = attr.ib(repr=False)  # pragma: no mutate
-    schema: "BaseSchema" = attr.ib()  # pragma: no mutate
-    verbose_name: str = attr.ib()  # pragma: no mutate
-    app: Any = attr.ib(default=None)  # pragma: no mutate
-    base_url: Optional[str] = attr.ib(default=None)  # pragma: no mutate
-    path_parameters: ParameterSet[P] = attr.ib(factory=ParameterSet)  # pragma: no mutate
-    headers: ParameterSet[P] = attr.ib(factory=ParameterSet)  # pragma: no mutate
-    cookies: ParameterSet[P] = attr.ib(factory=ParameterSet)  # pragma: no mutate
-    query: ParameterSet[P] = attr.ib(factory=ParameterSet)  # pragma: no mutate
-    body: PayloadAlternatives[P] = attr.ib(factory=PayloadAlternatives)  # pragma: no mutate
-    case_cls: Type[C] = attr.ib(default=Case)  # type: ignore
-
-    @verbose_name.default
-    def _verbose_name_default(self) -> str:
-        return f"{self.method.upper()} {self.full_path}"
-
-    @property
-    def full_path(self) -> str:
-        return self.schema.get_full_path(self.path)
-
-    @property
-    def links(self) -> Dict[str, Dict[str, Any]]:
-        return self.schema.get_links(self)
-
-    def add_parameter(self, parameter: P) -> None:
-        """Add a new processed parameter to an API operation.
-
-        :param parameter: A parameter that will be used with this operation.
-        :rtype: None
-        """
-        lookup_table = {
-            "path": self.path_parameters,
-            "header": self.headers,
-            "cookie": self.cookies,
-            "query": self.query,
-            "body": self.body,
-        }
-        # If the parameter has a typo, then by default, there will be an error from `jsonschema` earlier.
-        # But if the user wants to skip schema validation, we choose to ignore a malformed parameter.
-        # In this case, we still might generate some tests for an API operation, but without this parameter,
-        # which is better than skip the whole operation from testing.
-        if parameter.location in lookup_table:
-            container = lookup_table[parameter.location]
-            container.add(parameter)
-
-    def as_strategy(
-        self,
-        hooks: Optional["HookDispatcher"] = None,
-        data_generation_method: DataGenerationMethod = DataGenerationMethod.default(),
-    ) -> st.SearchStrategy:
-        """Turn this API operation into a Hypothesis strategy."""
-        strategy = self.schema.get_case_strategy(self, hooks, data_generation_method)
-
-        def _apply_hooks(dispatcher: HookDispatcher, _strategy: st.SearchStrategy[Case]) -> st.SearchStrategy[Case]:
-            for hook in dispatcher.get_all_by_name("before_generate_case"):
-                _strategy = hook(HookContext(self), _strategy)
-            return _strategy
-
-        strategy = _apply_hooks(GLOBAL_HOOK_DISPATCHER, strategy)
-        strategy = _apply_hooks(self.schema.hooks, strategy)
-        if hooks is not None:
-            strategy = _apply_hooks(hooks, strategy)
-        return strategy
-
-    def get_strategies_from_examples(self) -> List[st.SearchStrategy[Case]]:
-        """Get examples from the API operation."""
-        return self.schema.get_strategies_from_examples(self)
-
-    def get_stateful_tests(self, response: GenericResponse, stateful: Optional["Stateful"]) -> Sequence["StatefulTest"]:
-        return self.schema.get_stateful_tests(response, self, stateful)
-
-    def get_parameter_serializer(self, location: str) -> Optional[Callable]:
-        """Get a function that serializes parameters for the given location.
-
-        It handles serializing data into various `collectionFormat` options and similar.
-        Note that payload is handled by this function - it is handled by serializers.
-        """
-        return self.schema.get_parameter_serializer(self, location)
-
-    def prepare_multipart(self, form_data: FormData) -> Tuple[Optional[List], Optional[Dict[str, Any]]]:
-        return self.schema.prepare_multipart(form_data, self)
-
-    def get_request_payload_content_types(self) -> List[str]:
-        return self.schema.get_request_payload_content_types(self)
-
-    def partial_deepcopy(self) -> "APIOperation":
-        return self.__class__(
-            path=self.path,  # string, immutable
-            method=self.method,  # string, immutable
-            definition=deepcopy(self.definition),
-            schema=self.schema.clone(),  # shallow copy
-            app=self.app,  # not deepcopyable
-            base_url=self.base_url,  # string, immutable
-            path_parameters=deepcopy(self.path_parameters),
-            headers=deepcopy(self.headers),
-            cookies=deepcopy(self.cookies),
-            query=deepcopy(self.query),
-            body=deepcopy(self.body),
-        )
-
-    def clone(self, **components: Any) -> "APIOperation":
-        """Create a new instance of this API operation with updated components."""
-        return self.__class__(
-            path=self.path,
-            method=self.method,
-            verbose_name=self.verbose_name,
-            definition=self.definition,
-            schema=self.schema,
-            app=self.app,
-            base_url=self.base_url,
-            path_parameters=components["path_parameters"],
-            query=components["query"],
-            headers=components["headers"],
-            cookies=components["cookies"],
-            body=components["body"],
-        )
-
-    def make_case(
-        self,
-        *,
-        path_parameters: Optional[PathParameters] = None,
-        headers: Optional[Headers] = None,
-        cookies: Optional[Cookies] = None,
-        query: Optional[Query] = None,
-        body: Union[Body, NotSet] = NOT_SET,
-        media_type: Optional[str] = None,
-    ) -> C:
-        """Create a new example for this API operation.
-
-        The main use case is constructing Case instances completely manually, without data generation.
-        """
-        return self.schema.make_case(
-            case_cls=self.case_cls,
-            operation=self,
-            path_parameters=path_parameters,
-            headers=headers,
-            cookies=cookies,
-            query=query,
-            body=body,
-            media_type=media_type,
-        )
-
-    @property
-    def operation_reference(self) -> str:
-        path = self.path.replace("~", "~0").replace("/", "~1")
-        return f"#/paths/{path}/{self.method}"
-
-    def validate_response(self, response: GenericResponse) -> None:
-        """Validate API response for conformance.
-
-        :raises CheckFailed: If the response does not conform to the API schema.
-        """
-        return self.schema.validate_response(self, response)
-
-    def is_response_valid(self, response: GenericResponse) -> bool:
-        """Validate API response for conformance."""
-        try:
-            self.validate_response(response)
-            return True
-        except CheckFailed:
-            return False
-
-
-# backward-compatibility
-Endpoint = APIOperation
-
-
-class Status(str, Enum):
-    """Status of an action or multiple actions."""
-
-    success = "success"  # pragma: no mutate
-    failure = "failure"  # pragma: no mutate
-    error = "error"  # pragma: no mutate
-
-
-@attr.s(slots=True, repr=False)  # pragma: no mutate
-class Check:
-    """Single check run result."""
-
-    name: str = attr.ib()  # pragma: no mutate
-    value: Status = attr.ib()  # pragma: no mutate
-    response: Optional[GenericResponse] = attr.ib()  # pragma: no mutate
-    elapsed: float = attr.ib()  # pragma: no mutate
-    example: Case = attr.ib()  # pragma: no mutate
-    message: Optional[str] = attr.ib(default=None)  # pragma: no mutate
-    # Failure-specific context
-    context: Optional[FailureContext] = attr.ib(default=None)  # pragma: no mutate
-    request: Optional[requests.PreparedRequest] = attr.ib(default=None)  # pragma: no mutate
-
-
-@attr.s(slots=True, repr=False)  # pragma: no mutate
-class Request:
-    """Request data extracted from `Case`."""
-
-    method: str = attr.ib()  # pragma: no mutate
-    uri: str = attr.ib()  # pragma: no mutate
-    body: Optional[str] = attr.ib()  # pragma: no mutate
-    headers: Headers = attr.ib()  # pragma: no mutate
-
-    @classmethod
-    def from_case(cls, case: Case, session: requests.Session) -> "Request":
-        """Create a new `Request` instance from `Case`."""
-        base_url = case.get_full_base_url()
-        kwargs = case.as_requests_kwargs(base_url)
-        request = requests.Request(**kwargs)
-        prepared = session.prepare_request(request)  # type: ignore
-        return cls.from_prepared_request(prepared)
-
-    @classmethod
-    def from_prepared_request(cls, prepared: requests.PreparedRequest) -> "Request":
-        """A prepared request version is already stored in `requests.Response`."""
-        body = prepared.body
-
-        if isinstance(body, str):
-            # can be a string for `application/x-www-form-urlencoded`
-            body = body.encode("utf-8")
-
-        # these values have `str` type at this point
-        uri = cast(str, prepared.url)
-        method = cast(str, prepared.method)
-        return cls(
-            uri=uri,
-            method=method,
-            headers={key: [value] for (key, value) in prepared.headers.items()},
-            body=base64.b64encode(body).decode() if body is not None else body,
-        )
-
-
-def serialize_payload(payload: bytes) -> str:
-    return base64.b64encode(payload).decode()
-
-
-@attr.s(slots=True, repr=False)  # pragma: no mutate
-class Response:
-    """Unified response data."""
-
-    status_code: int = attr.ib()  # pragma: no mutate
-    message: str = attr.ib()  # pragma: no mutate
-    headers: Dict[str, List[str]] = attr.ib()  # pragma: no mutate
-    body: Optional[str] = attr.ib()  # pragma: no mutate
-    encoding: Optional[str] = attr.ib()  # pragma: no mutate
-    http_version: str = attr.ib()  # pragma: no mutate
-    elapsed: float = attr.ib()  # pragma: no mutate
-
-    @classmethod
-    def from_requests(cls, response: requests.Response) -> "Response":
-        """Create a response from requests.Response."""
-        headers = {name: response.raw.headers.getlist(name) for name in response.raw.headers.keys()}
-        # Similar to http.client:319 (HTTP version detection in stdlib's `http` package)
-        http_version = "1.0" if response.raw.version == 10 else "1.1"
-
-        def is_empty(_response: requests.Response) -> bool:
-            # Assume the response is empty if:
-            #   - no `Content-Length` header
-            #   - no chunks when iterating over its content
-            return "Content-Length" not in headers and list(_response.iter_content()) == []
-
-        body = None if is_empty(response) else serialize_payload(response.content)
-        return cls(
-            status_code=response.status_code,
-            message=response.reason,
-            body=body,
-            encoding=response.encoding,
-            headers=headers,
-            http_version=http_version,
-            elapsed=response.elapsed.total_seconds(),
-        )
-
-    @classmethod
-    def from_wsgi(cls, response: WSGIResponse, elapsed: float) -> "Response":
-        """Create a response from WSGI response."""
-        message = http.client.responses.get(response.status_code, "UNKNOWN")
-        headers = {name: response.headers.getlist(name) for name in response.headers.keys()}
-        # Note, this call ensures that `response.response` is a sequence, which is needed for comparison
-        data = response.get_data()
-        body = None if response.response == [] else serialize_payload(data)
-        encoding: Optional[str]
-        if body is not None:
-            encoding = response.mimetype_params.get("charset", response.charset)
-        else:
-            encoding = None
-        return cls(
-            status_code=response.status_code,
-            message=message,
-            body=body,
-            encoding=encoding,
-            headers=headers,
-            http_version="1.1",
-            elapsed=elapsed,
-        )
-
-
-@attr.s(slots=True)  # pragma: no mutate
-class Interaction:
-    """A single interaction with the target app."""
-
-    request: Request = attr.ib()  # pragma: no mutate
-    response: Response = attr.ib()  # pragma: no mutate
-    checks: List[Check] = attr.ib()  # pragma: no mutate
-    status: Status = attr.ib()  # pragma: no mutate
-    recorded_at: str = attr.ib(factory=lambda: datetime.datetime.now().isoformat())  # pragma: no mutate
-
-    @classmethod
-    def from_requests(cls, response: requests.Response, status: Status, checks: List[Check]) -> "Interaction":
-        return cls(
-            request=Request.from_prepared_request(response.request),
-            response=Response.from_requests(response),
-            status=status,
-            checks=checks,
-        )
-
-    @classmethod
-    def from_wsgi(
-        cls,
-        case: Case,
-        response: WSGIResponse,
-        headers: Dict[str, Any],
-        elapsed: float,
-        status: Status,
-        checks: List[Check],
-    ) -> "Interaction":
-        session = requests.Session()
-        session.headers.update(headers)
-        return cls(
-            request=Request.from_case(case, session),
-            response=Response.from_wsgi(response, elapsed),
-            status=status,
-            checks=checks,
-        )
-
-
-@attr.s(slots=True, repr=False)  # pragma: no mutate
-class TestResult:
-    """Result of a single test."""
-
-    __test__ = False
-
-    method: str = attr.ib()  # pragma: no mutate
-    path: str = attr.ib()  # pragma: no mutate
-    verbose_name: str = attr.ib()  # pragma: no mutate
-    data_generation_method: DataGenerationMethod = attr.ib()  # pragma: no mutate
-    checks: List[Check] = attr.ib(factory=list)  # pragma: no mutate
-    errors: List[Tuple[Exception, Optional[Case]]] = attr.ib(factory=list)  # pragma: no mutate
-    interactions: List[Interaction] = attr.ib(factory=list)  # pragma: no mutate
-    logs: List[LogRecord] = attr.ib(factory=list)  # pragma: no mutate
-    is_errored: bool = attr.ib(default=False)  # pragma: no mutate
-    seed: Optional[int] = attr.ib(default=None)  # pragma: no mutate
-    # To show a proper reproduction code if an error happens and there is no way to get actual headers that were
-    # sent over the network. Or there could be no actual requests at all
-    overridden_headers: Optional[Dict[str, Any]] = attr.ib(default=None)  # pragma: no mutate
-
-    def mark_errored(self) -> None:
-        self.is_errored = True
-
-    @property
-    def has_errors(self) -> bool:
-        return bool(self.errors)
-
-    @property
-    def has_failures(self) -> bool:
-        return any(check.value == Status.failure for check in self.checks)
-
-    @property
-    def has_logs(self) -> bool:
-        return bool(self.logs)
-
-    def add_success(self, name: str, example: Case, response: GenericResponse, elapsed: float) -> Check:
-        check = Check(
-            name=name, value=Status.success, response=response, elapsed=elapsed, example=example, request=None
-        )
-        self.checks.append(check)
-        return check
-
-    def add_failure(
-        self,
-        name: str,
-        example: Case,
-        response: Optional[GenericResponse],
-        elapsed: float,
-        message: str,
-        context: Optional[FailureContext],
-        request: Optional[requests.PreparedRequest] = None,
-    ) -> Check:
-        check = Check(
-            name=name,
-            value=Status.failure,
-            response=response,
-            elapsed=elapsed,
-            example=example,
-            message=message,
-            context=context,
-            request=request,
-        )
-        self.checks.append(check)
-        return check
-
-    def add_error(self, exception: Exception, example: Optional[Case] = None) -> None:
-        self.errors.append((exception, example))
-
-    def store_requests_response(self, response: requests.Response, status: Status, checks: List[Check]) -> None:
-        self.interactions.append(Interaction.from_requests(response, status, checks))
-
-    def store_wsgi_response(
-        self,
-        case: Case,
-        response: WSGIResponse,
-        headers: Dict[str, Any],
-        elapsed: float,
-        status: Status,
-        checks: List[Check],
-    ) -> None:
-        self.interactions.append(Interaction.from_wsgi(case, response, headers, elapsed, status, checks))
-
-
-@attr.s(slots=True, repr=False)  # pragma: no mutate
-class TestResultSet:
-    """Set of multiple test results."""
-
-    __test__ = False
-
-    results: List[TestResult] = attr.ib(factory=list)  # pragma: no mutate
-    generic_errors: List[InvalidSchema] = attr.ib(factory=list)  # pragma: no mutate
-
-    def __iter__(self) -> Iterator[TestResult]:
-        return iter(self.results)
-
-    @property
-    def is_empty(self) -> bool:
-        """If the result set contains no results."""
-        return len(self.results) == 0 and len(self.generic_errors) == 0
-
-    @property
-    def has_failures(self) -> bool:
-        """If any result has any failures."""
-        return any(result.has_failures for result in self)
-
-    @property
-    def has_errors(self) -> bool:
-        """If any result has any errors."""
-        return self.errored_count > 0
-
-    @property
-    def has_logs(self) -> bool:
-        """If any result has any captured logs."""
-        return any(result.has_logs for result in self)
-
-    def _count(self, predicate: Callable) -> int:
-        return sum(1 for result in self if predicate(result))
-
-    @property
-    def passed_count(self) -> int:
-        return self._count(lambda result: not result.has_errors and not result.has_failures)
-
-    @property
-    def failed_count(self) -> int:
-        return self._count(lambda result: result.has_failures and not result.is_errored)
-
-    @property
-    def errored_count(self) -> int:
-        return self._count(lambda result: result.has_errors or result.is_errored) + len(self.generic_errors)
-
-    @property
-    def total(self) -> Dict[str, Dict[Union[str, Status], int]]:
-        """An aggregated statistic about test results."""
-        output: Dict[str, Dict[Union[str, Status], int]] = {}
-        for item in self.results:
-            for check in item.checks:
-                output.setdefault(check.name, Counter())
-                output[check.name][check.value] += 1
-                output[check.name]["total"] += 1
-        # Avoid using Counter, since its behavior could harm in other places:
-        # `if not total["unknown"]:` - this will lead to the branch execution
-        # It is better to let it fail if there is a wrong key
-        return {key: dict(value) for key, value in output.items()}
-
-    def append(self, item: TestResult) -> None:
-        """Add a new item to the results list."""
-        self.results.append(item)
-
-
-CheckFunction = Callable[[GenericResponse, Case], Optional[bool]]  # pragma: no mutate