from Orange.data import Table, Domain
from Orange.widgets.tests.base import WidgetTest
from Orange.widgets.widget import OWWidget
from Orange.widgets.gui import comboBox
from Orange.widgets.settings import \
    ContextSetting, DomainContextHandler, PerfectDomainContextHandler
from Orange.widgets.utils.itemmodels import DomainModel

from orangewidget.utils.signals import Input


class OWProblem(OWWidget):
    name = "Context Problem"

    v1 = ContextSetting(None)
    v2 = ContextSetting(None)
    settingsHandler = DomainContextHandler()

    class Inputs:
        data = Input("Data", Table, default=True)

    def __init__(self):
        super().__init__()

        m1 = DomainModel(DomainModel.SEPARATED,
                         valid_types=DomainModel.PRIMITIVE)
        comboBox(self.controlArea, self, "v1", model=m1, label="All")
        m2 = DomainModel(DomainModel.METAS | DomainModel.CLASSES,
                         valid_types=DomainModel.PRIMITIVE)
        comboBox(self.controlArea, self, "v2", model=m2, label="Metas + Classes")

        self.contextAboutToBeOpened.connect(lambda x: self.init_interface(x[0]))

    def init_interface(self, data):
        domain = data.domain if data is not None else None
        self.controls.v1.model().set_domain(domain)
        self.controls.v2.model().set_domain(domain)

    @Inputs.data
    def set_data(self, data):
        self.closeContext()
        self.openContext(data)


class TestOWProblem(WidgetTest):

    def setUp(self):
        self.widget = self.create_widget(OWProblem)

    def test_v2_invalid_after_reloading(self):
        iris = Table("iris")
        petals_in_metas = Domain(iris.domain.attributes[:2],
                                 iris.domain.class_var,
                                 metas=iris.domain.attributes[2:])
        self.send_signal(self.widget.Inputs.data,
                         iris.transform(petals_in_metas))
        self.widget.v1 = iris.domain["sepal width"]
        # v2 can have only metas and classes thus this will be invalid
        # context for original iris
        self.widget.v2 = iris.domain["petal width"]
        self.send_signal(self.widget.Inputs.data, iris)  # crashes