schemathesis 4.0.0a9__py3-none-any.whl → 4.0.0a11__py3-none-any.whl

schemathesis/config/_error.py

@@ -0,0 +1,156 @@
+from __future__ import annotations
+
+import difflib
+from typing import TYPE_CHECKING
+
+from schemathesis.core.errors import SchemathesisError
+
+if TYPE_CHECKING:
+    from jsonschema import ValidationError
+
+
+class ConfigError(SchemathesisError):
+    """Invalid configuration."""
+
+    @classmethod
+    def from_validation_error(cls, error: ValidationError) -> ConfigError:
+        message = error.message
+        if error.validator == "enum":
+            message = _format_enum_error(error)
+        elif error.validator == "minimum":
+            message = _format_minimum_error(error)
+        elif error.validator == "required":
+            message = _format_required_error(error)
+        elif error.validator == "type":
+            message = _format_type_error(error)
+        elif error.validator == "additionalProperties":
+            message = _format_additional_properties_error(error)
+        elif error.validator == "anyOf":
+            message = _format_anyof_error(error)
+        return cls(message)
+
+
+def _format_minimum_error(error: ValidationError) -> str:
+    assert isinstance(error.validator_value, (int, float))
+    section = path_to_section_name(list(error.path)[:-1] if error.path else [])
+    assert error.path
+
+    prop_name = error.path[-1]
+    min_value = error.validator_value
+    actual_value = error.instance
+
+    return (
+        f"Error in {section} section:\n  Value too low:\n\n"
+        f"  - '{prop_name}' → Must be at least {min_value}, but got {actual_value}."
+    )
+
+
+def _format_required_error(error: ValidationError) -> str:
+    assert isinstance(error.validator_value, list)
+    missing_keys = sorted(set(error.validator_value) - set(error.instance))
+
+    section = path_to_section_name(list(error.path))
+
+    details = "\n".join(f"  - '{key}'" for key in missing_keys)
+    return f"Error in {section} section:\n  Missing required properties:\n\n{details}\n\n"
+
+
+def _format_enum_error(error: ValidationError) -> str:
+    assert isinstance(error.validator_value, list)
+    valid_values = sorted(error.validator_value)
+
+    path = list(error.path)
+
+    if path and isinstance(path[-1], int):
+        idx = path[-1]
+        prop_name = path[-2]
+        section_path = path[:-2]
+        description = f"Item #{idx} in the '{prop_name}' array"
+    else:
+        prop_name = path[-1] if path else "value"
+        section_path = path[:-1]
+        description = f"'{prop_name}'"
+
+    suggestion = ""
+    if isinstance(error.instance, str) and all(isinstance(v, str) for v in valid_values):
+        match = _find_closest_match(error.instance, valid_values)
+        if match:
+            suggestion = f" Did you mean '{match}'?"
+
+    section = path_to_section_name(section_path)
+    valid_values_str = ", ".join(repr(v) for v in valid_values)
+    return (
+        f"Error in {section} section:\n  Invalid value:\n\n"
+        f"  - {description} → '{error.instance}' is not a valid value.{suggestion}\n\n"
+        f"Valid values are: {valid_values_str}."
+    )
+
+
+def _format_type_error(error: ValidationError) -> str:
+    expected = error.validator_value
+    assert isinstance(expected, (str, list))
+    section = path_to_section_name(list(error.path)[:-1] if error.path else [])
+    assert error.path
+
+    type_phrases = {
+        "object": "an object",
+        "array": "an array",
+        "number": "a number",
+        "boolean": "a boolean",
+        "string": "a string",
+        "integer": "an integer",
+        "null": "null",
+    }
+    message = f"Error in {section} section:\n  Type error:\n\n  - '{error.path[-1]}' → Must be "
+
+    if isinstance(expected, list):
+        message += f"one of: {' or '.join(expected)}"
+    else:
+        message += type_phrases[expected]
+    actual = type(error.instance).__name__
+    message += f", but got {actual}: {error.instance}"
+    return message
+
+
+def _format_additional_properties_error(error: ValidationError) -> str:
+    valid = list(error.schema.get("properties", {}))
+    unknown = sorted(set(error.instance) - set(valid))
+    valid_list = ", ".join(f"'{prop}'" for prop in valid)
+    section = path_to_section_name(list(error.path))
+
+    details = []
+    for prop in unknown:
+        match = _find_closest_match(prop, valid)
+        if match:
+            details.append(f"- '{prop}' → Did you mean '{match}'?")
+        else:
+            details.append(f"- '{prop}'")
+
+    return (
+        f"Error in {section} section:\n  Unknown properties:\n\n"
+        + "\n".join(f"  {detail}" for detail in details)
+        + f"\n\nValid properties for {section} are: {valid_list}."
+    )
+
+
+def _format_anyof_error(error: ValidationError) -> str:
+    if list(error.schema_path) == ["properties", "operations", "items", "anyOf"]:
+        section = path_to_section_name(list(error.path))
+        return (
+            f"Error in {section} section:\n  At least one filter is required when defining [[operations]].\n\n"
+            "Please specify at least one include or exclude filter property (e.g., include-path, exclude-tag, etc.)."
+        )
+    return error.message
+
+
+def path_to_section_name(path: list[int | str]) -> str:
+    """Convert a JSON path to a TOML-like section name."""
+    if not path:
+        return "root"
+
+    return f"[{'.'.join(str(p) for p in path)}]"
+
+
+def _find_closest_match(value: str, variants: list[str]) -> str | None:
+    matches = difflib.get_close_matches(value, variants, n=1, cutoff=0.6)
+    return matches[0] if matches else None

