def mark_read(self, request): try: # To Pass Serializer request.data["sender"] = "a" request.data["content"] = "a" request.data["read"] = "False" request.data["urgency"] = "1" serializer = MessageSerializer(data=request.data) serializer.is_valid() update_msg = Dynamo.get_message(serializer.validated_data) update_msg.read = True response = Dynamo.update_message(update_msg) except Exception as e: print(e) return Response( { "Status": "Fail", "MessageRef": {"recipient": request.data["recipient"], "timestamp": request.data["timestamp"]}, } ) return Response( { "Status": "Success", "MessageRef": {"recipient": request.data["recipient"], "timestamp": request.data["timestamp"]}, } )
def test_message_mark(self): message = Dynamo.get_message_by_id(message_id=1000) message[0]['read'] = True Dynamo.update_message(message[0]) self.assertTrue(Dynamo.get_message_by_id(message_id=1000)[0]['read'])