ComboBoxSearch: Add tests

Author: ales-erjavec

Orange/widgets/utils/tests/test_combobox.py

@@ -0,0 +1,114 @@
+# 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())
+
+        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))
+
+        spy = QSignalSpy(cb.activated[int])
+        QTest.mouseClick(popup.viewport(), Qt.LeftButton, Qt.NoModifier,
+                         rect.center(), QApplication.doubleClickInterval())
+
+        self.assertEqual(spy[0], [1])
+        self.assertEqual(cb.currentIndex(), 1)
+        self.assertEqual(cb.currentText(), "Two")
+
+    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))
+        # QTest.mouseMove(popup.viewport(), rect.center())
+        mouseMove(popup.viewport(), rect.center())
+        self.assertEqual(popup.currentIndex().row(), 2)
+
+    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):
+    # Like QTest.mouseMove, but functional  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)