From 4d95b6bd9b32a83e4b31ab240ddf5535280693b4 Mon Sep 17 00:00:00 2001
From: Alex Manning <alexander.manning@stfc.ac.uk>
Date: Wed, 10 May 2023 13:54:04 +0100
Subject: [PATCH] Tweak API scheme.

---
 jasmin_services/api/views.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/jasmin_services/api/views.py b/jasmin_services/api/views.py
index 5f0b103..f760b3d 100644
--- a/jasmin_services/api/views.py
+++ b/jasmin_services/api/views.py
@@ -4,6 +4,7 @@
 import django.contrib.auth
 import django.db.models as dj_models
 import django.utils.timezone
+import drf_spectacular.types
 import drf_spectacular.utils
 import jasmin_django_utils.api.viewsets
 import rest_framework.decorators as rf_decorators
@@ -38,7 +39,8 @@ class ServicesViewSet(
                 type=dt.date,
                 description="ISO Date on which you would like to know the active roles for a service.",
             )
-        ]
+        ],
+        responses=serializers.RoleSerializer(many=True),
     )
     @rf_decorators.action(detail=True, required_scopes=["jasmin.services.serviceroles.all"])
     def roles(self, request, pk=None):
@@ -84,6 +86,9 @@ class UsersViewSet(
     action_serializers = {"services": serializers.ServiceListSerializer}
     required_scopes = ["jasmin.services.userservices.all"]
 
+    @drf_spectacular.utils.extend_schema(
+        responses=serializers.ServiceListSerializer(many=True),
+    )
     @rf_decorators.action(detail=True)
     def services(self, request, username=None):
         """List the services of a given user."""