test_oweditdomain: Add test for context menu

Author: ales-erjavec

Orange/widgets/data/tests/test_oweditdomain.py

@@ -12,7 +12,7 @@
 
 from AnyQt.QtCore import QItemSelectionModel, Qt, QItemSelection
 from AnyQt.QtWidgets import QAction, QComboBox, QLineEdit, \
-    QStyleOptionViewItem, QDialog
+    QStyleOptionViewItem, QDialog, QMenu
 from AnyQt.QtTest import QTest, QSignalSpy
 
 from Orange.widgets.utils import colorpalettes
@@ -38,6 +38,7 @@
     GroupItemsDialog)
 from Orange.widgets.data.owcolor import OWColor, ColorRole
 from Orange.widgets.tests.base import WidgetTest, GuiTest
+from Orange.widgets.tests.utils import contextMenu
 from Orange.tests import test_filename, assert_array_nanequal
 from Orange.widgets.utils.state_summary import format_summary_details
 
@@ -544,6 +545,23 @@ def test_discrete_editor_rename_selected_items_action(self):
             list(spy), [[]], 'variable_changed should emit exactly once'
         )
 
+    def test_discrete_editor_context_menu(self):
+        w = DiscreteVariableEditor()
+        v = Categorical("C", ("a", "b", "c"),
+                        (("A", "1"), ("B", "b")), False)
+        w.set_data_categorical(v, [])
+        view = w.values_edit
+        model = view.model()
+
+        pos = view.visualRect(model.index(0, 0)).center()
+        with patch.object(QMenu, "setVisible", return_value=None) as m:
+            contextMenu(view.viewport(), pos)
+            m.assert_called()
+
+        menu = view.findChild(QMenu)
+        self.assertIsNotNone(menu)
+        menu.close()
+
     def test_time_editor(self):
         w = TimeVariableEditor()
         self.assertEqual(w.get_data(), (None, []))

Orange/widgets/tests/utils.py

@@ -6,8 +6,8 @@
 
 from AnyQt.QtCore import Qt, QObject, QEventLoop, QTimer, QLocale, QPoint
 from AnyQt.QtTest import QTest
-from AnyQt.QtGui import QMouseEvent
-from AnyQt.QtWidgets import QApplication
+from AnyQt.QtGui import QMouseEvent, QContextMenuEvent
+from AnyQt.QtWidgets import QApplication, QWidget
 
 from Orange.data import Table, Domain, ContinuousVariable
 
@@ -323,6 +323,25 @@ def mouseMove(widget, pos=QPoint(), delay=-1):  # pragma: no-cover
     QApplication.sendEvent(widget, me)
 
 
+def contextMenu(
+        widget: QWidget, pos=QPoint(), reason=QContextMenuEvent.Mouse,
+        modifiers=Qt.NoModifier, delay=-1
+) -> None:
+    """
+    Simulate a context menu event on `widget`.
+
+    `pos` is the event origin specified in widget's local coordinates. If not
+    specified. Then widget.rect().center() is used instead.
+    """
+    if pos.isNull():
+        pos = widget.rect().center()
+    globalPos = widget.mapToGlobal(pos)
+    ev = QContextMenuEvent(reason, pos, globalPos, modifiers)
+    if delay >= 0:
+        QTest.qWait(delay)
+    QApplication.sendEvent(widget, ev)
+
+
 def table_dense_sparse(test_case):
     # type: (Callable) -> Callable
     """Run a single test case on both dense and sparse Orange tables.