class TestMTV(unittest.TestCase):
def setUp(self):
self.mtv = MTVItaly('mock_url://', 3)
def test_list_show(self):
shows = self.mtv.list_shows()
self.assertIsNotNone(shows)
titles = tuple(show[0] for show in shows)
self.assertIn('Il Testimone', titles)
def test_list_seasons(self):
seasons = self.mtv.list_seasons('il-testimone')
self.assertIsNotNone(seasons)
titles = tuple(season[0] for season in seasons)
self.assertIn('Stagione 1', titles)
def test_list_seasons_empty(self):
with self.assertRaises(Exception):
self.mtv.list_seasons('ballerini-dietro-il-sipario')
def test_list_episodes(self):
episodes = self.mtv.list_episodes('il-testimone', 's01')
self.assertIsNotNone(episodes)
titles = tuple(episode[0] for episode in episodes)
self.assertIn('Loco, loco, loco', titles)
@mock.patch('mtvit.urllib')
def test_quality_selection(self, mock_urllib):
fn = os.path.join(os.path.dirname(__file__), 'mocks', 'video_qualities_mock.xml')
mock_urllib.urlopen.return_value = fn
episode_qualities = self.mtv.list_video_qualities('il-testimone', 's01', 'il-testimone-s01e07')
self.assertIsNotNone(episode_qualities)
normalized_qualities = self.mtv.normalize_qualities(episode_qualities)
self.assertEqual(3, len(normalized_qualities))
li = xbmcgui.ListItem(season[0])
xbmcplugin.addDirectoryItem(handle=addon_handle,
url=season[1],
listitem=li,
isFolder=True)
xbmcplugin.endOfDirectory(addon_handle)
except:
dialog = xbmcgui.Dialog()
ok = dialog.ok('MTV Italy Ondemand',
addon.getLocalizedString(NO_EPISODES))
elif mode[0] == 'list_episodes':
season_id = args['season_id'][0]
show_id = args['show_id'][0]
for episode in mtv.list_episodes(show_id, season_id):
li = xbmcgui.ListItem(episode[0],
iconImage=episode[2],
thumbnailImage=episode[2])
xbmcplugin.addDirectoryItem(handle=addon_handle,
url=episode[3],
listitem=li,
isFolder=True)
xbmcplugin.endOfDirectory(addon_handle)
elif mode[0] == 'showvid':
episode_title = args['episode_title'][0]
episode_image = args['episode_image'][0]
show_id = args['show_id'][0]
season_id = args['season_id'][0]
elif mode[0] == 'list_seasons':
show_id = args['show_id'][0]
try:
for season in mtv.list_seasons(show_id):
li = xbmcgui.ListItem(season[0])
xbmcplugin.addDirectoryItem(handle = addon_handle, url = season[1], listitem = li, isFolder = True)
xbmcplugin.endOfDirectory(addon_handle)
except:
dialog = xbmcgui.Dialog()
ok = dialog.ok('MTV Italy Ondemand', addon.getLocalizedString(NO_EPISODES))
elif mode[0] == 'list_episodes':
season_id = args['season_id'][0]
show_id = args['show_id'][0]
for episode in mtv.list_episodes(show_id, season_id):
li = xbmcgui.ListItem(episode[0], iconImage=episode[2], thumbnailImage=episode[2])
xbmcplugin.addDirectoryItem(handle=addon_handle, url=episode[3], listitem=li, isFolder = True)
xbmcplugin.endOfDirectory(addon_handle)
elif mode[0] == 'showvid':
episode_title = args['episode_title'][0]
episode_image = args['episode_image'][0]
show_id = args['show_id'][0]
season_id = args['season_id'][0]
episode_id = args['episode_id'][0]
video_url = mtv.build_video_url(show_id, season_id, episode_id)
li = xbmcgui.ListItem(episode_title, iconImage=episode_image)
li.setInfo('video', {'Title' : episode_title})
xbmc.Player(xbmc.PLAYER_CORE_AUTO).play(video_url + ' swfurl=http://media.mtvnservices.com/player/prime/mediaplayerprime.2.7.14.swf swfvfy=true', li)
