schemathesis 3.18.5__py3-none-any.whl → 3.19.1__py3-none-any.whl

schemathesis/serializers.py

@@ -1,9 +1,9 @@
 import binascii
 import os
+from dataclasses import dataclass
 from io import BytesIO
 from typing import TYPE_CHECKING, Any, Callable, Collection, Dict, Generator, Optional, Type
 
-import attr
 import yaml
 from typing_extensions import Protocol, runtime_checkable
 
@@ -22,14 +22,14 @@ except ImportError:
 SERIALIZERS: Dict[str, Type["Serializer"]] = {}
 
 
-@attr.s(slots=True)  # pragma: no mutate
+@dataclass
 class SerializerContext:
     """The context for serialization process.
 
     :ivar Case case: Generated example that is being processed.
     """
 
-    case: "Case" = attr.ib()  # pragma: no mutate
+    case: "Case"
 
 
 @runtime_checkable

schemathesis/service/ci.py

@@ -1,8 +1,8 @@
 import enum
 import os
+from dataclasses import asdict, dataclass
 from typing import Dict, Optional
 
-import attr
 from typing_extensions import Protocol, runtime_checkable
 
 
@@ -54,52 +54,52 @@ def detect() -> Optional[CIProvider]:
     return None
 
 
-def asdict(env: Environment) -> Dict[str, Optional[str]]:
-    data = attr.asdict(env)
+def _asdict(env: Environment) -> Dict[str, Optional[str]]:
+    data = asdict(env)
     data["provider"] = env.provider.value
     return data
 
 
-@attr.s(slots=True)
+@dataclass
 class GitHubActionsEnvironment:
     """Useful data to capture from GitHub Actions environment."""
 
     provider = CIProvider.GITHUB
     variable_name = "GITHUB_ACTIONS"
     verbose_name = "GitHub Actions"
-    asdict = asdict
+    asdict = _asdict
 
     # GitHub API URL.
     # For example, `https://api.github.com`
-    api_url: str = attr.ib()
+    api_url: str
     # The owner and repository name.
     # For example, `schemathesis/schemathesis`.
-    repository: str = attr.ib()
+    repository: str
     # The name of the person or app that initiated the workflow.
     # For example, `Stranger6667`
-    actor: str = attr.ib()
+    actor: str
     # The commit SHA that triggered the workflow.
     # For example, `e56e13224f08469841e106449f6467b769e2afca`
-    sha: str = attr.ib()
+    sha: str
     # A unique number for each workflow run within a repository.
     # For example, `1658821493`.
-    run_id: str = attr.ib()
+    run_id: str
     # The name of the workflow.
     # For example, `My test workflow`.
-    workflow: str = attr.ib()
+    workflow: str
     # The head ref or source branch of the pull request in a workflow run.
     # For example, `dd/report-ci`.
-    head_ref: Optional[str] = attr.ib()
+    head_ref: Optional[str]
     # The name of the base ref or target branch of the pull request in a workflow run.
     # For example, `main`.
-    base_ref: Optional[str] = attr.ib()
+    base_ref: Optional[str]
     # The branch or tag ref that triggered the workflow run.
     # This is only set if a branch or tag is available for the event type.
     # For example, `refs/pull/1533/merge`
-    ref: Optional[str] = attr.ib()
+    ref: Optional[str]
     # The Schemathesis GitHub Action version.
     # For example `v1.0.1`
-    action_ref: Optional[str] = attr.ib()
+    action_ref: Optional[str]
 
     @classmethod
     def is_set(cls) -> bool:
@@ -135,42 +135,42 @@ class GitHubActionsEnvironment:
         }
 
 
-@attr.s(slots=True)
+@dataclass
 class GitLabCIEnvironment:
     """Useful data to capture from GitLab CI environment."""
 
     provider = CIProvider.GITLAB
     variable_name = "GITLAB_CI"
     verbose_name = "GitLab CI"
-    asdict = asdict
+    asdict = _asdict
 
     # GitLab API URL
     # For example, `https://gitlab.com/api/v4`
-    api_v4_url: str = attr.ib()
+    api_v4_url: str
     # The ID of the current project.
     # For example, `12345678`
