schemathesis 3.39.15__py3-none-any.whl → 4.0.0__py3-none-any.whl

schemathesis/engine/phases/unit/_executor.py

@@ -0,0 +1,416 @@
+from __future__ import annotations
+
+import time
+import unittest
+import uuid
+from typing import TYPE_CHECKING, Any, Callable
+from warnings import WarningMessage, catch_warnings
+
+import requests
+from hypothesis.errors import InvalidArgument
+from hypothesis_jsonschema._canonicalise import HypothesisRefResolutionError
+from jsonschema.exceptions import SchemaError as JsonSchemaError
+from jsonschema.exceptions import ValidationError
+from requests.exceptions import ChunkedEncodingError
+from requests.structures import CaseInsensitiveDict
+
+from schemathesis.checks import CheckContext, run_checks
+from schemathesis.config._generation import GenerationConfig
+from schemathesis.core.compat import BaseExceptionGroup
+from schemathesis.core.control import SkipTest
+from schemathesis.core.errors import (
+    SERIALIZERS_SUGGESTION_MESSAGE,
+    InternalError,
+    InvalidHeadersExample,
+    InvalidRegexPattern,
+    InvalidRegexType,
+    InvalidSchema,
+    MalformedMediaType,
+    SerializationNotPossible,
+)
+from schemathesis.core.failures import Failure, FailureGroup
+from schemathesis.core.transport import Response
+from schemathesis.engine import Status, events
+from schemathesis.engine.context import EngineContext
+from schemathesis.engine.errors import (
+    DeadlineExceeded,
+    TestingState,
+    UnexpectedError,
+    UnrecoverableNetworkError,
+    UnsupportedRecursiveReference,
+    clear_hypothesis_notes,
+    deduplicate_errors,
+    is_unrecoverable_network_error,
+)
+from schemathesis.engine.phases import PhaseName
+from schemathesis.engine.recorder import ScenarioRecorder
+from schemathesis.generation import metrics, overrides
+from schemathesis.generation.case import Case
+from schemathesis.generation.hypothesis.builder import (
+    InvalidHeadersExampleMark,
+    InvalidRegexMark,
+    NonSerializableMark,
+    UnsatisfiableExampleMark,
+)
+from schemathesis.generation.hypothesis.reporting import ignore_hypothesis_output
+
+if TYPE_CHECKING:
+    from schemathesis.schemas import APIOperation
+
+
+def run_test(
+    *,
+    operation: APIOperation,
+    test_function: Callable,
+    ctx: EngineContext,
+    phase: PhaseName,
+    suite_id: uuid.UUID,
+) -> events.EventGenerator:
+    """A single test run with all error handling needed."""
+    import hypothesis.errors
+
+    scenario_started = events.ScenarioStarted(label=operation.label, phase=phase, suite_id=suite_id)
+    yield scenario_started
+    errors: list[Exception] = []
+    skip_reason = None
+    test_start_time = time.monotonic()
+    recorder = ScenarioRecorder(label=operation.label)
+    state = TestingState()
+
+    def non_fatal_error(error: Exception, code_sample: str | None = None) -> events.NonFatalError:
+        return events.NonFatalError(
+            error=error, phase=phase, label=operation.label, related_to_operation=True, code_sample=code_sample
+        )
+
+    def scenario_finished(status: Status) -> events.ScenarioFinished:
+        return events.ScenarioFinished(
+            id=scenario_started.id,
+            suite_id=suite_id,
+            phase=phase,
+            label=operation.label,
+            recorder=recorder,
+            status=status,
+            elapsed_time=time.monotonic() - test_start_time,
+            skip_reason=skip_reason,
+            is_final=False,
+        )
+
+    phase_name = phase.value.lower()
+    assert phase_name in ("examples", "coverage", "fuzzing", "stateful")
+
+    operation_config = ctx.config.operations.get_for_operation(operation)
+    continue_on_failure = operation_config.continue_on_failure or ctx.config.continue_on_failure or False
+    generation = ctx.config.generation_for(operation=operation, phase=phase_name)
+    override = overrides.for_operation(ctx.config, operation=operation)
+    auth = ctx.config.auth_for(operation=operation)
+    headers = ctx.config.headers_for(operation=operation)
+    transport_kwargs = ctx.get_transport_kwargs(operation=operation)
+    check_ctx = CheckContext(
+        override=override,
+        auth=auth,
+        headers=CaseInsensitiveDict(headers) if headers else None,
+        config=ctx.config.checks_config_for(operation=operation, phase=phase_name),
+        transport_kwargs=transport_kwargs,
+        recorder=recorder,
+    )
+
+    try:
+        setup_hypothesis_database_key(test_function, operation)
+        with catch_warnings(record=True) as warnings, ignore_hypothesis_output():
+            test_function(
+                ctx=ctx,
+                state=state,
+                errors=errors,
+                check_ctx=check_ctx,
+                recorder=recorder,
+                generation=generation,
+                transport_kwargs=transport_kwargs,
+                continue_on_failure=continue_on_failure,
+            )
+        # Test body was not executed at all - Hypothesis did not generate any tests, but there is no error
+        status = Status.SUCCESS
+    except (SkipTest, unittest.case.SkipTest) as exc:
+        status = Status.SKIP
+        skip_reason = {"Hypothesis has been told to run no examples for this test.": "No examples in schema"}.get(
+            str(exc), str(exc)
+        )
+    except (FailureGroup, Failure):
+        status = Status.FAILURE
+    except UnexpectedError:
+        # It could be an error in user-defined extensions, network errors or internal Schemathesis errors
+        status = Status.ERROR
+        for idx, err in enumerate(errors):
+            if isinstance(err, MalformedMediaType):
+                errors[idx] = InvalidSchema(str(err))
+    except hypothesis.errors.Flaky as exc:
+        if isinstance(exc.__cause__, hypothesis.errors.DeadlineExceeded):
+            status = Status.ERROR
+            yield non_fatal_error(DeadlineExceeded.from_exc(exc.__cause__))
+        elif isinstance(exc, hypothesis.errors.FlakyFailure) and any(
+            isinstance(subexc, hypothesis.errors.DeadlineExceeded) for subexc in exc.exceptions
+        ):
+            for sub_exc in exc.exceptions:
+                if isinstance(sub_exc, hypothesis.errors.DeadlineExceeded):
+                    yield non_fatal_error(DeadlineExceeded.from_exc(sub_exc))
+            status = Status.ERROR
+        elif errors:
+            status = Status.ERROR
+        else:
+            status = Status.FAILURE
+    except BaseExceptionGroup:
+        status = Status.ERROR
+    except hypothesis.errors.Unsatisfiable:
+        # We need more clear error message here
+        status = Status.ERROR
+        yield non_fatal_error(hypothesis.errors.Unsatisfiable("Failed to generate test cases for this API operation"))
+    except KeyboardInterrupt:
+        yield scenario_finished(Status.INTERRUPTED)
+        yield events.Interrupted(phase=phase)
+        return
+    except AssertionError as exc:  # May come from `hypothesis-jsonschema` or `hypothesis`
+        status = Status.ERROR
+        try:
+            operation.schema.validate()
+            msg = "Unexpected error during testing of this API operation"
+            exc_msg = str(exc)
+            if exc_msg:
+                msg += f": {exc_msg}"
+            try:
+                raise InternalError(msg) from exc
+            except InternalError as exc:
+                yield non_fatal_error(exc)
+        except ValidationError as exc:
+            yield non_fatal_error(
+                InvalidSchema.from_jsonschema_error(
+                    exc,
+                    path=operation.path,
+                    method=operation.method,
+                    config=ctx.config.output,
+                )
+            )
+    except HypothesisRefResolutionError:
+        status = Status.ERROR
+        yield non_fatal_error(UnsupportedRecursiveReference())
+    except InvalidArgument as exc:
+        status = Status.ERROR
+        message = get_invalid_regular_expression_message(warnings)
+        if message:
+            # `hypothesis-jsonschema` emits a warning on invalid regular expression syntax
+            yield non_fatal_error(InvalidRegexPattern.from_hypothesis_jsonschema_message(message))
+        else:
+            yield non_fatal_error(exc)
+    except hypothesis.errors.DeadlineExceeded as exc:
+        status = Status.ERROR
+        yield non_fatal_error(DeadlineExceeded.from_exc(exc))
+    except JsonSchemaError as exc:
+        status = Status.ERROR
+        yield non_fatal_error(InvalidRegexPattern.from_schema_error(exc, from_examples=False))
+    except Exception as exc:
+        status = Status.ERROR
+        clear_hypothesis_notes(exc)
+        # Likely a YAML parsing issue. E.g. `00:00:00.00` (without quotes) is parsed as float `0.0`
+        if str(exc) == "first argument must be string or compiled pattern":
+            yield non_fatal_error(
+                InvalidRegexType(
+                    "Invalid `pattern` value: expected a string. "
+                    "If your schema is in YAML, ensure `pattern` values are quoted",
+                )
+            )
+        else:
+            code_sample: str | None = None
+            if state.unrecoverable_network_error is not None and state.unrecoverable_network_error.error is exc:
+                code_sample = state.unrecoverable_network_error.code_sample
+            yield non_fatal_error(exc, code_sample=code_sample)
+    if (
+        status == Status.SUCCESS
+        and continue_on_failure
+        and any(check.status == Status.FAILURE for checks in recorder.checks.values() for check in checks)
+    ):
+        status = Status.FAILURE
+    if UnsatisfiableExampleMark.is_set(test_function):
+        status = Status.ERROR
+        yield non_fatal_error(
+            hypothesis.errors.Unsatisfiable("Failed to generate test cases from examples for this API operation")
+        )
+    non_serializable = NonSerializableMark.get(test_function)
+    if non_serializable is not None and status != Status.ERROR:
+        status = Status.ERROR
+        media_types = ", ".join(non_serializable.media_types)
+        yield non_fatal_error(
+            SerializationNotPossible(
+                "Failed to generate test cases from examples for this API operation because of"
+                f" unsupported payload media types: {media_types}\n{SERIALIZERS_SUGGESTION_MESSAGE}",
+                media_types=non_serializable.media_types,
+            )
+        )
+
+    invalid_regex = InvalidRegexMark.get(test_function)
+    if invalid_regex is not None and status != Status.ERROR:
+        status = Status.ERROR
+        yield non_fatal_error(InvalidRegexPattern.from_schema_error(invalid_regex, from_examples=True))
+    invalid_headers = InvalidHeadersExampleMark.get(test_function)
+    if invalid_headers:
+        status = Status.ERROR
+        yield non_fatal_error(InvalidHeadersExample.from_headers(invalid_headers))
+    for error in deduplicate_errors(errors):
+        yield non_fatal_error(error)
+
+    yield scenario_finished(status)
+
+
+def setup_hypothesis_database_key(test: Callable, operation: APIOperation) -> None:
+    """Make Hypothesis use separate database entries for every API operation.
+
+    It increases the effectiveness of the Hypothesis database in the CLI.
+    """
+    # Hypothesis's function digest depends on the test function signature. To reflect it for the web API case,
+    # we use all API operation parameters in the digest.
+    extra = operation.label.encode("utf8")
+    for parameter in operation.iter_parameters():
+        extra += parameter.serialize(operation).encode("utf8")
+    test.hypothesis.inner_test._hypothesis_internal_add_digest = extra  # type: ignore
+
+
+def get_invalid_regular_expression_message(warnings: list[WarningMessage]) -> str | None:
+    for warning in warnings:
+        message = str(warning.message)
+        if "is not valid syntax for a Python regular expression" in message:
+            return message
+    return None
+
+
+def cached_test_func(f: Callable) -> Callable:
+    def wrapped(
+        *,
+        ctx: EngineContext,
+        state: TestingState,
+        case: Case,
+        errors: list[Exception],
+        check_ctx: CheckContext,
+        recorder: ScenarioRecorder,
+        generation: GenerationConfig,
+        transport_kwargs: dict[str, Any],
+        continue_on_failure: bool,
+    ) -> None:
+        try:
+            if ctx.has_to_stop:
+                raise KeyboardInterrupt
+            if generation.unique_inputs:
+                cached = ctx.get_cached_outcome(case)
+                if isinstance(cached, BaseException):
+                    raise cached
+                elif cached is None:
+                    return None
+                try:
+                    f(
+                        case=case,
+                        check_ctx=check_ctx,
+                        recorder=recorder,
+                        generation=generation,
+                        transport_kwargs=transport_kwargs,
+                        continue_on_failure=continue_on_failure,
+                    )
+                except BaseException as exc:
+                    ctx.cache_outcome(case, exc)
+                    raise
+                else:
+                    ctx.cache_outcome(case, None)
+            else:
+                f(
+                    case=case,
+                    check_ctx=check_ctx,
+                    recorder=recorder,
+                    generation=generation,
+                    transport_kwargs=transport_kwargs,
+                    continue_on_failure=continue_on_failure,
+                )
+        except (KeyboardInterrupt, Failure):
+            raise
+        except Exception as exc:
+            if isinstance(exc, (requests.ConnectionError, ChunkedEncodingError)) and is_unrecoverable_network_error(
+                exc
+            ):
+                # Server likely has crashed and does not accept any connections at all
+                # Don't report these error - only the original crash should be reported
+                if exc.request is not None:
+                    headers = {key: value[0] for key, value in exc.request.headers.items()}
+                else:
+                    headers = {**dict(case.headers or {}), **transport_kwargs.get("headers", {})}
+                verify = transport_kwargs.get("verify", True)
+                state.unrecoverable_network_error = UnrecoverableNetworkError(
+                    error=exc,
+                    code_sample=case.as_curl_command(headers=headers, verify=verify),
+                )
+                raise
+            errors.append(exc)
+            raise UnexpectedError from None
+
+    wrapped.__name__ = f.__name__
+
+    return wrapped
+
+
+@cached_test_func
+def test_func(
+    *,
+    case: Case,
+    check_ctx: CheckContext,
+    recorder: ScenarioRecorder,
+    generation: GenerationConfig,
+    transport_kwargs: dict[str, Any],
+    continue_on_failure: bool,
+) -> None:
+    recorder.record_case(parent_id=None, transition=None, case=case)
+    try:
+        response = case.call(**transport_kwargs)
+    except (requests.Timeout, requests.ConnectionError, ChunkedEncodingError) as error:
+        if isinstance(error.request, requests.Request):
+            recorder.record_request(case_id=case.id, request=error.request.prepare())
+        elif isinstance(error.request, requests.PreparedRequest):
+            recorder.record_request(case_id=case.id, request=error.request)
+        raise
+    recorder.record_response(case_id=case.id, response=response)
+    metrics.maximize(generation.maximize, case=case, response=response)
+    validate_response(
+        case=case,
+        ctx=check_ctx,
+        response=response,
+        continue_on_failure=continue_on_failure,
+        recorder=recorder,
+    )
+
+
+def validate_response(
+    *,
+    case: Case,
+    ctx: CheckContext,
+    response: Response,
+    continue_on_failure: bool,
+    recorder: ScenarioRecorder,
+) -> None:
+    failures = set()
+
+    def on_failure(name: str, collected: set[Failure], failure: Failure) -> None:
+        collected.add(failure)
+        failure_data = recorder.find_failure_data(parent_id=case.id, failure=failure)
+        recorder.record_check_failure(
+            name=name,
+            case_id=failure_data.case.id,
+            code_sample=failure_data.case.as_curl_command(headers=failure_data.headers, verify=failure_data.verify),
+            failure=failure,
+        )
+
+    def on_success(name: str, _case: Case) -> None:
+        recorder.record_check_success(name=name, case_id=_case.id)
+
+    failures = run_checks(
+        case=case,
+        response=response,
+        ctx=ctx,
+        checks=ctx._checks,
+        on_failure=on_failure,
+        on_success=on_success,
+    )
+
+    if failures and not continue_on_failure:
+        raise FailureGroup(list(failures)) from None

