enable testing with CELERY_ALWAYS_EAGER

Author: miraculixx

tpasync/resources.py

@@ -1,7 +1,10 @@
-from celery.result import AsyncResult
+from celery.result import AsyncResult, EagerResult
+from django.conf import settings
 from django.conf.urls import url
 from tastypie import resources, http, utils
 
+# use for dev/test only!
+EAGER_RESULTS={}
 
 class AsyncResourceMixin(object):
     def async_get_detail(self, request, **kwargs):
@@ -49,7 +52,10 @@ def async_state(self, request, task_id, **kwargs):
 
         Other methods are forbidden.
         """
-        task = AsyncResult(task_id)
+        if not getattr(settings, 'CELERY_ALWAYS_EAGER'):
+            task = AsyncResult(task_id)
+        else:
+            task = EAGER_RESULTS[task_id]
         if request.method == 'GET':
             data = {
                 'state': task.state, 'id': task.id,
@@ -82,7 +88,11 @@ def async_result(self, request, task_id, **kwargs):
         If request is not ready (or doesn't exist - we can't tell this),
         return Http 404 Not Found.
         """
-        task = AsyncResult(task_id)
+        # hack to allow local testing
+        if not getattr(settings, 'CELERY_ALWAYS_EAGER'):
+            task = AsyncResult(task_id)
+        else:
+            task = EAGER_RESULTS[task_id]
         if task.ready():
             try:
                 result = task.get()
@@ -163,6 +173,8 @@ def inner(self, request, **kwargs):
                 return http.HttpNotImplemented()
 
             if isinstance(result, AsyncResult):
+                if isinstance(result, EagerResult):
+                    EAGER_RESULTS[result.id] = result
                 response = http.HttpAccepted()
                 response['Location'] = self._build_reverse_url(
                     'api_async_state',