Merge pull request #598 from onaio/592-project-endpoint-refactor

592 project endpoint refactor

onadata/apps/api/tests/viewsets/test_abstract_viewset.py

@@ -166,6 +166,8 @@ def _project_create(self, project_data={}, merge=True):
         data['url'] = 'http://testserver/api/v1/projects/%s'\
             % self.project.pk
         self.assertDictContainsSubset(data, response.data)
+
+        request.user = self.user
         self.project_data = ProjectSerializer(
             self.project, context={'request': request}).data
 

onadata/apps/api/tests/viewsets/test_connect_viewset.py

@@ -5,7 +5,9 @@
     TestAbstractViewSet
 from onadata.apps.api.viewsets.connect_viewset import ConnectViewSet
 from onadata.apps.api.viewsets.project_viewset import ProjectViewSet
+
 from onadata.libs.authentication import DigestAuthentication
+from onadata.libs.serializers.project_serializer import ProjectSerializer
 
 
 class TestConnectViewSet(TestAbstractViewSet):
@@ -56,6 +58,10 @@ def test_get_starred_projects(self):
         response = view(request, user=self.user)
 
         self.assertEqual(response.status_code, 200)
+
+        request.user = self.user
+        self.project_data = ProjectSerializer(
+            self.project, context={'request': request}).data
         self.assertEqual(response.data, [self.project_data])
 
     def test_user_list_with_digest(self):

onadata/apps/api/tests/viewsets/test_project_viewset.py

@@ -61,6 +61,8 @@ def test_projects_tags(self):
 
         # check filter by tag
         request = self.factory.get('/', data={"tags": "hello"}, **self.extra)
+
+        request.user = self.user
         self.project_data = ProjectSerializer(
             self.project, context={'request': request}).data
         response = list_view(request, pk=project_id)
@@ -126,6 +128,7 @@ def test_publish_xls_form_to_project(self):
     def test_num_datasets(self):
         self._publish_xls_form_to_project()
         request = self.factory.post('/', data={}, **self.extra)
+        request.user = self.user
         self.project_data = ProjectSerializer(
             self.project, context={'request': request}).data
         self.assertEqual(self.project_data['num_datasets'], 1)
@@ -134,6 +137,7 @@ def test_last_submission_date(self):
         self._publish_xls_form_to_project()
         self._make_submissions()
         request = self.factory.post('/', data={}, **self.extra)
+        request.user = self.user
         self.project_data = ProjectSerializer(
             self.project, context={'request': request}).data
         date_created = self.xform.instances.order_by(

onadata/libs/serializers/project_serializer.py

@@ -23,6 +23,7 @@ class ProjectSerializer(serializers.HyperlinkedModelSerializer):
         lookup_field='username',
         read_only=True)
     metadata = JsonField(source='metadata', required=False)
+    starred = serializers.SerializerMethodField('is_starred_project')
     users = serializers.SerializerMethodField('get_project_permissions')
     forms = serializers.SerializerMethodField('get_project_forms')
     public = BooleanField(
@@ -93,3 +94,12 @@ def get_last_submission_date(self, obj):
                                                            flat=True)
 
             return last_submission and last_submission[0]
+
+    def is_starred_project(self, obj):
+        request = self.context['request']
+        user = request.user
+        user_stars = obj.user_stars.all()
+        if user in user_stars:
+            return True
+
+        return False