Merge pull request #853 from scieloorg/beta

Incorporação de códigos estáveis

scielomanager/api/resources.py

@@ -3,7 +3,7 @@
 
 from django.db.models import Q
 from django.contrib.auth.models import User
-from django.core.exceptions import ObjectDoesNotExist
+from django.core.exceptions import ObjectDoesNotExist, MultipleObjectsReturned
 from tastypie.resources import ModelResource, Resource
 from tastypie import fields
 from tastypie.contrib.contenttypes.fields import GenericForeignKeyField
@@ -26,6 +26,8 @@
     Article,
 )
 
+from scielomanager.utils import usercontext
+
 from articletrack.models import (
     Checkin,
     Notice,
@@ -36,6 +38,15 @@
 
 logger = logging.getLogger(__name__)
 
+
+def current_user_active_collection():
+    return usercontext.get_finder().get_current_user_active_collection()
+
+
+def current_user_collections():
+    return usercontext.get_finder().get_current_user_collections()
+
+
 class ApiKeyAuthMeta:
     authentication = ApiKeyAuthentication()
     authorization = DjangoAuthorization()
@@ -263,11 +274,19 @@ def dehydrate_collections(self, bundle):
             return ''
 
     def dehydrate_pub_status(self, bundle):
-        col = bundle.obj.collections.get()
+        try:
+            col = bundle.obj.collections.get()
+        except MultipleObjectsReturned:
+            col = current_user_active_collection()
+
         return bundle.obj.membership_info(col, 'status')
 
     def dehydrate_pub_status_reason(self, bundle):
-        col = bundle.obj.collections.get()
+        try:
+            col = bundle.obj.collections.get()
+        except MultipleObjectsReturned:
+            col = current_user_active_collection()
+
         return bundle.obj.membership_info(col, 'reason')
 
 

scielomanager/api/tests.py

@@ -303,6 +303,39 @@ def test_filter_eletronic_issn(self):
         self.assertEqual(len(json.loads(response.content)['objects']), 1)
         self.assertEqual(json.loads(response.content)['objects'][0]['eletronic_issn'], '1234-1234')
 
+    def test_dehydrate_pub_status_with_one_collections(self):
+        col = modelfactories.CollectionFactory()
+
+        col.add_user(self.user)
+
+        journal = modelfactories.JournalFactory.create()
+        journal.join(col, self.user)
+
+        response = self.app.get('/api/v1/journals/',
+            extra_environ=self.extra_environ).json
+
+        self.assertEqual(response['objects'][0]['pub_status'], u'inprogress')
+
+    def test_dehydrate_pub_status_with_multiple_collections(self):
+        col = modelfactories.CollectionFactory()
+        col2 = modelfactories.CollectionFactory()
+
+        col.add_user(self.user)
+        col2.add_user(self.user)
+
+        col.make_default_to_user(self.user)
+
+        journal = modelfactories.JournalFactory.create()
+        journal.join(col, self.user)
+        journal.join(col2, self.user, )
+
+        journal.change_status(col, u'current', u'yeah', self.user)
+
+        response = self.app.get('/api/v1/journals/',
+            extra_environ=self.extra_environ).json
+
+        self.assertEqual(response['objects'][0]['pub_status'], u'current')
+
 
 class CollectionRestAPITest(WebTest):