schemathesis/engine/phases/unit/_pool.py

@@ -0,0 +1,82 @@
+from __future__ import annotations
+
+import threading
+import uuid
+from queue import Queue
+from types import TracebackType
+from typing import TYPE_CHECKING, Callable
+
+from schemathesis.core.result import Result
+from schemathesis.engine.phases import PhaseName
+
+if TYPE_CHECKING:
+    from schemathesis.engine.context import EngineContext
+    from schemathesis.generation.hypothesis.builder import HypothesisTestMode
+
+
+class TaskProducer:
+    """Produces test tasks for workers to execute."""
+
+    def __init__(self, ctx: EngineContext) -> None:
+        self.operations = ctx.schema.get_all_operations()
+        self.lock = threading.Lock()
+
+    def next_operation(self) -> Result | None:
+        """Get next API operation in a thread-safe manner."""
+        with self.lock:
+            return next(self.operations, None)
+
+
+class WorkerPool:
+    """Manages a pool of worker threads."""
+
+    def __init__(
+        self,
+        workers_num: int,
+        producer: TaskProducer,
+        worker_factory: Callable,
+        ctx: EngineContext,
+        mode: HypothesisTestMode,
+        phase: PhaseName,
+        suite_id: uuid.UUID,
+    ) -> None:
+        self.workers_num = workers_num
+        self.producer = producer
+        self.worker_factory = worker_factory
+        self.ctx = ctx
+        self.mode = mode
+        self.phase = phase
+        self.suite_id = suite_id
+        self.workers: list[threading.Thread] = []
+        self.events_queue: Queue = Queue()
+
+    def start(self) -> None:
+        """Start all worker threads."""
+        for i in range(self.workers_num):
+            worker = threading.Thread(
+                target=self.worker_factory,
+                kwargs={
+                    "ctx": self.ctx,
+                    "mode": self.mode,
+                    "phase": self.phase,
+                    "events_queue": self.events_queue,
+                    "producer": self.producer,
+                    "suite_id": self.suite_id,
+                },
+                name=f"schemathesis_unit_tests_{i}",
+                daemon=True,
+            )
+            self.workers.append(worker)
+            worker.start()
+
+    def stop(self) -> None:
+        """Stop all workers gracefully."""
+        for worker in self.workers:
+            worker.join()
+
+    def __enter__(self) -> WorkerPool:
+        self.start()
+        return self
+
+    def __exit__(self, ty: type[BaseException] | None, value: BaseException | None, tb: TracebackType | None) -> None:
+        self.stop()