diff --git a/sidebar/model.py b/sidebar/model.py index 2958e42..c21fe7f 100644 --- a/sidebar/model.py +++ b/sidebar/model.py @@ -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()) diff --git a/sidebar/test_model.py b/sidebar/test_model.py index 4b31a2d..42b3169 100644 --- a/sidebar/test_model.py +++ b/sidebar/test_model.py @@ -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"))