addon/tests.py

# -*- coding: utf-8 -*-
import os
import sys
import unittest
import parameterized

from platformcode import config

librerias = os.path.join(config.get_runtime_path(), 'lib')
sys.path.insert(0, librerias)
from core.support import typo
from core.item import Item
import channelselector
from core import servertools
import re

validUrlRegex = re.compile(
    r'^(?:http|ftp)s?://'  # http:// or https://
    r'(?:(?:[A-Z0-9](?:[A-Z0-9-]{0,61}[A-Z0-9])?\.)+(?:[A-Z]{2,6}\.?|[A-Z0-9-]{2,}\.?)|'  # domain...
    r'localhost|'  # localhost...
    r'\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3})'  # ...or ip
    r'(?::\d+)?'  # optional port
    r'(?:/?|[/?]\S+)$', re.IGNORECASE)

chBlackList = ['url']
chNumRis = {
    'altadefinizione01': {
        'Film': 20
    },
    'altadefinizione01_link': {
        'Film': 16,
        'Serie TV': 16,
    },
    'altadefinizioneclick': {
        'Film': 36,
        'Serie TV': 12,
    },
    'casacinema': {
        'Film': 10,
        'Serie TV': 10,
    },
    'cineblog01': {
        'Film': 12,
        'Serie TV': 13
    },
    'cinemalibero': {
        'Film': 20,
        'Serie TV': 20,
    },
    'cinetecadibologna': {
        'Film': 10
    },
    'eurostreaming': {
        'Serie TV': 18
    },
    'Filmpertutti': {
        'Film': 24,
        'Serie TV': 24,
    },
    'guardaSerie TVclick': {
        'da controllare': 0
    },
    'hd4me': {
        'Film': 10
    },
    'ilgeniodellostreaming': {
        'Film': 30,
        'Serie TV': 30
    },
    'italiaserie': {
        'Serie TV': 20
    },
    'casacinemaInfo': {
        'Film': 150
    },
    'netfreex': {
        'Film': 30,
        'Serie TV': 30
    },
    'piratestreaming': {
        'Film': 24,
        'Serie TV': 24
    },
    'polpotv': {
        'Film': 12,
        'Serie TV': 12
    },
    'streamingaltadefinizione': {
        'Film': 30,
        'Serie TV': 30
    },
    'seriehd': {
        'Serie TV': 12
    },
    'serietvonline': {
        'Film': 35,
        'Serie TV': 35
    },
    'tantifilm': {
        'Film': 20,
        'Serie TV': 20
    },
}


def getChannels():
    channel_list = channelselector.filterchannels("all")
    ret = []
    for chItem in channel_list:
        ch = chItem.channel
        if ch not in chBlackList:
            ret.append({'ch': ch})
    return ret


from specials import news

dictNewsChannels, any_active = news.get_channels_list()

srvLinkDict = {
    "wstream": ["https://wstream.video/video6zvimpy52/dvvwxyfs32ab"],
    "akvideo": ["https://akvideo.stream/video.php?file_code=23god95lrtqv"]
}


def getServers():
    server_list = servertools.get_servers_list()
    ret = []
    for srv in server_list:
        if srv in srvLinkDict:
            ret.append({'srv': srv})
    return ret


@parameterized.parameterized_class(getChannels())
class GenericChannelTest(unittest.TestCase):
    def __init__(self, *args):
        self.module = __import__('channels.%s' % self.ch, fromlist=["channels.%s" % self.ch])
        super(GenericChannelTest, self).__init__(*args)

    def test_menuitems(self):
        hasChannelConfig = False
        mainlist = self.module.mainlist(Item())
        self.assertTrue(mainlist, 'channel ' + self.ch + ' has no menu')

        for it in mainlist:
            it.title = it.title.decode('ascii', 'ignore')
            if it.action == 'channel_config':
                hasChannelConfig = True
                continue
            if it.action == 'search':  # channel specific
                continue
            itemlist = getattr(self.module, it.action)(it)
            self.assertTrue(itemlist, 'channel ' + self.ch + ' -> ' + it.title + ' is empty')
            if self.ch in chNumRis:  # so a priori quanti risultati dovrebbe dare
                for content in chNumRis[self.ch]:
                    if content in it.title:
                        risNum = len(itemlist) - 1  # - nextpage
                        self.assertEqual(chNumRis[self.ch][content], risNum,
                                         'channel ' + self.ch + ' -> ' + it.title + ' returned wrong number of results')
                        break

            for resIt in itemlist:
                self.assertLess(len(resIt.fulltitle), 100,
                                'channel ' + self.ch + ' -> ' + it.title + ' might contain wrong titles\n' + resIt.fulltitle)
                if resIt.url:
                    self.assertIsNotNone(re.match(validUrlRegex, resIt.url),
                                         'channel ' + self.ch + ' -> ' + it.title + ' -> ' + resIt.title + ' might contain wrong url\n' + resIt.url)
                if 'year' in resIt.infoLabels and resIt.infoLabels['year']:
                    msgYear = 'channel ' + self.ch + ' -> ' + it.title + ' might contain wrong infolabels year\n' + str(
                        resIt.infoLabels['year'])
                    self.assert_(type(resIt.infoLabels['year']) is int or resIt.infoLabels['year'].isdigit(), msgYear)
                    self.assert_(int(resIt.infoLabels['year']) > 1900 and int(resIt.infoLabels['year']) < 2100, msgYear)

                if resIt.title == typo(config.get_localized_string(30992), 'color kod bold'):  # next page
                    nextPageItemlist = getattr(self.module, resIt.action)(resIt)
                    self.assertTrue(nextPageItemlist,
                                    'channel ' + self.ch + ' -> ' + it.title + ' has nextpage not working')
        self.assertTrue(hasChannelConfig, 'channel ' + self.ch + ' has no channel config')

    def test_newest(self):
        for cat in dictNewsChannels:
            for ch in dictNewsChannels[cat]:
                if self.ch == ch[0]:
                    itemlist = self.module.newest(cat)
                    self.assertTrue(itemlist, 'channel ' + self.ch + ' returned no news for category ' + cat)
                    break


#
# @parameterized.parameterized_class(getServers())
# class GenericServerTest(unittest.TestCase):
#     def __init__(self, *args):
#         self.module = __import__('servers.%s' % self.srv, fromlist=["servers.%s" % self.srv])
#         super(GenericServerTest, self).__init__(*args)
#
#     def test_resolve(self):
#         for link in srvLinkDict[self.srv]:
#             find = servertools.findvideosbyserver(link, self.srv)
#             self.assertTrue(find, 'link ' + link + ' not recognised')
#             page_url = find[0][1]
#             if self.module.test_video_exists(page_url)[0]:
#                 urls = self.module.get_video_url(page_url)
#                 print urls
#                 for u in urls:
#                     directUrl, headersUrl = u[1].split('|')
#                     headers = {}
#                     for name in headersUrl.split('&'):
#                         h, v = name.split('=')
#                         headers[h] = v
#                     print headers
#                     self.assertEqual(requests.head(directUrl, headers=headers, timeout=15).status_code, 200, self.srv + ' scraper did not return valid url for link ' + link)

if __name__ == '__main__':
    config.set_setting('tmdb_active', False)
    unittest.main()