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)
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)