schemathesis/config/_generation.py

@@ -0,0 +1,150 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import TYPE_CHECKING, Any
+
+from schemathesis.config._diff_base import DiffBase
+from schemathesis.generation.modes import GenerationMode
+
+if TYPE_CHECKING:
+    from schemathesis.generation.targets import TargetFunction
+
+
+@dataclass(repr=False)
+class GenerationConfig(DiffBase):
+    modes: list[GenerationMode]
+    max_examples: int | None
+    no_shrink: bool
+    deterministic: bool
+    # Allow generating `\x00` bytes in strings
+    allow_x00: bool
+    # Generate strings using the given codec
+    codec: str | None
+    maximize: list[TargetFunction]
+    # Whether to generate security parameters
+    with_security_parameters: bool
+    # Allowing using `null` for optional arguments in GraphQL queries
+    graphql_allow_null: bool
+    database: str | None
+    unique_inputs: bool
+    exclude_header_characters: str | None
+
+    __slots__ = (
+        "modes",
+        "max_examples",
+        "no_shrink",
+        "deterministic",
+        "allow_x00",
+        "codec",
+        "maximize",
+        "with_security_parameters",
+        "graphql_allow_null",
+        "database",
+        "unique_inputs",
+        "exclude_header_characters",
+    )
+
+    def __init__(
+        self,
+        *,
+        modes: list[GenerationMode] | None = None,
+        max_examples: int | None = None,
+        no_shrink: bool = False,
+        deterministic: bool = False,
+        allow_x00: bool = True,
+        codec: str | None = "utf-8",
+        maximize: list[TargetFunction] | None = None,
+        with_security_parameters: bool = True,
+        graphql_allow_null: bool = True,
+        database: str | None = None,
+        unique_inputs: bool = False,
+        exclude_header_characters: str | None = None,
+    ) -> None:
+        from schemathesis.generation import GenerationMode
+
+        # TODO: Switch to `all` by default.
+        self.modes = modes or [GenerationMode.POSITIVE]
+        self.max_examples = max_examples
+        self.no_shrink = no_shrink
+        self.deterministic = deterministic
+        self.allow_x00 = allow_x00
+        self.codec = codec
+        self.maximize = maximize or []
+        self.with_security_parameters = with_security_parameters
+        self.graphql_allow_null = graphql_allow_null
+        self.database = database
+        self.unique_inputs = unique_inputs
+        self.exclude_header_characters = exclude_header_characters
+
+    @classmethod
+    def from_dict(cls, data: dict[str, Any]) -> GenerationConfig:
+        mode_raw = data.get("mode")
+        if mode_raw == "all":
+            modes = GenerationMode.all()
+        elif mode_raw is not None:
+            modes = [GenerationMode(mode_raw)]
+        else:
+            modes = None
+        maximize = _get_maximize(data.get("maximize"))
+        return cls(
+            modes=modes,
+            max_examples=data.get("max-examples"),
+            no_shrink=data.get("no-shrink", False),
+            deterministic=data.get("deterministic", False),
+            allow_x00=data.get("allow-x00", True),
+            codec=data.get("codec", "utf-8"),
+            maximize=maximize,
+            with_security_parameters=data.get("with-security-parameters", True),
+            graphql_allow_null=data.get("graphql-allow-null", True),
+            database=data.get("database"),
+            unique_inputs=data.get("unique-inputs", False),
+            exclude_header_characters=data.get("exclude-header-characters"),
+        )
+
+    def update(
+        self,
+        *,
+        modes: list[GenerationMode] | None = None,
+        max_examples: int | None = None,
+        no_shrink: bool = False,
+        deterministic: bool | None = None,
+        allow_x00: bool = True,
+        codec: str | None = None,
+        maximize: list[TargetFunction] | None = None,
+        with_security_parameters: bool | None = None,
+        graphql_allow_null: bool = True,
+        database: str | None = None,
+        unique_inputs: bool = False,
+        exclude_header_characters: str | None = None,
+    ) -> None:
+        if modes is not None:
+            self.modes = modes
+        if max_examples is not None:
+            self.max_examples = max_examples
+        self.no_shrink = no_shrink
+        self.deterministic = deterministic or False
+        self.allow_x00 = allow_x00
+        if codec is not None:
+            self.codec = codec
+        if maximize is not None:
+            self.maximize = maximize
+        if with_security_parameters is not None:
+            self.with_security_parameters = with_security_parameters
+        self.graphql_allow_null = graphql_allow_null
+        if database is not None:
+            self.database = database
+        self.unique_inputs = unique_inputs
+        if exclude_header_characters is not None:
+            self.exclude_header_characters = exclude_header_characters
+
+
+def _get_maximize(value: Any) -> list[TargetFunction]:
+    from schemathesis.generation.targets import TARGETS
+
+    if isinstance(value, list):
+        targets = value
+    elif isinstance(value, str):
+        targets = [value]
+    else:
+        targets = []
+    return TARGETS.get_by_names(targets)

