-#!/usr/bin/env python
+#!/usr/bin/env python3
from __future__ import unicode_literals
import unittest
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+from yt_dlp import YoutubeDL
+from yt_dlp.compat import compat_shlex_quote
from yt_dlp.postprocessor import (
+ ExecAfterDownloadPP,
FFmpegThumbnailsConvertorPP,
MetadataFromFieldPP,
- MetadataFromTitlePP,
+ MetadataParserPP,
)
class TestMetadataFromField(unittest.TestCase):
- def test_format_to_regex(self):
- pp = MetadataFromFieldPP(None, ['title:%(title)s - %(artist)s'])
- self.assertEqual(pp._data[0]['regex'], r'(?P<title>.+)\ \-\ (?P<artist>.+)')
-
- def test_field_to_outtmpl(self):
- pp = MetadataFromFieldPP(None, ['title:%(title)s : %(artist)s'])
- self.assertEqual(pp._data[0]['tmpl'], '%(title)s')
- def test_in_out_seperation(self):
- pp = MetadataFromFieldPP(None, ['%(title)s \\: %(artist)s:%(title)s : %(artist)s'])
- self.assertEqual(pp._data[0]['in'], '%(title)s : %(artist)s')
- self.assertEqual(pp._data[0]['out'], '%(title)s : %(artist)s')
+ def test_format_to_regex(self):
+ self.assertEqual(
+ MetadataParserPP.format_to_regex('%(title)s - %(artist)s'),
+ r'(?P<title>.+)\ \-\ (?P<artist>.+)')
+ self.assertEqual(MetadataParserPP.format_to_regex(r'(?P<x>.+)'), r'(?P<x>.+)')
+ def test_field_to_template(self):
+ self.assertEqual(MetadataParserPP.field_to_template('title'), '%(title)s')
+ self.assertEqual(MetadataParserPP.field_to_template('1'), '1')
+ self.assertEqual(MetadataParserPP.field_to_template('foo bar'), 'foo bar')
+ self.assertEqual(MetadataParserPP.field_to_template(' literal'), ' literal')
-class TestMetadataFromTitle(unittest.TestCase):
- def test_format_to_regex(self):
- pp = MetadataFromTitlePP(None, '%(title)s - %(artist)s')
- self.assertEqual(pp._titleregex, r'(?P<title>.+)\ \-\ (?P<artist>.+)')
+ def test_metadatafromfield(self):
+ self.assertEqual(
+ MetadataFromFieldPP.to_action('%(title)s \\: %(artist)s:%(title)s : %(artist)s'),
+ (MetadataParserPP.Actions.INTERPRET, '%(title)s : %(artist)s', '%(title)s : %(artist)s'))
class TestConvertThumbnail(unittest.TestCase):
for _, out in tests:
os.remove(file.format(out))
+
+
+class TestExecAfterDownload(unittest.TestCase):
+ def test_parse_cmd(self):
+ pp = ExecAfterDownloadPP(YoutubeDL(), '')
+ info = {'filepath': 'file name'}
+ quoted_filepath = compat_shlex_quote(info['filepath'])
+
+ self.assertEqual(pp.parse_cmd('echo', info), 'echo %s' % quoted_filepath)
+ self.assertEqual(pp.parse_cmd('echo.{}', info), 'echo.%s' % quoted_filepath)
+ self.assertEqual(pp.parse_cmd('echo "%(filepath)s"', info), 'echo "%s"' % info['filepath'])