fixes to the my_ views, and their tests

companies/serializers.py

@@ -2,7 +2,7 @@ from rest_framework import serializers
 from companies.models import Company
 
 from drf_extra_fields.geo_fields import PointField
-
+from tagulous.serializers.json import Serializer as TagSerializer
 from utils.tag_serializers import TagListSerializerField, TaggitSerializer
 
 class CompanySerializer(TaggitSerializer, serializers.ModelSerializer):

companies/tests.py

@@ -259,14 +259,23 @@ class MyCompanyViewTest(APITestCase):
         self.user.save()
 
     def test_auth_user_gets_data(self):
+        # create instance
+        user_instances = [
+            self.factory(creator=self.user),
+            self.factory(creator=self.user),
+        ]
+
         # Authenticate
         token = get_tokens_for_user(self.user)
         self.client.credentials(HTTP_AUTHORIZATION=f"Bearer {token['access']}")
 
         # Query endpoint
         response = self.client.get(self.endpoint)
+        payload = response.json()
+
         # Assert forbidden code
         self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertEquals(len(user_instances), len(payload))
 
     def test_anon_user_cannot_access(self):
         # send in request

companies/views.py

@@ -125,5 +125,5 @@ class CompanyViewSet(viewsets.ModelViewSet):
 @permission_classes([IsAuthenticated,])
 def my_company(request):
       qs = Company.objects.filter(creator=request.user)
-      company_serializer = CompanySerializer(qs)
+      company_serializer = CompanySerializer(qs, many=True)
       return Response(data=company_serializer.data)