Anna Schumaker
c982fe624b
I can't inherit from Gtk.TreeListModel since the new() constructor is the only way to set the create_child_func. Instead, use this wrapper function to do the same thing. Signed-off-by: Anna Schumaker <Anna@NoWheyCreamery.com>
54 lines
1.8 KiB
Python
54 lines
1.8 KiB
Python
# Copyright 2021 (c) Anna Schumaker.
|
|
import datetime
|
|
import db
|
|
import unittest
|
|
from gi.repository import GObject
|
|
from gi.repository import Gio
|
|
from gi.repository import Gtk
|
|
from . import model
|
|
|
|
class TestChildModel(unittest.TestCase):
|
|
def items_changed(self, table, pos, rm, add):
|
|
self.changed = (pos, rm, add)
|
|
|
|
def setUp(self):
|
|
db.reset()
|
|
|
|
def test_init(self):
|
|
artist = db.artist.Table.find("Test Artist", "Test Sort")
|
|
child = model.ChildModel(artist)
|
|
|
|
self.assertIsInstance(child, GObject.GObject)
|
|
self.assertIsInstance(child, Gio.ListModel)
|
|
|
|
self.assertEqual(child.parent, artist)
|
|
self.assertEqual(child.get_item_type(), GObject.TYPE_PYOBJECT)
|
|
self.assertEqual(child.get_n_items(), 0)
|
|
|
|
def test_children(self):
|
|
artist = db.artist.Table.find("Test Artist", "Test Sort")
|
|
child = model.ChildModel(artist)
|
|
child.connect("items-changed", self.items_changed)
|
|
|
|
album = artist.find_album("Test Album", datetime.date.today())
|
|
self.assertEqual(child.get_n_items(), 1)
|
|
self.assertEqual(child.get_item(0), album)
|
|
self.assertEqual(self.changed, (0, 0, 1))
|
|
|
|
|
|
class TestTreeListModel(unittest.TestCase):
|
|
def test_init(self):
|
|
tree = model.TreeListModel(db.artist.Table)
|
|
self.assertIsInstance(tree, Gtk.TreeListModel)
|
|
self.assertEqual(tree.get_model(), db.artist.Table)
|
|
self.assertFalse(tree.get_autoexpand())
|
|
self.assertFalse(tree.get_passthrough())
|
|
|
|
def test_create_child_func(self):
|
|
artist = db.artist.Table.find("Test Artist", "Test Sort")
|
|
child = model.create_child_func(artist)
|
|
self.assertEqual(child.parent, artist)
|
|
|
|
genre = db.genre.Table.find("Test Genre")
|
|
self.assertIsNone(model.create_child_func(genre))
|