Finish Typing

mocket/__init__.py

@@ -1,20 +1,48 @@
-from mocket.async_mocket import async_mocketize
-from mocket.entry import MocketEntry
-from mocket.mocket import Mocket
-from mocket.mocketizer import Mocketizer, mocketize
-from mocket.ssl.context import MocketSSLContext
-
-# NOTE this is here for backwards-compat to keep old import-paths working
-from mocket.ssl.context import MocketSSLContext as FakeSSLContext
+from mocket.bytes import (
+    MocketBytesEntry,
+    MocketBytesRequest,
+    MocketBytesResponse,
+)
+from mocket.compat import FakeSSLContext, MocketEntry
+from mocket.core.async_mocket import async_mocketize
+from mocket.core.mocket import Mocket
+from mocket.core.mocketizer import Mocketizer, mocketize
+from mocket.core.socket import MocketSocket
+from mocket.core.ssl.context import MocketSSLContext
+from mocket.core.ssl.socket import MocketSSLSocket
+from mocket.http import (
+    MocketHttpEntry,
+    MocketHttpMethod,
+    MocketHttpRequest,
+    MocketHttpResponse,
+)
+from mocket.redis import (
+    MocketRedisEntry,
+    MocketRedisRequest,
+    MocketRedisResponse,
+)
 
-__all__ = (
-    "async_mocketize",
-    "mocketize",
+__all__ = [
     "Mocket",
-    "MocketEntry",
-    "Mocketizer",
+    "MocketBytesEntry",
+    "MocketBytesRequest",
+    "MocketBytesResponse",
+    "MocketHttpEntry",
+    "MocketHttpMethod",
+    "MocketHttpRequest",
+    "MocketHttpResponse",
+    "MocketRedisEntry",
+    "MocketRedisRequest",
+    "MocketRedisResponse",
     "MocketSSLContext",
+    "MocketSSLSocket",
+    "MocketSocket",
+    "Mocketizer",
+    "async_mocketize",
+    "mocketize",
+    # NOTE this is here for backwards-compat to keep old import-paths working
     "FakeSSLContext",
-)
+    "MocketEntry",
+]
 
 __version__ = "3.13.2"

mocket/async_mocket.py

@@ -1,22 +1,6 @@
-from mocket.mocketizer import Mocketizer
-from mocket.utils import get_mocketize
+from mocket.core.async_mocket import async_mocketize
 
-
-async def wrapper(
-    test,
-    truesocket_recording_dir=None,
-    strict_mode=False,
-    strict_mode_allowed=None,
-    *args,
-    **kwargs,
-):
-    async with Mocketizer.factory(
-        test, truesocket_recording_dir, strict_mode, strict_mode_allowed, args
-    ):
-        return await test(*args, **kwargs)
-
-
-async_mocketize = get_mocketize(wrapper_=wrapper)
-
-
-__all__ = ("async_mocketize",)
+# NOTE this is here for backwards-compat to keep old import-paths working
+__all__ = [
+    "async_mocketize",
+]

mocket/bytes.py

@@ -0,0 +1,79 @@
+from __future__ import annotations
+
+from typing import Sequence
+
+from typing_extensions import Self
+
+from mocket.core.entry import MocketBaseEntry, MocketBaseRequest, MocketBaseResponse
+from mocket.core.mocket import Mocket
+from mocket.core.types import Address
+
+
+class MocketBytesRequest(MocketBaseRequest):
+    def __init__(self) -> None:
+        self._data = b""
+
+    @property
+    def data(self) -> bytes:
+        return self._data
+
+    @classmethod
+    def from_data(cls: type[Self], data: bytes) -> Self:
+        request = cls()
+        request._data = data
+        return request
+
+
+class MocketBytesResponse(MocketBaseResponse):
+    def __init__(self, data: bytes | str | bool) -> None:
+        if isinstance(data, str):
+            data = data.encode()
+        elif isinstance(data, bool):
+            data = bytes(data)
+        self._data = data
+
+    @property
+    def data(self) -> bytes:
+        return self._data
+
+
+class MocketBytesEntry(MocketBaseEntry):
+    request_cls = MocketBytesRequest
+    response_cls = MocketBytesResponse
+
+    def __init__(
+        self,
+        address: Address,
+        responses: Sequence[MocketBytesResponse | Exception],
+    ) -> None:
+        if not len(responses):
+            responses = [MocketBytesResponse(data=b"")]
+
+        super().__init__(
+            address=address,
+            responses=responses,
+        )
+
+    @classmethod
+    def register_response(
+        cls,
+        address: Address,
+        response: MocketBytesResponse | Exception,
+    ) -> None:
+        entry = cls(
+            address=address,
+            responses=[response],
+        )
+        Mocket.register(entry)
+
+    @classmethod
+    def register_responses(
+        cls,
+        address: Address,
+        responses: Sequence[MocketBytesResponse | Exception],
+    ) -> None:
+        entry = cls(
+            address=address,
+            responses=responses,
+        )
+        Mocket.register(entry)

mocket/compat/__init__.py

