test: BI-6602 add TestingHttpxClient (#1253)

* test: BI-6602 add TestingHttpxClient

* fix: drop redundant contexts

* fix: naming

Author: ovsds

lib/dl_httpx/dl_httpx/__init__.py

@@ -17,6 +17,7 @@
     TypedSchemaDictAnnotation,
     TypedSchemaListAnnotation,
 )
+from .testing import TestingHttpxClient
 
 
 __all__ = [
@@ -35,4 +36,5 @@
     "HttpStatusHttpxClientException",
     "RequestHttpxClientException",
     "NoRetriesHttpxClientException",
+    "TestingHttpxClient",
 ]

lib/dl_httpx/dl_httpx/testing/__init__.py

@@ -0,0 +1,13 @@
+import pathlib
+
+import dl_testing
+
+
+dl_testing.register_all_assert_rewrites(__name__, pathlib.Path(__file__).parent)
+
+from .client import TestingHttpxClient
+
+
+__all__ = [
+    "TestingHttpxClient",
+]

lib/dl_httpx/dl_httpx/testing/client.py

@@ -0,0 +1,58 @@
+import asyncio
+from typing import (
+    Any,
+    Awaitable,
+    Callable,
+)
+
+import attrs
+
+
+@attrs.define(kw_only=True, auto_attribs=True, frozen=True)
+class TestingHttpxClient:
+    sync_client: Any
+    async_client: Any
+
+    def _sync_method_wrapper(
+        self,
+        sync_client_method: Callable[..., Any],
+        async_client_method: Callable[..., Any],
+    ) -> Callable[..., Any]:
+        def wrapper(*args: Any, **kwargs: Any) -> Any:
+            sync_response = sync_client_method(*args, **kwargs)
+            async_response = async_client_method(*args, **kwargs)
+
+            assert sync_response == async_response
+
+            return sync_response
+
+        return wrapper
+
+    def _async_method_wrapper(
+        self,
+        sync_client_method: Callable[..., Any],
+        async_client_method: Callable[..., Awaitable[Any]],
+    ) -> Callable[..., Awaitable[Any]]:
+        async def wrapper(*args: Any, **kwargs: Any) -> Any:
+            async_response = await async_client_method(*args, **kwargs)
+            sync_response = sync_client_method(*args, **kwargs)
+
+            assert async_response == sync_response
+
+            return async_response
+
+        return wrapper
+
+    def __getattr__(self, name: str) -> Callable[..., Any]:
+        sync_client_method = getattr(self.sync_client, name)
+        async_client_method = getattr(self.async_client, name)
+
+        if asyncio.iscoroutinefunction(async_client_method):
+            return self._async_method_wrapper(sync_client_method, async_client_method)
+
+        return self._sync_method_wrapper(sync_client_method, async_client_method)
+
+
+__all__ = [
+    "TestingHttpxClient",
+]

lib/dl_httpx/pyproject.toml

@@ -12,6 +12,7 @@ attrs = "*"
 dl-configs = {path = "../dl_configs"}
 dl-pydantic = {path = "../dl_pydantic"}
 dl-retrier = {path = "../dl_retrier"}
+dl-testing = {path = "../dl_testing"}
 httpx = "*"
 pydantic = "*"
 python = ">=3.10, <3.13"