# (C) Copyright 2004-2021 Enthought, Inc., Austin, TX
# All rights reserved.
#
# This software is provided without warranty under the terms of the BSD
# license included in LICENSE.txt and may be redistributed only under
# the conditions described in the aforementioned license. The license
# is also available online at http://www.enthought.com/licenses/BSD.txt
#
# Thanks for using Enthought open source!

import unittest

from pyface.timer.api import CallbackTimer
from traits.api import HasTraits, Instance, Int
from traitsui.api import Handler, Item, UIInfo, View, toolkit

from ._tools import (
    BaseTestMixin, GuiTestAssistant, is_qt, no_gui_test_assistant
)


class SimpleModel(HasTraits):
    cell = Int()


class ClosableHandler(Handler):

    info = Instance(UIInfo)

    def init(self, info):
        self.info = info
        return True

    def test_close(self):
        self.info.ui.dispose()


simple_view = View(
    Item("cell"), title="Enter IDs and conditions", buttons=["OK", "Cancel"]
)


@unittest.skipIf(no_gui_test_assistant, "No GuiTestAssistant")
class TestViewApplication(BaseTestMixin, GuiTestAssistant, unittest.TestCase):
    def setUp(self):
        BaseTestMixin.setUp(self)
        GuiTestAssistant.setUp(self)
        self.model = SimpleModel()
        self.handler = ClosableHandler()
        self.event_loop_timeout = False
        self.closed = False

        if is_qt():
            if len(self.qt_app.topLevelWidgets()) > 0:
                with self.event_loop_with_timeout(repeat=5):
                    self.gui.invoke_later(self.qt_app.closeAllWindows)

    def tearDown(self):
        GuiTestAssistant.tearDown(self)
        BaseTestMixin.tearDown(self)

    def view_application(self, kind, button=None):
        if button is None:
            self.gui.invoke_later(self.close_dialog)
        else:
            self.gui.invoke_later(self.click_button, text=button)

        timer = CallbackTimer.single_shot(
            callback=self.stop_event_loop, interval=1.0
        )
        try:
            self.result = toolkit().view_application(
                context=self.model,
                view=simple_view,
                kind=kind,
                handler=self.handler,
            )
        finally:
            timer.stop()

    def view_application_event_loop(self, kind):
        with self.event_loop_until_condition(lambda: self.closed):
            self.gui.invoke_later(
                toolkit().view_application,
                context=self.model,
                view=simple_view,
                kind=kind,
                handler=self.handler,
            )
            self.gui.invoke_after(100, self.close_dialog)

    def close_dialog(self):
        if is_qt():
            self.handler.info.ui.control.close()
            self.closed = True
        else:
            raise NotImplementedError("Can't close current backend")

    def click_button(self, text):
        if is_qt():
            from pyface.qt.QtGui import QPushButton
            from pyface.ui.qt4.util.testing import find_qt_widget

            button = find_qt_widget(
                self.handler.info.ui.control,
                QPushButton,
                lambda widget: widget.text() == text,
            )
            if button is None:
                raise RuntimeError("Can't find {} button".format(text))
            button.click()
            self.closed = True
        else:
            raise NotImplementedError("Can't click current backend")

    def stop_event_loop(self):
        self.gui.stop_event_loop()
        self.event_loop_timeout = True

    def test_modal_view_application_close(self):
        self.view_application("modal")

        self.assertTrue(self.closed)
        self.assertFalse(self.event_loop_timeout)
        self.assertFalse(self.result)

    def test_nonmodal_view_application_close(self):
        self.view_application("nonmodal")

        self.assertTrue(self.closed)
        self.assertFalse(self.event_loop_timeout)
        self.assertTrue(self.result)

    def test_livemodal_view_application_close(self):
        self.view_application("livemodal")

        self.assertTrue(self.closed)
        self.assertFalse(self.event_loop_timeout)
        self.assertFalse(self.result)

    def test_live_view_application_close(self):
        self.view_application("live")

        self.assertTrue(self.closed)
        self.assertFalse(self.event_loop_timeout)
        self.assertTrue(self.result)

    def test_modal_view_application_ok(self):
        self.view_application("modal", button="OK")

        self.assertTrue(self.closed)
        self.assertFalse(self.event_loop_timeout)
        self.assertTrue(self.result)

    def test_nonmodal_view_application_ok(self):
        self.view_application("nonmodal", button="OK")

        self.assertTrue(self.closed)
        self.assertFalse(self.event_loop_timeout)
        self.assertTrue(self.result)

    def test_livemodal_view_application_ok(self):
        self.view_application("livemodal", button="OK")

        self.assertTrue(self.closed)
        self.assertFalse(self.event_loop_timeout)
        self.assertTrue(self.result)

    def test_live_view_application_ok(self):
        self.view_application("live", button="OK")

        self.assertTrue(self.closed)
        self.assertFalse(self.event_loop_timeout)
        self.assertTrue(self.result)

    def test_modal_view_application_cancel(self):
        self.view_application("modal", button="Cancel")

        self.assertTrue(self.closed)
        self.assertFalse(self.event_loop_timeout)
        self.assertFalse(self.result)

    def test_nonmodal_view_application_cancel(self):
        self.view_application("nonmodal", button="Cancel")

        self.assertTrue(self.closed)
        self.assertFalse(self.event_loop_timeout)
        self.assertFalse(self.result)

    def test_livemodal_view_application_cancel(self):
        self.view_application("livemodal", button="Cancel")

        self.assertTrue(self.closed)
        self.assertFalse(self.event_loop_timeout)
        self.assertFalse(self.result)

    def test_live_view_application_cancel(self):
        self.view_application("live", button="Cancel")

        self.assertTrue(self.closed)
        self.assertFalse(self.event_loop_timeout)
        self.assertFalse(self.result)

    def test_modal_view_application_eventloop_close(self):
        self.view_application_event_loop("modal")

        self.assertTrue(self.closed)
        self.assertFalse(self.event_loop_timeout)

    def test_nonmodal_view_application_eventloop_close(self):
        self.view_application_event_loop("nonmodal")

        self.assertTrue(self.closed)
        self.assertFalse(self.event_loop_timeout)

    def test_livemodal_view_application_eventloop_close(self):
        self.view_application_event_loop("livemodal")

        self.assertTrue(self.closed)
        self.assertFalse(self.event_loop_timeout)

    def test_live_view_application_eventloop_close(self):
        self.view_application_event_loop("live")

        self.assertTrue(self.closed)
        self.assertFalse(self.event_loop_timeout)
