Added queries that calculate counts for organization resources

awx/main/tests/functional/api/test_organization_counts.py

@@ -0,0 +1,54 @@
+import pytest
+
+from django.core.urlresolvers import reverse
+
+@pytest.fixture
+def resourced_organization(organization, project, user):
+    admin_user = user('test-admin', True)
+    member_user = user('org-member')
+
+    # Associate one resource of every type with the organization
+    organization.users.add(member_user)
+    organization.admins.add(admin_user)
+    organization.projects.add(project)
+    organization.teams.create(name='org-team')
+    inventory = organization.inventories.create(name="associated-inv")
+    inventory.jobtemplates.create(name="test-jt",
+                                  description="test-job-template-desc",
+                                  project=project,
+                                  playbook="test_playbook.yml")
+
+    return organization
+
+@pytest.mark.django_db
+def test_org_counts_admin(resourced_organization, user, get):
+    # Check that all types of resources are counted by a superuser
+    external_admin = user('admin', True)
+    response = get(reverse('api:organization_list', args=[]), external_admin)
+    counts = response.data['results'][0]['summary_fields']['counts']
+
+    assert counts == {
+        'users': 1,
+        'admins': 1,
+        'job_templates': 1,
+        'projects': 1,
+        'inventories': 1,
+        'teams': 1
+    }
+
+@pytest.mark.django_db
+def test_org_counts_member(resourced_organization, get):
+    # Check that a non-admin user can only see the full project and
+    #   user count, consistent with the RBAC rules
+    member_user = resourced_organization.users.get(username='org-member')
+    response = get(reverse('api:organization_list', args=[]), member_user)
+    counts = response.data['results'][0]['summary_fields']['counts']
+
+    assert counts == {
+        'users': 1,  # User can see themselves
+        'admins': 0,
+        'job_templates': 0,
+        'projects': 1,  # Projects are shared with all the organization
+        'inventories': 0,
+        'teams': 0
+    }