schemathesis 3.39.7__py3-none-any.whl → 4.0.0a2__py3-none-any.whl

schemathesis/service/serialization.py

@@ -1,173 +0,0 @@
-from __future__ import annotations
-
-from dataclasses import asdict
-from typing import TYPE_CHECKING, Any, Callable, Dict, Optional, TypeVar, cast
-
-from ..internal.transformation import merge_recursively
-from ..runner import events
-from ..runner.serialization import _serialize_check
-
-if TYPE_CHECKING:
-    from ..stateful import events as stateful_events
-
-S = TypeVar("S", bound=events.ExecutionEvent)
-SerializeFunc = Callable[[S], Optional[Dict[str, Any]]]
-
-
-def serialize_initialized(event: events.Initialized) -> dict[str, Any] | None:
-    return {
-        "operations_count": event.operations_count,
-        "location": event.location or "",
-        "base_url": event.base_url,
-    }
-
-
-def serialize_before_probing(_: events.BeforeProbing) -> None:
-    return None
-
-
-def serialize_after_probing(event: events.AfterProbing) -> dict[str, Any] | None:
-    probes = event.probes or []
-    return {"probes": [probe.serialize() for probe in probes]}
-
-
-def serialize_before_analysis(_: events.BeforeAnalysis) -> None:
-    return None
-
-
-def serialize_after_analysis(event: events.AfterAnalysis) -> dict[str, Any] | None:
-    return event._serialize()
-
-
-def serialize_before_execution(event: events.BeforeExecution) -> dict[str, Any] | None:
-    return {
-        "correlation_id": event.correlation_id,
-        "verbose_name": event.verbose_name,
-        "data_generation_method": event.data_generation_method,
-    }
-
-
-def serialize_after_execution(event: events.AfterExecution) -> dict[str, Any] | None:
-    return {
-        "correlation_id": event.correlation_id,
-        "verbose_name": event.verbose_name,
-        "status": event.status,
-        "elapsed_time": event.elapsed_time,
-        "data_generation_method": event.data_generation_method,
-        "result": {
-            "checks": [_serialize_check(check) for check in event.result.checks],
-            "errors": [asdict(error) for error in event.result.errors],
-            "skip_reason": event.result.skip_reason,
-        },
-    }
-
-
-def serialize_interrupted(_: events.Interrupted) -> dict[str, Any] | None:
-    return None
-
-
-def serialize_internal_error(event: events.InternalError) -> dict[str, Any] | None:
-    return {
-        "type": event.type.value,
-        "subtype": event.subtype.value if event.subtype else event.subtype,
-        "title": event.title,
-        "message": event.message,
-        "extras": event.extras,
-        "exception_type": event.exception_type,
-        "exception": event.exception,
-        "exception_with_traceback": event.exception_with_traceback,
-    }
-
-
-def serialize_finished(event: events.Finished) -> dict[str, Any] | None:
-    return {
-        "generic_errors": [
-            {
-                "exception": error.exception,
-                "exception_with_traceback": error.exception_with_traceback,
-                "title": error.title,
-            }
-            for error in event.generic_errors
-        ],
-        "running_time": event.running_time,
-    }
-
-
-def serialize_stateful_event(event: events.StatefulEvent) -> dict[str, Any] | None:
-    return _serialize_stateful_event(event.data)
-
-
-def _serialize_stateful_event(event: stateful_events.StatefulEvent) -> dict[str, Any] | None:
-    return {"data": {event.__class__.__name__: event.asdict()}}
-
-
-def serialize_after_stateful_execution(event: events.AfterStatefulExecution) -> dict[str, Any] | None:
-    return {
-        "status": event.status,
-        "data_generation_method": event.data_generation_method,
-        "result": asdict(event.result),
-        "elapsed_time": event.elapsed_time,
-    }
-
-
-SERIALIZER_MAP = {
-    events.Initialized: serialize_initialized,
-    events.BeforeProbing: serialize_before_probing,
-    events.AfterProbing: serialize_after_probing,
-    events.BeforeAnalysis: serialize_before_analysis,
-    events.AfterAnalysis: serialize_after_analysis,
-    events.BeforeExecution: serialize_before_execution,
-    events.AfterExecution: serialize_after_execution,
-    events.Interrupted: serialize_interrupted,
-    events.InternalError: serialize_internal_error,
-    events.StatefulEvent: serialize_stateful_event,
-    events.AfterStatefulExecution: serialize_after_stateful_execution,
-    events.Finished: serialize_finished,
-}
-
-
-def serialize_event(
-    event: events.ExecutionEvent,
-    *,
-    on_initialized: SerializeFunc | None = None,
-    on_before_probing: SerializeFunc | None = None,
-    on_after_probing: SerializeFunc | None = None,
-    on_before_analysis: SerializeFunc | None = None,
-    on_after_analysis: SerializeFunc | None = None,
-    on_before_execution: SerializeFunc | None = None,
-    on_after_execution: SerializeFunc | None = None,
-    on_interrupted: SerializeFunc | None = None,
-    on_internal_error: SerializeFunc | None = None,
-    on_stateful_event: SerializeFunc | None = None,
-    on_after_stateful_execution: SerializeFunc | None = None,
-    on_finished: SerializeFunc | None = None,
-    extra: dict[str, Any] | None = None,
-) -> dict[str, dict[str, Any] | None]:
-    """Turn an event into JSON-serializable structure."""
-    # Use the explicitly provided serializer for this event and fallback to default one if it is not provided
-    serializer = {
-        events.Initialized: on_initialized,
-        events.BeforeProbing: on_before_probing,
-        events.AfterProbing: on_after_probing,
-        events.BeforeAnalysis: on_before_analysis,
-        events.AfterAnalysis: on_after_analysis,
-        events.BeforeExecution: on_before_execution,
-        events.AfterExecution: on_after_execution,
-        events.Interrupted: on_interrupted,
-        events.InternalError: on_internal_error,
-        events.StatefulEvent: on_stateful_event,
-        events.AfterStatefulExecution: on_after_stateful_execution,
-        events.Finished: on_finished,
-    }.get(event.__class__)
-    if serializer is None:
-        serializer = cast(SerializeFunc, SERIALIZER_MAP[event.__class__])
-    data = serializer(event)
-    if extra is not None:
-        # If `extra` is present, then merge it with the serialized data. If serialized data is empty, then replace it
-        # with `extra` value
-        if data is None:
-            data = extra
-        else:
-            data = merge_recursively(data, extra)
-    # Externally tagged structure
-    return {event.__class__.__name__: data}

