plugin.video.viervijfzes/tests/test_epg.py

# -*- coding: utf-8 -*-
""" Tests for EPG API """

# pylint: disable=missing-docstring,no-self-use

from __future__ import absolute_import, division, print_function, unicode_literals

import logging
import unittest
from datetime import date

from resources.lib import kodiutils
from resources.lib.viervijfzes.content import ContentApi, Episode
from resources.lib.viervijfzes.epg import EpgApi, EpgProgram

_LOGGER = logging.getLogger(__name__)


class TestEpg(unittest.TestCase):
    def __init__(self, *args, **kwargs):
        super(TestEpg, self).__init__(*args, **kwargs)
        self._epg = EpgApi()

    def test_vier_today(self):
        programs = self._epg.get_epg('Play4', date.today().strftime('%Y-%m-%d'))
        self.assertIsInstance(programs, list)
        self.assertIsInstance(programs[0], EpgProgram)

    def test_vijf_today(self):
        programs = self._epg.get_epg('Play5', date.today().strftime('%Y-%m-%d'))
        self.assertIsInstance(programs, list)
        self.assertIsInstance(programs[0], EpgProgram)

    def test_zes_today(self):
        programs = self._epg.get_epg('Play6', date.today().strftime('%Y-%m-%d'))
        self.assertIsInstance(programs, list)
        self.assertIsInstance(programs[0], EpgProgram)

    def test_unknown_today(self):
        with self.assertRaises(Exception):
            self._epg.get_epg('vtm', date.today().strftime('%Y-%m-%d'))

    def test_vier_out_of_range(self):
        programs = self._epg.get_epg('Play4', '2020-01-01')
        self.assertEqual(programs, [])

    def test_play_video_from_epg(self):
        epg_programs = self._epg.get_epg('Play4', 'yesterday')
        epg_program = [program for program in epg_programs if program.video_url][0]

        # Lookup the Episode data since we don't have an UUID
        api = ContentApi(cache_path=kodiutils.get_cache_path())
        episode = api.get_episode(epg_program.video_url)
        self.assertIsInstance(episode, Episode)

    # def test_map_epg_genre(self):
    #     genres = []
    #     for channel in ['vier', 'vijf', 'zes']:
    #         for day in ['yesterday', 'today', 'tomorrow']:
    #             programs = self._epg.get_epg(channel, day)
    #
    #         for program in programs:
    #             if program.genre not in genres:
    #                 genres.append(program.genre)
    #
    #     print(genres)
    #


if __name__ == '__main__':
    unittest.main()