-    project_id: str = attr.ib()
+    project_id: str
     # The username of the user who started the job.
     # For example, `Stranger6667`
-    user_login: str = attr.ib()
+    user_login: str
     # The commit revision the project is built for.
     # For example, `e56e13224f08469841e106449f6467b769e2afca`
-    commit_sha: str = attr.ib()
+    commit_sha: str
     # NOTE: `commit_branch` and `merge_request_source_branch_name` may mean the same thing, but they are available
     # in different context. There are also a couple of `CI_BUILD_*` variables that could be used, but they are
     # not documented.
     # The commit branch name. Not available in merge request pipelines or tag pipelines.
     # For example, `dd/report-ci`.
-    commit_branch: Optional[str] = attr.ib()
+    commit_branch: Optional[str]
     # The source branch name of the merge request. Only available in merge request pipelines.
     # For example, `dd/report-ci`.
-    merge_request_source_branch_name: Optional[str] = attr.ib()
+    merge_request_source_branch_name: Optional[str]
     # The target branch name of the merge request.
     # For example, `main`.
-    merge_request_target_branch_name: Optional[str] = attr.ib()
+    merge_request_target_branch_name: Optional[str]
     # The project-level internal ID of the merge request.
     # For example, `42`.
-    merge_request_iid: Optional[str] = attr.ib()
+    merge_request_iid: Optional[str]
 
     @classmethod
     def is_set(cls) -> bool:

schemathesis/service/client.py

@@ -1,9 +1,9 @@
 import hashlib
 import http
+from dataclasses import asdict
 from typing import Any, Optional, Union
 from urllib.parse import urljoin
 
-import attr
 import requests
 from requests.adapters import HTTPAdapter, Retry
 
@@ -51,7 +51,7 @@ class ServiceClient(requests.Session):
 
     def login(self, metadata: Metadata) -> AuthResponse:
         """Send a login request."""
-        response = self.post("/auth/cli/login/", json={"metadata": attr.asdict(metadata)})
+        response = self.post("/auth/cli/login/", json={"metadata": asdict(metadata)})
         data = response.json()
         return AuthResponse(username=data["username"])
 

schemathesis/service/events.py

@@ -1,7 +1,6 @@
+from dataclasses import dataclass
 from typing import Optional
 
-import attr
-
 from ..utils import format_exception
 from . import ci
 
@@ -17,40 +16,40 @@ class Event:
         return self.__class__.__name__.upper()
 
 
-@attr.s(slots=True)
+@dataclass
 class Metadata(Event):
     """Meta-information about the report."""
 
-    size: int = attr.ib()
-    ci_environment: Optional[ci.Environment] = attr.ib()
+    size: int
+    ci_environment: Optional[ci.Environment]
 
 
-@attr.s(slots=True)
+@dataclass
 class Completed(Event):
     """Report uploaded successfully."""
 
-    message: str = attr.ib()
-    next_url: str = attr.ib()
+    message: str
+    next_url: str
 
 
-@attr.s(slots=True)
+@dataclass
 class Error(Event):
     """Internal error inside the Schemathesis.io handler."""
 
-    exception: Exception = attr.ib()
+    exception: Exception
 
     def get_message(self, include_traceback: bool = False) -> str:
         return format_exception(self.exception, include_traceback=include_traceback)
 
 
-@attr.s(slots=True)
+@dataclass
 class Failed(Event):
     """A client-side error which should be displayed to the user."""
 
