from sima.lib.db import SimaDB
from sima.lib.track import Track
-from sima.lib.meta import Album
+from sima.lib.meta import Album, Artist, MetaContainer
DEVOLT = {
'albumartist': 'Devolt',
'albumartistsort': 'Devolt',
'artist': 'Devolt',
+ 'genre': ['Rock'],
'date': '2011-12-01',
'disc': '1/1',
'file': 'music/Devolt/2011-Grey/03-Devolt - Crazy.mp3',
'track': '3/6'}
DB_FILE = 'file::memory:?cache=shared'
-KEEP_FILE = False # File db in file to ease debug
+KEEP_FILE = True # File db in file to ease debug
if KEEP_FILE:
DB_FILE = '/dev/shm/unittest.sqlite'
CURRENT = datetime.datetime.utcnow()
# recent first, oldest last
hist = list()
for i in range(1, 5): # starts at 1 to ensure records are in the past
- trk = Track(file=f'/foo/bar.{i}', name='{i}-baz', album='foolbum')
+ trk = Track(file=f'/foo/bar.{i}', name=f'{i}-baz',
+ album='foolbum', genre=f'{i}')
hist.append(trk)
last = CURRENT - datetime.timedelta(minutes=i)
self.db.add_history(trk, date=last)
#
trk03 = Track(file='03', **tr)
self.db.add_history(trk03, CURRENT-datetime.timedelta(hours=1))
- # got multiple tracks, same artist, got artist history len == 1
+ # got multiple tracks, same artist/album, got artist/album history len = 1
art_history = self.db.fetch_artists_history()
+ alb_history = self.db.fetch_albums_history()
self.assertEqual(len(art_history), 1)
+ self.assertEqual(len(alb_history), 1)
self.assertEqual(art_history, [trk01.Artist])
# Now add new artist to history
trk04.artist, trk03.artist],
art_history)
- def test_04_triggers(self):
+ def test_04_filtering_history(self):
+ # Controls artist history filtering
+ for i in range(0, 5):
+ trk = Track(file=f'/foo/bar.{i}', name=f'{i}-baz',
+ artist=f'{i}-art', album=f'{i}-lbum')
+ last = CURRENT - datetime.timedelta(minutes=i)
+ self.db.add_history(trk, date=last)
+ if i == 5: # bounce latest record
+ self.db.add_history(trk, date=last)
+ art_history = self.db.fetch_artists_history()
+ # Already checked but to be sure, we should have 5 artists in history
+ self.assertEqual(len(art_history), 5)
+ for needle in ['4-art', Artist(name='4-art')]:
+ art_history = self.db.fetch_artists_history(needle=needle)
+ self.assertEqual(art_history, [needle])
+ needle = Artist(name='not-art')
+ art_history = self.db.fetch_artists_history(needle=needle)
+ self.assertEqual(art_history, [])
+ # Controls artists history filtering
+ # for a list of Artist objects
+ needle_list = [Artist(name='3-art'), Artist(name='4-art')]
+ art_history = self.db.fetch_artists_history(needle=needle_list)
+ self.assertEqual(art_history, needle_list)
+ # for a MetaContainer
+ needle_meta = MetaContainer(needle_list)
+ art_history = self.db.fetch_artists_history(needle=needle_meta)
+ self.assertEqual(art_history, needle_list)
+ # for a list of string (Meta object handles Artist/str comparison)
+ art_history = self.db.fetch_artists_history(['3-art', '4-art'])
+ self.assertEqual(art_history, needle_list)
+
+ # Controls album history filtering
+ needle = Artist(name='4-art')
+ alb_history = self.db.fetch_albums_history(needle=needle)
+ self.assertEqual(alb_history, [Album(name='4-lbum')])
+
+ def test_05_triggers(self):
self.db.purge_history(duration=0)
tracks_ids = list()
# Add a first track
if i == 3:
self.db.get_bl_artist(trk.Artist)
- def test_blocklist_triggers(self):
+ def test_blocklist_triggers_00(self):
+ trk01 = Track(file='01', name='01', artist='artist A', album='album A')
+ blart01_id = self.db.get_bl_artist(trk01.Artist)
+ blalb01_id = self.db.get_bl_album(Album(name=trk01.album, mbid=trk01.musicbrainz_albumid))
+ conn = self.db.get_database_connection()
+ self.db._remove_blocklist_id(blart01_id, with_connection=conn)
+ self.db._remove_blocklist_id(blalb01_id, with_connection=conn)
+ albums = conn.execute('SELECT albums.name FROM albums;').fetchall()
+ artists = conn.execute('SELECT artists.name FROM artists;').fetchall()
+ conn.close()
+ self.assertNotIn((trk01.album,), albums)
+ self.assertNotIn((trk01.artist,), artists)
+
+ def test_blocklist_triggers_01(self):
trk01 = Track(file='01', name='01', artist='artist A', album='album A')
trk02 = Track(file='02', name='01', artist='artist A', album='album B')
trk01_id = self.db.get_bl_track(trk01)
conn.close()
+class Test_02Genre(Main):
+
+ def test_genre(self):
+ conn = self.db.get_database_connection()
+ self.db.get_genre('Post-Rock', with_connection=conn)
+ genres = list()
+ for i in range(1, 15): # starts at 1 to ensure records are in the past
+ trk = Track(file=f'/foo/bar.{i}', name=f'{i}-baz',
+ album='foolbum', artist=f'{i}-art', genre=f'{i}')
+ genres.append(f'{i}')
+ last = CURRENT - datetime.timedelta(minutes=i)
+ self.db.add_history(trk, date=last)
+ genre_hist = self.db.fetch_genres_history(limit=10)
+ self.assertEqual([g[0] for g in genre_hist], genres[:10])
+
# VIM MODLINE
# vim: ai ts=4 sw=4 sts=4 expandtab fileencoding=utf8