Implemented __str__ for Cohort model

RamiroAlvaro · renzon · commit 3ddda3d73df5 · 2020-10-07T14:42:31.000-03:00
close #2889
diff --git a/pythonpro/cohorts/models.py b/pythonpro/cohorts/models.py
@@ -19,6 +19,9 @@ class Cohort(models.Model):
     def get_absolute_url(self):
         return reverse('cohorts:detail', kwargs={'slug': self.slug})
 
+    def __str__(self):
+        return f'Turma: {self.title}'
+
     @property
     def recorded_webinars(self):
         return [w for w in self.webinars if w.vimeo_id]
diff --git a/pythonpro/cohorts/tests/test_cohorts.py b/pythonpro/cohorts/tests/test_cohorts.py
@@ -58,6 +58,10 @@ def test_status_code(resp):
     assert 200 == resp.status_code
 
 
+def test_str(cohort):
+    assert str(cohort) == f'Turma: {cohort.title}'
+
+
 @pytest.mark.parametrize('property_name', 'title forum_post'.split())
 def test_cohort_properties(cohort, resp, property_name):
     dj_assert_contains(resp, getattr(cohort, property_name))