-    detail: str = attr.ib()
+    detail: str
 
 
-@attr.s(slots=True)
+@dataclass
 class Timeout(Event):
     """The handler did not finish its work in time.
 

schemathesis/service/hosts.py

@@ -1,10 +1,10 @@
 """Work with stored auth data."""
 import enum
 import tempfile
+from dataclasses import dataclass
 from pathlib import Path
 from typing import Any, Dict, Optional
 
-import attr
 import tomli
 import tomli_w
 
@@ -12,12 +12,12 @@ from ..types import PathLike
 from .constants import DEFAULT_HOSTNAME, DEFAULT_HOSTS_PATH, HOSTS_FORMAT_VERSION
 
 
-@attr.s(slots=True)
+@dataclass
 class HostData:
     """Stored data related to a host."""
 
-    hostname: str = attr.ib()
-    hosts_file: PathLike = attr.ib()
+    hostname: str
+    hosts_file: PathLike
 
     def load(self) -> Dict[str, Any]:
         return load(self.hosts_file).get(self.hostname, {})

schemathesis/service/metadata.py

@@ -1,42 +1,41 @@
 """Useful info to collect from CLI usage."""
 import platform
-
-import attr
+from dataclasses import dataclass, field
 
 from ..constants import __version__
 
 
-@attr.s(slots=True)
+@dataclass
 class PlatformMetadata:
     # System / OS name, e.g. "Linux" or "Windows".
-    system: str = attr.ib(factory=platform.system)
+    system: str = field(default_factory=platform.system)
     # System release, e.g. "5.14" or "NT".
-    release: str = attr.ib(factory=platform.release)
+    release: str = field(default_factory=platform.release)
     # Machine type, e.g. "i386".
-    machine: str = attr.ib(factory=platform.machine)
+    machine: str = field(default_factory=platform.machine)
 
 
-@attr.s(slots=True)
+@dataclass
 class InterpreterMetadata:
     # The Python version as "major.minor.patch".
-    version: str = attr.ib(factory=platform.python_version)
+    version: str = field(default_factory=platform.python_version)
     # Python implementation, e.g. "CPython" or "PyPy".
-    implementation: str = attr.ib(factory=platform.python_implementation)
+    implementation: str = field(default_factory=platform.python_implementation)
 
 
-@attr.s(slots=True)
+@dataclass
 class CliMetadata:
     # Schemathesis package version.
-    version: str = attr.ib(default=__version__)
+    version: str = __version__
 
 
-@attr.s(slots=True)
+@dataclass
 class Metadata:
     """CLI environment metadata."""
 
     # Information about the host platform.
-    platform: PlatformMetadata = attr.ib(factory=PlatformMetadata)
+    platform: PlatformMetadata = field(default_factory=PlatformMetadata)
     # Python interpreter info.
-    interpreter: InterpreterMetadata = attr.ib(factory=InterpreterMetadata)
+    interpreter: InterpreterMetadata = field(default_factory=InterpreterMetadata)
     # CLI info itself.
-    cli: CliMetadata = attr.ib(factory=CliMetadata)
+    cli: CliMetadata = field(default_factory=CliMetadata)

schemathesis/service/models.py

@@ -1,26 +1,25 @@
+from dataclasses import dataclass
 from typing import Optional
 
-import attr
 
-
-@attr.s(slots=True)
+@dataclass
 class ApiDetails:
-    location: str = attr.ib()
-    base_url: Optional[str] = attr.ib()
+    location: str
+    base_url: Optional[str]
 
 
-@attr.s(slots=True)
+@dataclass
 class AuthResponse:
-    username: str = attr.ib()
+    username: str
 
 
-@attr.s(slots=True)
+@dataclass
 class UploadResponse:
-    message: str = attr.ib()
-    next_url: str = attr.ib()
-    correlation_id: str = attr.ib()
+    message: str
+    next_url: str
+    correlation_id: str
 
 
-@attr.s(slots=True)
+@dataclass
 class FailedUploadResponse:
-    detail: str = attr.ib()
+    detail: str

schemathesis/service/report.py

@@ -5,11 +5,11 @@ import tarfile
 import threading
 import time
 from contextlib import suppress
+from dataclasses import asdict, dataclass, field
 from io import BytesIO
 from queue import Queue
 from typing import Any, Dict, Optional
 
-import attr
 import click
 
 from ..cli.context import ExecutionContext
@@ -23,15 +23,15 @@ from .models import UploadResponse
 from .serialization import serialize_event
 
 
-@attr.s(slots=True)
+@dataclass
 class ReportWriter:
     """Schemathesis.io test run report.
 
     Simplifies adding new files to the archive.
     """
 
-    _tar: tarfile.TarFile = attr.ib()
-    _events_count: int = attr.ib(default=0)
+    _tar: tarfile.TarFile
+    _events_count: int = 0
 
     def add_json_file(self, name: str, data: Any) -> None:
         buffer = BytesIO()
@@ -63,7 +63,7 @@ class ReportWriter:
             # The time that the test run began
             "started_at": started_at,
             # Metadata about CLI environment
-            "environment": attr.asdict(metadata),
+            "environment": asdict(metadata),
             # Environment variables specific for CI providers
             "ci": ci_environment.asdict() if ci_environment is not None else None,
             # CLI usage statistic
@@ -80,7 +80,6 @@ class ReportWriter:
         self.add_json_file(filename, serialize_event(event))
 
 
-@attr.s(slots=True)  # pragma: no mutate
 class BaseReportHandler(EventHandler):
     in_queue: Queue
     worker: threading.Thread
@@ -96,20 +95,20 @@ class BaseReportHandler(EventHandler):
         self.worker.join(WORKER_JOIN_TIMEOUT)
 
 
-@attr.s(slots=True)  # pragma: no mutate
+@dataclass
 class ServiceReportHandler(BaseReportHandler):
-    client: ServiceClient = attr.ib()  # pragma: no mutate
-    host_data: HostData = attr.ib()  # pragma: no mutate
-    api_name: Optional[str] = attr.ib()  # pragma: no mutate
-    location: str = attr.ib()  # pragma: no mutate
-    base_url: Optional[str] = attr.ib()  # pragma: no mutate
-    started_at: str = attr.ib()  # pragma: no mutate
-    telemetry: bool = attr.ib()  # pragma: no mutate
-    out_queue: Queue = attr.ib()  # pragma: no mutate
-    in_queue: Queue = attr.ib(factory=Queue)  # pragma: no mutate
-    worker: threading.Thread = attr.ib(init=False)  # pragma: no mutate
-
-    def __attrs_post_init__(self) -> None:
+    client: ServiceClient
+    host_data: HostData
+    api_name: Optional[str]
+    location: str
+    base_url: Optional[str]
+    started_at: str
+    telemetry: bool
+    out_queue: Queue
+    in_queue: Queue = field(default_factory=Queue)
+    worker: threading.Thread = field(init=False)
+
+    def __post_init__(self) -> None:
         self.worker = threading.Thread(
             target=write_remote,
             kwargs={
@@ -191,19 +190,19 @@ def write_remote(
         out_queue.put(events.Error(exc))
 
 
-@attr.s(slots=True)  # pragma: no mutate
+@dataclass
 class FileReportHandler(BaseReportHandler):
-    file_handle: click.utils.LazyFile = attr.ib()  # pragma: no mutate
-    api_name: Optional[str] = attr.ib()  # pragma: no mutate
-    location: str = attr.ib()  # pragma: no mutate
-    base_url: Optional[str] = attr.ib()  # pragma: no mutate
-    started_at: str = attr.ib()  # pragma: no mutate
-    telemetry: bool = attr.ib()  # pragma: no mutate
-    out_queue: Queue = attr.ib()  # pragma: no mutate
-    in_queue: Queue = attr.ib(factory=Queue)  # pragma: no mutate
-    worker: threading.Thread = attr.ib(init=False)  # pragma: no mutate
-
-    def __attrs_post_init__(self) -> None:
+    file_handle: click.utils.LazyFile
+    api_name: Optional[str]
+    location: str
+    base_url: Optional[str]
+    started_at: str
+    telemetry: bool
+    out_queue: Queue
+    in_queue: Queue = field(default_factory=Queue)
+    worker: threading.Thread = field(init=False)
+
+    def __post_init__(self) -> None:
         self.worker = threading.Thread(
             target=write_file,
             kwargs={

schemathesis/service/serialization.py

@@ -1,7 +1,6 @@
+from dataclasses import asdict
 from typing import Any, Callable, Dict, List, Optional, TypeVar, cast
 
-import attr
-
 from ..models import Response
 from ..runner import events
 from ..runner.serialization import SerializedCase
@@ -70,7 +69,7 @@ def serialize_after_execution(event: events.AfterExecution) -> Optional[Dict[str
                     "response": _serialize_response(check.response) if check.response is not None else None,
                     "example": _serialize_case(check.example),
                     "message": check.message,
-                    "context": attr.asdict(check.context) if check.context is not None else None,
+                    "context": asdict(check.context) if check.context is not None else None,
                     "history": [
                         {"case": _serialize_case(entry.case), "response": _serialize_response(entry.response)}
                         for entry in check.history
@@ -138,7 +137,6 @@ def serialize_event(
     extra: Optional[Dict[str, Any]] = None,
 ) -> Dict[str, Optional[Dict[str, Any]]]:
     """Turn an event into JSON-serializable structure."""
-    # Due to https://github.com/python-attrs/attrs/issues/864 it is easier to implement filtration manually
     # Use the explicitly provided serializer for this event and fallback to default one if it is not provided
     serializer = {
         events.Initialized: on_initialized,

schemathesis/specs/graphql/loaders.py

@@ -5,6 +5,7 @@ import backoff
 import graphql
 import requests
 from graphql import ExecutionResult
+from pyrate_limiter import Limiter
 from starlette.applications import Starlette
 from starlette_testclient import TestClient as ASGIClient
 from werkzeug import Client
@@ -13,6 +14,7 @@ from yarl import URL
 from ...constants import DEFAULT_DATA_GENERATION_METHODS, WAIT_FOR_SCHEMA_INTERVAL, CodeSampleStyle
 from ...exceptions import HTTPError
 from ...hooks import HookContext, dispatch
+from ...throttling import build_limiter
 from ...types import DataGenerationMethodInput, PathLike
 from ...utils import WSGIResponse, prepare_data_generation_methods, require_relative_url, setup_headers
 from .schemas import GraphQLSchema
@@ -28,6 +30,7 @@ def from_path(
     base_url: Optional[str] = None,
     data_generation_methods: DataGenerationMethodInput = DEFAULT_DATA_GENERATION_METHODS,
     code_sample_style: str = CodeSampleStyle.default().name,
+    rate_limit: Optional[str] = None,
     encoding: str = "utf8",
 ) -> GraphQLSchema:
     """Load GraphQL schema via a file from an OS path.
