from tests import TestCase, add import os from quodlibet.parse import FileFromPattern, XMLFromPattern, Pattern class _TPattern(TestCase): from quodlibet.formats._audio import AudioFile def setUp(self): s1 = { 'tracknumber': '5/6', 'artist':'Artist', 'title':'Title5', '~filename':'/path/to/a.mp3', 'xmltest': "<&>"} s2 = { 'tracknumber': '6', 'artist':'Artist', 'title':'Title6', '~filename': '/path/to/b.ogg', 'discnumber':'2', 'unislash': u"foo\uff0fbar" } s3 = { 'title': 'test/subdir', 'genre':'/\n/', '~filename':'/one/more/a.flac', 'version': 'Instrumental'} s4 = { 'performer': 'a\nb', 'artist': 'foo\nbar'} s5 = { 'tracknumber': '7/1234','artist':'Artist', 'title':'Title7', '~filename':'/path/to/e.mp3'} self.a = self.AudioFile(s1) self.b = self.AudioFile(s2) self.c = self.AudioFile(s3) self.d = self.AudioFile(s4) self.e = self.AudioFile(s5) class TPattern(_TPattern): from quodlibet.formats._audio import AudioFile def test_conditional_number_dot_title(s): pat = Pattern('. >') s.assertEquals(pat.format(s.a), '5/6. Title5') s.assertEquals(pat.format(s.b), '6. Title6') s.assertEquals(pat.format(s.c), 'test/subdir') def test_conditional_other_number_dot_title(s): pat = Pattern('<tracknumber|<tracknumber>|00>. <title>') s.assertEquals(pat.format(s.a), '5/6. Title5') s.assertEquals(pat.format(s.b), '6. Title6') s.assertEquals(pat.format(s.c), '00. test/subdir') def test_conditional_other_other(s): # FIXME: was <tracknumber|a|b|c>.. but we can't put <>| in the format # string since it would break the XML pattern formater. s.assertEqual(Pattern('<tracknumber|a|b|c>').format(s.a), "") def test_conditional_genre(s): pat = Pattern('<genre|<genre>|music>') s.assertEquals(pat.format(s.a), 'music') s.assertEquals(pat.format(s.b), 'music') s.assertEquals(pat.format(s.c), '/, /') def test_conditional_unknown(s): pat = Pattern('<album|foo|bar>') s.assertEquals(pat.format(s.a), 'bar') def test_conditional_notfile(s): pat = Pattern('<tracknumber|<tracknumber>|00>') s.assertEquals(pat.format(s.a), '5/6') s.assertEquals(pat.format(s.b), '6') s.assertEquals(pat.format(s.c), '00') def test_conditional_subdir(s): pat = Pattern('/a<genre|/<genre>>/<title>') s.assertEquals(pat.format(s.a), '/a/Title5') s.assertEquals(pat.format(s.b), '/a/Title6') s.assertEquals(pat.format(s.c), '/a//, //test/subdir') def test_number_dot_title(s): pat = Pattern('<tracknumber>. <title>') s.assertEquals(pat.format(s.a), '5/6. Title5') s.assertEquals(pat.format(s.b), '6. Title6') s.assertEquals(pat.format(s.c), '. test/subdir') def test_recnumber_dot_title(s): pat = Pattern('\<<tracknumber>\>. <title>') s.assertEquals(pat.format(s.a), '<5/6>. Title5') s.assertEquals(pat.format(s.b), '<6>. Title6') s.assertEquals(pat.format(s.c), '<>. test/subdir') def test_generated(s): pat = Pattern('<~basename>') s.assertEquals(pat.format(s.a), os.path.basename(s.a["~filename"])) def test_generated_and_not_generated(s): pat = Pattern('<~basename> <title>') res = pat.format(s.a) s.assertEquals( res, os.path.basename(s.a["~filename"]) + " " + s.a["title"]) def test_number_dot_title_dot(s): pat = Pattern('<tracknumber>. <title>.') s.assertEquals(pat.format(s.a), '5/6. Title5.') s.assertEquals(pat.format(s.b), '6. Title6.') s.assertEquals(pat.format(s.c), '. test/subdir.') def test_number_dot_genre(s): pat = Pattern('<tracknumber>. <genre>') s.assertEquals(pat.format(s.a), '5/6. ') s.assertEquals(pat.format(s.b), '6. ') s.assertEquals(pat.format(s.c), '. /, /') def test_unicode_with_int(s): song = s.AudioFile({"tracknumber": "5/6", "title": "\xe3\x81\x99\xe3\x81\xbf\xe3\x82\x8c".decode('utf-8')}) pat = Pattern('<~#track>. <title>') s.assertEquals(pat.format(song), "5. \xe3\x81\x99\xe3\x81\xbf\xe3\x82\x8c".decode('utf-8')) class TFileFromPattern(_TPattern): def test_escape_slash(s): fpat = FileFromPattern('<~filename>') pat = Pattern('<~filename>') wpat = FileFromPattern(r'\\<artist>\\ "<title>') s.assertEquals(fpat.format(s.a), "_path_to_a.mp3") s.assertEquals(pat.format(s.a), "/path/to/a.mp3") s.assertEquals(wpat.format(s.a), "\\Artist\\ \"Title5.mp3") def test_specialcase_anti_ext(s): p1 = FileFromPattern('<~filename>') p2 = FileFromPattern('<~dirname>_<~basename>') s.assertEquals(p1.format(s.a), p2.format(s.a)) s.assertEquals(p1.format(s.a), '_path_to_a.mp3') s.assertEquals(p1.format(s.b), p2.format(s.b)) s.assertEquals(p1.format(s.b), '_path_to_b.ogg') s.assertEquals(p1.format(s.c), p2.format(s.c)) s.assertEquals(p1.format(s.c), '_one_more_a.flac') def test_number_dot_title_dot(s): pat = FileFromPattern('<tracknumber>. <title>.') s.assertEquals(pat.format(s.a), '05. Title5..mp3') s.assertEquals(pat.format(s.b), '06. Title6..ogg') s.assertEquals(pat.format(s.c), '. test_subdir..flac') def test_tracknumber_decimals(s): pat = FileFromPattern('<tracknumber>. <title>') s.assertEquals(pat.format(s.a), '05. Title5.mp3') s.assertEquals(pat.format(s.e), '0007. Title7.mp3') def test_raw_slash_preservation(s): pat = FileFromPattern('/a/b/<genre>') s.assertEquals(pat.format(s.a), '/a/b/.mp3') s.assertEquals(pat.format(s.b), '/a/b/.ogg') s.assertEquals(pat.format(s.c), '/a/b/_, _.flac') def test_unicode_slash_removal(s): pat = FileFromPattern(u'/a/b/<unislash>') s.assertEquals(pat.format(s.b), '/a/b/foo_bar.ogg') def test_directory_rooting(s): s.assertRaises(ValueError, FileFromPattern, 'a/<b>') s.assertRaises(ValueError, FileFromPattern, '<a>/<b>') FileFromPattern('/<a>/<b>') def test_ext_case_preservation(s): x = s.AudioFile({ '~filename':'/tmp/Xx.Flac', 'title':'Xx' }) # If pattern has a particular ext, preserve case of ext p1 = FileFromPattern('<~basename>') s.assertEquals(p1.format(x), 'Xx.Flac') p2 = FileFromPattern('<title>.FLAC') s.assertEquals(p2.format(x), 'Xx.FLAC') # If pattern doesn't have a particular ext, lowercase ext p3 = FileFromPattern('<title>') s.assertEquals(p3.format(x), 'Xx.flac') def test_backslash_conversion_win32(s): if os.name == 'nt': pat = FileFromPattern(r'Z:\<artist>\<title>') s.assertEquals(pat.format(s.a), 'Z:/Artist/Title5.mp3') class TXMLFromPattern(_TPattern): def test_markup_passthrough(s): pat = XMLFromPattern(r'\<b\><<title>>\</b\>') s.assertEquals(pat.format(s.a), '<b><Title5></b>') s.assertEquals(pat.format(s.b), '<b><Title6></b>') s.assertEquals(pat.format(s.c), '<b><test/subdir></b>') def test_escape(s): pat = XMLFromPattern(r'\<b\><<xmltest>>\</b\>') s.assertEquals(pat.format(s.a), '<b><<&>></b>') def test_cond_markup(s): pat = XMLFromPattern(r'<title|\<b\><title> woo\</b\>>') s.assertEquals(pat.format(s.a), '<b>Title5 woo</b>') class TRealTags(TestCase): def test_empty(self): self.failUnlessEqual(Pattern("").tags, set([])) def test_both(self): pat = "<foo|<~bar~fuu> - <fa>|<bar>>" self.failUnlessEqual(Pattern(pat).tags, set(["bar", "fuu", "fa"])) class TPatternFormatList(_TPattern): def test_same(s): pat = Pattern('<~basename> <title>') s.failUnlessEqual(pat.format_list(s.a), set([pat.format(s.a)])) pat = Pattern('/a<genre|/<genre>>/<title>') s.failUnlessEqual(pat.format_list(s.a), set([pat.format(s.a)])) def test_same2(s): fpat = FileFromPattern('<~filename>') pat = Pattern('<~filename>') s.assertEquals(fpat.format_list(s.a), set([fpat.format(s.a)])) s.assertEquals(pat.format_list(s.a), set([pat.format(s.a)])) def test_tied(s): pat = Pattern('<genre>') s.failUnlessEqual(pat.format_list(s.c), set(['/', '/'])) pat = Pattern('<performer>') s.failUnlessEqual(pat.format_list(s.d), set(['a', 'b'])) pat = Pattern('<performer><performer>') s.failUnlessEqual(set(pat.format_list(s.d)), set(['aa', 'ab', 'ba', 'bb'])) pat = Pattern('<~performer~artist>') s.failUnlessEqual(pat.format_list(s.d), set(['a - foo', 'b - foo', 'a - bar', 'b - bar'])) pat = Pattern('<performer~artist>') s.failUnlessEqual(pat.format_list(s.d), set(['a - foo', 'b - foo', 'a - bar', 'b - bar'])) pat = Pattern('<artist|<artist>.|<performer>>') s.failUnlessEqual(pat.format_list(s.d), set(['foo.', 'bar.'])) pat = Pattern('<artist|<artist|<artist>.|<performer>>>') s.failUnlessEqual(pat.format_list(s.d), set(['foo.', 'bar.'])) add(TPattern) add(TFileFromPattern) add(TXMLFromPattern) add(TRealTags) add(TPatternFormatList)