schemathesis 3.15.4__py3-none-any.whl → 4.4.2__py3-none-any.whl

schemathesis/specs/openapi/serialization.py

@@ -1,7 +1,10 @@
+from __future__ import annotations
+
 import json
-from typing import Any, Callable, Dict, Generator, List, Optional, Tuple, Union
+from typing import Any, Callable, Dict, Generator, List
 
-from ...utils import compose
+from schemathesis.core.parameters import LOCATION_TO_CONTAINER
+from schemathesis.schemas import APIOperation
 
 Generated = Dict[str, Any]
 Definition = Dict[str, Any]
@@ -9,21 +12,38 @@ DefinitionList = List[Definition]
 MapFunction = Callable[[Generated], Generated]
 
 
+def get_serializers_for_operation(operation: APIOperation) -> dict[str, Callable]:
+    serializers = {}
+    for location, container in LOCATION_TO_CONTAINER.items():
+        serializer = operation.get_parameter_serializer(location)
+        if serializer is not None:
+            serializers[container] = serializer
+    return serializers
+
+
 def make_serializer(
-    func: Callable[[DefinitionList], Generator[Optional[Callable], None, None]]
-) -> Callable[[DefinitionList], Optional[Callable]]:
+    func: Callable[[DefinitionList], Generator[Callable | None, None, None]],
+) -> Callable[[DefinitionList], Callable | None]:
     """A maker function to avoid code duplication."""
 
-    def _wrapper(definitions: DefinitionList) -> Optional[Callable]:
-        conversions = list(func(definitions))
-        if conversions:
-            return compose(*[conv for conv in conversions if conv is not None])
-        return None
+    def _wrapper(definitions: DefinitionList) -> Callable | None:
+        functions = list(func(definitions))
+        if not functions:
+            return None
+
+        def composed(x: Any) -> Any:
+            result = x
+            for func in reversed(functions):
+                if func is not None:
+                    result = func(result)
+            return result
+
+        return composed
 
     return _wrapper
 
 
-def _serialize_openapi3(definitions: DefinitionList) -> Generator[Optional[Callable], None, None]:
+def _serialize_openapi3(definitions: DefinitionList) -> Generator[Callable | None, None, None]:
     """Different collection styles for Open API 3.0."""
     for definition in definitions:
         name = definition["name"]
