call pook.off in the end of pook.activate wrapper function

src/pook/activate_async.py

@@ -24,5 +24,6 @@ async def wrapper(*args, **kw):
                 fn(*args, **kw)
         finally:
             _engine.disable()
+            _engine.reset()
 
     return wrapper

src/pook/api.py

@@ -138,6 +138,7 @@ def wrapper(*args, **kw):
             fn(*args, **kw)
         finally:
             _engine.disable()
+            _engine.reset()
 
     return wrapper
 

tests/unit/api_test.py

@@ -1,3 +1,4 @@
+import asyncio
 import pytest
 
 from pook import api
@@ -47,3 +48,28 @@ def test_mock_contructors(engine):
 
     assert len(engine.mocks) == 0
     assert engine.active is False
+
+
+def test_activate_as_decorator(engine):
+    @api.activate
+    def test_activate():
+        api.get("foo.com")
+        assert engine.active is True
+        assert engine.isdone() is False
+
+    test_activate()
+    assert engine.active is False
+    assert engine.isdone() is True
+
+
+async def test_activate_as_decorator_for_async(engine):
+    @api.activate
+    async def test_activate():
+        await asyncio.sleep(0)
+        api.get("foo.com")
+        assert engine.active is True
+        assert engine.isdone() is False
+
+    await test_activate()
+    assert engine.active is False
+    assert engine.isdone() is True