diff --git a/contentcuration/contentcuration/tests/viewsets/test_invitation.py b/contentcuration/contentcuration/tests/viewsets/test_invitation.py index ac4afd794f..a14aae8ccf 100644 --- a/contentcuration/contentcuration/tests/viewsets/test_invitation.py +++ b/contentcuration/contentcuration/tests/viewsets/test_invitation.py @@ -286,10 +286,11 @@ def test_update_invitation_accept(self): response = self.client.post(reverse("invitation-accept", kwargs={"pk": invitation.id})) self.assertEqual(response.status_code, 200, response.content) try: - models.Invitation.objects.get(id=invitation.id) + invitation = models.Invitation.objects.get(id=invitation.id) except models.Invitation.DoesNotExist: self.fail("Invitation was deleted") + self.assertTrue(invitation.accepted) self.assertTrue(self.channel.editors.filter(pk=self.invited_user.id).exists()) self.assertTrue( models.Invitation.objects.filter( diff --git a/contentcuration/contentcuration/viewsets/invitation.py b/contentcuration/contentcuration/viewsets/invitation.py index 56cd27ddd8..55507297ef 100644 --- a/contentcuration/contentcuration/viewsets/invitation.py +++ b/contentcuration/contentcuration/viewsets/invitation.py @@ -141,6 +141,8 @@ def perform_update(self, serializer): def accept(self, request, pk=None): invitation = self.get_object() invitation.accept() + invitation.accepted = True + invitation.save() Change.create_change( generate_update_event( invitation.id, INVITATION, {"accepted": True}, channel_id=invitation.channel_id