mxpv-podsync/cmd/resolver/function_test.py

import function as ytdl
import unittest


class TestYtdl(unittest.TestCase):
    def test_resolve(self):
        self.assertIsNotNone(
            ytdl._resolve('https://youtube.com/watch?v=ygIUF678y40',
                          {'format': 'video', 'quality': 'low', 'provider': 'youtube'}))
        self.assertIsNotNone(
            ytdl._resolve('https://youtube.com/watch?v=WyaEiO4hyik',
                          {'format': 'audio', 'quality': 'high', 'provider': 'youtube'}))
        self.assertIsNotNone(
            ytdl._resolve('https://youtube.com/watch?v=5mjUF2j9dgA',
                          {'format': 'video', 'quality': 'low', 'provider': 'youtube'})
        )
        self.assertIsNotNone(
            ytdl._resolve('https://www.youtube.com/watch?v=2nH7xAMqD2g',
                          {'format': 'video', 'quality': 'high', 'provider': 'youtube'})
        )

    def test_vimeo(self):
        self.assertIsNotNone(
            ytdl._resolve('https://vimeo.com/237715420', {'format': 'video', 'quality': 'low', 'provider': 'vimeo'}))
        self.assertIsNotNone(
            ytdl._resolve('https://vimeo.com/275211960', {'format': 'video', 'quality': 'high', 'provider': 'vimeo'})
        )


class TestDynamo(unittest.TestCase):
    def test_metadata(self):
        item = ytdl._get_metadata('86qZ')
        self.assertIsNotNone(item)
        self.assertIsNotNone(item['format'])
        self.assertIsNotNone(item['quality'])
        self.assertIsNotNone(item['provider'])

    def test_counter(self):
        counter = ytdl._update_resolve_counter('86qZ')
        self.assertEqual(counter, 1)
        counter = ytdl._update_resolve_counter('86qZ')
        self.assertEqual(counter, 2)

    def test_download(self):
        url = ytdl.download('86qZ', '7XJYLq3gviY')
        self.assertIsNotNone(url)

    def test_quota_check(self):
        with self.assertRaises(ytdl.QuotaExceeded):
            ytdl.download('xro548QlJ', 'j51NFs0bZ9c')