from unittest import TestCase

from traits.api import HasTraits, Str, Int, List
from traitsui.api import View, Group, Item, TableEditor, ObjectColumn, RGBColor
from traitsui.color_column import ColorColumn

from traitsui.tests._tools import skip_if_null, store_exceptions_on_all_threads


class MyEntry(HasTraits):
    name = Str()
    value = Int(0)
    color = RGBColor()

    entry_view = View(Group(Item("name"), Item("value"), Item("color")))


my_editor = TableEditor(
    columns=[
        ObjectColumn(name="name"),
        ObjectColumn(name="value"),
        ColorColumn(name="color", style="readonly"),
    ],
    orientation="vertical",
    show_toolbar=True,
    row_factory=MyEntry,
)


class MyData(HasTraits):
    data_list = List(MyEntry)

    view = View(Item("data_list", editor=my_editor, show_label=False))


class TestColorColumn(TestCase):
    @skip_if_null
    def test_color_column(self):
        # Behaviour: column ui should display without error

        with store_exceptions_on_all_threads():
            d1 = MyEntry(name="a", value=2, color=(1.0, 0.3, 0.1))
            d2 = MyEntry(name="b", value=3, color=(0.1, 0.0, 0.9))
            data = MyData(data_list=[d1, d2])

            ui = data.edit_traits()

            ui.dispose()
