add filtering cloud host api by service form service env

allegro · Aug 21, 2020 · ca53e67 · ca53e67
1 parent 2e9864f
commit ca53e67
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/src/ralph/virtual/api.py b/src/ralph/virtual/api.py
@@ -255,6 +255,12 @@ class CloudHostViewSet(BaseObjectViewSetMixin, RalphAPIViewSet):
         ),
     ]
 
+    filter_fields = [
+        'service_env__service__uid',
+        'service_env__service__name',
+        'service_env__service__id',
+    ]
+
 
 class CloudProjectViewSet(RalphAPIViewSet):
     queryset = CloudProject.objects.all()

diff --git a/src/ralph/virtual/tests/test_api.py b/src/ralph/virtual/tests/test_api.py
@@ -140,6 +140,18 @@ def test_get_cloudhost_detail(self):
             response.data['technical_owners'][0]['username'], 'user2'
         )
 
+    def test_filter_cloudhost_by_service_uid(self):
+        cloud_host = CloudHostFactory()
+        url = (
+            reverse('cloudhost-list') +
+            '?service_env__service__uid={}'.format(cloud_host.service_env.service.uid)
+        )
+        response = self.client.get(url, format='json')
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertEqual(
+            response.data['count'], 1
+        )
+
     def test_get_cloudproject_detail(self):
         url = reverse('cloudproject-detail', args=(self.cloud_project.id,))
         response = self.client.get(url, format='json')