svtplay-dl/lib/svtplay_dl/tests/formatname.py

from __future__ import absolute_import
import unittest

from svtplay_dl.utils.output import _formatname
from svtplay_dl.service import Service
from svtplay_dl.utils.parser import setup_defaults


class formatnameTest(unittest.TestCase):
    all_combo = [
        [
            'test-{title}-{service}-{ext}-{episode}-{id}-{season}-{episodename}',
            {'title': 'title', 'season': 99, 'episode': 21, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "test-title-service-mp4-21-0xdeadface-99-episodename",
        ],
        [
            'test-{title}-{service}-{ext}-{episode}-{id}-{season}-{episodename}',
            {'title': 'title', 'season': 99, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "test-title-service-mp4-0xdeadface-99-episodename"
        ],
        [
            'test-{title}-{service}-{ext}-{episode}-{id}-{season}-{episodename}',
            {'title': 'title', 'season': 99, 'id': '0xdeadface', 'ext': 'ext'},
            "test-title-service-mp4-0xdeadface-99"
        ],
        [
            'test-{title}-{service}-{ext}-{episode}-{id}-{season}-{episodename}',
            {'title': 'title', 'episode': 21, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "test-title-service-mp4-21-0xdeadface-episodename"
        ],
        [
            'test-{title}-{service}-{ext}-{episode}-{id}-{season}-{episodename}',
            {'title': 'title', 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "test-title-service-mp4-0xdeadface-episodename"
        ],
        [
            'test-{title}-{service}-{ext}-{episode}-{id}-{season}-{episodename}',
            {'title': 'title', 'episodename': 'episodename', 'id': '0xdeadface'},
            "test-title-service-mp4-0xdeadface-episodename"
        ],
        [
            'test-{title}-{episode}-{season}-{service}-{episodename}-{id}-{ext}',
            {'title': 'title', 'season': 99, 'episode': 21, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "test-title-21-99-service-episodename-0xdeadface-mp4"
        ],
        [
            'test-{title}-{episode}-{season}-{service}-{episodename}-{id}-{ext}',
            {'title': 'title', 'season': 99, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "test-title-99-service-episodename-0xdeadface-mp4"
        ],
        [
            'test-{title}-{episode}-{season}-{service}-{episodename}-{id}-{ext}',
            {'title': 'title', 'season': 99, 'id': '0xdeadface', 'ext': 'ext'},
            "test-title-99-service-0xdeadface-mp4"
        ],
        [
            'test-{title}-{episode}-{season}-{service}-{episodename}-{id}-{ext}',
            {'title': 'title', 'episode': 21, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "test-title-21-service-episodename-0xdeadface-mp4"
        ],
        [
            'test-{title}-{episode}-{season}-{service}-{episodename}-{id}-{ext}',
            {'title': 'title', 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "test-title-service-episodename-0xdeadface-mp4"
        ],
        [
            'test-{title}-{episode}-{season}-{service}-{episodename}-{id}-{ext}',
            {'title': 'title', 'episodename': 'episodename', 'id': '0xdeadface'},
            "test-title-service-episodename-0xdeadface-mp4"
        ],
        [
            '{id}-{season}-{ext}-{episode}-{episodename}-{title}-{service}',
            {'title': 'title', 'season': 99, 'episode': 21, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "0xdeadface-99-mp4-21-episodename-title-service"
        ],
        [
            '{id}-{season}-{ext}-{episode}-{episodename}-{title}-{service}',
            {'title': 'title', 'season': 99, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "0xdeadface-99-mp4-episodename-title-service"
        ],
        [
            '{id}-{season}-{ext}-{episode}-{episodename}-{title}-{service}',
            {'title': 'title', 'season': 99, 'id': '0xdeadface', 'ext': 'ext'},
            "0xdeadface-99-mp4-title-service"
        ],
        [
            '{id}-{season}-{ext}-{episode}-{episodename}-{title}-{service}',
            {'title': 'title', 'episode': 21, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "0xdeadface-mp4-21-episodename-title-service"
        ],
        [
            '{id}-{season}-{ext}-{episode}-{episodename}-{title}-{service}',
            {'title': 'title', 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "0xdeadface-mp4-episodename-title-service"
        ],
        [
            '{id}-{season}-{ext}-{episode}-{episodename}-{title}-{service}',
            {'title': 'title', 'episodename': 'episodename', 'id': '0xdeadface'},
            "0xdeadface-mp4-episodename-title-service"
        ],
        [
            '{service}-{ext}-{season}-{id}-{title}-{episodename}-{episode}',
            {'title': 'title', 'season': 99, 'episode': 21, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "service-mp4-99-0xdeadface-title-episodename-21"
        ],
        [
            '{service}-{ext}-{season}-{id}-{title}-{episodename}-{episode}',
            {'title': 'title', 'season': 99, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "service-mp4-99-0xdeadface-title-episodename"
        ],
        [
            '{service}-{ext}-{season}-{id}-{title}-{episodename}-{episode}',
            {'title': 'title', 'season': 99, 'id': '0xdeadface', 'ext': 'ext'},
            "service-mp4-99-0xdeadface-title"
        ],
        [
            '{service}-{ext}-{season}-{id}-{title}-{episodename}-{episode}',
            {'title': 'title', 'episode': 21, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "service-mp4-0xdeadface-title-episodename-21"
        ],
        [
            '{service}-{ext}-{season}-{id}-{title}-{episodename}-{episode}',
            {'title': 'title', 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "service-mp4-0xdeadface-title-episodename"
        ],
        [
            '{service}-{ext}-{season}-{id}-{title}-{episodename}-{episode}',
            {'title': 'title', 'episodename': 'episodename', 'id': '0xdeadface'},
            "service-mp4-0xdeadface-title-episodename"
        ],
        [
            'test-{title}-{service}-{ext}-{episode}-{id}-{season}',
            {'title': 'title', 'season': 99, 'episode': 21, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "test-title-service-mp4-21-0xdeadface-99"
        ],
        [
            'test-{title}-{service}-{ext}-{episode}-{id}-{season}',
            {'title': 'title', 'season': 99, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "test-title-service-mp4-0xdeadface-99"
        ],
        [
            'test-{title}-{service}-{ext}-{episode}-{id}-{season}',
            {'title': 'title', 'season': 99, 'id': '0xdeadface', 'ext': 'ext'},
            "test-title-service-mp4-0xdeadface-99"
        ],
        [
            'test-{title}-{service}-{ext}-{episode}-{id}-{season}',
            {'title': 'title', 'episode': 21, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "test-title-service-mp4-21-0xdeadface"
        ],
        [
            'test-{title}-{service}-{ext}-{episode}-{id}-{season}',
            {'title': 'title', 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "test-title-service-mp4-0xdeadface"
        ],
        [
            'test-{title}-{service}-{ext}-{episode}-{id}-{season}',
            {'title': 'title', 'episodename': 'episodename', 'id': '0xdeadface'},
            "test-title-service-mp4-0xdeadface"
        ],
        [
            '{title}-{ext}-{episode}-{id}-{season}',
            {'title': 'title', 'season': 99, 'episode': 21, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "title-mp4-21-0xdeadface-99"
        ],
        [
            '{title}-{ext}-{episode}-{id}-{season}',
            {'title': 'title', 'season': 99, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "title-mp4-0xdeadface-99"
        ],
        [
            '{title}-{ext}-{episode}-{id}-{season}',
            {'title': 'title', 'season': 99, 'id': '0xdeadface', 'ext': 'ext'},
            "title-mp4-0xdeadface-99"
        ],
        [
            '{title}-{ext}-{episode}-{id}-{season}',
            {'title': 'title', 'episode': 21, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "title-mp4-21-0xdeadface"
        ],
        [
            '{title}-{ext}-{episode}-{id}-{season}',
            {'title': 'title', 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "title-mp4-0xdeadface"
        ],
        [
            '{title}-{ext}-{episode}-{id}-{season}',
            {'title': 'title', 'episodename': 'episodename', 'id': '0xdeadface'},
            "title-mp4-0xdeadface"
        ],
        [
            '{title}-{ext}.{episode}-{id}.{season}',
            {'title': 'title', 'season': 99, 'episode': 21, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "title-mp4.21-0xdeadface.99"
        ],
        [
            '{title}-{ext}.{episode}-{id}.{season}',
            {'title': 'title', 'season': 99, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "title-mp4-0xdeadface.99"
        ],
        [
            '{title}-{ext}.{episode}-{id}.{season}',
            {'title': 'title', 'season': 99, 'id': '0xdeadface', 'ext': 'ext'},
            "title-mp4-0xdeadface.99"
        ],
        [
            '{title}-{ext}.{episode}-{id}.{season}',
            {'title': 'title', 'episode': 21, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "title-mp4.21-0xdeadface"
        ],
        [
            '{title}-{ext}.{episode}-{id}.{season}',
            {'title': 'title', 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "title-mp4-0xdeadface"
        ],
        [
            '{title}-{ext}.{episode}-{id}.{season}',
            {'title': 'title', 'episodename': 'episodename', 'id': '0xdeadface'},
            "title-mp4-0xdeadface"
        ],
        [
            '{id}-{season}{ext}-{episode}{episodename}-{title}-{service}',
            {'title': 'title', 'season': 99, 'episode': 21, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "0xdeadface-99mp4-21episodename-title-service"
        ],
        [
            '{id}-{season}{ext}-{episode}{episodename}-{title}-{service}',
            {'title': 'title', 'season': 99, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "0xdeadface-99mp4episodename-title-service"
        ],
        [
            '{id}-{season}{ext}-{episode}{episodename}-{title}-{service}',
            {'title': 'title', 'season': 99, 'id': '0xdeadface', 'ext': 'ext'},
            "0xdeadface-99mp4-title-service"
        ],
        [
            '{id}-{season}{ext}-{episode}{episodename}-{title}-{service}',
            {'title': 'title', 'episode': 21, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "0xdeadfacemp4-21episodename-title-service"
        ],
        [
            '{id}-{season}{ext}-{episode}{episodename}-{title}-{service}',
            {'title': 'title', 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "0xdeadfacemp4episodename-title-service"
        ],
        [
            '{id}-{season}{ext}-{episode}{episodename}-{title}-{service}',
            {'title': 'title', 'episodename': 'episodename', 'id': '0xdeadface'},
            "0xdeadfacemp4episodename-title-service"
        ],
        [
            '{episodename}a{title}-{service}',
            {'title': 'title', 'season': 99, 'episode': 21, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "episodenameatitle-service"
        ],
        [
            '{episodename}a{title}-{service}',
            {'title': 'title', 'season': 99, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "episodenameatitle-service"
        ],
        [
            '{episodename}a{title}-{service}',
            {'title': 'title', 'season': 99, 'id': '0xdeadface', 'ext': 'ext'},
            "atitle-service"
        ],
        [
            '{episodename}a{title}-{service}',
            {'title': 'title', 'episode': 21, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "episodenameatitle-service"
        ],
        [
            '{episodename}a{title}-{service}',
            {'title': 'title', 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "episodenameatitle-service"
        ],
        [
            '{episodename}a{title}-{service}',
            {'title': 'title', 'episodename': 'episodename', 'id': '0xdeadface'},
            "episodenameatitle-service"
        ],
        [
            '{title}.{episode}.{episodename}-{id}-{service}.{ext}',
            {'title': 'title', 'season': 99, 'episode': 21, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "title.21.episodename-0xdeadface-service.mp4"
        ],
        [
            '{title}.{episode}.{episodename}-{id}-{service}.{ext}',
            {'title': 'title', 'season': 99, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "title.episodename-0xdeadface-service.mp4"
        ],
        [
            '{title}.{episode}.{episodename}-{id}-{service}.{ext}',
            {'title': 'title', 'season': 99, 'id': '0xdeadface', 'ext': 'ext'},
            "title-0xdeadface-service.mp4"
        ],
        [
            '{title}.{episode}.{episodename}-{id}-{service}.{ext}',
            {'title': 'title', 'episode': 21, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "title.21.episodename-0xdeadface-service.mp4"
        ],
        [
            '{title}.{episode}.{episodename}-{id}-{service}.{ext}',
            {'title': 'title', 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "title.episodename-0xdeadface-service.mp4"
        ],
        [
            '{title}.{episode}.{episodename}-{id}-{service}.{ext}',
            {'title': 'title', 'episodename': 'episodename', 'id': '0xdeadface'},
            "title.episodename-0xdeadface-service.mp4"
        ],
        [
            '{title}.s{season}e{episode}.{episodename}-{id}-{service}.{ext}',
            {'title': 'title', 'season': 99, 'episode': 21, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "title.s99e21.episodename-0xdeadface-service.mp4"
        ],
        [
            '{title}.s{season}e{episode}.{episodename}-{id}-{service}.{ext}',
            {'title': 'title', 'season': 99, 'episodename': 'episodename', 'id': '0xdeadface', 'ext': 'ext'},
            "title.s99.episodename-0xdeadface-service.mp4"
        ]
    ]

    def test_formatname(self):
        for item in self.all_combo:
            config = setup_defaults()
            config.set("filename", item[0])
            service = Service(config, "localhost")
            service.output.update(item[1])
            self.assertEqual(_formatname(service.output, config, "mp4"), item[2])