Add tests for FileWidget ContextMenu (#923)

* Fix Copy Display Properties
danielhrisca · Oct 10, 2023 · 7671fad · 7671fad
1 parent c14390e
commit 7671fad
Show file tree

Hide file tree

Showing 3 changed files with 435 additions and 7 deletions.
diff --git a/src/asammdf/gui/widgets/tree.py b/src/asammdf/gui/widgets/tree.py
@@ -773,12 +773,15 @@ def keyPressEvent(self, event):
             self.plot.plot.update()
 
         elif modifiers == (QtCore.Qt.ControlModifier | QtCore.Qt.ShiftModifier) and key == QtCore.Qt.Key_C:
-            selected_items = [item for item in self.selectedItems() if item.type() == ChannelsTreeItem.Channel]
-            if not selected_items:
-                return
-            else:
+            selected_items = [
+                item
+                for item in self.selectedItems()
+                if item.type() in (ChannelsTreeItem.Channel, ChannelsTreeItem.Group)
+            ]
+            if selected_items:
                 item = selected_items[0]
-            QtWidgets.QApplication.instance().clipboard().setText(item.get_display_properties())
+                clipboard_text = item.get_display_properties()
+                QtWidgets.QApplication.instance().clipboard().setText(clipboard_text)
 
         elif modifiers == (QtCore.Qt.ControlModifier | QtCore.Qt.ShiftModifier) and key == QtCore.Qt.Key_V:
             info = QtWidgets.QApplication.instance().clipboard().text()
@@ -1930,6 +1933,10 @@ def get_color_using_ranges(self, value, pen=False):
         return get_color_using_ranges(value, self.get_ranges(), self.signal.color, pen=pen)
 
     def get_display_properties(self):
+        if self.type() == ChannelsTreeItem.Group:
+            if self.childCount():
+                return self.child(0).get_display_properties()
+            return ""
         info = {
             "color": self.color.name(),
             "precision": self.precision,

diff --git a/test/asammdf/gui/widgets/test_BasePlotWidget.py b/test/asammdf/gui/widgets/test_BasePlotWidget.py
@@ -63,6 +63,7 @@ def add_channel_to_plot(self, plot=None, channel_name=None, channel_index=None):
         return plot_channel
 
     def context_menu(self, action_text, position=None):
+        self.processEvents()
         with mock.patch("asammdf.gui.widgets.tree.QtWidgets.QMenu", wraps=QMenuWrap):
             mo_action = mock.MagicMock()
             mo_action.text.return_value = action_text