sidebar: Create a UserSelection list model

Signed-off-by: Anna Schumaker <Anna@NoWheyCreamery.com>
This commit is contained in:
Anna Schumaker 2021-11-04 16:01:42 -04:00
parent 1d5a52623c
commit 59d9c3b484
2 changed files with 14 additions and 0 deletions

View File

@ -82,3 +82,9 @@ class FilterUserModel(Gtk.FilterListModel):
Gtk.FilterListModel.__init__(self)
self.set_model(db.user.Table)
self.set_filter(UserFilter())
class UserSelection(Gtk.SingleSelection):
def __init__(self):
Gtk.SingleSelection.__init__(self)
self.set_model(FilterUserModel())

View File

@ -118,3 +118,11 @@ class TestFilterUserModel(unittest.TestCase):
self.assertFalse(filter.do_match(db.user.Table.find("Previous")))
self.assertTrue(filter.do_match(db.user.Table.find("Queued Tracks")))
self.assertTrue(filter.do_match(db.user.Table.find("Test Playlist")))
class TestUserSelectionModel(unittest.TestCase):
def test_init(self):
selection = model.UserSelection()
self.assertIsInstance(selection, Gtk.SingleSelection)
self.assertIsInstance(selection.get_model(), model.FilterUserModel)
self.assertEqual(selection.get_selected_item(), db.user.Table.find("Favorites"))