asyncly 0.3.4__tar.gz → 0.4.0__tar.gz

{asyncly-0.3.4 → asyncly-0.4.0}/.gitignore

@@ -161,4 +161,5 @@ cython_debug/
 #  and can be added to the global gitignore or merged into this file.  For a more nuclear
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 .idea/
-.vscode/
+.vscode/
+.DS_Store

{asyncly-0.3.4 → asyncly-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: asyncly
-Version: 0.3.4
+Version: 0.4.0
 Summary: Simple HTTP client and server for your integrations based on aiohttp
 Project-URL: Homepage, https://github.com/andy-takker/asyncly
 Project-URL: Source, https://github.com/andy-takker/asyncly
@@ -30,7 +30,7 @@ Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: <4,>=3.10
 Requires-Dist: aiohttp<4,>=3.9.5
 Provides-Extra: msgspec
-Requires-Dist: msgspec<0.19,>=0.18.6; extra == 'msgspec'
+Requires-Dist: msgspec<0.20,>=0.19.0; extra == 'msgspec'
 Provides-Extra: orjson
 Requires-Dist: orjson<4,>=3.10.6; extra == 'orjson'
 Provides-Extra: pydantic
@@ -98,7 +98,7 @@ Quick start guide
 -----------------
 
 HttpClient
-~~~~~~~~~~~~~~
+~~~~~~~~~~
 
 Simple HTTP Client for `https://catfact.ninja`. See full example in `examples/catfact_client.py`_
 
@@ -130,6 +130,9 @@ Simple HTTP Client for `https://catfact.ninja`. See full example in `examples/ca
 Test Async Server for client
 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 
+Example
+*******
+
 For the HTTP client, we create a server to which he will go and simulate real
 responses. You can dynamically change the responses from the server in
 a specific test.
@@ -191,12 +194,43 @@ Now we can use them in tests. See full example in `examples/test_catfact_client.
         with pytest.raises(asyncio.TimeoutError):
             await catfact_client.fetch_random_cat_fact(timeout=1)
 
+Useful responses and serializers
+********************************
+
+- JsonResponse_: simple JSON response from any object.
+  You can setup status code and serializer for it. Using JsonSerializer_
+
+- MsgpackResponse_: response in msgpack_ format with It's like JSON.
+  But fast and small. Using MsgpackSerializer_.
+
+- SequenceResponse_: useful response if you want return different responses
+  on next request. Accepts BaseMockResponse_'s input.
+
+- TimeoutResponse_: response with latency. For slow testing
+
+- TomlResponse_: return TOML format text response. Using TomlSerializer_.
+
+- YamlResponse_: return YAML format text response. Using YamlSerializer_.
 
 .. _PyPI: https://pypi.org/
 .. _aiohttp: https://pypi.org/project/aiohttp/
+.. _msgpack: https://msgpack.org
 .. _msgspec: https://github.com/jcrist/msgspec
 .. _orjson: https://github.com/ijl/orjson
 .. _pydantic: https://github.com/pydantic/pydantic
 
 .. _examples/catfact_client.py: https://github.com/andy-takker/asyncly/blob/master/examples/catfact_client.py
-.. _examples/test_catfact_client.py: https://github.com/andy-takker/asyncly/blob/master/examples/test_catfact_client.py
+.. _examples/test_catfact_client.py: https://github.com/andy-takker/asyncly/blob/master/examples/test_catfact_client.py
+
+.. _BaseMockResponse: https://github.com/andy-takker/asyncly/blob/master/asyncly/srvmocker/responses/base.py
+.. _JsonResponse: https://github.com/andy-takker/asyncly/blob/master/asyncly/srvmocker/responses/json.py
+.. _MsgpackResponse: https://github.com/andy-takker/asyncly/blob/master/asyncly/srvmocker/responses/msgpack.py
+.. _SequenceResponse: https://github.com/andy-takker/asyncly/blob/master/asyncly/srvmocker/responses/sequence.py
+.. _TimeoutResponse: https://github.com/andy-takker/asyncly/blob/master/asyncly/srvmocker/responses/timeout.py
+.. _TomlResponse: https://github.com/andy-takker/asyncly/blob/master/asyncly/srvmocker/responses/toml.py
+.. _YamlResponse: https://github.com/andy-takker/asyncly/blob/master/asyncly/srvmocker/responses/yaml.py
+
+.. _JsonSerializer: https://github.com/andy-takker/asyncly/blob/master/asyncly/srvmocker/serialization/json.py
+.. _MsgpackSerializer: https://github.com/andy-takker/asyncly/blob/master/asyncly/srvmocker/serialization/msgpack.py
+.. _TomlSerializer: https://github.com/andy-takker/asyncly/blob/master/asyncly/srvmocker/serialization/toml.py
+.. _YamlSerializer: https://github.com/andy-takker/asyncly/blob/master/asyncly/srvmocker/serialization/yaml.py

{asyncly-0.3.4 → asyncly-0.4.0}/README.rst

@@ -59,7 +59,7 @@ Quick start guide
 -----------------
 
 HttpClient
-~~~~~~~~~~~~~~
+~~~~~~~~~~
 
 Simple HTTP Client for `https://catfact.ninja`. See full example in `examples/catfact_client.py`_
 
@@ -91,6 +91,9 @@ Simple HTTP Client for `https://catfact.ninja`. See full example in `examples/ca
 Test Async Server for client
 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 
+Example
+*******
+
 For the HTTP client, we create a server to which he will go and simulate real
 responses. You can dynamically change the responses from the server in
 a specific test.
@@ -152,12 +155,43 @@ Now we can use them in tests. See full example in `examples/test_catfact_client.
         with pytest.raises(asyncio.TimeoutError):
             await catfact_client.fetch_random_cat_fact(timeout=1)
 
+Useful responses and serializers
+********************************
+
+- JsonResponse_: simple JSON response from any object.
+  You can setup status code and serializer for it. Using JsonSerializer_
+
+- MsgpackResponse_: response in msgpack_ format with It's like JSON.
+  But fast and small. Using MsgpackSerializer_.
+
+- SequenceResponse_: useful response if you want return different responses
+  on next request. Accepts BaseMockResponse_'s input.
+
+- TimeoutResponse_: response with latency. For slow testing
+
+- TomlResponse_: return TOML format text response. Using TomlSerializer_.
+
+- YamlResponse_: return YAML format text response. Using YamlSerializer_.
 
 .. _PyPI: https://pypi.org/
 .. _aiohttp: https://pypi.org/project/aiohttp/
+.. _msgpack: https://msgpack.org
 .. _msgspec: https://github.com/jcrist/msgspec
 .. _orjson: https://github.com/ijl/orjson
 .. _pydantic: https://github.com/pydantic/pydantic
 
 .. _examples/catfact_client.py: https://github.com/andy-takker/asyncly/blob/master/examples/catfact_client.py
-.. _examples/test_catfact_client.py: https://github.com/andy-takker/asyncly/blob/master/examples/test_catfact_client.py
+.. _examples/test_catfact_client.py: https://github.com/andy-takker/asyncly/blob/master/examples/test_catfact_client.py
+
+.. _BaseMockResponse: https://github.com/andy-takker/asyncly/blob/master/asyncly/srvmocker/responses/base.py
+.. _JsonResponse: https://github.com/andy-takker/asyncly/blob/master/asyncly/srvmocker/responses/json.py
+.. _MsgpackResponse: https://github.com/andy-takker/asyncly/blob/master/asyncly/srvmocker/responses/msgpack.py
+.. _SequenceResponse: https://github.com/andy-takker/asyncly/blob/master/asyncly/srvmocker/responses/sequence.py
+.. _TimeoutResponse: https://github.com/andy-takker/asyncly/blob/master/asyncly/srvmocker/responses/timeout.py
+.. _TomlResponse: https://github.com/andy-takker/asyncly/blob/master/asyncly/srvmocker/responses/toml.py
+.. _YamlResponse: https://github.com/andy-takker/asyncly/blob/master/asyncly/srvmocker/responses/yaml.py
+
+.. _JsonSerializer: https://github.com/andy-takker/asyncly/blob/master/asyncly/srvmocker/serialization/json.py
+.. _MsgpackSerializer: https://github.com/andy-takker/asyncly/blob/master/asyncly/srvmocker/serialization/msgpack.py
+.. _TomlSerializer: https://github.com/andy-takker/asyncly/blob/master/asyncly/srvmocker/serialization/toml.py
+.. _YamlSerializer: https://github.com/andy-takker/asyncly/blob/master/asyncly/srvmocker/serialization/yaml.py

{asyncly-0.3.4 → asyncly-0.4.0}/asyncly/client/base.py

@@ -1,3 +1,4 @@
+import sys
 from typing import Any, Literal
 
 from aiohttp import ClientSession
@@ -10,6 +11,13 @@ from asyncly.client.handlers.base import (
 )
 from asyncly.client.timeout import TimeoutType, get_timeout
 
+if sys.version_info >= (3, 11):
+    from http import HTTPMethod
+
+    MethodType = HTTPMethod | Literal["GET", "POST", "PUT", "DELETE", "PATCH", "HEAD"]
+else:
+    MethodType = Literal["GET", "POST", "PUT", "DELETE", "PATCH", "HEAD"]
+
 
 class BaseHttpClient:
     __slots__ = ("_url", "_session", "_client_name")
@@ -31,7 +39,7 @@ class BaseHttpClient:
 
     async def _make_req(
         self,
-        method: Literal["GET", "POST", "PUT", "DELETE", "PATCH", "HEAD"],
+        method: MethodType,
         url: URL,
         handlers: ResponseHandlersType,
         timeout: TimeoutType = DEFAULT_TIMEOUT,

{asyncly-0.3.4 → asyncly-0.4.0}/asyncly/client/handlers/base.py

@@ -17,7 +17,7 @@ async def apply_handler(
     handler = _find_handler(handlers=handlers, status=response.status)
     if not handler:
         raise UnhandledStatusException(
-            f"Unexpected resposne {response.status} from {response.url}",
+            f"Unexpected response {response.status} from {response.url}",
             status=response.status,
             url=response.url,
             client_name=client_name,

asyncly-0.4.0/asyncly/client/handlers/msgspec.py

@@ -0,0 +1,49 @@
+from collections.abc import Awaitable, Callable
+from typing import Any, Literal, Protocol, TypeVar
+
+from aiohttp import ClientResponse
+from msgspec import Struct
+from msgspec.json import decode as decode_json
+from msgspec.msgpack import decode as decode_msgpack
+from msgspec.toml import decode as decode_toml
+from msgspec.yaml import decode as decode_yaml
+
+T = TypeVar("T", bound=Struct)
+
+DataFormat = Literal["json", "msgpack", "toml", "yaml"]
+
+
+class DataFormatDecode(Protocol):
+    def __call__(
+        self,
+        buf: bytes | str,
+        *,
+        type: type[T] = ...,
+        strict: bool = True,
+        dec_hook: Callable[[type, Any], Any] | None = None,
+    ) -> Any: ...
+
+
+def parse_struct(
+    struct: type[T],
+    data_format: DataFormat = "json",
+    strict: bool = True,
+) -> Callable[[ClientResponse], Awaitable[T]]:
+    decode = _choose_decoder(data_format)
+
+    async def _parse(response: ClientResponse) -> T:
+        return decode(await response.read(), type=struct, strict=strict)
+
+    return _parse
+
+
+def _choose_decoder(data_format: DataFormat) -> DataFormatDecode:
+    if data_format == "json":
+        return decode_json  # type: ignore[return-value]
+    elif data_format == "msgpack":
+        return decode_msgpack  # type: ignore[return-value]
+    elif data_format == "toml":
+        return decode_toml
+    elif data_format == "yaml":
+        return decode_yaml
+    return decode_json

asyncly-0.4.0/asyncly/srvmocker/__init__.py

@@ -0,0 +1,14 @@
+from asyncly.srvmocker.models import MockRoute, MockService
+from asyncly.srvmocker.responses.base import BaseMockResponse
+from asyncly.srvmocker.responses.content import ContentResponse
+from asyncly.srvmocker.responses.json import JsonResponse
+from asyncly.srvmocker.service import start_service
+
+__all__ = (
+    "BaseMockResponse",
+    "ContentResponse",
+    "MockRoute",
+    "MockService",
+    "JsonResponse",
+    "start_service",
+)

{asyncly-0.3.4 → asyncly-0.4.0}/asyncly/srvmocker/models.py

@@ -1,11 +1,11 @@
-from abc import ABC, abstractmethod
 from collections.abc import MutableMapping, MutableSequence
 from dataclasses import dataclass
 
 from aiohttp.web_request import Request
-from aiohttp.web_response import Response
 from yarl import URL
 
+from asyncly.srvmocker.responses.base import BaseMockResponse
+
 
 @dataclass(frozen=True)
 class MockRoute:
@@ -20,12 +20,6 @@ class RequestHistory:
     body: bytes
 
 
-class BaseMockResponse(ABC):
-    @abstractmethod
-    async def response(self, request: Request) -> Response:
-        pass
-
-
 @dataclass(frozen=True)
 class MockService:
     history: MutableSequence[RequestHistory]

asyncly-0.4.0/asyncly/srvmocker/responses/base.py

@@ -0,0 +1,10 @@
+from abc import ABC, abstractmethod
+
+from aiohttp.web_request import Request
+from aiohttp.web_response import Response
+
+
+class BaseMockResponse(ABC):
+    @abstractmethod
+    async def response(self, request: Request) -> Response:
+        pass

asyncly-0.4.0/asyncly/srvmocker/responses/content.py

@@ -0,0 +1,36 @@
+from collections.abc import Mapping, MutableMapping
+from dataclasses import dataclass
+from http import HTTPStatus
+from typing import Any
+
+from aiohttp import hdrs
+from aiohttp.web_request import Request
+from aiohttp.web_response import Response
+
+from asyncly.srvmocker.responses.base import BaseMockResponse
+from asyncly.srvmocker.serialization.base import Serializer
+
+
+@dataclass
+class ContentResponse(BaseMockResponse):
+    body: Any = None
+    status: int = HTTPStatus.OK
+    headers: Mapping[str, str] | None = None
+    serializer: Serializer | None = None
+
+    async def response(self, request: Request) -> Response:
+        headers: MutableMapping[str, str] = dict()
+        if self.headers:
+            headers.update(self.headers)
+        if self.serializer:
+            headers[hdrs.CONTENT_TYPE] = self.serializer.content_type
+        return Response(
+            status=self.status,
+            body=self.serialize(),
+            headers=headers,
+        )
+
+    def serialize(self) -> Any:
+        if not self.serializer:
+            return self.body
+        return self.serializer.dumps(self.body)

asyncly-0.4.0/asyncly/srvmocker/responses/json.py

@@ -0,0 +1,30 @@
+from collections.abc import Mapping
+from http import HTTPStatus
+from typing import Any
+
+from aiohttp.web_request import Request
+from aiohttp.web_response import Response
+
+from asyncly.srvmocker.responses.base import BaseMockResponse
+from asyncly.srvmocker.responses.content import ContentResponse
+from asyncly.srvmocker.serialization.json import JsonSerializer
+
+
+class JsonResponse(BaseMockResponse):
+    _content: ContentResponse
+
+    def __init__(
+        self,
+        body: Any,
+        status: int = HTTPStatus.OK,
+        headers: Mapping[str, str] | None = None,
+    ) -> None:
+        self._content = ContentResponse(
+            body=body,
+            status=status,
+            headers=headers,
+            serializer=JsonSerializer,
+        )
+
+    async def response(self, request: Request) -> Response:
+        return await self._content.response(request)

asyncly-0.4.0/asyncly/srvmocker/responses/msgpack.py

@@ -0,0 +1,30 @@
+from collections.abc import Mapping
+from http import HTTPStatus
+from typing import Any
+
+from aiohttp.web_request import Request
+from aiohttp.web_response import Response
+
+from asyncly.srvmocker.responses.base import BaseMockResponse
+from asyncly.srvmocker.responses.content import ContentResponse
+from asyncly.srvmocker.serialization.msgpack import MsgpackSerializer
+
+
+class MsgpackResponse(BaseMockResponse):
+    __content: ContentResponse
+
+    def __init__(
+        self,
+        body: Any,
+        status: int = HTTPStatus.OK,
+        headers: Mapping[str, str] | None = None,
+    ) -> None:
+        self.__content = ContentResponse(
+            body=body,
+            status=status,
+            headers=headers,
+            serializer=MsgpackSerializer,
+        )
+
+    async def response(self, request: Request) -> Response:
+        return await self.__content.response(request)

asyncly-0.4.0/asyncly/srvmocker/responses/sequence.py

@@ -0,0 +1,17 @@
+from collections.abc import Iterable, Iterator
+
+from aiohttp.web_request import Request
+from aiohttp.web_response import Response
+
+from asyncly.srvmocker.responses.base import BaseMockResponse
+
+
+class SequenceResponse(BaseMockResponse):
+    responses: Iterator[BaseMockResponse]
+
+    def __init__(self, responses: Iterable[BaseMockResponse]) -> None:
+        self.responses = iter(responses)
+
+    async def response(self, request: Request) -> Response:
+        resp = next(self.responses)
+        return await resp.response(request)

asyncly-0.4.0/asyncly/srvmocker/responses/timeout.py

@@ -0,0 +1,19 @@
+from asyncio import sleep
+from dataclasses import dataclass
+
+from aiohttp.web_request import Request
+from aiohttp.web_response import Response
+
+from asyncly.srvmocker.responses.base import BaseMockResponse
+
+TimeoutType = int | float
+
+
+@dataclass
+class LatencyResponse(BaseMockResponse):
+    wrapped: BaseMockResponse
+    latency: TimeoutType
+
+    async def response(self, request: Request) -> Response:
+        await sleep(self.latency)
+        return await self.wrapped.response(request)

asyncly-0.4.0/asyncly/srvmocker/responses/toml.py

@@ -0,0 +1,30 @@
+from collections.abc import Mapping
+from http import HTTPStatus
+from typing import Any
+
+from aiohttp.web_request import Request
+from aiohttp.web_response import Response
+
+from asyncly.srvmocker.responses.base import BaseMockResponse
+from asyncly.srvmocker.responses.content import ContentResponse
+from asyncly.srvmocker.serialization.toml import TomlSerializer
+
+
+class TomlResponse(BaseMockResponse):
+    _content: ContentResponse
+
+    def __init__(
+        self,
+        body: Any,
+        status: int = HTTPStatus.OK,
+        headers: Mapping[str, str] | None = None,
+    ) -> None:
+        self._content = ContentResponse(
+            body=body,
+            status=status,
+            headers=headers,
+            serializer=TomlSerializer,
+        )
+
+    async def response(self, request: Request) -> Response:
+        return await self._content.response(request)

asyncly-0.4.0/asyncly/srvmocker/responses/yaml.py

@@ -0,0 +1,30 @@
+from collections.abc import Mapping
+from http import HTTPStatus
+from typing import Any
+
+from aiohttp.web_request import Request
+from aiohttp.web_response import Response
+
+from asyncly.srvmocker.responses.base import BaseMockResponse
+from asyncly.srvmocker.responses.content import ContentResponse
+from asyncly.srvmocker.serialization.yaml import YamlSerializer
+
+
+class YamlResponse(BaseMockResponse):
+    _content: ContentResponse
+
+    def __init__(
+        self,
+        body: Any,
+        status: int = HTTPStatus.OK,
+        headers: Mapping[str, str] | None = None,
+    ) -> None:
+        self._content = ContentResponse(
+            body=body,
+            status=status,
+            headers=headers,
+            serializer=YamlSerializer,
+        )
+
+    async def response(self, request: Request) -> Response:
+        return await self._content.response(request)

asyncly-0.4.0/asyncly/srvmocker/serialization/base.py

@@ -0,0 +1,9 @@
+from collections.abc import Callable
+from dataclasses import dataclass
+from typing import Any
+
+
+@dataclass(frozen=True)
+class Serializer:
+    dumps: Callable[[Any], str | bytes]
+    content_type: str

asyncly-0.4.0/asyncly/srvmocker/serialization/json.py

@@ -0,0 +1,9 @@
+import json
+from typing import Final
+
+from asyncly.srvmocker.serialization.base import Serializer
+
+JsonSerializer: Final = Serializer(
+    dumps=json.dumps,
+    content_type="application/json",
+)

asyncly-0.4.0/asyncly/srvmocker/serialization/msgpack.py

@@ -0,0 +1,10 @@
+from typing import Final
+
+import msgspec
+
+from asyncly.srvmocker.serialization.base import Serializer
+
+MsgpackSerializer: Final = Serializer(
+    dumps=msgspec.msgpack.encode,
+    content_type="application/msgpack",
+)

asyncly-0.4.0/asyncly/srvmocker/serialization/toml.py

@@ -0,0 +1,13 @@
+from typing import Final
+
+try:
+    import toml
+except ImportError:
+    raise ImportError("toml is not installed")
+
+from asyncly.srvmocker.serialization.base import Serializer
+
+TomlSerializer: Final = Serializer(
+    dumps=toml.dumps,
+    content_type="application/toml",
+)

asyncly-0.4.0/asyncly/srvmocker/serialization/yaml.py

@@ -0,0 +1,10 @@
+from typing import Final
+
+import yaml
+
+from asyncly.srvmocker.serialization.base import Serializer
+
+YamlSerializer: Final = Serializer(
+    dumps=yaml.dump,
+    content_type="application/yaml",
+)

{asyncly-0.3.4 → asyncly-0.4.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "asyncly"
-version = "0.3.4"
+version = "0.4.0"
 description = "Simple HTTP client and server for your integrations based on aiohttp"
 authors = [{ name = "Sergey Natalenko", email = "sergey.natalenko@mail.ru" }]
 requires-python = ">=3.10, <4"
@@ -35,7 +35,7 @@ classifiers = [
 dependencies = ["aiohttp>=3.9.5,<4"]
 
 [project.optional-dependencies]
-msgspec = ["msgspec>=0.18.6,<0.19"]
+msgspec = ["msgspec>=0.19.0,<0.20"]
 pydantic = ["pydantic>=2.8.2,<3"]
 orjson = ["orjson>=3.10.6,<4"]
 
@@ -47,13 +47,16 @@ Source = "https://github.com/andy-takker/asyncly"
 [dependency-groups]
 dev = [
     "pre-commit>=3.7.1,<5.0.0",
-    "mypy>=1.10.1,<2",
+    "mypy>=1.17.1,<2",
     "ruff>=0.5.2,<0.9.0",
     "restructuredtext-lint>=1.4.0,<2",
     "pygments>=2.18.0,<3",
     "pytest>=8.3.3,<9",
     "pytest-asyncio>=0.24.0,<0.25",
     "pytest-cov>=6.0.0,<7",
+    "types-pyyaml>=6.0.12.20250822",
+    "types-toml>=0.10.8.20240310",
+    "toml>=0.10.2",
 ]
 
 [tool.hatch.build.targets.sdist]

asyncly-0.3.4/asyncly/client/handlers/msgspec.py

@@ -1,15 +0,0 @@
-from collections.abc import Awaitable, Callable
-from typing import TypeVar
-
-from aiohttp import ClientResponse
-from msgspec import Struct
-from msgspec.json import decode
-
-T = TypeVar("T", bound=Struct)
-
-
-def parse_struct(struct: type[T]) -> Callable[[ClientResponse], Awaitable[T]]:
-    async def _parse(response: ClientResponse) -> T:
-        return decode(await response.read(), type=struct)
-
-    return _parse

asyncly-0.3.4/asyncly/srvmocker/__init__.py

@@ -1,10 +0,0 @@
-from asyncly.srvmocker.models import MockRoute, MockService
-from asyncly.srvmocker.responses import JsonResponse
-from asyncly.srvmocker.service import start_service
-
-__all__ = (
-    "MockRoute",
-    "MockService",
-    "JsonResponse",
-    "start_service",
-)

asyncly-0.3.4/asyncly/srvmocker/responses.py

@@ -1,80 +0,0 @@
-from asyncio import sleep
-from collections.abc import Iterable, Iterator, Mapping, MutableMapping
-from dataclasses import dataclass
-from http import HTTPStatus
-from typing import Any
-
-from aiohttp import hdrs
-from aiohttp.web_request import Request
-from aiohttp.web_response import Response
-
-from asyncly.srvmocker.models import BaseMockResponse
-from asyncly.srvmocker.serialization import JsonSerializer, Serializer
-
-TimeoutType = int | float
-
-
-@dataclass
-class ContentResponse(BaseMockResponse):
-    body: Any = None
-    status: int = HTTPStatus.OK
-    headers: Mapping[str, str] | None = None
-    serializer: Serializer | None = None
-
-    async def response(self, request: Request) -> Response:
-        headers: MutableMapping[str, str] = dict()
-        if self.headers:
-            headers.update(self.headers)
-        if self.serializer:
-            headers[hdrs.CONTENT_TYPE] = self.serializer.content_type
-        return Response(
-            status=self.status,
-            body=self.serialize(),
-            headers=headers,
-        )
-
-    def serialize(self) -> Any:
-        if not self.serializer:
-            return self.body
-        return self.serializer.dumps(self.body)
-
-
-@dataclass
-class LatencyResponse(BaseMockResponse):
-    wrapped: BaseMockResponse
-    latency: TimeoutType
-
-    async def response(self, request: Request) -> Response:
-        await sleep(self.latency)
-        return await self.wrapped.response(request)
-
-
-class MockSeqResponse(BaseMockResponse):
-    responses: Iterator[BaseMockResponse]
-
-    def __init__(self, responses: Iterable[BaseMockResponse]) -> None:
-        self.responses = iter(responses)
-
-    async def response(self, request: Request) -> Response:
-        resp = next(self.responses)
-        return await resp.response(request)
-
-
-class JsonResponse(BaseMockResponse):
-    __content: ContentResponse
-
-    def __init__(
-        self,
-        body: Any,
-        status: int = HTTPStatus.OK,
-        headers: Mapping[str, str] | None = None,
-    ) -> None:
-        self.__content = ContentResponse(
-            body=body,
-            status=status,
-            headers=headers,
-            serializer=JsonSerializer,
-        )
-
-    async def response(self, request: Request) -> Response:
-        return await self.__content.response(request)

asyncly-0.3.4/asyncly/srvmocker/serialization.py

@@ -1,16 +0,0 @@
-import json
-from collections.abc import Callable
-from dataclasses import dataclass
-from typing import Any, Final
-
-
-@dataclass(frozen=True)
-class Serializer:
-    dumps: Callable[[Any], str]
-    content_type: str
-
-
-JsonSerializer: Final = Serializer(
-    dumps=json.dumps,
-    content_type="application/json",
-)