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)
