ComboBoxSearch: Add tests

Author: ales-erjavec

Orange/widgets/utils/tests/test_combobox.py

@@ -0,0 +1,146 @@
+# pylint: disable=all
+
+import unittest
+
+from AnyQt.QtCore import Qt, QPoint, QRect
+from AnyQt.QtGui import QMouseEvent
+from AnyQt.QtWidgets import QListView, QApplication
+from AnyQt.QtTest import QTest, QSignalSpy
+from Orange.widgets.tests.base import GuiTest
+
+from Orange.widgets.utils import combobox
+
+
+class TestComboBoxSearch(GuiTest):
+    def setUp(self):
+        super().setUp()
+        cb = combobox.ComboBoxSearch()
+        cb.addItem("One")
+        cb.addItem("Two")
+        cb.addItem("Three")
+        cb.insertSeparator(cb.count())
+        cb.addItem("Four")
+        self.cb = cb
+
+    def tearDown(self):
+        super().tearDown()
+        self.cb.deleteLater()
+        self.cb = None
+
+    def test_combobox(self):
+        cb = self.cb
+        cb.grab()
+        cb.showPopup()
+        popup = cb.findChild(QListView)  # type: QListView
+        # run through paint code for coverage
+        popup.grab()
+        cb.grab()
+
+        model = popup.model()
+        self.assertEqual(model.rowCount(), cb.count())
+        QTest.keyClick(popup, Qt.Key_E)
+        self.assertEqual(model.rowCount(), 2)
+        QTest.keyClick(popup, Qt.Key_Backspace)
+        self.assertEqual(model.rowCount(), cb.count())
+        QTest.keyClick(popup, Qt.Key_F)
+        self.assertEqual(model.rowCount(), 1)
+        popup.setCurrentIndex(model.index(0, 0))
+        spy = QSignalSpy(cb.activated[int])
+        QTest.keyClick(popup, Qt.Key_Enter)
+
+        self.assertEqual(spy[0], [4])
+        self.assertEqual(cb.currentIndex(), 4)
+        self.assertEqual(cb.currentText(), "Four")
+        self.assertFalse(popup.isVisible())
+
+    def test_combobox_navigation(self):
+        cb = self.cb
+        cb.setCurrentIndex(4)
+        self.assertTrue(cb.currentText(), "Four")
+        cb.showPopup()
+        popup = cb.findChild(QListView)  # type: QListView
+        self.assertEqual(popup.currentIndex().row(), 4)
+
+        QTest.keyClick(popup, Qt.Key_Up)
+        self.assertEqual(popup.currentIndex().row(), 2)
+        QTest.keyClick(popup, Qt.Key_Escape)
+        self.assertFalse(popup.isVisible())
+        self.assertEqual(cb.currentIndex(), 4)
+        cb.hidePopup()
+
+    def test_click(self):
+        cb = self.cb
+        cb.showPopup()
+        popup = cb.findChild(QListView)  # type: QListView
+        model = popup.model()
+        rect = popup.visualRect(model.index(1, 0))
+        if hasattr(Qt, "WA_DontShowOnScreen"):
+            popup.window().setAttribute(Qt.WA_DontShowOnScreen, True)
+        spy = QSignalSpy(cb.activated[int])
+        QTest.mouseClick(popup.viewport(), Qt.LeftButton, Qt.NoModifier,
+                         rect.center(), QApplication.doubleClickInterval() + 10)
+
+        self.assertEqual(spy[0], [1])
+        self.assertEqual(cb.currentIndex(), 1)
+        self.assertEqual(cb.currentText(), "Two")
+
+    def test_focus_out(self):
+        cb = self.cb
+        cb.showPopup()
+        popup = cb.findChild(QListView)
+        # Activate some other window to simulate focus out
+        w = QListView()
+        w.show()
+        w.activateWindow()
+        w.hide()
+        self.assertFalse(popup.isVisible())
+
+    def test_track(self):
+        cb = self.cb
+        cb.setStyleSheet("combobox-list-mousetracking: 1")
+        cb.showPopup()
+        popup = cb.findChild(QListView)  # type: QListView
+        model = popup.model()
+        rect = popup.visualRect(model.index(2, 0))
+        mouseMove(popup.viewport(), rect.center())
+        self.assertEqual(popup.currentIndex().row(), 2)
+        cb.hidePopup()
+
+    def test_empty(self):
+        cb = self.cb
+        cb.clear()
+        cb.showPopup()
+        popup = cb.findChild(QListView)  # type: QListView
+        self.assertIsNone(popup)
+
+    def test_popup_util(self):
+        geom = QRect(10, 10, 100, 400)
+        screen = QRect(0, 0, 600, 600)
+        g1 = combobox.dropdown_popup_geometry(
+            geom, QRect(200, 100, 100, 20), screen
+        )
+        self.assertEqual(g1, QRect(200, 120, 100, 400))
+        g2 = combobox.dropdown_popup_geometry(
+            geom, QRect(-10, 0, 100, 20), screen
+        )
+        self.assertEqual(g2, QRect(0, 20, 100, 400))
+        g3 = combobox.dropdown_popup_geometry(
+            geom, QRect(590, 0, 100, 20), screen
+        )
+        self.assertEqual(g3, QRect(600 - 100, 20, 100, 400))
+        g4 = combobox.dropdown_popup_geometry(
+            geom, QRect(0, 500, 100, 20), screen
+        )
+        self.assertEqual(g4, QRect(0, 500 - 400, 100, 400))
+
+
+def mouseMove(widget, pos=QPoint(), delay=-1):  # pragma: no-cover
+    # Like QTest.mouseMove, but functional without QCursor.setPos
+    if pos.isNull():
+        pos = widget.rect().center()
+    me = QMouseEvent(QMouseEvent.MouseMove, pos, widget.mapToGlobal(pos),
+                     Qt.NoButton, Qt.MouseButtons(0), Qt.NoModifier)
+    if delay > 0:
+        QTest.qWait(delay)
+
+    QApplication.sendEvent(widget, me)