mirror of
https://github.com/jopohl/urh.git
synced 2026-03-07 00:36:47 +01:00
211 lines
9.3 KiB
Python
211 lines
9.3 KiB
Python
import copy
|
|
|
|
from PyQt5.QtCore import QPoint, Qt
|
|
from PyQt5.QtTest import QTest
|
|
from PyQt5.QtWidgets import QApplication
|
|
|
|
from tests.QtTestCase import QtTestCase
|
|
|
|
|
|
class TestAnalysisTabGUI(QtTestCase):
|
|
def setUp(self):
|
|
super().setUp()
|
|
self.cfc = self.form.compare_frame_controller
|
|
self.add_signal_to_form("two_participants.complex")
|
|
QApplication.instance().processEvents()
|
|
self.signal = self.form.signal_tab_controller.signal_frames[0].signal
|
|
self.signal.noise_threshold = 0.0175
|
|
self.signal.qad_center = 0
|
|
self.signal.bit_len = 100
|
|
self.signal.tolerance = 5
|
|
|
|
|
|
def test_analyze_button_fsk(self):
|
|
self.add_signal_to_form("fsk.complex")
|
|
self.cfc.ui.btnAnalyze.click()
|
|
self.assertTrue(True)
|
|
|
|
def test_analyze_button_enocean(self):
|
|
self.add_signal_to_form("enocean.complex")
|
|
w = self.form.signal_tab_controller.signal_frames[1].ui.spinBoxCenterOffset
|
|
w.setValue(0)
|
|
QTest.keyClick(w, Qt.Key_Enter)
|
|
w = self.form.signal_tab_controller.signal_frames[1].ui.spinBoxNoiseTreshold
|
|
w.setValue(0.0111)
|
|
QTest.keyClick(w, Qt.Key_Enter)
|
|
self.cfc.ui.btnAnalyze.click()
|
|
self.assertTrue(True)
|
|
|
|
def test_table_selection(self):
|
|
self.form.ui.tabWidget.setCurrentIndex(1)
|
|
self.cfc.ui.cbProtoView.setCurrentIndex(0)
|
|
self.cfc.ui.tblViewProtocol.selectRow(1)
|
|
QApplication.instance().processEvents()
|
|
self.assertEqual(self.cfc.ui.lBitsSelection.text(), self.cfc.proto_analyzer.messages[1].plain_bits_str)
|
|
|
|
self.cfc.ui.tblViewProtocol.clearSelection()
|
|
QApplication.instance().processEvents()
|
|
self.assertEqual("", self.cfc.ui.lBitsSelection.text())
|
|
|
|
self.cfc.ui.tblViewProtocol.select(0, 0, 0, 3)
|
|
QApplication.instance().processEvents()
|
|
self.assertEqual("1010", self.cfc.ui.lBitsSelection.text())
|
|
self.cfc.ui.cbProtoView.setCurrentIndex(1)
|
|
QApplication.instance().processEvents()
|
|
|
|
min_row, max_row, start, end = self.cfc.ui.tblViewProtocol.selection_range()
|
|
self.assertEqual(min_row, 0)
|
|
self.assertEqual(max_row, 0)
|
|
self.assertEqual(start, 0)
|
|
self.assertEqual(end, 1)
|
|
|
|
def test_search(self):
|
|
search_str = "100110001"
|
|
self.cfc.ui.cbProtoView.setCurrentIndex(0)
|
|
self.cfc.ui.tblViewProtocol.clearSelection()
|
|
self.cfc.ui.lineEditSearch.setText(search_str)
|
|
self.cfc.ui.btnSearchSelectFilter.click()
|
|
|
|
selected_now = self.cfc.ui.tblViewProtocol.selectedIndexes()
|
|
self.assertEqual(len(self.cfc.ui.tblViewProtocol.selectedIndexes()), len(search_str))
|
|
|
|
self.cfc.ui.btnNextSearch.click()
|
|
self.assertNotEqual(selected_now, self.cfc.ui.tblViewProtocol.selectedIndexes())
|
|
|
|
self.cfc.ui.btnPrevSearch.click()
|
|
self.assertEqual(selected_now, self.cfc.ui.tblViewProtocol.selectedIndexes())
|
|
|
|
self.cfc.select_action.trigger()
|
|
self.assertEqual(self.cfc.ui.btnSearchSelectFilter.text(), "Select all")
|
|
self.cfc.ui.btnSearchSelectFilter.click()
|
|
self.assertGreater(len(self.cfc.ui.tblViewProtocol.selectedIndexes()), len(selected_now))
|
|
|
|
self.cfc.filter_action.trigger()
|
|
self.assertEqual(self.cfc.ui.btnSearchSelectFilter.text(), "Filter")
|
|
self.cfc.ui.btnSearchSelectFilter.click()
|
|
hidden_rows = [i for i in range(self.cfc.protocol_model.row_count)
|
|
if self.cfc.ui.tblViewProtocol.isRowHidden(i)]
|
|
|
|
self.assertEqual(hidden_rows, [0, 5, 6, 10, 13, 14, 16, 17])
|
|
|
|
def test_show_diff(self):
|
|
hidden_columns_before = [i for i in range(self.cfc.protocol_model.col_count)
|
|
if self.cfc.ui.tblViewProtocol.isColumnHidden(i)]
|
|
self.assertEqual(len(hidden_columns_before), 0)
|
|
|
|
self.cfc.ui.chkBoxShowOnlyDiffs.click()
|
|
self.assertTrue(self.cfc.ui.cbShowDiffs.isChecked())
|
|
|
|
hidden_columns_now = [i for i in range(self.cfc.protocol_model.col_count)
|
|
if self.cfc.ui.tblViewProtocol.isColumnHidden(i)]
|
|
|
|
self.assertNotEqual(hidden_columns_before, hidden_columns_now)
|
|
|
|
self.cfc.ui.chkBoxOnlyShowLabelsInProtocol.click()
|
|
|
|
hidden_columns_now = [i for i in range(self.cfc.protocol_model.col_count)
|
|
if self.cfc.ui.tblViewProtocol.isColumnHidden(i)]
|
|
|
|
self.assertEqual(len(hidden_columns_now), self.cfc.protocol_model.col_count)
|
|
|
|
def test_add_message_type(self):
|
|
self.assertEqual(len(self.cfc.proto_analyzer.message_types), 1)
|
|
self.cfc.ui.btnAddMessagetype.click()
|
|
self.assertEqual(len(self.cfc.proto_analyzer.message_types), 2)
|
|
self.cfc.ui.btnRemoveMessagetype.click()
|
|
self.assertEqual(len(self.cfc.proto_analyzer.message_types), 1)
|
|
|
|
def test_create_context_menu(self):
|
|
# Add protocol label should be disabled if table is empty
|
|
self.cfc.proto_tree_model.rootItem.child(0).show = False
|
|
QApplication.instance().processEvents()
|
|
|
|
self.assertEqual(self.cfc.protocol_model.rowCount(), 0)
|
|
self.cfc.ui.tblViewProtocol.context_menu_pos = QPoint(0, 0)
|
|
QApplication.instance().processEvents()
|
|
|
|
menu = self.cfc.ui.tblViewProtocol.create_context_menu()
|
|
|
|
create_label_action = next(a for a in menu.actions() if a.text() == "Add protocol label")
|
|
|
|
self.assertFalse(create_label_action.isEnabled())
|
|
|
|
def test_show_in_interpretation(self):
|
|
self.form.ui.tabWidget.setCurrentIndex(1)
|
|
self.assertEqual(self.form.ui.tabWidget.currentIndex(), 1)
|
|
self.cfc.ui.cbProtoView.setCurrentIndex(0)
|
|
|
|
self.cfc.ui.tblViewProtocol.selectRow(1)
|
|
min_row, max_row, start, end = self.cfc.ui.tblViewProtocol.selection_range()
|
|
self.cfc.ui.tblViewProtocol.show_interpretation_clicked.emit(min_row, start, max_row, end - 1)
|
|
self.assertEqual(self.form.ui.tabWidget.currentIndex(), 0)
|
|
self.assertFalse(self.form.signal_tab_controller.signal_frames[0].ui.gvSignal.selection_area.is_empty)
|
|
|
|
def test_hide_row(self):
|
|
num_messages = len(self.cfc.proto_analyzer.messages)
|
|
self.form.ui.tabWidget.setCurrentIndex(1)
|
|
self.assertGreater(num_messages, 0)
|
|
self.assertEqual(self.cfc.protocol_model.rowCount(), num_messages)
|
|
self.cfc.ui.tblViewProtocol.hide_row(0)
|
|
self.assertTrue(self.cfc.ui.tblViewProtocol.isRowHidden(0))
|
|
self.assertEqual(len(self.cfc.protocol_model.hidden_rows), 1)
|
|
|
|
for msg in range(1, num_messages):
|
|
self.assertFalse(self.cfc.ui.tblViewProtocol.isRowHidden(msg))
|
|
|
|
self.form.ui.tabWidget.setCurrentIndex(2)
|
|
QApplication.instance().processEvents()
|
|
self.form.ui.tabWidget.setCurrentIndex(1)
|
|
QApplication.instance().processEvents()
|
|
self.assertEqual(self.cfc.protocol_model.rowCount(), num_messages)
|
|
self.assertTrue(self.cfc.ui.tblViewProtocol.isRowHidden(0))
|
|
|
|
for msg in range(1, num_messages):
|
|
self.assertFalse(self.cfc.ui.tblViewProtocol.isRowHidden(msg))
|
|
|
|
self.assertEqual(len(self.cfc.protocol_model.hidden_rows), 1)
|
|
|
|
def test_refresh_existing_decodings(self):
|
|
self.assertEqual(self.cfc.proto_analyzer.messages[0].decoder, self.cfc.decodings[0])
|
|
decoder = copy.deepcopy(self.cfc.proto_analyzer.messages[0].decoder)
|
|
decoder.chain.append(decoder.code_invert)
|
|
self.cfc.proto_analyzer.messages[0].decoder = decoder
|
|
self.assertNotEqual(self.cfc.proto_analyzer.messages[0].decoder, self.cfc.decodings[0])
|
|
|
|
self.cfc.refresh_existing_encodings()
|
|
self.assertEqual(self.cfc.proto_analyzer.messages[0].decoder, self.cfc.decodings[0])
|
|
|
|
def test_get_labels_from_selection(self):
|
|
self.cfc.ui.tblViewProtocol.selectRow(1)
|
|
self.assertEqual(len(self.cfc.get_labels_from_selection(*self.cfc.ui.tblViewProtocol.selection_range())), 0)
|
|
|
|
def test_refresh_field_types_for_labels(self):
|
|
self.cfc.add_protocol_label(0, 10, 0, 0, edit_label_name=False)
|
|
n = len(self.cfc.field_types)
|
|
self.cfc.refresh_field_types_for_labels()
|
|
self.assertEqual(len(self.cfc.field_types), n)
|
|
|
|
def test_tree_view_selection_changed(self):
|
|
self.cfc.proto_tree_model.addGroup()
|
|
self.cfc.proto_tree_model.addGroup()
|
|
QApplication.instance().processEvents()
|
|
self.assertEqual(len(self.cfc.active_group_ids), 1)
|
|
self.cfc.ui.treeViewProtocols.selectAll()
|
|
self.cfc.ui.treeViewProtocols.selection_changed.emit()
|
|
QApplication.instance().processEvents()
|
|
self.assertEqual(len(self.cfc.active_group_ids), 1)
|
|
|
|
def test_label_selection_changed(self):
|
|
self.assertEqual(self.cfc.ui.tblViewProtocol.horizontalScrollBar().value(), 0)
|
|
self.cfc.add_protocol_label(40, 60, 2, 0, edit_label_name=False)
|
|
self.assertEqual(self.cfc.protocol_label_list_model.rowCount(), 1)
|
|
self.cfc.ui.listViewLabelNames.selectAll()
|
|
self.assertEqual(len(self.cfc.ui.listViewLabelNames.selectedIndexes()), 1)
|
|
self.assertGreater(self.cfc.ui.tblViewProtocol.horizontalScrollBar().value(), 0)
|
|
|
|
def test_remove_label(self):
|
|
self.cfc.add_protocol_label(10, 20, 2, 0, edit_label_name=False)
|
|
self.assertEqual(self.cfc.protocol_label_list_model.rowCount(), 1)
|
|
self.cfc.protocol_label_list_model.delete_label_at(0)
|
|
self.assertEqual(self.cfc.protocol_label_list_model.rowCount(), 0)
|