@@ -43,6 +46,7 @@ def from_path(
             data_generation_methods=data_generation_methods,
             code_sample_style=code_sample_style,
             location=pathlib.Path(path).absolute().as_uri(),
+            rate_limit=rate_limit,
         )
 
 
@@ -55,6 +59,7 @@ def from_url(
     data_generation_methods: DataGenerationMethodInput = DEFAULT_DATA_GENERATION_METHODS,
     code_sample_style: str = CodeSampleStyle.default().name,
     wait_for_schema: Optional[float] = None,
+    rate_limit: Optional[str] = None,
     **kwargs: Any,
 ) -> GraphQLSchema:
     """Load GraphQL schema from the network.
@@ -68,8 +73,10 @@ def from_url(
     """
     setup_headers(kwargs)
     kwargs.setdefault("json", {"query": INTROSPECTION_QUERY})
-    if not base_url and port:
-        base_url = str(URL(url).with_port(port))
+    if port:
+        url = str(URL(url).with_port(port))
+        if not base_url:
+            base_url = url
 
     if wait_for_schema is not None:
 
@@ -94,6 +101,7 @@ def from_url(
         app=app,
         data_generation_methods=data_generation_methods,
         code_sample_style=code_sample_style,
+        rate_limit=rate_limit,
     )
 
 
