mirror of
https://github.com/beetbox/beets.git
synced 2025-12-06 08:39:17 +01:00
71b9fd785c
This reverts commit 9c41c39913.
That commit used byte strings for the `if __name__ == '__main__'` pattern,
which was necessary when we were doing unicode_literals. But it is wrong on
Python 3, and now that we're liberated from unicode_literals, we need to go
back to native strings for this comparison.
54 lines
1.8 KiB
Python
54 lines
1.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
"""Tests for the 'ihate' plugin"""
|
|
|
|
from __future__ import division, absolute_import, print_function
|
|
|
|
from test._common import unittest
|
|
from beets import importer
|
|
from beets.library import Item
|
|
from beetsplug.ihate import IHatePlugin
|
|
|
|
|
|
class IHatePluginTest(unittest.TestCase):
|
|
|
|
def test_hate(self):
|
|
|
|
match_pattern = {}
|
|
test_item = Item(
|
|
genre=u'TestGenre',
|
|
album=u'TestAlbum',
|
|
artist=u'TestArtist')
|
|
task = importer.SingletonImportTask(None, test_item)
|
|
|
|
# Empty query should let it pass.
|
|
self.assertFalse(IHatePlugin.do_i_hate_this(task, match_pattern))
|
|
|
|
# 1 query match.
|
|
match_pattern = [u"artist:bad_artist", u"artist:TestArtist"]
|
|
self.assertTrue(IHatePlugin.do_i_hate_this(task, match_pattern))
|
|
|
|
# 2 query matches, either should trigger.
|
|
match_pattern = [u"album:test", u"artist:testartist"]
|
|
self.assertTrue(IHatePlugin.do_i_hate_this(task, match_pattern))
|
|
|
|
# Query is blocked by AND clause.
|
|
match_pattern = [u"album:notthis genre:testgenre"]
|
|
self.assertFalse(IHatePlugin.do_i_hate_this(task, match_pattern))
|
|
|
|
# Both queries are blocked by AND clause with unmatched condition.
|
|
match_pattern = [u"album:notthis genre:testgenre",
|
|
u"artist:testartist album:notthis"]
|
|
self.assertFalse(IHatePlugin.do_i_hate_this(task, match_pattern))
|
|
|
|
# Only one query should fire.
|
|
match_pattern = [u"album:testalbum genre:testgenre",
|
|
u"artist:testartist album:notthis"]
|
|
self.assertTrue(IHatePlugin.do_i_hate_this(task, match_pattern))
|
|
|
|
|
|
def suite():
|
|
return unittest.TestLoader().loadTestsFromName(__name__)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main(defaultTest='suite')
|