class TPluginsSongsMenu(PluginTestCase):
def setUp(self):
config.init()
self.h = SongsMenuPluginHandler()
library = SongLibrary()
library.librarian = SongLibrarian()
self.lib = library
self.parent = Gtk.Window()
def tearDown(self):
self.lib.destroy()
self.parent.destroy()
config.quit()
def test_init(self):
for id_, plugin in self.plugins.items():
if self.h.plugin_handle(plugin):
self.h.plugin_enable(plugin)
self.h.handle(id_, None, None, [])
self.h.plugin_disable(plugin)
def test_handle_single(self):
self.skipTest("Pops up windows and needs user input.. so disabled."
"Still worth keeping whilst we don't have unit tests "
"for all plugins.")
# Ignored...
for id_, plugin in self.plugins.items():
if self.h.plugin_handle(plugin):
self.h.plugin_enable(plugin, None)
self.h.handle(id_, self.lib, self.parent, SONGS)
self.h.plugin_disable(plugin)
def test_handles_albums(self):
for id_, plugin in self.plugins.items():
if isinstance(plugin, SongsMenuPlugin):
ha = plugin.handles_albums
self.failIf(hasattr(plugin, "plugin_single_album") and not ha)
self.failIf(hasattr(plugin, "plugin_plugin_album") and not ha)
self.failIf(hasattr(plugin, "plugin_albums") and not ha)
class TPluginsSongsMenu(PluginTestCase):
def setUp(self):
config.init()
self.h = SongsMenuPluginHandler()
library = SongLibrary()
library.librarian = SongLibrarian()
self.lib = library
self.parent = Gtk.Window()
def tearDown(self):
self.lib.destroy()
self.parent.destroy()
config.quit()
def test_init(self):
for id_, plugin in self.plugins.iteritems():
if self.h.plugin_handle(plugin):
self.h.plugin_enable(plugin)
self.h.handle(id_, None, None, [])
self.h.plugin_disable(plugin)
def test_handle_single(self):
self.skipTest("Pops up windows and needs user input.. so disabled."
"Still worth keeping whilst we don't have unit tests "
"for all plugins.")
# Ignored...
for id_, plugin in self.plugins.iteritems():
if self.h.plugin_handle(plugin):
self.h.plugin_enable(plugin, None)
self.h.handle(id_, self.lib, self.parent, SONGS)
self.h.plugin_disable(plugin)
def test_handles_albums(self):
for id_, plugin in self.plugins.iteritems():
if isinstance(plugin, SongsMenuPlugin):
ha = plugin.handles_albums
self.failIf(hasattr(plugin, "plugin_single_album") and not ha)
self.failIf(hasattr(plugin, "plugin_plugin_album") and not ha)
self.failIf(hasattr(plugin, "plugin_albums") and not ha)
class TSongsMenuPlugins(TestCase):
def _confirmer(self, *args):
self.confirmed = True
def setUp(self):
self.tempdir = mkdtemp()
self.pm = PluginManager(folders=[self.tempdir])
self.confirmed = False
self.handler = SongsMenuPluginHandler(self._confirmer, self._confirmer)
self.pm.register_handler(self.handler)
self.pm.rescan()
self.assertEquals(self.pm.plugins, [])
self.library = SongLibrary('foo')
def tearDown(self):
self.library.destroy()
self.pm.quit()
for f in os.listdir(self.tempdir):
os.remove(os.path.join(self.tempdir, f))
os.rmdir(self.tempdir)
def create_plugin(self, id='', name='', desc='', icon='',
funcs=None, mod=False):
fd, fn = mkstemp(suffix='.py', text=True, dir=self.tempdir)
file = os.fdopen(fd, 'w')
if mod:
indent = ''
else:
file.write(
"from quodlibet.plugins.songsmenu import SongsMenuPlugin\n")
file.write("class %s(SongsMenuPlugin):\n" % name)
indent = ' '
file.write("%spass\n" % indent)
if name:
file.write("%sPLUGIN_ID = %r\n" % (indent, name))
if name:
file.write("%sPLUGIN_NAME = %r\n" % (indent, name))
if desc:
file.write("%sPLUGIN_DESC = %r\n" % (indent, desc))
if icon:
file.write("%sPLUGIN_ICON = %r\n" % (indent, icon))
for f in (funcs or []):
if f in ["__init__"]:
file.write("%sdef %s(self, *args): super(%s, self).__init__("
"*args); raise Exception(\"as expected\")\n"
% (indent, f, name))
else:
file.write("%sdef %s(*args): return args\n" % (indent, f))
file.flush()
file.close()
def test_empty_has_no_plugins(self):
self.pm.rescan()
self.assertEquals(self.pm.plugins, [])
def test_name_and_desc_plus_func_is_one(self):
self.create_plugin(name='Name', desc='Desc', funcs=['plugin_song'])
self.pm.rescan()
self.assertEquals(len(self.pm.plugins), 1)
def test_additional_functions_still_only_one(self):
self.create_plugin(name='Name', desc='Desc',
funcs=['plugin_song', 'plugin_songs'])
self.pm.rescan()
self.assertEquals(len(self.pm.plugins), 1)
def test_two_plugins_are_two(self):
self.create_plugin(name='Name', desc='Desc', funcs=['plugin_song'])
self.create_plugin(name='Name2', desc='Desc2',
funcs=['plugin_albums'])
self.pm.rescan()
self.assertEquals(len(self.pm.plugins), 2)
def test_disables_plugin(self):
self.create_plugin(name='Name', desc='Desc', funcs=['plugin_song'])
self.pm.rescan()
self.failIf(self.pm.enabled(self.pm.plugins[0]))
def test_enabledisable_plugin(self):
self.create_plugin(name='Name', desc='Desc', funcs=['plugin_song'])
self.pm.rescan()
plug = self.pm.plugins[0]
self.pm.enable(plug, True)
self.failUnless(self.pm.enabled(plug))
self.pm.enable(plug, False)
self.failIf(self.pm.enabled(plug))
def test_ignores_broken_plugin(self):
self.create_plugin(name="Broken", desc="Desc",
funcs=["__init__", "plugin_song"])
self.pm.rescan()
plug = self.pm.plugins[0]
self.pm.enable(plug, True)
with capture_output():
menu = self.handler.Menu(None, [AudioFile()])
self.failIf(menu and menu.get_children())
def test_Menu(self):
self.create_plugin(name='Name', desc='Desc', funcs=['plugin_song'])
self.handler.Menu(None, [AudioFile()])
def test_handling_songs_without_confirmation(self):
plugin = Plugin(FakeSongsMenuPlugin)
self.handler.plugin_enable(plugin)
MAX = FakeSongsMenuPlugin.MAX_INVOCATIONS
songs = [AudioFile({'~filename': "/tmp/%s" % x, 'artist': 'foo'})
for x in range(MAX)]
self.handler.handle(plugin.id, self.library, None, songs)
self.failIf(self.confirmed, ("Wasn't expecting a confirmation for %d"
" invocations" % len(songs)))
def test_handling_lots_of_songs_with_confirmation(self):
plugin = Plugin(FakeSongsMenuPlugin)
self.handler.plugin_enable(plugin)
MAX = FakeSongsMenuPlugin.MAX_INVOCATIONS
songs = [AudioFile({'~filename': "/tmp/%s" % x, 'artist': 'foo'})
for x in range(MAX + 1)]
self.handler.handle(plugin.id, self.library, None, songs)
self.failUnless(self.confirmed,
("Should have confirmed %d invocations (Max=%d)."
% (len(songs), MAX)))
class TSongsMenuPlugins(TestCase):
def _confirmer(self, *args):
self.confirmed = True
def setUp(self):
self.tempdir = mkdtemp()
self.pm = PluginManager(folders=[self.tempdir])
self.confirmed = False
self.handler = SongsMenuPluginHandler(self._confirmer, self._confirmer)
self.pm.register_handler(self.handler)
self.pm.rescan()
self.assertEquals(self.pm.plugins, [])
self.library = SongLibrary('foo')
def tearDown(self):
self.library.destroy()
self.pm.quit()
for f in os.listdir(self.tempdir):
os.remove(os.path.join(self.tempdir, f))
os.rmdir(self.tempdir)
def create_plugin(self, id='', name='', desc='', icon='',
funcs=None, mod=False):
fd, fn = mkstemp(suffix='.py', text=True, dir=self.tempdir)
file = os.fdopen(fd, 'w')
if mod:
indent = ''
else:
file.write(
"from quodlibet.plugins.songsmenu import SongsMenuPlugin\n")
file.write("class %s(SongsMenuPlugin):\n" % name)
indent = ' '
file.write("%spass\n" % indent)
if name:
file.write("%sPLUGIN_ID = %r\n" % (indent, name))
if name:
file.write("%sPLUGIN_NAME = %r\n" % (indent, name))
if desc:
file.write("%sPLUGIN_DESC = %r\n" % (indent, desc))
if icon:
file.write("%sPLUGIN_ICON = %r\n" % (indent, icon))
for f in (funcs or []):
if f in ["__init__"]:
file.write("%sdef %s(self, *args): super(%s, self).__init__("
"*args); raise Exception(\"as expected\")\n"
% (indent, f, name))
else:
file.write("%sdef %s(*args): return args\n" % (indent, f))
file.flush()
file.close()
def test_empty_has_no_plugins(self):
self.pm.rescan()
self.assertEquals(self.pm.plugins, [])
def test_name_and_desc_plus_func_is_one(self):
self.create_plugin(name='Name', desc='Desc', funcs=['plugin_song'])
self.pm.rescan()
self.assertEquals(len(self.pm.plugins), 1)
def test_additional_functions_still_only_one(self):
self.create_plugin(name='Name', desc='Desc',
funcs=['plugin_song', 'plugin_songs'])
self.pm.rescan()
self.assertEquals(len(self.pm.plugins), 1)
def test_two_plugins_are_two(self):
self.create_plugin(name='Name', desc='Desc', funcs=['plugin_song'])
self.create_plugin(name='Name2', desc='Desc2',
funcs=['plugin_albums'])
self.pm.rescan()
self.assertEquals(len(self.pm.plugins), 2)
def test_disables_plugin(self):
self.create_plugin(name='Name', desc='Desc', funcs=['plugin_song'])
self.pm.rescan()
self.failIf(self.pm.enabled(self.pm.plugins[0]))
def test_enabledisable_plugin(self):
self.create_plugin(name='Name', desc='Desc', funcs=['plugin_song'])
self.pm.rescan()
plug = self.pm.plugins[0]
self.pm.enable(plug, True)
self.failUnless(self.pm.enabled(plug))
self.pm.enable(plug, False)
self.failIf(self.pm.enabled(plug))
def test_ignores_broken_plugin(self):
self.create_plugin(name="Broken", desc="Desc",
funcs=["__init__", "plugin_song"])
self.pm.rescan()
plug = self.pm.plugins[0]
self.pm.enable(plug, True)
with capture_output():
menu = self.handler.Menu(None, [AudioFile()])
self.failIf(menu and menu.get_children())
def test_Menu(self):
self.create_plugin(name='Name', desc='Desc', funcs=['plugin_song'])
self.handler.Menu(None, [AudioFile()])
def test_handling_songs_without_confirmation(self):
plugin = Plugin(FakeSongsMenuPlugin)
self.handler.plugin_enable(plugin)
MAX = FakeSongsMenuPlugin.MAX_INVOCATIONS
songs = [AudioFile({'~filename': "/tmp/%s" % x, 'artist': 'foo'})
for x in range(MAX)]
self.handler.handle(plugin.id, self.library, None, songs)
self.failIf(self.confirmed, ("Wasn't expecting a confirmation for %d"
" invocations" % len(songs)))
def test_handling_lots_of_songs_with_confirmation(self):
plugin = Plugin(FakeSongsMenuPlugin)
self.handler.plugin_enable(plugin)
MAX = FakeSongsMenuPlugin.MAX_INVOCATIONS
songs = [AudioFile({'~filename': "/tmp/%s" % x, 'artist': 'foo'})
for x in range(MAX + 1)]
self.handler.handle(plugin.id, self.library, None, songs)
self.failUnless(self.confirmed,
("Should have confirmed %d invocations (Max=%d)."
% (len(songs), MAX)))
