def test_get_file__none(self): """ If there are no files, get_file() should return None. """ video = Video("http://www.youtube.com/watch?v=J_DV9b0x7v4") video.files = None self.assertTrue(video.get_file() is None) video.files = [] self.assertTrue(video.get_file() is None)
def test_get_file__no_mimetypes(self): """ If none of the videos have mime types, the first file should be returned. """ video = Video("http://www.youtube.com/watch?v=J_DV9b0x7v4") file1 = VideoFile(url='http://google.com') file2 = VideoFile(url='http://xkcd.com') file3 = VideoFile(url='http://example.com') video.files = [file1, file2, file3] self.assertEqual(video.get_file(), file1) video.files = [file3, file2, file1] self.assertEqual(video.get_file(), file3)
def test_get_file__open(self): """ Tests that open video formats are preferred over proprietary. """ video = Video("http://www.youtube.com/watch?v=J_DV9b0x7v4") file1 = VideoFile(url='http://google.com', mime_type="video/ogg") file2 = VideoFile(url='http://xkcd.com', mime_type="application/x-shockwave-flash") file3 = VideoFile(url='http://example.com', mime_type="video/mp4") video.files = [file1, file2, file3] self.assertEqual(video.get_file(), file1) video.files = [file3, file2, file1] self.assertEqual(video.get_file(), file1)