diff --git a/hypha/apply/api/v1/screening/tests/test_views.py b/hypha/apply/api/v1/screening/tests/test_views.py index 289da06509e26624fb08ac562f19e514757bb786..3093eace8dfdd7f1f089f23448afd7f342e8b8f5 100644 --- a/hypha/apply/api/v1/screening/tests/test_views.py +++ b/hypha/apply/api/v1/screening/tests/test_views.py @@ -129,13 +129,15 @@ class SubmissionScreeningStatusViewSetTests(APITestCase): data={'id': self.yes_screening_status.id} ) self.assertEqual(response.status_code, status.HTTP_201_CREATED) - self.assertEqual(len(response.json()), self.submission.screening_statuses.count()) - first_submission_screening_status = self.submission.screening_statuses.order_by('-id').last() + self.assertEqual(len(response.json()), 2) + + first_submission_screening_status = self.yes_default_screening_status self.assertEqual(response.json()[0]['id'], first_submission_screening_status.id) self.assertEqual(response.json()[0]['title'], first_submission_screening_status.title) self.assertEqual(response.json()[0]['yes'], first_submission_screening_status.yes) self.assertEqual(response.json()[0]['default'], first_submission_screening_status.default) - self.assertEqual(response.json()[1]['id'], self.submission.screening_statuses.order_by('id').last().id) + + self.assertEqual(response.json()[1]['id'], self.yes_screening_status.id) def test_staff_can_list_submission_screening_statuses(self): user = StaffFactory()