From 8b86be81f377aec5f048afa81c7e9c03b8aa7886 Mon Sep 17 00:00:00 2001 From: Antti Ajanki Date: Mon, 8 Aug 2011 21:04:00 +0300 Subject: Improve Subtv test --- src/unittest/testwebvi.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) (limited to 'src') diff --git a/src/unittest/testwebvi.py b/src/unittest/testwebvi.py index 53f61ab..d50126b 100644 --- a/src/unittest/testwebvi.py +++ b/src/unittest/testwebvi.py @@ -417,15 +417,21 @@ class TestServiceModules(unittest.TestCase): links = self.downloadAndExtractLinks(ref, 4, 'series') # Program page - links = self.downloadAndExtractLinks(links[0].ref, 1, 'program') + plinks = [] + for li in links: + plinks = self.downloadAndExtractLinks(li.ref, 0, 'program') + if plinks: + break + if not plinks: + self.assertTrue(False, 'None of the program pages has links') # Video link - videolink = links[0] + videolink = plinks[0] self.assertNotEqual(videolink.stream, None, 'No media object in a video link') self.assertNotEqual(videolink.ref, None, 'No description page in a video link') # Direct video page link - queries, params = self.extractQueryParams(links[0].stream) + queries, params = self.extractQueryParams(plinks[0].stream) self.assertTrue('srcurl' in queries and 'pid' in params, 'Required parameter missing in video link') videopageurl = queries['srcurl'] + '?' + params['pid'] videopageref = self.urlToWvtref(videopageurl) -- cgit v1.2.3