schemathesis/service/usage.py

@@ -1,66 +0,0 @@
-from __future__ import annotations
-
-import sys
-from typing import Any
-
-import click
-from click.types import StringParamType
-
-from .. import cli, hooks
-
-
-def collect(args: list[str] | None = None) -> dict[str, Any] | None:
-    """Collect anonymized CLI usage data."""
-    context: click.Context | None = click.get_current_context(silent=True)
-    if context is not None and not sys.argv[0].endswith("pytest"):
-        args = args or sys.argv[2:]
-        parameters, _, types = parse_cli_args(context, args)
-        parameters_data: dict[str, dict[str, Any]] = {}
-        used_headers: list[str] = []
-        schema = parameters["schema"]
-        app = parameters.get("app")
-        if not schema:
-            schema_kind = None
-        else:
-            schema_kind = cli.callbacks.parse_schema_kind(schema, app).name
-        usage = {
-            "schema_kind": schema_kind,
-            "parameters": parameters_data,
-            "used_headers": used_headers,
-            "hooks": hooks.collect_statistic(),
-        }
-        types_iter = iter(types)
-        for option, value in parameters.items():
-            option_type = next(types_iter)
-            if isinstance(option_type, click.Argument):
-                continue
-            if option_type.multiple:
-                # Forward the iterator to the next option type
-                for _ in range(len(value) - 1):
-                    next(types_iter)
-            entry = _collect_option(option, option_type, used_headers, value)
-            if entry:
-                parameters_data[option] = entry
-        return usage
-    return None
-
-
-def _collect_option(option: str, option_type: click.Parameter, used_headers: list[str], value: Any) -> dict[str, Any]:
-    entry = {}
-    if isinstance(option_type.type, (StringParamType, click.types.File)):
-        if option == "headers" and value:
-            used_headers.extend(header.split(":", 1)[0] for header in value)
-        else:
-            # Free-form values are replaced with their number of occurrences, to avoid sending sensitive info
-            if option_type.multiple:
-                entry["count"] = len(value)
-            else:
-                entry["count"] = 1
-    else:
-        entry["value"] = value
-    return entry
-
-
-def parse_cli_args(context: click.Context, args: list[str]) -> tuple[dict[str, Any], list, list[click.Parameter]]:
-    parser = cli.run.make_parser(context)
-    return parser.parse_args(args=args)

schemathesis/specs/graphql/loaders.py

