def setUp(self):
self.underlying = self.UnderlyingLibrary()
self.added = []
self.changed = []
self.removed = []
self._sigs = [
connect_obj(self.underlying, 'added', list.extend, self.added),
connect_obj(self.underlying, 'changed', list.extend, self.changed),
connect_obj(self.underlying, 'removed', list.extend, self.removed),
]
self.library = AlbumLibrary(self.underlying)
# Populate for every test
self.underlying.add(self.Frange(12))
def setUp(self):
self.underlying = self.UnderlyingLibrary()
self.added = []
self.changed = []
self.removed = []
self._sigs = [
connect_obj(self.underlying, 'added', list.extend, self.added),
connect_obj(self.underlying,
'changed', list.extend, self.changed),
connect_obj(self.underlying,
'removed', list.extend, self.removed),
]
self.library = AlbumLibrary(self.underlying)
# Populate for every test
self.underlying.add(self.Frange(12))
class TAlbumLibrary(TestCase):
Fake = FakeSong
Frange = staticmethod(ASrange)
UnderlyingLibrary = Library
def setUp(self):
self.underlying = self.UnderlyingLibrary()
self.added = []
self.changed = []
self.removed = []
self._sigs = [
connect_obj(self.underlying, 'added', list.extend, self.added),
connect_obj(self.underlying,
'changed', list.extend, self.changed),
connect_obj(self.underlying,
'removed', list.extend, self.removed),
]
self.library = AlbumLibrary(self.underlying)
# Populate for every test
self.underlying.add(self.Frange(12))
def tearDown(self):
for s in self._sigs:
self.underlying.disconnect(s)
self.underlying.destroy()
self.library.destroy()
def test_get(self):
key = self.underlying.get("file_1.mp3").album_key
self.failUnlessEqual(self.library.get(key).title, "Album 1")
album = self.library.get(key)
self.failUnlessEqual(album.key, key)
self.failUnlessEqual(len(album.songs), 4)
key = self.underlying.get("file_2.mp3").album_key
self.failUnlessEqual(self.library.get(key).title, "Album 2")
def test_getitem(self):
key = self.underlying.get("file_4.mp3").album_key
self.failUnlessEqual(self.library[key].key, key)
def test_keys(self):
self.failUnless(len(self.library.keys()), 3)
def test_has_key(self):
key = self.underlying.get("file_1.mp3").album_key
self.failUnless(self.library.has_key(key))
def test_misc_collection(self):
self.failUnless(itervalues(self.library))
self.failUnless(iteritems(self.library))
def test_items(self):
self.failUnlessEqual(len(self.library.items()), 3)
def test_items_2(self):
albums = self.library.values()
self.failUnlessEqual(len(albums), 3)
songs = self.underlying._contents.values()
# Make sure "all the songs' albums" == "all the albums", roughly
self.failUnlessEqual({a.key for a in albums},
{s.album_key for s in songs})
def test_remove(self):
key = self.underlying.get("file_1.mp3").album_key
songs = self.underlying._contents
# Remove all songs in Album 1
for i in range(1, 12, 3):
song = songs["file_%d.mp3" % i]
self.underlying.remove([song])
# Album 1 is all gone...
self.failUnlessEqual(self.library.get(key), None)
# ...but Album 2 is fine
key = self.underlying.get("file_2.mp3").album_key
album2 = self.library[key]
self.failUnlessEqual(album2.key, key)
self.failUnlessEqual(len(album2.songs), 4)
def test_misc(self):
# It shouldn't implement FileLibrary etc
self.failIf(getattr(self.library, "filename", None))