@@ -105,6 +113,7 @@ def from_file(
     data_generation_methods: DataGenerationMethodInput = DEFAULT_DATA_GENERATION_METHODS,
     code_sample_style: str = CodeSampleStyle.default().name,
     location: Optional[str] = None,
+    rate_limit: Optional[str] = None,
 ) -> GraphQLSchema:
     """Load GraphQL schema from a file descriptor or a string.
 
@@ -130,6 +139,7 @@ def from_file(
         data_generation_methods=data_generation_methods,
         code_sample_style=code_sample_style,
         location=location,
+        rate_limit=rate_limit,
     )
 
 
@@ -141,6 +151,7 @@ def from_dict(
     location: Optional[str] = None,
     data_generation_methods: DataGenerationMethodInput = DEFAULT_DATA_GENERATION_METHODS,
     code_sample_style: str = CodeSampleStyle.default().name,
+    rate_limit: Optional[str] = None,
 ) -> GraphQLSchema:
     """Load GraphQL schema from a Python dictionary.
 
@@ -153,6 +164,9 @@ def from_dict(
     _code_sample_style = CodeSampleStyle.from_str(code_sample_style)
     hook_context = HookContext()
     dispatch("before_load_schema", hook_context, raw_schema)
+    rate_limiter: Optional[Limiter] = None
+    if rate_limit is not None:
+        rate_limiter = build_limiter(rate_limit)
     instance = GraphQLSchema(
         raw_schema,
         location=location,
@@ -160,6 +174,7 @@ def from_dict(
         app=app,
         data_generation_methods=prepare_data_generation_methods(data_generation_methods),
         code_sample_style=_code_sample_style,
+        rate_limiter=rate_limiter,
     )  # type: ignore
     dispatch("after_load_schema", hook_context, instance)
     return instance
@@ -172,6 +187,7 @@ def from_wsgi(
     base_url: Optional[str] = None,
     data_generation_methods: DataGenerationMethodInput = DEFAULT_DATA_GENERATION_METHODS,
     code_sample_style: str = CodeSampleStyle.default().name,
+    rate_limit: Optional[str] = None,
     **kwargs: Any,
 ) -> GraphQLSchema:
     """Load GraphQL schema from a WSGI app.
@@ -194,6 +210,7 @@ def from_wsgi(
         app=app,
         data_generation_methods=data_generation_methods,
         code_sample_style=code_sample_style,
+        rate_limit=rate_limit,
     )
 
 
@@ -204,6 +221,7 @@ def from_asgi(
     base_url: Optional[str] = None,
     data_generation_methods: DataGenerationMethodInput = DEFAULT_DATA_GENERATION_METHODS,
     code_sample_style: str = CodeSampleStyle.default().name,
+    rate_limit: Optional[str] = None,
     **kwargs: Any,
 ) -> GraphQLSchema:
     """Load GraphQL schema from an ASGI app.
@@ -225,6 +243,7 @@ def from_asgi(
         app=app,
         data_generation_methods=data_generation_methods,
         code_sample_style=code_sample_style,
+        rate_limit=rate_limit,
     )
 
 

schemathesis/specs/graphql/schemas.py

@@ -1,9 +1,9 @@
 import enum
+from dataclasses import dataclass
 from enum import unique
 from typing import Any, Callable, Dict, Generator, List, Optional, Sequence, Tuple, Type, TypeVar, Union, cast
 from urllib.parse import urlsplit
 
-import attr
 import graphql
 import requests
 from hypothesis import strategies as st
@@ -31,7 +31,7 @@ class RootType(enum.Enum):
     MUTATION = enum.auto()
 
 
-@attr.s(slots=True, repr=False)  # pragma: no mutate
+@dataclass(repr=False)
 class GraphQLCase(Case):
     def as_requests_kwargs(
         self, base_url: Optional[str] = None, headers: Optional[Dict[str, str]] = None
@@ -84,19 +84,19 @@ class GraphQLCase(Case):
 C = TypeVar("C", bound=Case)
 
 
-@attr.s()
+@dataclass
 class GraphQLOperationDefinition(OperationDefinition):
-    field_name: str = attr.ib()
-    type_: graphql.GraphQLType = attr.ib()
-    root_type: RootType = attr.ib()
+    field_name: str
+    type_: graphql.GraphQLType
+    root_type: RootType
 
 
-@attr.s()  # pragma: no mutate
+@dataclass
 class GraphQLSchema(BaseSchema):
     def get_full_path(self, path: str) -> str:
         return self.base_path
 
-    @property  # pragma: no mutate
+    @property
     def verbose_name(self) -> str:
         return "GraphQL"