Merge pull request #769 from weni-ai/feature/get-all-company-data

return all organization with new user has permission; fix unit test

connect/api/v1/account/views.py

@@ -283,30 +283,28 @@ def receive_emails(self, request, **kwargs):
     )
     def get_user_company_info(self, request, **kwargs):
         user = self.request.user
-        authorizations = user.authorizations_user
-        if authorizations.count() == 1:
-            organization = authorizations.first().organization
+        authorizations = user.authorizations_user.all()
+        response = []
+
+        for authorization in authorizations:
+
+            organization = authorization.organization
             first_user = organization.authorizations.order_by("created_at").first().user
+
             if first_user.id != user.id:
-                organization = dict(
+                organization_data = dict(
                     uuid=str(organization.uuid),
                     name=organization.name,
-                    authorization=authorizations.first().role
-                )
-                company = dict(
-                    company_name=first_user.company_name,
-                    company_segment=first_user.company_segment,
-                    company_sector=first_user.company_sector,
-                    number_people=first_user.number_people,
-                    weni_helps=first_user.weni_helps
+                    authorization=authorization.role
                 )
-                data = dict(
-                    organization=organization,
-                    company=company
+                response.append(
+                    dict(
+                        organization=organization_data,
+                        company=first_user.get_company_data
+                    )
                 )
-                return Response(status=status.HTTP_200_OK, data=data)
 
-        return Response(status=status.HTTP_200_OK, data={})
+        return Response(status=status.HTTP_200_OK, data=response)
 
 
 class SearchUserViewSet(mixins.ListModelMixin, GenericViewSet):  # pragma: no cover

connect/api/v1/tests/test_account.py

@@ -171,5 +171,5 @@ def request(self, token):
 
     def test_okay(self):
         response, content_data = self.request(self.user_token)
-        self.assertEquals(list(content_data.keys()), ['organization', 'company'])
+        self.assertEquals(list(content_data[0].keys()), ['organization', 'company'])
         self.assertEqual(response.status_code, status.HTTP_200_OK)

connect/authentication/models.py

@@ -264,6 +264,16 @@ def set_verify_email(self):
 
         self.save(update_fields=["first_login"])
 
+    @property
+    def get_company_data(self):
+        return dict(
+            company_name=self.company_name,
+            company_segment=self.company_segment,
+            company_sector=self.company_sector,
+            number_people=self.number_people,
+            weni_helps=self.weni_helps
+        )
+
 
 class UserEmailSetup(models.Model):
     user = models.OneToOneField(User, on_delete=models.CASCADE, related_name="email_setup")