emmental/db/test_db.py
Anna Schumaker c658e873a6 db: Create a function to find playlists based on state id
So we can write the current playlist stateid to the settings file and
look it up later.

Implements: #36 (Find playlists based on playlist state id)
Signed-off-by: Anna Schumaker <Anna@NoWheyCreamery.com>
2021-11-18 10:20:24 -05:00

21 lines
770 B
Python

# Copyright 2021 (c) Anna Schumaker.
import db
import unittest
class TestDB(unittest.TestCase):
def test_new_db(self):
db.reset()
self.assertTrue(db.new_db())
db.make_fake_track(1, 1, "Test Track", "/a/b/c/1.ogg")
self.assertFalse(db.new_db())
def test_find_playlist(self):
track = db.make_fake_track(1, 1, "Test Track", "/a/b/c/1.ogg")
collection = db.user.Table.find("Collection")
genre = db.genre.Table.find("Test Genre")
self.assertIsNone(db.find_playlist(123456))
for plist in [ collection, track.artist, track.album, track.disc,
genre, track.decade, track.year, track.library ]:
self.assertEqual(db.find_playlist(plist.plist_state.rowid), plist)