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