def test_library_create(self): if os.path.exists(DB_PATH): os.remove(DB_PATH) sqlstore = SQLStore(DB_PATH) lib = Library(sqlstore) uid = lib.insert(artist="artist1", album='album1', title='title1', path='/path') song = lib.songFromId(uid) # check required fields self.assertEqual(song['artist'], 'artist1') self.assertEqual(song['album'], 'album1') self.assertEqual(song['title'], 'title1') self.assertEqual(song['path'], '/path') # check default fields self.assertEqual(song['playcount'], 0) # both these values should be 1 artists = list(lib.getArtists()) self.assertEqual(len(artists), 1) art = artists[0]['uid'] albums = list(lib.getAlbums(art)) self.assertEqual(len(albums), 1) lib.update(uid, artist="The Artist", album="Album", title="Title") song = lib.songFromId(uid) self.assertEqual(song['artist'], 'The Artist') self.assertEqual(song['album'], 'Album') self.assertEqual(song['title'], 'Title') lib.increment(uid, 'playcount', 5) song = lib.songFromId(uid) self.assertEqual(song['playcount'], 5) lib.increment(uid, 'playcount', -2) song = lib.songFromId(uid) self.assertEqual(song['playcount'], 3) # both these values should be 1, after updating artist,album artists = list(lib.getArtists()) self.assertEqual(len(artists), 1) art = artists[0]['uid'] albums = list(lib.getAlbums(art)) self.assertEqual(len(albums), 1)
def test_library_delete(self): if os.path.exists(DB_PATH): os.remove(DB_PATH) sqlstore = SQLStore(DB_PATH) lib = Library(sqlstore) uid = lib.insert(artist="artist1", album='album1', title='title1', path='/path') lib.remove(uid) with self.assertRaises(KeyError): lib.songFromId(uid)
def test_history_import_playback(self): if os.path.exists(DB_PATH): os.remove(DB_PATH) sqlstore = SQLStore(DB_PATH) lib = Library(sqlstore) h = History(sqlstore) h.setLogEnabled(True) h.setUpdateEnabled(True) lib.history = h uid = lib.insert(artist="artist1", album='album1', title='title1', path='/path', last_played=100000) songb = lib.songFromId(uid) record = { 'date': 50000, 'uid': uid, 'column': Song.playtime, 'value': None } lib.import_record(record) songa = lib.songFromId(uid) self.assertEqual(songa[Song.last_played], songb[Song.last_played]) self.assertEqual(songa[Song.play_count], 1 + songb[Song.play_count]) new_date = 300000 record['date'] = new_date lib.import_record(record) songa = lib.songFromId(uid) self.assertNotEqual(songa[Song.frequency], 0) self.assertEqual(songa[Song.last_played], new_date) self.assertEqual(songa[Song.play_count], 2 + songb[Song.play_count])
def test_history_import_update_int(self): """ create a song, then a record which will update the rating import the record and verify that the artist name was changed successfully """ if os.path.exists(DB_PATH): os.remove(DB_PATH) sqlstore = SQLStore(DB_PATH) lib = Library(sqlstore) h = History(sqlstore) h.setLogEnabled(True) h.setUpdateEnabled(True) lib.history = h rate1 = 5 rate2 = 10 uid = lib.insert(artist="artist1", album='album1', title='title1', rating=rate1, path='/path') songb = lib.songFromId(uid) record = { 'date': 0, 'uid': uid, 'column': Song.rating, 'value': "%s" % rate2 } lib.import_record(record) songa = lib.songFromId(uid) self.assertNotEqual(songb[Song.rating], songa[Song.rating]) self.assertEqual(songa[Song.rating], rate2)