def on_edit_labels_clicked(self):
view_type = self.ui.cbViewType.currentIndex()
protocol_label_dialog = ProtocolLabelDialog(preselected_index=0,
message=self.ui.tblViewMessage.selected_message,
viewtype=view_type, parent=self)
protocol_label_dialog.finished.connect(self.on_protocol_label_dialog_finished)
protocol_label_dialog.showMaximized()
def on_edit_labels_clicked(self):
view_type = self.ui.cbViewType.currentIndex()
protocol_label_dialog = ProtocolLabelDialog(preselected_index=0,
message=self.ui.tblViewMessage.selected_message,
viewtype=view_type, parent=self)
protocol_label_dialog.finished.connect(self.on_protocol_label_dialog_finished)
protocol_label_dialog.showMaximized()
def on_edit_label_triggered(self, label_index: int):
view_type = self.ui.cbViewType.currentIndex()
protocol_label_dialog = ProtocolLabelDialog(
message=self.ui.tblViewMessage.selected_message,
viewtype=view_type,
selected_index=label_index,
parent=self)
protocol_label_dialog.finished.connect(
self.on_protocol_label_dialog_finished)
protocol_label_dialog.showMaximized()
def setUp(self):
super().setUp()
self.form.add_protocol_file(get_path_for_data_file("protocol.proto.xml"))
self.cframe = self.form.compare_frame_controller
self.cframe.add_protocol_label(9, 19, 0, 0, edit_label_name=False) # equals 10-20 in view
self.cframe.add_protocol_label(39, 54, 1, 0, edit_label_name=False) # equals 40-55 in view
self.assertEqual(len(self.cframe.proto_analyzer.protocol_labels), 2)
self.dialog = ProtocolLabelDialog(message=self.cframe.proto_analyzer.messages[0],
viewtype=0, parent=self.cframe)
if self.SHOW:
self.dialog.show()
def setUp(self):
super().setUp()
QApplication.instance().processEvents()
QTest.qWait(self.WAIT_TIMEOUT_BEFORE_NEW)
self.form.add_protocol_file(get_path_for_data_file("protocol.proto.xml"))
self.cframe = self.form.compare_frame_controller
self.cframe.add_protocol_label(9, 19, 0, 0, edit_label_name=False) # equals 10-20 in view
self.cframe.add_protocol_label(39, 54, 1, 0, edit_label_name=False) # equals 40-55 in view
self.assertEqual(len(self.cframe.proto_analyzer.protocol_labels), 2)
self.dialog = ProtocolLabelDialog(preselected_index=1,
message=self.cframe.proto_analyzer.messages[0],
viewtype=0, parent=self.cframe)
if self.SHOW:
self.dialog.show()
def test_crc_widget_in_protocol_label_dialog(self):
mt = MessageType("test")
mt.append(
ChecksumLabel("test_crc", 8, 16, 0,
FieldType("test_crc", FieldType.Function.CHECKSUM)))
self.dialog = ProtocolLabelDialog(0, Message([0] * 100, 0, mt), 0)
self.assertEqual(self.dialog.ui.tabWidgetAdvancedSettings.count(), 1)
self.assertEqual(self.dialog.ui.tabWidgetAdvancedSettings.tabText(0),
"test_crc")
def on_edit_label_triggered(self, label_index: int):
view_type = self.ui.cbViewType.currentIndex()
protocol_label_dialog = ProtocolLabelDialog(message=self.ui.tblViewMessage.selected_message,
viewtype=view_type, selected_index=label_index, parent=self)
protocol_label_dialog.finished.connect(self.on_protocol_label_dialog_finished)
protocol_label_dialog.showMaximized()
class TestProtocolLabelDialog(QtTestCase):
def setUp(self):
super().setUp()
QApplication.instance().processEvents()
QTest.qWait(self.WAIT_TIMEOUT_BEFORE_NEW)
self.form.add_protocol_file(get_path_for_data_file("protocol.proto.xml"))
self.cframe = self.form.compare_frame_controller
self.cframe.add_protocol_label(9, 19, 0, 0, edit_label_name=False) # equals 10-20 in view
self.cframe.add_protocol_label(39, 54, 1, 0, edit_label_name=False) # equals 40-55 in view
self.assertEqual(len(self.cframe.proto_analyzer.protocol_labels), 2)
self.dialog = ProtocolLabelDialog(preselected_index=1,
message=self.cframe.proto_analyzer.messages[0],
viewtype=0, parent=self.cframe)
if self.SHOW:
self.dialog.show()
def test_protocol_label_dialog(self):
self.assertIn(self.cframe.proto_analyzer.default_message_type.name, self.dialog.windowTitle())
table_model = self.dialog.ui.tblViewProtoLabels.model()
self.assertEqual(table_model.rowCount(), 2)
label = table_model.message_type[0]
table_model.setData(table_model.index(0, 0), "testname")
self.assertEqual(label.name, "testname")
table_model.setData(table_model.index(0, 1), 15)
self.assertEqual(label.start, 15 - 1)
self.dialog.ui.tblViewProtoLabels.openPersistentEditor(table_model.index(0, 1))
table_model.setData(table_model.index(0, 2), 30)
self.assertEqual(label.end, 30)
self.dialog.ui.tblViewProtoLabels.openPersistentEditor(table_model.index(0, 2))
table_model.setData(table_model.index(0, 3), 4)
self.assertEqual(label.color_index, 4)
self.dialog.ui.tblViewProtoLabels.openPersistentEditor(table_model.index(0, 3))
table_model.setData(table_model.index(0, 4), False)
self.assertEqual(label.apply_decoding, False)
self.dialog.ui.tblViewProtoLabels.openPersistentEditor(table_model.index(0, 4))
def test_change_view_type(self):
table_model = self.dialog.ui.tblViewProtoLabels.model()
# Bit View
self.assertEqual(table_model.data(table_model.index(0, 1)), 10)
self.assertEqual(table_model.data(table_model.index(0, 2)), 20)
self.assertEqual(table_model.data(table_model.index(1, 1)), 40)
self.assertEqual(table_model.data(table_model.index(1, 2)), 55)
self.dialog.ui.cbProtoView.setCurrentIndex(1)
self.assertEqual(table_model.data(table_model.index(0, 1)), 4)
self.assertEqual(table_model.data(table_model.index(0, 2)), 6)
self.assertEqual(table_model.data(table_model.index(1, 1)), 12)
self.assertEqual(table_model.data(table_model.index(1, 2)), 15)
label = table_model.message_type[0]
table_model.setData(table_model.index(0, 1), 2)
table_model.setData(table_model.index(0, 2), 5)
self.assertEqual(label.start, 4)
self.assertEqual(label.end, 17)
def test_remove_labels(self):
self.dialog.ui.tblViewProtoLabels.selectAll()
self.assertEqual(self.dialog.ui.tblViewProtoLabels.model().rowCount(), 2)
remove_action = self.dialog.ui.tblViewProtoLabels.create_context_menu().actions()[0]
remove_action.trigger()
self.assertEqual(self.dialog.ui.tblViewProtoLabels.model().rowCount(), 0)
self.assertEqual(len(self.dialog.ui.tblViewProtoLabels.create_context_menu().actions()), 0)