@@ -1,364 +0,0 @@
-from __future__ import annotations
-
-import json
-import pathlib
-from functools import lru_cache
-from json import JSONDecodeError
-from typing import IO, TYPE_CHECKING, Any, Callable, Dict, NoReturn, cast
-
-from ...code_samples import CodeSampleStyle
-from ...constants import DEFAULT_RESPONSE_TIMEOUT, WAIT_FOR_SCHEMA_INTERVAL
-from ...exceptions import SchemaError, SchemaErrorType
-from ...generation import (
-    DEFAULT_DATA_GENERATION_METHODS,
-    DataGenerationMethod,
-    DataGenerationMethodInput,
-    GenerationConfig,
-)
-from ...hooks import HookContext, dispatch
-from ...internal.output import OutputConfig
-from ...internal.validation import require_relative_url
-from ...loaders import load_schema_from_url
-from ...throttling import build_limiter
-from ...transports.headers import setup_default_headers
-from ...types import PathLike, Specification
-
-if TYPE_CHECKING:
-    from graphql import DocumentNode
-    from pyrate_limiter import Limiter
-
-    from ...transports.responses import GenericResponse
-    from .schemas import GraphQLSchema
-
-
-@lru_cache
-def get_introspection_query() -> str:
-    import graphql
-
-    return graphql.get_introspection_query()
-
-
-@lru_cache
-def get_introspection_query_ast() -> DocumentNode:
-    import graphql
-
-    query = get_introspection_query()
-    return graphql.parse(query)
-
-
-def from_path(
-    path: PathLike,
-    *,
-    app: Any = None,
-    base_url: str | None = None,
-    data_generation_methods: DataGenerationMethodInput = DEFAULT_DATA_GENERATION_METHODS,
-    generation_config: GenerationConfig | None = None,
-    output_config: OutputConfig | None = None,
-    code_sample_style: str = CodeSampleStyle.default().name,
-    rate_limit: str | None = None,
-    encoding: str = "utf8",
-    sanitize_output: bool = True,
-) -> GraphQLSchema:
-    """Load GraphQL schema via a file from an OS path.
-
-    :param path: A path to the schema file.
-    :param encoding: The name of the encoding used to decode the file.
-    """
-    with open(path, encoding=encoding) as fd:
-        return from_file(
-            fd,
-            app=app,
-            base_url=base_url,
-            data_generation_methods=data_generation_methods,
-            code_sample_style=code_sample_style,
-            generation_config=generation_config,
-            output_config=output_config,
-            location=pathlib.Path(path).absolute().as_uri(),
-            rate_limit=rate_limit,
-            sanitize_output=sanitize_output,
-        )
-
-
-def extract_schema_from_response(response: GenericResponse) -> dict[str, Any]:
-    from requests import Response
-
-    try:
-        if isinstance(response, Response):
-            decoded = response.json()
-        else:
-            decoded = response.json
-    except JSONDecodeError as exc:
-        raise SchemaError(
-            SchemaErrorType.UNEXPECTED_CONTENT_TYPE,
-            "Received unsupported content while expecting a JSON payload for GraphQL",
-        ) from exc
-    return decoded
-
-
-def from_url(
-    url: str,
-    *,
-    app: Any = None,
-    base_url: str | None = None,
-    port: int | None = None,
-    data_generation_methods: DataGenerationMethodInput = DEFAULT_DATA_GENERATION_METHODS,
-    generation_config: GenerationConfig | None = None,
-    code_sample_style: str = CodeSampleStyle.default().name,
-    wait_for_schema: float | None = None,
-    rate_limit: str | None = None,
-    sanitize_output: bool = True,
-    **kwargs: Any,
-) -> GraphQLSchema:
-    """Load GraphQL schema from the network.
-
-    :param url: Schema URL.
-    :param Optional[str] base_url: Base URL to send requests to.
-    :param Optional[int] port: An optional port if you don't want to pass the ``base_url`` parameter, but only to change
-                               port in ``url``.
-    :param app: A WSGI app instance.
-    :return: GraphQLSchema
-    """
-    import backoff
-    import requests
-
-    setup_default_headers(kwargs)
-    kwargs.setdefault("json", {"query": get_introspection_query()})
-    if port:
-        from yarl import URL
-
-        url = str(URL(url).with_port(port))
-        if not base_url:
-            base_url = url
-
-    if wait_for_schema is not None:
-
-        @backoff.on_exception(  # type: ignore
-            backoff.constant,
-            requests.exceptions.ConnectionError,
-            max_time=wait_for_schema,
-            interval=WAIT_FOR_SCHEMA_INTERVAL,
-        )
-        def _load_schema(_uri: str, **_kwargs: Any) -> requests.Response:
-            return requests.post(_uri, **_kwargs)
-
-    else:
-        _load_schema = requests.post
-
-    kwargs.setdefault("timeout", DEFAULT_RESPONSE_TIMEOUT / 1000)
-    response = load_schema_from_url(lambda: _load_schema(url, **kwargs))
-    raw_schema = extract_schema_from_response(response)
-    return from_dict(
-        raw_schema=raw_schema,
-        location=url,
-        base_url=base_url,
-        app=app,
-        data_generation_methods=data_generation_methods,
-        code_sample_style=code_sample_style,
-        rate_limit=rate_limit,
-        sanitize_output=sanitize_output,
-    )
-
-
-def from_file(
-    file: IO[str] | str,
-    *,
-    app: Any = None,
-    base_url: str | None = None,
-    data_generation_methods: DataGenerationMethodInput = DEFAULT_DATA_GENERATION_METHODS,
-    generation_config: GenerationConfig | None = None,
-    output_config: OutputConfig | None = None,
-    code_sample_style: str = CodeSampleStyle.default().name,
-    location: str | None = None,
-    rate_limit: str | None = None,
-    sanitize_output: bool = True,
-) -> GraphQLSchema:
-    """Load GraphQL schema from a file descriptor or a string.
-
-    :param file: Could be a file descriptor, string or bytes.
-    """
-    import graphql
-
-    if isinstance(file, str):
-        data = file
-    else:
-        data = file.read()
-    try:
-        document = graphql.build_schema(data)
-        result = graphql.execute(document, get_introspection_query_ast())
-        # TYPES: We don't pass `is_awaitable` above, therefore `result` is of the `ExecutionResult` type
-        result = cast(graphql.ExecutionResult, result)
-        # TYPES:
-        #  - `document` is a valid schema, because otherwise `build_schema` will rise an error;
-        #  - `INTROSPECTION_QUERY` is a valid query - it is known upfront;
-        # Therefore the execution result is always valid at this point and `result.data` is not `None`
-        raw_schema = cast(Dict[str, Any], result.data)
-    except Exception as exc:
-        try:
-            raw_schema = json.loads(data)
-            if not isinstance(raw_schema, dict) or "__schema" not in raw_schema:
-                _on_invalid_schema(exc)
-        except json.JSONDecodeError:
-            _on_invalid_schema(exc, extras=[entry for entry in str(exc).splitlines() if entry])
-    return from_dict(
-        raw_schema,
-        app=app,
-        base_url=base_url,
-        data_generation_methods=data_generation_methods,
-        generation_config=generation_config,
-        output_config=output_config,
-        code_sample_style=code_sample_style,
-        location=location,
-        rate_limit=rate_limit,
-        sanitize_output=sanitize_output,
-    )
-
-
-def _on_invalid_schema(exc: Exception, extras: list[str] | None = None) -> NoReturn:
-    raise SchemaError(
-        SchemaErrorType.GRAPHQL_INVALID_SCHEMA,
-        "The provided API schema does not appear to be a valid GraphQL schema",
-        extras=extras or [],
-    ) from exc
-
-
-def from_dict(
-    raw_schema: dict[str, Any],
-    *,
-    app: Any = None,
-    base_url: str | None = None,
-    location: str | None = None,
-    data_generation_methods: DataGenerationMethodInput = DEFAULT_DATA_GENERATION_METHODS,
-    generation_config: GenerationConfig | None = None,
-    output_config: OutputConfig | None = None,
-    code_sample_style: str = CodeSampleStyle.default().name,
-    rate_limit: str | None = None,
-    sanitize_output: bool = True,
-) -> GraphQLSchema:
-    """Load GraphQL schema from a Python dictionary.
-
-    :param dict raw_schema: A schema to load.
-    :param Optional[str] location: Optional schema location. Either a full URL or a filesystem path.
-    :param Optional[str] base_url: Base URL to send requests to.
-    :param app: A WSGI app instance.
-    :return: GraphQLSchema
-    """
-    from ... import transports
-    from .schemas import GraphQLSchema
-
-    _code_sample_style = CodeSampleStyle.from_str(code_sample_style)
-    hook_context = HookContext()
-    if "data" in raw_schema:
-        raw_schema = raw_schema["data"]
-    dispatch("before_load_schema", hook_context, raw_schema)
-    rate_limiter: Limiter | None = None
-    if rate_limit is not None:
-        rate_limiter = build_limiter(rate_limit)
-    instance = GraphQLSchema(
-        raw_schema,
-        specification=Specification.GRAPHQL,
-        location=location,
-        base_url=base_url,
-        app=app,
-        data_generation_methods=DataGenerationMethod.ensure_list(data_generation_methods),
-        generation_config=generation_config or GenerationConfig(),
-        output_config=output_config or OutputConfig(),
-        code_sample_style=_code_sample_style,
-        rate_limiter=rate_limiter,
-        sanitize_output=sanitize_output,
-        transport=transports.get(app),
-    )  # type: ignore
-    dispatch("after_load_schema", hook_context, instance)
-    return instance
-
-
-def from_wsgi(
-    schema_path: str,
-    app: Any,
-    *,
-    base_url: str | None = None,
-    data_generation_methods: DataGenerationMethodInput = DEFAULT_DATA_GENERATION_METHODS,
-    generation_config: GenerationConfig | None = None,
-    output_config: OutputConfig | None = None,
-    code_sample_style: str = CodeSampleStyle.default().name,
-    rate_limit: str | None = None,
-    sanitize_output: bool = True,
-    **kwargs: Any,
-) -> GraphQLSchema:
-    """Load GraphQL schema from a WSGI app.
-
-    :param str schema_path: An in-app relative URL to the schema.
-    :param app: A WSGI app instance.
-    :param Optional[str] base_url: Base URL to send requests to.
-    :return: GraphQLSchema
-    """
-    from werkzeug import Client
-
-    from ...transports.responses import WSGIResponse
-
-    require_relative_url(schema_path)
-    setup_default_headers(kwargs)
-    kwargs.setdefault("json", {"query": get_introspection_query()})
-    client = Client(app, WSGIResponse)
-    response = load_schema_from_url(lambda: client.post(schema_path, **kwargs))
-    raw_schema = extract_schema_from_response(response)
-    return from_dict(
-        raw_schema=raw_schema,
-        location=schema_path,
-        base_url=base_url,
-        app=app,
-        data_generation_methods=data_generation_methods,
-        generation_config=generation_config,
-        output_config=output_config,
-        code_sample_style=code_sample_style,
-        rate_limit=rate_limit,
-        sanitize_output=sanitize_output,
-    )
-
-
-def from_asgi(
-    schema_path: str,
-    app: Any,
-    *,
-    base_url: str | None = None,
-    data_generation_methods: DataGenerationMethodInput = DEFAULT_DATA_GENERATION_METHODS,
-    generation_config: GenerationConfig | None = None,
-    output_config: OutputConfig | None = None,
-    code_sample_style: str = CodeSampleStyle.default().name,
-    rate_limit: str | None = None,
-    sanitize_output: bool = True,
-    **kwargs: Any,
-) -> GraphQLSchema:
-    """Load GraphQL schema from an ASGI app.
-
-    :param str schema_path: An in-app relative URL to the schema.
-    :param app: An ASGI app instance.
-    :param Optional[str] base_url: Base URL to send requests to.
-    """
-    from starlette_testclient import TestClient as ASGIClient
-
-    require_relative_url(schema_path)
-    setup_default_headers(kwargs)
-    kwargs.setdefault("json", {"query": get_introspection_query()})
-    client = ASGIClient(app)
-    response = load_schema_from_url(lambda: client.post(schema_path, **kwargs))
-    raw_schema = extract_schema_from_response(response)
-    return from_dict(
-        raw_schema=raw_schema,
-        location=schema_path,
-        base_url=base_url,
-        app=app,
-        data_generation_methods=data_generation_methods,
-        generation_config=generation_config,
-        output_config=output_config,
-        code_sample_style=code_sample_style,
-        rate_limit=rate_limit,
-        sanitize_output=sanitize_output,
-    )
-
-
-def get_loader_for_app(app: Any) -> Callable:
-    from ...transports.asgi import is_asgi_app
-
-    if is_asgi_app(app):
-        return from_asgi
-    return from_wsgi

schemathesis/specs/openapi/expressions/context.py

@@ -1,16 +0,0 @@
-from __future__ import annotations
-
-from dataclasses import dataclass
-from typing import TYPE_CHECKING
-
-if TYPE_CHECKING:
-    from ....models import Case
-    from ....transports.responses import GenericResponse
-
-
-@dataclass
-class ExpressionContext:
-    """Context in what an expression are evaluated."""
-
-    response: GenericResponse
-    case: Case