Add task filtering by playbook id

Change-Id: Ibcafe4ad574a7a55c48b99a38f23702f10f2c8a1
This commit is contained in:
David Moreau Simard 2018-10-16 10:45:02 -04:00
parent 864e0b481b
commit 85e76d68db
No known key found for this signature in database
GPG Key ID: CBEB466764A9E621
2 changed files with 11 additions and 0 deletions

View File

@ -125,6 +125,15 @@ class TaskTestCase(APITestCase):
request = self.client.get("/api/v1/tasks/%s" % task.id)
self.assertEqual(task.name, request.data["name"])
def test_get_tasks_by_playbook(self):
playbook = factories.PlaybookFactory()
task = factories.TaskFactory(name="task1", playbook=playbook)
factories.TaskFactory(name="task2", playbook=playbook)
request = self.client.get("/api/v1/tasks?playbook=%s" % playbook.id)
self.assertEqual(2, len(request.data["results"]))
self.assertEqual(task.name, request.data["results"][0]["name"])
self.assertEqual("task2", request.data["results"][1]["name"])
def test_get_task_duration(self):
started = timezone.now()
ended = started + datetime.timedelta(hours=1)

View File

@ -55,6 +55,8 @@ class PlayViewSet(viewsets.ModelViewSet):
class TaskViewSet(viewsets.ModelViewSet):
queryset = models.Task.objects.all()
serializer_class = serializers.TaskSerializer
filter_backends = (DjangoFilterBackend,)
filter_fields = ("playbook",)
class HostViewSet(viewsets.ModelViewSet):