def test_get_document_url(self): document = DocumentFileFactory.create() serializer = FileSerializer(document, context={ 'user': document.creator }) self.assertEqual(serializer.get_url(document), document.document.url)
def test_get_youtube_link(self): video = VideoFileFactory.create() serializer = FileSerializer(video, context={'user': video.creator}) self.assertEqual(serializer.get_url(video), video.youtube_link)
def test_get_image_url(self): image = ImageFileFactory.create() serializer = FileSerializer(image, context={'user': image.creator}) self.assertEqual(serializer.get_url(image), image.image.url)