Move test client to emmett-core RSGI

emmett/app.py

@@ -324,9 +324,8 @@ class App(_App):
 
     config_class = Config
     modules_class = AppModule
-    # routers_class = (HTTPRouter, WebsocketRouter)
     signals_class = Signals
-    test_client_class = None
+    test_client_class = EmmettTestClient
 
     def __init__(
         self,
@@ -457,11 +456,6 @@ def config_from_yaml(self, filename: str, namespace: Optional[str] = None):
     def use_template_extension(self, ext_cls, **config):
         return self.templater.use_extension(ext_cls, **config)
 
-    # TODO: move to emmett_core
-    def test_client(self, use_cookies: bool = True, **kwargs) -> EmmettTestClient:
-        tclass = self.test_client_class or EmmettTestClient
-        return tclass(self, use_cookies=use_cookies, **kwargs)
-
     def module(
         self,
         import_name: str,

emmett/rsgi/handlers.py

@@ -72,5 +72,4 @@ async def _debug_handler(self) -> str:
 
 
 class WSHandler(_WSHandler):
-    __slots__ = ['pre_handler', '__dict__']
     wrapper_cls = Websocket

emmett/testing.py

@@ -0,0 +1,31 @@
+from emmett_core.protocols.rsgi.test_client.client import EmmettTestClient as _EmmettTestClient, ClientContext as _ClientContext, ClientHTTPHandlerMixin
+
+from .ctx import current
+from .rsgi.handlers import HTTPHandler
+from .wrappers.response import Response
+
+
+class ClientContextResponse(Response):
+    def __init__(self, original_response: Response):
+        super().__init__()
+        self.status = original_response.status
+        self.headers._data.update(original_response.headers._data)
+        self.cookies.update(original_response.cookies.copy())
+        self.__dict__.update(original_response.__dict__)
+
+
+class ClientContext(_ClientContext):
+    _response_wrap_cls = ClientContextResponse
+
+    def __init__(self, ctx):
+        super().__init__(ctx)
+        self.T = current.T
+
+
+class ClientHTTPHandler(ClientHTTPHandlerMixin, HTTPHandler):
+    _client_ctx_cls = ClientContext
+
+
+class EmmettTestClient(_EmmettTestClient):
+    _current = current
+    _handler_cls = ClientHTTPHandler