recordsansible/ara
Code Issues Proposed changes

Merge "Add play filtering by playbook id"

Browse Source
This commit is contained in:
Zuul 2018-10-16 19:58:31 +00:00 committed by Gerrit Code Review
commit 53118108f7
2 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
ara/api/tests/tests_play.py
View File

@ -89,6 +89,13 @@ class PlayTestCase(APITestCase):
request = self.client.get("/api/v1/plays/%s" % play.id) request = self.client.get("/api/v1/plays/%s" % play.id)
self.assertEqual(play.name, request.data["name"]) self.assertEqual(play.name, request.data["name"])
def test_get_play_by_playbook(self):
play = factories.PlayFactory(name="play1")
factories.PlayFactory(name="play2")
request = self.client.get("/api/v1/plays?playbook=1")
self.assertEqual(1, len(request.data["results"]))
self.assertEqual(play.name, request.data["results"][0]["name"])
def test_get_play_duration(self): def test_get_play_duration(self):
started = timezone.now() started = timezone.now()
ended = started + datetime.timedelta(hours=1) ended = started + datetime.timedelta(hours=1)

2
ara/api/views.py
View File

@ -48,6 +48,8 @@ class PlaybookFilesDetail(NestedViewSetMixin, viewsets.ModelViewSet):
class PlayViewSet(viewsets.ModelViewSet): class PlayViewSet(viewsets.ModelViewSet):
queryset = models.Play.objects.all() queryset = models.Play.objects.all()
serializer_class = serializers.PlaySerializer serializer_class = serializers.PlaySerializer
filter_backends = (DjangoFilterBackend,)
filter_fields = ("playbook",)
class TaskViewSet(viewsets.ModelViewSet): class TaskViewSet(viewsets.ModelViewSet):