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

schemathesis/__init__.py

@@ -1,11 +1,11 @@
 from __future__ import annotations
 
-from schemathesis import auths, contrib, engine, errors, experimental, graphql, hooks, openapi, pytest, python
+from schemathesis import auths, contrib, engine, errors, graphql, hooks, openapi, pytest, python
 from schemathesis.checks import CheckContext, CheckFunction, check
-from schemathesis.core.output import OutputConfig, sanitization
+from schemathesis.core.output import sanitization
 from schemathesis.core.transport import Response
 from schemathesis.core.version import SCHEMATHESIS_VERSION
-from schemathesis.generation import GenerationConfig, GenerationMode, HeaderConfig
+from schemathesis.generation import GenerationMode
 from schemathesis.generation.case import Case
 from schemathesis.generation.targets import TargetContext, TargetFunction, target
 from schemathesis.hooks import HookContext
@@ -22,9 +22,6 @@ __all__ = [
     "CheckContext",
     "CheckFunction",
     "GenerationMode",
-    "GenerationConfig",
-    "HeaderConfig",
-    "OutputConfig",
     "Response",
     "TargetContext",
     "TargetFunction",
@@ -36,7 +33,6 @@ __all__ = [
     "contrib",
     "engine",
     "errors",
-    "experimental",
     "graphql",
     "hook",
     "hooks",

schemathesis/checks.py

@@ -3,12 +3,12 @@ from __future__ import annotations
 import json
 from typing import TYPE_CHECKING, Any, Callable, Iterable, Iterator, Optional
 
+from schemathesis.config import ChecksConfig
 from schemathesis.core.failures import (
     CustomFailure,
     Failure,
     FailureGroup,
     MalformedJson,
-    MaxResponseTimeConfig,
     ResponseTimeExceeded,
     ServerError,
 )
@@ -23,7 +23,6 @@ if TYPE_CHECKING:
     from schemathesis.generation.case import Case
 
 CheckFunction = Callable[["CheckContext", "Response", "Case"], Optional[bool]]
-ChecksConfig = dict[CheckFunction, Any]
 
 
 class CheckContext:
@@ -38,8 +37,9 @@ class CheckContext:
     config: ChecksConfig
     transport_kwargs: dict[str, Any] | None
     recorder: ScenarioRecorder | None
+    checks: list[CheckFunction]
 
-    __slots__ = ("override", "auth", "headers", "config", "transport_kwargs", "recorder")
+    __slots__ = ("override", "auth", "headers", "config", "transport_kwargs", "recorder", "checks")
 
     def __init__(
         self,
@@ -56,6 +56,13 @@ class CheckContext:
         self.config = config
         self.transport_kwargs = transport_kwargs
         self.recorder = recorder
+        self.checks = []
+        for check in CHECKS.get_all():
+            name = check.__name__
+            if self.config.get_by_name(name=name).enabled:
+                self.checks.append(check)
+        if self.config.max_response_time.enabled:
+            self.checks.append(max_response_time)
 
     def find_parent(self, *, case_id: str) -> Case | None:
         if self.recorder is not None:
@@ -102,15 +109,18 @@ def not_a_server_error(ctx: CheckContext, response: Response, case: Case) -> boo
     return None
 
 
+DEFAULT_MAX_RESPONSE_TIME = 10.0
+
+
 def max_response_time(ctx: CheckContext, response: Response, case: Case) -> bool | None:
-    config = ctx.config.get(max_response_time, MaxResponseTimeConfig())
+    limit = ctx.config.max_response_time.limit or DEFAULT_MAX_RESPONSE_TIME
     elapsed = response.elapsed
-    if elapsed > config.limit:
+    if elapsed > limit:
         raise ResponseTimeExceeded(
             operation=case.operation.label,
-            message=f"Actual: {elapsed:.2f}ms\nLimit: {config.limit * 1000:.2f}ms",
+            message=f"Actual: {elapsed:.2f}ms\nLimit: {limit * 1000:.2f}ms",
             elapsed=elapsed,
-            deadline=config.limit,
+            deadline=limit,
         )
     return None
 

schemathesis/cli/commands/__init__.py

@@ -4,20 +4,68 @@ from dataclasses import dataclass
 from typing import Any
 
 import click
+from tomli import TOMLDecodeError
 
-from schemathesis.cli import hooks
+from schemathesis.cli.commands.data import Data
 from schemathesis.cli.commands.run import run as run_command
+from schemathesis.cli.commands.run.handlers.output import display_header
+from schemathesis.cli.constants import EXTENSIONS_DOCUMENTATION_URL
 from schemathesis.cli.core import get_terminal_width
 from schemathesis.cli.ext.groups import CommandWithGroupedOptions, GroupedOption
+from schemathesis.config import ConfigError, SchemathesisConfig
+from schemathesis.core.errors import HookError, format_exception
+from schemathesis.core.version import SCHEMATHESIS_VERSION
 
 CONTEXT_SETTINGS = {"help_option_names": ["-h", "--help"]}
 
 
 @click.group(context_settings=CONTEXT_SETTINGS)  # type: ignore[misc]
+@click.option(  # type: ignore[misc]
+    "--config-file",
+    "config_file",
+    help="The path to `schemathesis.toml` file to use for configuration",
+    metavar="PATH",
+    type=str,
+)
+@click.pass_context  # type: ignore[misc]
 @click.version_option()  # type: ignore[misc]
-def schemathesis() -> None:
+def schemathesis(ctx: click.Context, config_file: str | None) -> None:
     """Property-based API testing for OpenAPI and GraphQL."""
-    hooks.load()
+    try:
+        if config_file is not None:
+            config = SchemathesisConfig.from_path(config_file)
+        else:
+            config = SchemathesisConfig.discover()
+    except (TOMLDecodeError, ConfigError) as exc:
+        display_header(SCHEMATHESIS_VERSION)
+        click.secho(
+            f"❌  Failed to load configuration file{f' from {config_file}' if config_file else ''}",
+            fg="red",
+            bold=True,
+        )
+        if isinstance(exc, TOMLDecodeError):
+            detail = "The configuration file content is not valid TOML"
+        else:
+            detail = "The loaded configuration is incorrect"
+        click.echo(f"\n{detail}\n\n{exc}")
+        ctx.exit(1)
+    except HookError as exc:
+        click.secho("Unable to load Schemathesis extension hooks", fg="red", bold=True)
+        formatted_module_name = click.style(f"'{exc.module_path}'", bold=True)
+        cause = exc.__cause__
+        assert isinstance(cause, Exception)
+        if isinstance(cause, ModuleNotFoundError) and cause.name == exc.module_path:
+            click.echo(
+                f"\nAn attempt to import the module {formatted_module_name} failed because it could not be found."
+            )
+            click.echo("\nEnsure the module name is correctly spelled and reachable from the current directory.")
+        else:
+            click.echo(f"\nAn error occurred while importing the module {formatted_module_name}. Traceback:")
+            message = format_exception(cause, with_traceback=True, skip_frames=1)
+            click.secho(f"\n{message}", fg="red")
+        click.echo(f"\nFor more information on how to work with hooks, visit {EXTENSIONS_DOCUMENTATION_URL}")
+        ctx.exit(1)
+    ctx.obj = Data(config=config)
 
 
 @dataclass

schemathesis/cli/commands/data.py

@@ -0,0 +1,10 @@
+from dataclasses import dataclass
+
+from schemathesis.config import SchemathesisConfig
+
+
+@dataclass
+class Data:
+    config: SchemathesisConfig
+
+    __slots__ = ("config",)