schemathesis/config/_health_check.py

@@ -0,0 +1,24 @@
+from __future__ import annotations
+
+from enum import Enum, unique
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    import hypothesis
+
+
+@unique
+class HealthCheck(str, Enum):
+    data_too_large = "data_too_large"
+    filter_too_much = "filter_too_much"
+    too_slow = "too_slow"
+    large_base_example = "large_base_example"
+    all = "all"
+
+    def as_hypothesis(self) -> list[hypothesis.HealthCheck]:
+        from hypothesis import HealthCheck
+
+        if self.name == "all":
+            return list(HealthCheck)
+
+        return [HealthCheck[self.name]]

schemathesis/config/_operations.py

@@ -0,0 +1,313 @@
+from __future__ import annotations
+
+import re
+from contextlib import contextmanager
+from dataclasses import dataclass
+from typing import TYPE_CHECKING, Any, Callable, Generator
+
+from schemathesis.config._auth import AuthConfig
+from schemathesis.config._checks import ChecksConfig
+from schemathesis.config._diff_base import DiffBase
+from schemathesis.config._env import resolve
+from schemathesis.config._error import ConfigError
+from schemathesis.config._generation import GenerationConfig
+from schemathesis.config._parameters import load_parameters
+from schemathesis.config._phases import PhasesConfig
+from schemathesis.config._rate_limit import build_limiter
+from schemathesis.core.errors import IncorrectUsage
+from schemathesis.filters import FilterSet, HasAPIOperation, expression_to_filter_function, is_deprecated
+
+if TYPE_CHECKING:
+    from pyrate_limiter import Limiter
+
+    from schemathesis.schemas import APIOperation
+
+FILTER_ATTRIBUTES = [
+    ("name", "name"),
+    ("method", "method"),
+    ("path", "path"),
+    ("tag", "tag"),
+    ("operation-id", "operation_id"),
+]
+
+
+@contextmanager
+def reraise_filter_error(attr: str) -> Generator:
+    try:
+        yield
+    except IncorrectUsage as exc:
+        if str(exc) == "Filter already exists":
+            raise ConfigError(
+                f"Filter for '{attr}' already exists. You can't simultaneously include and exclude the same thing."
+            ) from None
+        raise
+    except re.error as exc:
+        raise ConfigError(
+            f"Filter for '{attr}' contains an invalid regular expression: {exc.pattern!r}\n\n  {exc}"
+        ) from None
+
+
+@dataclass
+class OperationsConfig(DiffBase):
+    operations: list[OperationConfig]
+
+    __slots__ = ("operations",)
+
+    def __init__(self, *, operations: list[OperationConfig] | None = None):
+        self.operations = operations or []
+
+    def __repr__(self) -> str:
+        if self.operations:
+            return f"[{', '.join(DiffBase.__repr__(cfg) for cfg in self.operations)}]"
+        return "[]"
+
+    @classmethod
+    def from_hierarchy(cls, configs: list[OperationsConfig]) -> OperationsConfig:  # type: ignore
+        return cls(operations=sum([config.operations for config in reversed(configs)], []))
+
+    def get_for_operation(self, operation: APIOperation) -> OperationConfig:
+        configs = [config for config in self.operations if config._filter_set.applies_to(operation)]
+        return OperationConfig.from_hierarchy(configs)
+
+    def create_filter_set(
+        self,
+        *,
+        include_path: tuple[str, ...],
+        include_method: tuple[str, ...],
+        include_name: tuple[str, ...],
+        include_tag: tuple[str, ...],
+        include_operation_id: tuple[str, ...],
+        include_path_regex: str | None,
+        include_method_regex: str | None,
+        include_name_regex: str | None,
+        include_tag_regex: str | None,
+        include_operation_id_regex: str | None,
+        exclude_path: tuple[str, ...],
+        exclude_method: tuple[str, ...],
+        exclude_name: tuple[str, ...],
+        exclude_tag: tuple[str, ...],
+        exclude_operation_id: tuple[str, ...],
+        exclude_path_regex: str | None,
+        exclude_method_regex: str | None,
+        exclude_name_regex: str | None,
+        exclude_tag_regex: str | None,
+        exclude_operation_id_regex: str | None,
+        include_by: Callable | None,
+        exclude_by: Callable | None,
+        exclude_deprecated: bool,
+    ) -> FilterSet:
+        # Build explicit include filters
+        include_set = FilterSet()
+        if include_by:
+            include_set.include(include_by)
+        for name_ in include_name:
+            include_set.include(name=name_)
+        for method in include_method:
+            include_set.include(method=method)
+        for path in include_path:
+            include_set.include(path=path)
+        for tag in include_tag:
+            include_set.include(tag=tag)
+        for operation_id in include_operation_id:
+            include_set.include(operation_id=operation_id)
+        if (
+            include_name_regex
+            or include_method_regex
+            or include_path_regex
+            or include_tag_regex
+            or include_operation_id_regex
+        ):
+            include_set.include(
+                name_regex=include_name_regex,
+                method_regex=include_method_regex,
+                path_regex=include_path_regex,
+                tag_regex=include_tag_regex,
+                operation_id_regex=include_operation_id_regex,
+            )
+
+        # Build explicit exclude filters
+        exclude_set = FilterSet()
+        if exclude_by:
+            exclude_set.include(exclude_by)
+        for name_ in exclude_name:
+            exclude_set.include(name=name_)
+        for method in exclude_method:
+            exclude_set.include(method=method)
+        for path in exclude_path:
+            exclude_set.include(path=path)
+        for tag in exclude_tag:
+            exclude_set.include(tag=tag)
+        for operation_id in exclude_operation_id:
+            exclude_set.include(operation_id=operation_id)
+        if (
+            exclude_name_regex
+            or exclude_method_regex
+            or exclude_path_regex
+            or exclude_tag_regex
+            or exclude_operation_id_regex
+        ):
+            exclude_set.include(
+                name_regex=exclude_name_regex,
+                method_regex=exclude_method_regex,
+                path_regex=exclude_path_regex,
+                tag_regex=exclude_tag_regex,
+                operation_id_regex=exclude_operation_id_regex,
+            )
+
+        # Add deprecated operations to exclude filters if requested
+        if exclude_deprecated:
+            exclude_set.include(is_deprecated)
+
+        # Also update operations list for consistency with config structure
+        if not include_set.is_empty():
+            self.operations.insert(0, OperationConfig(filter_set=include_set, enabled=True))
+        if not exclude_set.is_empty():
+            self.operations.insert(0, OperationConfig(filter_set=exclude_set, enabled=False))
+
+        final = FilterSet()
+
+        # Get a stable reference to operations
+        operations = list(self.operations)
+
+        # Define a closure that implements our priority logic
+        def priority_filter(ctx: HasAPIOperation) -> bool:
+            """Filter operations according to CLI and config priority."""
+            # 1. CLI includes override everything if present
+            if not include_set.is_empty():
+                return include_set.match(ctx)
+
+            # 2. CLI excludes take precedence over config
+            if not exclude_set.is_empty() and exclude_set.match(ctx):
+                return False
+
+            # 3. Check config operations in priority order (first match wins)
+            for op_config in operations:
+                if op_config._filter_set.match(ctx):
+                    return op_config.enabled
+
+            # 4. Default to include if no rule matches
+            return True
+
+        # Add our priority function as the filter
+        final.include(priority_filter)
+
+        return final
+
+
+@dataclass
+class OperationConfig(DiffBase):
+    _filter_set: FilterSet
+    enabled: bool
+    headers: dict | None
+    proxy: str | None
+    continue_on_failure: bool | None
+    tls_verify: bool | str | None
+    rate_limit: Limiter | None
+    request_timeout: float | int | None
+    request_cert: str | None
+    request_cert_key: str | None
+    parameters: dict[str, Any]
+    auth: AuthConfig
+    checks: ChecksConfig
+    phases: PhasesConfig
+    generation: GenerationConfig
+
+    __slots__ = (
+        "_filter_set",
+        "enabled",
+        "headers",
+        "proxy",
+        "continue_on_failure",
+        "tls_verify",
+        "rate_limit",
+        "_rate_limit",
+        "request_timeout",
+        "request_cert",
+        "request_cert_key",
+        "parameters",
+        "auth",
+        "checks",
+        "phases",
+        "generation",
+    )
+
+    def __init__(
+        self,
+        *,
+        filter_set: FilterSet | None = None,
+        enabled: bool = True,
+        headers: dict | None = None,
+        proxy: str | None = None,
+        continue_on_failure: bool | None = None,
+        tls_verify: bool | str | None = None,
+        rate_limit: str | None = None,
+        request_timeout: float | int | None = None,
+        request_cert: str | None = None,
+        request_cert_key: str | None = None,
+        parameters: dict[str, Any] | None = None,
+        auth: AuthConfig | None = None,
+        checks: ChecksConfig | None = None,
+        phases: PhasesConfig | None = None,
+        generation: GenerationConfig | None = None,
+    ) -> None:
+        self._filter_set = filter_set or FilterSet()
+        self.enabled = enabled
+        self.headers = headers
+        self.proxy = proxy
+        self.continue_on_failure = continue_on_failure
+        self.tls_verify = tls_verify
+        if rate_limit is not None:
+            self.rate_limit = build_limiter(rate_limit)
+        else:
+            self.rate_limit = rate_limit
+        self._rate_limit = rate_limit
+        self.request_timeout = request_timeout
+        self.request_cert = request_cert
+        self.request_cert_key = request_cert_key
+        self.parameters = parameters or {}
+        self.auth = auth or AuthConfig()
+        self.checks = checks or ChecksConfig()
+        self.phases = phases or PhasesConfig()
+        self.generation = generation or GenerationConfig()
+
+    @classmethod
+    def from_dict(cls, data: dict[str, Any]) -> OperationConfig:
+        filter_set = FilterSet()
+        for key_suffix, arg_suffix in (("", ""), ("-regex", "_regex")):
+            for attr, arg_name in FILTER_ATTRIBUTES:
+                key = f"include-{attr}{key_suffix}"
+                if key in data:
+                    with reraise_filter_error(attr):
+                        filter_set.include(**{f"{arg_name}{arg_suffix}": data[key]})
+                key = f"exclude-{attr}{key_suffix}"
+                if key in data:
+                    with reraise_filter_error(attr):
+                        filter_set.exclude(**{f"{arg_name}{arg_suffix}": data[key]})
+        for key, method in (("include-by", filter_set.include), ("exclude-by", filter_set.exclude)):
+            if key in data:
+                expression = data[key]
+                try:
+                    func = expression_to_filter_function(expression)
+                    method(func)
+                except ValueError:
+                    raise ConfigError(f"Invalid filter expression: '{expression}'") from None
+
+        return cls(
+            filter_set=filter_set,
+            enabled=data.get("enabled", True),
+            headers={resolve(key): resolve(value) for key, value in data.get("headers", {}).items()}
+            if "headers" in data
+            else None,
+            proxy=resolve(data.get("proxy")),
+            continue_on_failure=data.get("continue-on-failure", None),
+            tls_verify=resolve(data.get("tls-verify")),
+            rate_limit=resolve(data.get("rate-limit")),
+            request_timeout=data.get("request-timeout"),
+            request_cert=resolve(data.get("request-cert")),
+            request_cert_key=resolve(data.get("request-cert-key")),
+            parameters=load_parameters(data),
+            auth=AuthConfig.from_dict(data.get("auth", {})),
+            checks=ChecksConfig.from_dict(data.get("checks", {})),
+            phases=PhasesConfig.from_dict(data.get("phases", {})),
+            generation=GenerationConfig.from_dict(data.get("generation", {})),
+        )