Refactor.

Author: mindflayer

mocket/async_mocket.py

@@ -1,5 +1,3 @@
-import decorator
-
 from mocket import Mocketizer
 
 
@@ -12,25 +10,8 @@ async def __aexit__(*args, **kwargs):
             return Mocketizer.__exit__(*args, **kwargs)
 
         @staticmethod
-        def async_wrap(test=None, truesocket_recording_dir=None):
-            async def wrapper(t, *args, **kw):
-                instance = args[0] if args else None
-                namespace = ".".join(
-                    (
-                        instance.__class__.__module__,
-                        instance.__class__.__name__,
-                        t.__name__,
-                    )
-                )
-                async with AsyncMocketizer(
-                    instance,
-                    namespace=namespace,
-                    truesocket_recording_dir=truesocket_recording_dir,
-                ):
-                    await t(*args, **kw)
-                return wrapper
-
-            return decorator.decorator(wrapper, test)
+        def async_wrap(*args, **kwargs):
+            return Mocketizer.wrap(*args, **kwargs)
 
     return AsyncMocketizer.async_wrap