21 lines
770 B
Python
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)
|