Fix API for getting organizations (HumanSignal#997)

* Return organizations where user belongs to * Add tests Co-authored-by: nik <[email protected]>
prathameshk · Jun 3, 2021 · 8741d20 · 8741d20
1 parent 4e1cb57
commit 8741d20
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/label_studio/organizations/api.py b/label_studio/organizations/api.py
@@ -40,7 +40,7 @@ def get_object(self):
         return org
 
     def get_queryset(self):
-        return Organization.objects.filter(created_by=self.request.user)
+        return Organization.objects.filter(users=self.request.user).distinct()
 
     @swagger_auto_schema(tags=['Organizations'])
     def get(self, request, *args, **kwargs):

diff --git a/..._studio/tests/test_active_organization.py → label_studio/tests/test_organizations.py b/..._studio/tests/test_active_organization.py → label_studio/tests/test_organizations.py
@@ -8,3 +8,11 @@ def test_active_organization_filled(business_client):
     response = business_client.get('/api/users/')
     response_data = response.json()
     assert response_data[0]['active_organization'] == business_client.organization.id
+
+
+@pytest.mark.django_db
+def test_api_list_organizations(business_client):
+    response = business_client.get('/api/organizations/')
+    response_data = response.json()
+    assert len(response_data) == 1
+    assert response_data[0]['id'] == business_client.organization.id