emmental/playlist/test_column.py

# Copyright 2021 (c) Anna Schumaker.
from lib import settings
from . import column
from gi.repository import Gtk
import unittest

class FakeListItem:
    def __init__(self): self.child = None
    def get_item(self): return None
    def get_child(self): return self.child
    def set_child(self, child): self.child = child


class TestLabelFactory(unittest.TestCase):
    def test_init(self):
        factory = column.LabelFactory(0.5)
        self.assertIsInstance(factory, Gtk.SignalListItemFactory)
        self.assertEqual(factory.xalign, 0.5)

    def test_setup_teardown(self):
        fake = FakeListItem()
        factory = column.LabelFactory(0.5)

        factory.on_setup(factory, fake)
        self.assertIsInstance(fake.child, Gtk.Label)
        self.assertEqual(fake.child.get_xalign(), 0.5)

        factory.on_teardown(factory, fake)
        self.assertIsNone(fake.child)

    def test_bind_unbind(self):
        fake = FakeListItem()
        factory = column.LabelFactory(0.5)
        factory.on_setup(factory, fake)

        with self.assertRaises(NotImplementedError):
            factory.on_bind(factory, fake)

        fake.child.set_text("abcde")
        factory.on_unbind(factory, fake)
        self.assertEqual(fake.child.get_text(), "")


class TestPlaylistColumn(unittest.TestCase):
    def tearDown(self):
        settings.reset()

    def test_playlist_column_init(self):
        col = column.Column("Test", "test")
        self.assertIsInstance(col, Gtk.ColumnViewColumn)
        self.assertIsInstance(col.factory, Gtk.SignalListItemFactory)

        self.assertEqual(col.get_title(), "Test")
        self.assertEqual(col.field, "test")
        self.assertEqual(col.align, 0)
        self.assertEqual(col.get_factory(), col.factory)
        self.assertTrue(col.get_resizable())
        self.assertFalse(col.get_expand())

        col = column.Column("Test2", "test", expand=True)
        self.assertTrue(col.get_expand())

    def test_playlist_column_width(self):
        col = column.Column("Test", "test", width=100)

        self.assertEqual(settings.get_int("column.Test"), 100)
        col.set_fixed_width(200)
        self.assertEqual(settings.get_int("column.Test"), 200)