class TestVideo(unittest.TestCase): @classmethod def setUpClass(self): self.key = os.environ.get('YT_KEY') self.yt = YoutubeDataApi(self.key) self.video_id = 'wmxDZeh8W34' self.video_id_list = ['wmxDZeh8W34', 'PIXQdfiRZNk', 'nvEFb_dWJdQ'] self.fake_vid = '12345' def test_valid_caption(self): ''' #Verified by Megan Brown on 11/30/2018''' resp = self.yt.get_captions(self.video_id) self.assertEqual(type(resp), dict) self.assertEqual(type(resp['video_id']), str) def test_valid_list_of_captions(self): '''#Written by Megan Brown on 11/30/2018''' resp = self.yt.get_captions(self.video_id_list) self.assertEqual(type(resp), list) self.assertEqual(type(resp[0]['video_id']), str) def test_invalid_short_caption(self): '''#Written by Megan Brown on 11/30/2018''' with self.assertRaises(Exception): resp = self.yt.get_captions(self.fake_vid) def test_list_of_captions_with_invalid_string(self): '''#Written by Megan Brown on 11/30/2018''' error_list = self.video_id_list.copy() error_list.append(self.fake_vid) with self.assertRaises(Exception): resp = self.yt.get_captions(error_list)
class TestVideo(unittest.TestCase): def __init__(): self.key = os.environ.get('YT_KEY') self.yt = YoutubeDataApi(self.key) self.video_id = 'wmxDZeh8W34' self.video_id_list = ['wmxDZeh8W34', 'PIXQdfiRZNk','nvEFb_dWJdQ'] self.fake_vid = '12345' #Verified by Megan Brown on 11/30/2018 def test_valid_caption(self): resp = self.yt.get_captions(self.video_id) self.assertEqual(type(resp), dict) self.assertEqual(type(resp['video_id']), str) #Written by Megan Brown on 11/30/2018 def test_valid_list_of_captions(self): resp = self.yt.get_captions(self.video_id_list) self.assertEqual(type(resp), list) self.assertEqual(type(resp[0]['video_id']), str) #Written by Megan Brown on 11/30/2018 def test_invalid_short_caption(self): with self.assertRaises(Exception): resp = self.yt.get_captions(self.fake_vid) #Written by Megan Brown on 11/30/2018 def test_list_of_captions_with_invalid_string(self): error_list = self.video_id_list error_list.append(self.fake_vid) with self.assertRaises(Exception): resp = self.yt.get_captions(error_list) #Written by Megan Brown on 12/20/2018 def test_caption_parser_w_raw_json(self): captions = self.yt.get_captions(self.video_id, parser=P.raw_json) self.assertTrue(isinstance(self.captions, dict))