Fix AttributeError when @acomic decorator is used in Class Based Vi…

…ews (#22)

* #21: @atomic not applicable for the Class Based Views

* CHANGES

CHANGES/21.bugfix

@@ -0,0 +1 @@
+Fix AttributeError when `@acomic` decorator is used in Class Based Views.

aiojobs/aiohttp.py

@@ -1,5 +1,7 @@
 from functools import wraps
 
+from aiohttp.web import View
+
 from . import create_scheduler
 
 __all__ = ('setup', 'spawn', 'get_scheduler', 'get_scheduler_from_app',
@@ -25,6 +27,10 @@ async def spawn(request, coro):
 def atomic(coro):
     @wraps(coro)
     async def wrapper(request):
+        if isinstance(request, View):
+            # Class Based View decorated.
+            request = request.request
+
         job = await spawn(request, coro(request))
         return await job.wait()
     return wrapper

tests/test_aiohttp.py

@@ -66,3 +66,24 @@ async def handler(request):
 
     assert scheduler.active_count == 0
     assert scheduler.pending_count == 0
+
+
+async def test_atomic_from_view(test_client):
+    app = web.Application()
+
+    class MyView(web.View):
+        @atomic
+        async def get(self):
+            return web.Response()
+
+    app.router.add_route("*", "/", MyView)
+    aiojobs_setup(app)
+
+    client = await test_client(app)
+    resp = await client.get('/')
+    assert resp.status == 200
+
+    scheduler = get_scheduler_from_app(app)
+
+    assert scheduler.active_count == 0
+    assert scheduler.pending_count == 0