From a6068e695cb768a17d772185a20fbd77f73b8e2a Mon Sep 17 00:00:00 2001 From: 7x11x13 Date: Tue, 7 Jan 2025 15:19:52 -0500 Subject: [PATCH] Update tests --- test/test_YoutubeDL.py | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) diff --git a/test/test_YoutubeDL.py b/test/test_YoutubeDL.py index 48164a7b5a..bc3b63233d 100644 --- a/test/test_YoutubeDL.py +++ b/test/test_YoutubeDL.py @@ -697,6 +697,7 @@ def test_add_extra_info(self): 'title3': 'foo/bar\\test', 'title4': 'foo "bar" test', 'title5': 'áéí 𝐀', + 'title6': 'あ' * 10, 'timestamp': 1618488000, 'duration': 100000, 'playlist_index': 1, @@ -712,6 +713,8 @@ def test_add_extra_info(self): def test_prepare_outtmpl_and_filename(self): def test(tmpl, expected, *, info=None, **params): + if 'trim_file_name' not in params: + params['trim_file_name'] = 0 # disable trimming params['outtmpl'] = tmpl ydl = FakeYDL(params) ydl._num_downloads = 1 @@ -791,7 +794,7 @@ def expect_same_infodict(out): self.assertEqual(got_dict.get(info_field), expected, info_field) return True - test('%()j', (expect_same_infodict, None), trim_file_name=0) + test('%()j', (expect_same_infodict, None)) # NA placeholder NA_TEST_OUTTMPL = '%(uploader_date)s-%(width)d-%(x|def)s-%(id)s.%(ext)s' @@ -924,6 +927,16 @@ def gen(): test('%(title3)s', ('foo/bar\\test', 'foo⧸bar⧹test')) test('folder/%(title3)s', ('folder/foo/bar\\test', f'folder{os.path.sep}foo⧸bar⧹test')) + # --trim-filenames + test('%(title6)s.%(ext)s', 'あ' * 10 + '.mp4') + test('%(title6)s.%(ext)s', 'あ' * 3 + '.mp4', trim_file_name='3c') + test('%(title6)s.%(ext)s', 'あ' * 3 + '.mp4', trim_file_name='9b', filesystem_encoding='utf-8') + test('%(title6)s.%(ext)s', 'あ' * 3 + '.mp4', trim_file_name='10b', filesystem_encoding='utf-8') + test('%(title6)s.%(ext)s', 'あ' * 3 + '.mp4', trim_file_name='11b', filesystem_encoding='utf-8') + test('%(title6)s.%(ext)s', 'あ' * 4 + '.mp4', trim_file_name='12b', filesystem_encoding='utf-8') + test('%(title6)s.%(ext)s', 'あ' * 6 + '.mp4', trim_file_name='12b', filesystem_encoding='utf-16le') + test('folder/%(title6)s.%(ext)s', f'folder{os.path.sep}あああ.mp4', trim_file_name='3c') + def test_format_note(self): ydl = YoutubeDL() self.assertEqual(ydl._format_note({}), '')