@@ -0,0 +1,8 @@
+from mocket.compat.entry import MocketEntry, Response
+from mocket.core.ssl.context import MocketSSLContext as FakeSSLContext
+
+__all__ = [
+    "FakeSSLContext",
+    "MocketEntry",
+    "Response",
+]

mocket/compat/entry.py

@@ -0,0 +1,36 @@
+from __future__ import annotations
+
+from mocket.bytes import MocketBytesEntry, MocketBytesResponse
+from mocket.core.types import Address
+
+
+class Response(MocketBytesResponse):
+    def __init__(self, data: bytes | str | bool) -> None:
+        if isinstance(data, str):
+            data = data.encode()
+        elif isinstance(data, bool):
+            data = bytes(data)
+        self._data = data
+
+
+class MocketEntry(MocketBytesEntry):
+    def __init__(
+        self,
+        location: Address,
+        responses: list[MocketBytesResponse | Exception | bytes | str | bool]
+        | MocketBytesResponse
+        | Exception
+        | bytes
+        | str
+        | bool,
+    ) -> None:
+        if not isinstance(responses, list):
+            responses = [responses]
+
+        _responses = []
+        for response in responses:
+            if not isinstance(response, (MocketBytesResponse, Exception)):
+                response = MocketBytesResponse(response)
+            _responses.append(response)
+
+        super().__init__(address=location, responses=_responses)

mocket/compat/mockhttp.py

@@ -0,0 +1,139 @@
+from __future__ import annotations
+
+from io import BufferedReader
+from typing import Any
+
+from mocket.http import (
+    MocketHttpEntry,
+    MocketHttpMethod,
+    MocketHttpRequest,
+    MocketHttpResponse,
+)
+from mocket.mocket import Mocket
+
+
+class Response(MocketHttpResponse):
+    def __init__(
+        self,
+        body: str | bytes | BufferedReader = b"",
+        status: int = 200,
+        headers: dict[str, str] | None = None,
+    ) -> None:
+        super().__init__(
+            status_code=status,
+            headers=headers,
+            body=body,
+        )
+
+    @property
+    def status(self) -> int:
+        return self.status_code
+
+
+class Request(MocketHttpRequest):
+    @property
+    def body(self) -> str | None:  # type: ignore
+        body = super().body
+        if body is None:
+            return None
+        return body.decode()
+
+
+class Entry(MocketHttpEntry):
+    request_cls = Request
+    response_cls = Response  # type: ignore[assignment]
+
+    CONNECT = MocketHttpMethod.CONNECT
+    DELETE = MocketHttpMethod.DELETE
+    GET = MocketHttpMethod.GET
+    HEAD = MocketHttpMethod.HEAD
+    OPTIONS = MocketHttpMethod.OPTIONS
+    PATCH = MocketHttpMethod.PATCH
+    POST = MocketHttpMethod.POST
+    PUT = MocketHttpMethod.PUT
+    TRACE = MocketHttpMethod.TRACE
+
+    METHODS = list(MocketHttpMethod)
+
+    def __init__(
+        self,
+        uri: str,
+        method: MocketHttpMethod,
+        responses: list[Response | Exception],
+        match_querystring: bool = True,
+        add_trailing_slash: bool = True,
+    ) -> None:
+        super().__init__(
+            method=method,
+            uri=uri,
+            responses=responses,
+            match_querystring=match_querystring,
+            add_trailing_slash=add_trailing_slash,
+        )
+
+    def __repr__(self) -> str:
+        return (
+            f"{self.__class__.__name__}("
+            f"method='{self.method.name}', "
+            f"schema='{self.schema}', "
+            f"location={self.address}, "
+            f"path='{self.path}', "
+            f"query='{self.query}'"
+            ")"
+        )
+
+    @property
+    def schema(self) -> str:
+        return self.scheme
+
+    @classmethod
+    def register(
+        cls,
+        method: MocketHttpMethod,
+        uri: str,
+        *responses: Response | Exception,
+        **config: Any,
+    ) -> None:
+        if "body" in config or "status" in config:
+            raise AttributeError("Did you mean `Entry.single_register(...)`?")
+
+        if isinstance(config, dict):
+            match_querystring = config.get("match_querystring", True)
+            add_trailing_slash = config.get("add_trailing_slash", True)
+
+        entry = cls(
+            method=method,
+            uri=uri,
+            responses=list(responses),
+            match_querystring=match_querystring,
+            add_trailing_slash=add_trailing_slash,
+        )
+        Mocket.register(entry)
+
+    @classmethod
+    def single_register(
+        cls,
+        method: MocketHttpMethod,
+        uri: str,
+        body: str | bytes | BufferedReader = b"",
+        status: int = 200,
+        headers: dict[str, str] | None = None,
+        match_querystring: bool = True,
+        exception: Exception | None = None,
+    ) -> None:
+        response: Response | Exception
+        if exception is not None:
+            response = exception
+        else:
+            response = Response(
+                body=body,
+                status=status,
+                headers=headers,
+            )
+
+        cls.register(
+            method,
+            uri,
+            response,
+            match_querystring=match_querystring,
+        )