recordsansible
/
ara-server
Code Issues Proposed changes

Merge "Add host filtering by playbook id"

This commit is contained in:
Zuul 2018-10-16 22:46:03 +00:00 committed by Gerrit Code Review
parent cbf81c8abd 2e863887eb
commit 2d2942f0ec
2 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
ara/api/tests/tests_host.py
View File

@ -97,3 +97,12 @@ class HostTestCase(APITestCase):
host = factories.HostFactory()
request = self.client.get("/api/v1/hosts/%s" % host.id)
self.assertEqual(host.name, request.data["name"])
def test_get_hosts_by_playbook(self):
playbook = factories.PlaybookFactory()
host = factories.HostFactory(name="host1", playbook=playbook)
factories.HostFactory(name="host2", playbook=playbook)
request = self.client.get("/api/v1/hosts?playbook=%s" % playbook.id)
self.assertEqual(2, len(request.data["results"]))
self.assertEqual(host.name, request.data["results"][0]["name"])
self.assertEqual("host2", request.data["results"][1]["name"])

2
ara/api/views.py
View File

@ -62,6 +62,8 @@ class TaskViewSet(viewsets.ModelViewSet):
class HostViewSet(viewsets.ModelViewSet):
queryset = models.Host.objects.all()
serializer_class = serializers.HostSerializer
filter_backends = (DjangoFilterBackend,)
filter_fields = ("playbook",)
class ResultViewSet(viewsets.ModelViewSet):