plugin.video.viervijfzes/tests/test_api.py

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

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

from __future__ import absolute_import, division, print_function, unicode_literals

import logging
import unittest

import resources.lib.kodiutils as kodiutils
from resources.lib.viervijfzes.auth import AuthApi
from resources.lib.viervijfzes.content import ContentApi, Program, Episode, Category, CACHE_PREVENT

_LOGGER = logging.getLogger(__name__)


class TestApi(unittest.TestCase):
    def __init__(self, *args, **kwargs):
        super(TestApi, self).__init__(*args, **kwargs)
        auth = AuthApi(kodiutils.get_setting('username'), kodiutils.get_setting('password'), kodiutils.get_tokens_path())
        self._api = ContentApi(auth, cache_path=kodiutils.get_cache_path())

    def test_programs(self):
        for channel in ['vier', 'vijf', 'zes']:
            programs = self._api.get_programs(channel)
            self.assertIsInstance(programs, list)
            self.assertIsInstance(programs[0], Program)

    def test_categories(self):
        for channel in ['vier', 'vijf', 'zes']:
            categories = self._api.get_categories(channel)
        self.assertIsInstance(categories, list)
        if categories:
            self.assertIsInstance(categories[0], Category)

    def test_episodes(self):
        for channel, program in [('vier', 'auwch'), ('vijf', 'zo-man-zo-vrouw')]:
            program = self._api.get_program(channel, program, cache=CACHE_PREVENT)
            self.assertIsInstance(program, Program)
            self.assertIsInstance(program.seasons, dict)
            self.assertIsInstance(program.episodes, list)
            self.assertIsInstance(program.episodes[0], Episode)

    def test_clips(self):
        for channel, program in [('vier', 'gert-late-night'), ('zes', 'macgyver')]:
            program = self._api.get_program(channel, program, extract_clips=True, cache=CACHE_PREVENT)

            self.assertIsInstance(program.clips, list)
            self.assertIsInstance(program.clips[0], Episode)

            episode = self._api.get_episode(channel, program.clips[0].path, cache=CACHE_PREVENT)
            self.assertIsInstance(episode, Episode)

    @unittest.skipUnless(kodiutils.get_setting('username') and kodiutils.get_setting('password'), 'Skipping since we have no credentials.')
    def test_get_stream(self):
        program = self._api.get_program('vier', 'auwch')
        self.assertIsInstance(program, Program)

        episode = program.episodes[0]
        video = self._api.get_stream_by_uuid(episode.uuid)
        self.assertTrue(video)


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