add cohort to user me task serializer

gustavomm19 · Nov 15, 2024 · 1e85dbd · 1e85dbd
1 parent 039a8c1
commit 1e85dbd
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 0 deletions.
diff --git a/breathecode/assignments/serializers.py b/breathecode/assignments/serializers.py
@@ -62,6 +62,7 @@ class TaskGETSerializer(serpy.Serializer):
     user = UserSmallSerializer()
     opened_at = serpy.Field()
     delivered_at = serpy.Field()
+    cohort = CohortSmallSerializer(required=False)
     assignment_telemetry = serpy.MethodField()
 
     created_at = serpy.Field()

diff --git a/breathecode/assignments/tests/urls/tests_task.py b/breathecode/assignments/tests/urls/tests_task.py
@@ -31,6 +31,7 @@ def get_serializer(self, task, user):
         "opened_at": self.bc.datetime.to_iso_string(task.opened_at) if task.opened_at else task.opened_at,
         "delivered_at": self.bc.datetime.to_iso_string(task.delivered_at) if task.delivered_at else task.delivered_at,
         "user": {"first_name": user.first_name, "id": user.id, "last_name": user.last_name},
+        "cohort": {"id": task.cohort.id, "name": task.cohort.name, "slug": task.cohort.slug},
     }
 
 

diff --git a/breathecode/assignments/tests/urls/tests_task_id.py b/breathecode/assignments/tests/urls/tests_task_id.py
@@ -30,6 +30,7 @@ def get_serializer(self, task, user):
         "opened_at": self.bc.datetime.to_iso_string(task.opened_at) if task.opened_at else task.opened_at,
         "delivered_at": self.bc.datetime.to_iso_string(task.delivered_at) if task.delivered_at else task.delivered_at,
         "user": {"first_name": user.first_name, "id": user.id, "last_name": user.last_name},
+        "cohort": {"id": task.cohort.id, "name": task.cohort.name, "slug": task.cohort.slug},
     }
 
 

diff --git a/breathecode/assignments/tests/urls/tests_user_me_task.py b/breathecode/assignments/tests/urls/tests_user_me_task.py
@@ -81,6 +81,7 @@ def get_serializer(self, task, user):
         "opened_at": self.bc.datetime.to_iso_string(task.opened_at) if task.opened_at else task.opened_at,
         "delivered_at": self.bc.datetime.to_iso_string(task.delivered_at) if task.delivered_at else task.delivered_at,
         "user": {"first_name": user.first_name, "id": user.id, "last_name": user.last_name},
+        "cohort": {"id": task.cohort.id, "name": task.cohort.name, "slug": task.cohort.slug},
     }