@@ -37,7 +57,11 @@ def _serialize_openapi3(definitions: DefinitionList) -> Generator[Optional[Calla
             # Simple serialization
             style = definition.get("style")
             explode = definition.get("explode")
-            type_ = definition.get("schema", {}).get("type")
+            schema = definition.get("schema", {})
+            if isinstance(schema, dict):
+                type_ = schema.get("type")
+            else:
+                type_ = None
             if definition["in"] == "path":
                 yield from _serialize_path_openapi3(name, type_, style, explode)
             elif definition["in"] == "query":
@@ -49,9 +73,8 @@ def _serialize_openapi3(definitions: DefinitionList) -> Generator[Optional[Calla
 
 
 def _serialize_path_openapi3(
-    name: str, type_: str, style: Optional[str], explode: Optional[bool]
-) -> Generator[Optional[Callable], None, None]:
-    # pylint: disable=too-many-branches
+    name: str, type_: str | None, style: str | None, explode: bool | None
+) -> Generator[Callable | None, None, None]:
     if style == "simple":
         if type_ == "object":
             if explode is False:
@@ -77,8 +100,8 @@ def _serialize_path_openapi3(
 
 
 def _serialize_query_openapi3(
-    name: str, type_: str, style: Optional[str], explode: Optional[bool]
-) -> Generator[Optional[Callable], None, None]:
+    name: str, type_: str | None, style: str | None, explode: bool | None
+) -> Generator[Callable | None, None, None]:
     if type_ == "object":
         if style == "deepObject":
             yield deep_object(name)
@@ -97,8 +120,8 @@ def _serialize_query_openapi3(
 
 
 def _serialize_header_openapi3(
-    name: str, type_: str, explode: Optional[bool]
-) -> Generator[Optional[Callable], None, None]:
+    name: str, type_: str | None, explode: bool | None
+) -> Generator[Callable | None, None, None]:
     # Headers should be coerced to a string so we can check it for validity later
     yield to_string(name)
     # Header parameters always use the "simple" style, that is, comma-separated values
@@ -112,8 +135,8 @@ def _serialize_header_openapi3(
 
 
 def _serialize_cookie_openapi3(
-    name: str, type_: str, explode: Optional[bool]
-) -> Generator[Optional[Callable], None, None]:
+    name: str, type_: str | None, explode: bool | None
+) -> Generator[Callable | None, None, None]:
     # Cookies should be coerced to a string so we can check it for validity later
     yield to_string(name)
     # Cookie parameters always use the "form" style
@@ -130,7 +153,7 @@ def _serialize_cookie_openapi3(
             yield comma_delimited_object(name)
 
 
-def _serialize_swagger2(definitions: DefinitionList) -> Generator[Optional[Callable], None, None]:
+def _serialize_swagger2(definitions: DefinitionList) -> Generator[Callable | None, None, None]:
     """Different collection formats for Open API 2.0."""
     for definition in definitions:
         name = definition["name"]
@@ -166,11 +189,11 @@ def conversion(func: Callable[..., None]) -> Callable:
     return _wrapper
 
 
-def make_delimited(data: Optional[Dict[str, Any]], delimiter: str = ",") -> str:
+def make_delimited(data: dict[str, Any] | None, delimiter: str = ",") -> str:
     return delimiter.join(f"{key}={value}" for key, value in force_dict(data or {}).items())
 
 
-def force_iterable(value: Any) -> Union[List, Tuple]:
+def force_iterable(value: Any) -> list | tuple:
     """Converts the value to a list or a tuple.
 
     Only relevant for negative test scenarios where the original types might be changed.
@@ -180,7 +203,7 @@ def force_iterable(value: Any) -> Union[List, Tuple]:
     return [value]
 
 
-def force_dict(value: Any) -> Dict:
+def force_dict(value: Any) -> dict:
     """Converts the value to a dictionary.
 
     Only relevant for negative test scenarios where the original types might be changed.
@@ -198,7 +221,7 @@ def to_json(item: Generated, name: str) -> None:
 
 @conversion
 def delimited(item: Generated, name: str, delimiter: str) -> None:
-    item[name] = delimiter.join(map(str, force_iterable(item[name] or ())))
+    item[name] = delimiter.join(map(str, force_iterable(item[name] if item[name] is not None else ())))
 
 
 @conversion
@@ -228,7 +251,7 @@ def delimited_object(item: Generated, name: str) -> None:
 def extracted_object(item: Generated, name: str) -> None:
     """Merge a child node to the parent one."""
     generated = item.pop(name)
-    if generated:
+    if generated and isinstance(generated, dict):
         item.update(generated)
     else:
         item[name] = ""
@@ -248,7 +271,7 @@ def label_primitive(item: Generated, name: str) -> None:
 
 
 @conversion
-def label_array(item: Generated, name: str, explode: Optional[bool]) -> None:
+def label_array(item: Generated, name: str, explode: bool | None) -> None:
     """Serialize an array with the `label` style.
 
     Explode=True
@@ -271,7 +294,7 @@ def label_array(item: Generated, name: str, explode: Optional[bool]) -> None:
 
 
 @conversion
-def label_object(item: Generated, name: str, explode: Optional[bool]) -> None:
+def label_object(item: Generated, name: str, explode: bool | None) -> None:
     """Serialize an object with the `label` style.
 
     Explode=True
@@ -307,7 +330,7 @@ def matrix_primitive(item: Generated, name: str) -> None:
 
 
 @conversion
-def matrix_array(item: Generated, name: str, explode: Optional[bool]) -> None:
+def matrix_array(item: Generated, name: str, explode: bool | None) -> None:
     """Serialize an array with the `matrix` style.
 
     Explode=True
@@ -329,7 +352,7 @@ def matrix_array(item: Generated, name: str, explode: Optional[bool]) -> None:
 
 
 @conversion
-def matrix_object(item: Generated, name: str, explode: Optional[bool]) -> None:
+def matrix_object(item: Generated, name: str, explode: bool | None) -> None:
     """Serialize an object with the `matrix` style.
 
     Explode=True