'Tests the function that returns the appropriate extension depending on whether it is
a video from youtube, or one uploaded to the EdX server.'

def test_get_extension(self):

    expectedyoutube = 'video/youtube'
    expectednotyoutube = 'video/mp4'
    result1 = self.mod._get_extension(self.sample_sourceurl)
    result2 = self.mod._get_extension(self.sample_youtubeurl)
    self.assertEqual(expectedyoutube, result2)
    self.assertEqual(expectednotyoutube, result1)
