class TestMaincontrollerGUI(unittest.TestCase): def setUp(self): tests.utils_testing.short_wait() self.form = MainController() tests.utils_testing.short_wait() def tearDown(self): self.form.close() self.form.setParent(None) self.form.deleteLater() tests.utils_testing.short_wait() def test_open_recent(self): self.form.add_signalfile( tests.utils_testing.get_path_for_data_file("esaver.complex")) tests.utils_testing.short_wait() self.assertEqual(len(self.form.signal_tab_controller.signal_frames), 1) self.form.close_all() tests.utils_testing.short_wait() self.assertEqual(len(self.form.signal_tab_controller.signal_frames), 0) self.form.recentFileActionList[0].trigger() self.assertEqual(len(self.form.signal_tab_controller.signal_frames), 1) def test_update_decodings(self): self.form.update_decodings() self.assertTrue(True) def test_options_changed(self): self.form.add_signalfile( tests.utils_testing.get_path_for_data_file("esaver.complex")) self.form.on_options_changed({ "rel_symbol_length": 0, "show_pause_as_time": True, "default_view": 0 }) self.assertEqual( self.form.signal_tab_controller.signal_frames[0].ui.cbProtoView. currentIndex(), 0)
class TestProjectManager(unittest.TestCase): def setUp(self): logger.debug("Init form") self.form = MainController() logger.debug("Initialized form") app.processEvents() QTest.qWait(25) self.form.project_manager.set_project_folder(get_path_for_data_file(""), ask_for_new_project=False) self.cframe = self.form.compare_frame_controller self.gframe = self.form.generator_tab_controller self.dialog = ProjectDialogController(project_manager=self.form.project_manager, parent=self.form) def tearDown(self): self.dialog.close() self.dialog.setParent(None) self.dialog.deleteLater() app.processEvents() QTest.qWait(10) self.form.close() self.form.deleteLater() app.processEvents() QTest.qWait(10) def test_save_modulations(self): self.gframe.modulators[0].name = "Test" amplitude = random.random() self.gframe.modulators[0].carrier_amplitude = amplitude self.gframe.modulators[0].carrier_freq_hz = 1337 self.gframe.modulators[0].carrier_phase_deg = 42 self.gframe.modulators[0].modulation_type = 1 self.gframe.modulators[0].sample_rate = 10 ** 3 self.gframe.modulators.append(Modulator("test 2")) self.gframe.modulators = self.gframe.modulators[:2] # Take only the first two self.form.project_manager.saveProject() loaded_mods = self.form.project_manager.read_modulators_from_project_file() self.assertEqual(len(loaded_mods), 2) self.assertEqual(loaded_mods[0].name, "Test") self.assertEqual(loaded_mods[1].name, "test 2") self.assertEqual(loaded_mods[0].carrier_freq_hz, 1337) self.assertEqual(loaded_mods[0].carrier_phase_deg, 42) self.assertEqual(loaded_mods[0].modulation_type, 1) self.assertEqual(loaded_mods[0].sample_rate, 10 ** 3) self.gframe.modulators.clear() self.assertEqual(len(self.gframe.modulators), 0) self.form.project_manager.set_project_folder(self.form.project_manager.project_path) self.assertEqual(len(self.gframe.modulators), 2) def test_close_all(self): self.form.close_all() app.processEvents() QTest.qWait(10) self.assertEqual(self.form.signal_tab_controller.num_signals, 0) self.form.add_signalfile(get_path_for_data_file("ask.complex")) self.form.add_signalfile(get_path_for_data_file("fsk.complex")) self.assertEqual(self.form.signal_tab_controller.num_signals, 2) self.form.close_all() app.processEvents() QTest.qWait(10) self.assertEqual(self.form.signal_tab_controller.num_signals, 0) self.assertEqual(self.form.project_manager.project_file, None) def test_project_dialog(self): frequency = 1e9 sample_rate = 10e9 bandwidth = 10 gain = 42 descr = "URH rockz." self.dialog.ui.spinBoxFreq.setValue(frequency) self.assertEqual(self.dialog.freq, frequency) self.dialog.ui.spinBoxSampleRate.setValue(sample_rate) self.assertEqual(self.dialog.sample_rate, sample_rate) self.dialog.ui.spinBoxBandwidth.setValue(bandwidth) self.assertEqual(self.dialog.bandwidth, bandwidth) self.dialog.ui.spinBoxGain.setValue(gain) self.assertEqual(self.dialog.gain, gain) self.dialog.ui.txtEdDescription.setPlainText(descr) self.assertEqual(self.dialog.description, descr) self.dialog.ui.lineEditBroadcastAddress.setText("abcd") self.dialog.ui.lineEditBroadcastAddress.textEdited.emit("abcd") self.assertEqual(self.dialog.broadcast_address_hex, "abcd") if len(self.dialog.participants) == 0: self.dialog.ui.btnAddParticipant.click() self.assertEqual(len(self.dialog.participants), 1) model = self.dialog.participant_table_model model.setData(model.index(0, 0), "Testing") model.setData(model.index(0, 1), "T") model.setData(model.index(0, 2), 5) model.setData(model.index(0, 3), 0) model.setData(model.index(0, 4), "aaaa") participant = self.dialog.participants[0] self.assertEqual(participant.name, "Testing") self.assertEqual(participant.shortname, "T") self.assertEqual(participant.color_index, 5) self.assertEqual(participant.relative_rssi, 0) self.assertEqual(participant.address_hex, "aaaa") num_participants = len(self.dialog.participants) self.dialog.ui.btnAddParticipant.click() self.dialog.ui.btnAddParticipant.click() self.dialog.ui.btnAddParticipant.click() self.assertEqual(len(self.dialog.participants), num_participants + 3) self.dialog.ui.btnRemoveParticipant.click() self.dialog.ui.btnRemoveParticipant.click() self.dialog.ui.btnRemoveParticipant.click() self.assertEqual(len(self.dialog.participants), num_participants) test_path = os.path.join(QDir.tempPath(), "urh_test") self.dialog.ui.lineEdit_Path.setText(test_path) self.dialog.ui.lineEdit_Path.textEdited.emit(test_path) self.assertEqual(self.dialog.path, test_path) self.dialog.ui.btnOK.click() self.assertTrue(os.path.isdir(test_path)) self.form.project_manager.from_dialog(self.dialog) self.dialog = ProjectDialogController(project_manager=self.form.project_manager, parent=self.form, new_project=False) self.assertEqual(self.dialog.ui.spinBoxFreq.value(), frequency) self.assertEqual(self.dialog.ui.spinBoxSampleRate.value(), sample_rate) self.assertEqual(self.dialog.ui.spinBoxBandwidth.value(), bandwidth) self.assertEqual(self.dialog.ui.spinBoxGain.value(), gain) self.assertEqual(self.dialog.ui.txtEdDescription.toPlainText(), descr) self.assertFalse(self.dialog.ui.lineEdit_Path.isEnabled())
class TestOptionsGUI(unittest.TestCase): def setUp(self): self.form = MainController() app.processEvents() QTest.qWait(10) self.dialog = OptionsController( self.form.plugin_manager.installed_plugins, parent=self.form) def tearDown(self): self.dialog.close() self.dialog.deleteLater() app.processEvents() QTest.qWait(10) self.form.close() self.form.deleteLater() app.processEvents() QTest.qWait(10) def test_interpretation_tab(self): self.dialog.ui.tabWidget.setCurrentIndex(0) if self.dialog.ui.chkBoxEnableSymbols.isChecked(): self.assertNotEqual(self.dialog.ui.lSymbolLength.text(), "0%") else: self.assertEqual(self.dialog.ui.lSymbolLength.text(), "0%") self.dialog.ui.chkBoxEnableSymbols.click() if self.dialog.ui.chkBoxEnableSymbols.isChecked(): self.assertNotEqual(self.dialog.ui.lSymbolLength.text(), "0%") else: self.assertEqual(self.dialog.ui.lSymbolLength.text(), "0%") self.dialog.ui.chkBoxEnableSymbols.click() if self.dialog.ui.chkBoxEnableSymbols.isChecked(): self.assertNotEqual(self.dialog.ui.lSymbolLength.text(), "0%") else: self.assertEqual(self.dialog.ui.lSymbolLength.text(), "0%") def test_generation_tab(self): self.dialog.ui.tabWidget.setCurrentIndex(0) self.assertEqual( self.dialog.ui.checkBoxDefaultFuzzingPause.isChecked(), self.dialog.ui.doubleSpinBoxFuzzingPause.isEnabled()) self.dialog.ui.checkBoxDefaultFuzzingPause.click() self.assertEqual( self.dialog.ui.checkBoxDefaultFuzzingPause.isChecked(), self.dialog.ui.doubleSpinBoxFuzzingPause.isEnabled()) self.dialog.ui.checkBoxDefaultFuzzingPause.click() self.assertEqual( self.dialog.ui.checkBoxDefaultFuzzingPause.isChecked(), self.dialog.ui.doubleSpinBoxFuzzingPause.isEnabled()) def test_plugins_tab(self): self.dialog.ui.tabWidget.setCurrentIndex(4) self.assertEqual(self.dialog.ui.tabWidget.tabText(4), "Plugins") list_view = self.dialog.plugin_controller.ui.listViewPlugins model = list_view.model() self.assertIsInstance(model, PluginListModel) self.assertEqual(model.rowCount(), len(PluginManager().installed_plugins)) for i in range(model.rowCount()): descr = self.dialog.plugin_controller.ui.txtEditPluginDescription.toPlainText( ) list_view.setCurrentIndex(model.index(i, 0)) self.assertNotEqual( descr, self.dialog.plugin_controller.ui.txtEditPluginDescription. toPlainText()) def test_device_tab(self): self.dialog.ui.tabWidget.setCurrentIndex(5) self.assertEqual(self.dialog.ui.tabWidget.tabText(5), "Device") self.dialog.ui.listWidgetDevices.setCurrentRow(0) dev_name = self.dialog.ui.listWidgetDevices.currentItem().text() for i in range(1, self.dialog.ui.listWidgetDevices.count()): self.dialog.ui.listWidgetDevices.setCurrentRow(i) self.assertNotEqual( dev_name, self.dialog.ui.listWidgetDevices.currentItem().text()) dev_name = self.dialog.ui.listWidgetDevices.currentItem().text() self.dialog.ui.radioButtonPython2Interpreter.click() self.assertTrue(self.dialog.ui.lineEditPython2Interpreter.isEnabled()) self.assertFalse(self.dialog.ui.lineEditGnuradioDirectory.isEnabled()) self.dialog.ui.radioButtonGnuradioDirectory.click() self.assertFalse(self.dialog.ui.lineEditPython2Interpreter.isEnabled()) self.assertTrue(self.dialog.ui.lineEditGnuradioDirectory.isEnabled()) self.dialog.ui.radioButtonPython2Interpreter.click() self.assertFalse( self.dialog.ui.radioButtonGnuradioDirectory.isChecked()) self.assertFalse(self.dialog.ui.lineEditGnuradioDirectory.isEnabled()) self.assertTrue(self.dialog.ui.lineEditPython2Interpreter.isEnabled())
class TestPlugins(unittest.TestCase): def setUp(self): self.old_sym_len = constants.SETTINGS.value('rel_symbol_length', type=int) constants.SETTINGS.setValue('rel_symbol_length', 0) # Disable Symbols for this Test logger.debug("Init form") self.form = MainController() logger.debug("Initalized form") self.form.add_signalfile(get_path_for_data_file("esaver.complex")) self.sframe = self.form.signal_tab_controller.signal_frames[0] self.cframe = self.form.compare_frame_controller self.gframe = self.form.generator_tab_controller self.form.ui.tabWidget.setCurrentIndex(1) self.assertEqual(self.cframe.protocol_model.row_count, 3) self.signal = self.sframe.signal def tearDown(self): self.form.close_all() tests.utils_testing.short_wait() self.form.close() self.form.setParent(None) self.form.deleteLater() tests.utils_testing.short_wait() def test_message_break_plugin(self): bp = MessageBreakPlugin() action = bp.get_action(self.cframe.ui.tblViewProtocol, self.cframe.protocol_undo_stack, (1, 1, 4, 4), self.cframe.proto_analyzer, 0) self.assertEqual(self.cframe.protocol_model.row_count, 3) action.trigger() self.assertEqual(self.cframe.protocol_model.row_count, 4) self.cframe.protocol_undo_stack.undo() self.assertEqual(self.cframe.protocol_model.row_count, 3) def test_zero_hide_plugin_gui(self): self.assertEqual( len(self.cframe.proto_analyzer.decoded_proto_bits_str[0]), 377) zh = ZeroHidePlugin() zh.following_zeros = 188 action = zh.get_action(self.cframe.ui.tblViewProtocol, self.cframe.protocol_undo_stack, (), self.cframe.proto_analyzer, 0) action.trigger() self.assertEqual( len(self.cframe.proto_analyzer.decoded_proto_bits_str[0]), 377 - 188) self.cframe.protocol_undo_stack.undo() self.assertEqual( len(self.cframe.proto_analyzer.decoded_proto_bits_str[0]), 377) def test_zero_hide_plugin_function(self): zh = ZeroHidePlugin() zh.following_zeros = 3 self.form.add_signalfile(get_path_for_data_file("ask.complex")) self.form.ui.tabWidget.setCurrentIndex(1) test_bits = "10110010010110110110110110110110110001000000" self.assertEqual(self.cframe.proto_analyzer.decoded_proto_bits_str[3], test_bits) action = zh.get_action(self.cframe.ui.tblViewProtocol, self.cframe.protocol_undo_stack, (), self.cframe.proto_analyzer, 0) action.trigger() self.assertEqual(self.cframe.proto_analyzer.decoded_proto_bits_str[3], "10110010010110110110110110110110111") def test_sdr_interface_plugin(self): si = NetworkSDRInterfacePlugin() test_bits = [ "10101011111", "1010100011000111110001011001010101010101", "1010100011000111110001011001010100100", "1101010101011000011", "11010101010110000110", "11100010101001110000", "111100000011011101010101010000101010101010100001010011010101010011" ] for bits in test_bits: byte_vals = si.bit_str_to_bytearray(bits) self.assertEqual(len(byte_vals), int(math.ceil(len(bits) / 8)), msg=bits) recalculated = si.bytearray_to_bit_str(byte_vals) if len(bits) % 8 == 0: self.assertEqual(bits, recalculated) elif bits.endswith("1"): self.assertEqual(bits, recalculated.rstrip("0")) else: self.assertTrue(recalculated.startswith(bits)) def test_insert_sine_plugin(self): insert_sine_plugin = self.sframe.ui.gvSignal.insert_sine_plugin num_samples = 10000 tests.utils_testing.short_wait() dialog = insert_sine_plugin.get_insert_sine_dialog( original_data=self.signal.data, position=2000, sample_rate=self.signal.sample_rate, num_samples=num_samples) graphics_view = dialog.graphicsViewSineWave # type: ZoomableGraphicView while not dialog.doubleSpinBoxAmplitude.isEnabled(): tests.utils_testing.short_wait() self.assertEqual(int(graphics_view.sceneRect().width()), self.signal.num_samples + num_samples) self.assertEqual(insert_sine_plugin.insert_indicator.rect().width(), num_samples) self.assertEqual(insert_sine_plugin.insert_indicator.rect().x(), 2000) dialog.doubleSpinBoxAmplitude.setValue(0.1) dialog.doubleSpinBoxAmplitude.editingFinished.emit() self.assertEqual(insert_sine_plugin.amplitude, 0.1) while not dialog.doubleSpinBoxAmplitude.isEnabled(): tests.utils_testing.short_wait() dialog.doubleSpinBoxFrequency.setValue(1e6) dialog.doubleSpinBoxFrequency.editingFinished.emit() self.assertEqual(insert_sine_plugin.frequency, 1e6) while not dialog.doubleSpinBoxAmplitude.isEnabled(): tests.utils_testing.short_wait() dialog.doubleSpinBoxPhase.setValue(100) dialog.doubleSpinBoxPhase.editingFinished.emit() self.assertEqual(insert_sine_plugin.phase, 100) while not dialog.doubleSpinBoxAmplitude.isEnabled(): tests.utils_testing.short_wait() dialog.doubleSpinBoxSampleRate.setValue(2e6) dialog.doubleSpinBoxSampleRate.editingFinished.emit() self.assertEqual(insert_sine_plugin.sample_rate, 2e6) while not dialog.doubleSpinBoxAmplitude.isEnabled(): tests.utils_testing.short_wait() dialog.doubleSpinBoxNSamples.setValue(0.5e6) dialog.doubleSpinBoxNSamples.editingFinished.emit() self.assertEqual(insert_sine_plugin.num_samples, 0.5e6) while not dialog.doubleSpinBoxAmplitude.isEnabled(): tests.utils_testing.short_wait() sep = Formatter.local_decimal_seperator() self.assertEqual(dialog.lineEditTime.text(), "250" + sep + "000m")
class TestModulatorGUI(unittest.TestCase): def setUp(self): logger.debug("Init Form") tests.utils_testing.short_wait(interval=10) self.form = MainController() logger.debug("Add Signal") tests.utils_testing.short_wait() self.form.add_signalfile(get_path_for_data_file("esaver.complex")) logger.debug("Added Signal") self.signal = self.form.signal_tab_controller.signal_frames[0].signal self.gframe = self.form.generator_tab_controller self.form.ui.tabWidget.setCurrentIndex(2) tests.utils_testing.short_wait(interval=10) logger.debug("Create modulation dialog") self.dialog, _ = self.gframe.prepare_modulation_dialog() self.gframe.initialize_modulation_dialog("1111", self.dialog) def tearDown(self): self.dialog.close() self.dialog.setParent(None) self.dialog.deleteLater() tests.utils_testing.short_wait() self.form.close() self.form.setParent(None) self.form.deleteLater() tests.utils_testing.short_wait() def test_add_remove_modulator(self): self.assertEqual(len(self.dialog.modulators), 1) self.dialog.ui.btnAddModulation.click() self.assertEqual(len(self.dialog.modulators), 2) self.dialog.ui.btnAddModulation.click() self.assertEqual(len(self.dialog.modulators), 3) app.processEvents() self.dialog.ui.btnRemoveModulation.click() self.assertEqual(len(self.dialog.modulators), 2) self.dialog.ui.btnRemoveModulation.click() self.assertEqual(len(self.dialog.modulators), 1) self.assertFalse(self.dialog.ui.btnRemoveModulation.isEnabled()) def test_edit_carrier(self): self.dialog.ui.doubleSpinBoxCarrierFreq.setValue(1e9) self.dialog.ui.doubleSpinBoxCarrierFreq.editingFinished.emit() self.assertEqual(self.dialog.current_modulator.carrier_freq_hz, 1e9) self.dialog.ui.doubleSpinBoxCarrierPhase.setValue(100) self.dialog.ui.doubleSpinBoxCarrierPhase.editingFinished.emit() self.assertEqual(self.dialog.current_modulator.carrier_phase_deg, 100) def test_edit_data(self): bits = self.dialog.current_modulator.display_bits self.dialog.ui.linEdDataBits.setText("10101010") self.dialog.ui.linEdDataBits.editingFinished.emit() self.assertEqual(self.dialog.current_modulator.display_bits, "10101010") self.dialog.ui.btnRestoreBits.click() self.dialog.ui.linEdDataBits.editingFinished.emit() self.assertEqual(self.dialog.current_modulator.display_bits, bits) self.dialog.ui.spinBoxBitLength.setValue(1337) self.dialog.ui.spinBoxBitLength.editingFinished.emit() self.assertEqual(self.dialog.current_modulator.samples_per_bit, 1337) self.dialog.ui.spinBoxSampleRate.setValue(5e6) self.dialog.ui.spinBoxSampleRate.editingFinished.emit() self.assertEqual(self.dialog.current_modulator.sample_rate, 5e6) def test_zoom(self): self.dialog.ui.gVModulated.zoom(1.1) self.assertEqual(int(self.dialog.ui.gVModulated.view_rect().width()), int(self.dialog.ui.gVCarrier.view_rect().width())) self.assertEqual(int(self.dialog.ui.gVModulated.view_rect().width()), int(self.dialog.ui.gVData.view_rect().width())) self.dialog.ui.gVModulated.zoom(1.01) self.assertEqual(int(self.dialog.ui.gVModulated.view_rect().width()), int(self.dialog.ui.gVCarrier.view_rect().width())) self.assertEqual(int(self.dialog.ui.gVModulated.view_rect().width()), int(self.dialog.ui.gVData.view_rect().width())) def test_edit_modulation(self): self.dialog.ui.comboBoxModulationType.setCurrentText( "Amplitude Shift Keying (ASK)") self.assertEqual(self.dialog.ui.lParameterfor0.text(), "Amplitude for 0:") self.assertEqual(self.dialog.ui.lParameterfor1.text(), "Amplitude for 1:") self.dialog.ui.comboBoxModulationType.setCurrentText( "Frequency Shift Keying (FSK)") self.assertEqual(self.dialog.ui.lParameterfor0.text(), "Frequency for 0:") self.assertEqual(self.dialog.ui.lParameterfor1.text(), "Frequency for 1:") self.dialog.ui.comboBoxModulationType.setCurrentText( "Gaussian Frequency Shift Keying (GFSK)") self.assertEqual(self.dialog.ui.lParameterfor0.text(), "Frequency for 0:") self.assertEqual(self.dialog.ui.lParameterfor1.text(), "Frequency for 1:") self.dialog.ui.spinBoxGaussBT.setValue(0.5) self.dialog.ui.spinBoxGaussBT.editingFinished.emit() self.assertEqual(self.dialog.current_modulator.gauss_bt, 0.5) self.dialog.ui.spinBoxGaussFilterWidth.setValue(5) self.dialog.ui.spinBoxGaussFilterWidth.editingFinished.emit() self.assertEqual(self.dialog.current_modulator.gauss_filter_width, 5) self.dialog.ui.comboBoxModulationType.setCurrentText( "Phase Shift Keying (PSK)") self.assertEqual(self.dialog.ui.lParameterfor0.text(), "Phase (degree) for 0:") self.assertEqual(self.dialog.ui.lParameterfor1.text(), "Phase (degree) for 1:") self.dialog.ui.comboBoxModulationType.setCurrentText( "Amplitude Shift Keying (ASK)") self.assertEqual(self.dialog.ui.lParameterfor0.text(), "Amplitude for 0:") self.assertEqual(self.dialog.ui.lParameterfor1.text(), "Amplitude for 1:") self.assertEqual(int(self.dialog.ui.lSamplesInViewModulated.text()), int(self.dialog.ui.gVModulated.view_rect().width())) def test_signal_view(self): #self.dialog.show() # if on KDE tree_view = self.dialog.ui.treeViewSignals tree_model = tree_view.model() item = tree_model.rootItem.children[0].children[0] index = tree_model.createIndex(0, 0, item) rect = tree_view.visualRect(index) QTest.mousePress(tree_view.viewport(), Qt.LeftButton, pos=rect.center()) tree_view.selectAll() self.assertEqual(tree_view.selectedIndexes()[0], index) mime_data = tree_model.mimeData(tree_view.selectedIndexes()) drag_drop = QDropEvent(rect.center(), Qt.CopyAction | Qt.MoveAction, mime_data, Qt.LeftButton, Qt.NoModifier) drag_drop.acceptProposedAction() self.dialog.ui.gVOriginalSignal.dropEvent(drag_drop) self.assertEqual(self.dialog.ui.gVOriginalSignal.sceneRect().width(), self.signal.num_samples) self.dialog.ui.cbShowDataBitsOnly.click() self.dialog.ui.chkBoxLockSIV.click() self.assertEqual( int(self.dialog.ui.gVOriginalSignal.view_rect().width()), int(self.dialog.ui.gVModulated.view_rect().width())) freq = self.dialog.ui.doubleSpinBoxCarrierFreq.value() self.dialog.ui.btnAutoDetect.click() self.assertNotEqual(freq, self.dialog.ui.doubleSpinBoxCarrierFreq.value()) self.dialog.ui.comboBoxModulationType.setCurrentText( "Frequency Shift Keying (FSK)") self.dialog.ui.btnAutoDetect.click() self.assertEqual(self.dialog.ui.lCurrentSearchResult.text(), "1") self.dialog.ui.btnSearchNext.click() self.assertEqual(self.dialog.ui.lCurrentSearchResult.text(), "2") self.dialog.ui.btnSearchPrev.click() self.assertEqual(self.dialog.ui.lCurrentSearchResult.text(), "1")
class TestProtocolLabelDialog(unittest.TestCase): def setUp(self): constants.SETTINGS.setValue("align_labels", True) logger.debug("Init form") tests.utils_testing.short_wait() self.form = MainController() tests.utils_testing.short_wait() logger.debug("Add signal") self.form.add_protocol_file(get_path_for_data_file("protocol.proto")) logger.debug("added signal") tests.utils_testing.short_wait() 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 = ProtocolLabelController( preselected_index=1, message=self.cframe.proto_analyzer.messages[0], viewtype=0, parent=self.cframe) def tearDown(self): self.dialog.close() self.dialog.setParent(None) self.dialog.deleteLater() tests.utils_testing.short_wait() self.form.close() self.form.setParent(None) self.form.deleteLater() tests.utils_testing.short_wait() 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) table_model.setData(table_model.index(0, 2), 30) self.assertEqual(label.end, 30) table_model.setData(table_model.index(0, 3), 4) self.assertEqual(label.color_index, 4) table_model.setData(table_model.index(0, 4), False) self.assertEqual(label.apply_decoding, False) 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)
class TestSendRecvDialog(unittest.TestCase): SEND_RECV_TIMEOUT = 300 def setUp(self): constants.SETTINGS.setValue("NetworkSDRInterface", True) tests.utils_testing.short_wait() logger.debug("init form") self.form = MainController() self.signal = Signal(get_path_for_data_file("esaver.complex"), "testsignal") self.form.ui.tabWidget.setCurrentIndex(2) def tearDown(self): self.form.close() self.form.setParent(None) self.form.deleteLater() tests.utils_testing.short_wait(interval=10) def __get_recv_dialog(self): logger.debug("Creating Receive Dialog") tests.utils_testing.short_wait() receive_dialog = ReceiveDialogController(self.form.project_manager, testing_mode=True, parent=self.form) return receive_dialog def __get_send_dialog(self): logger.debug("Creating Send Dialog") tests.utils_testing.short_wait() send_dialog = SendDialogController(self.form.project_manager, modulated_data=self.signal.data, testing_mode=True, parent=self.form) send_dialog.graphics_view.show_full_scene(reinitialize=True) return send_dialog def __get_spectrum_dialog(self): logger.debug("Creating Spectrum Dialog") tests.utils_testing.short_wait() spectrum_dialog = SpectrumDialogController(self.form.project_manager, testing_mode=True, parent=self.form) return spectrum_dialog def __get_sniff_dialog(self): logger.debug("Creating Sniff Dialog") tests.utils_testing.short_wait() sniff_dialog = ProtocolSniffDialogController( self.form.project_manager, self.signal.noise_threshold, self.signal.qad_center, self.signal.bit_len, self.signal.tolerance, self.signal.modulation_type, testing_mode=True, parent=self.form) return sniff_dialog def __get_all_dialogs(self): yield self.__get_recv_dialog() yield self.__get_send_dialog() yield self.__get_spectrum_dialog() yield self.__get_sniff_dialog() def test_network_sdr_enabled(self): for dialog in self.__get_all_dialogs(): items = [ dialog.ui.cbDevice.itemText(i) for i in range(dialog.ui.cbDevice.count()) ] if isinstance(dialog, SpectrumDialogController): self.assertNotIn(NetworkSDRInterfacePlugin.NETWORK_SDR_NAME, items) else: self.assertIn(NetworkSDRInterfacePlugin.NETWORK_SDR_NAME, items) tests.utils_testing.short_wait(interval=10) def test_receive(self): receive_dialog = self.__get_recv_dialog() receive_dialog.ui.cbDevice.setCurrentText( NetworkSDRInterfacePlugin.NETWORK_SDR_NAME) receive_dialog.device.set_server_port(2222) receive_dialog.ui.btnStart.click() data = np.array([complex( 1, 2), complex(3, 4), complex(5, 6)], dtype=np.complex64) sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) sock.connect(("127.0.0.1", 2222)) sock.sendall(data.tostring()) sock.shutdown(socket.SHUT_RDWR) sock.close() app.processEvents() QTest.qWait(self.SEND_RECV_TIMEOUT) self.assertEqual(receive_dialog.device.current_index, 3) self.assertTrue(np.array_equal(receive_dialog.device.data[:3], data)) receive_dialog.ui.btnStop.click() receive_dialog.ui.btnClear.click() self.assertEqual(receive_dialog.device.current_index, 0) def test_send(self): receive_dialog = self.__get_recv_dialog() receive_dialog.ui.cbDevice.setCurrentText( NetworkSDRInterfacePlugin.NETWORK_SDR_NAME) receive_dialog.device.set_server_port(3333) receive_dialog.ui.btnStart.click() send_dialog = self.__get_send_dialog() send_dialog.ui.cbDevice.setCurrentText( NetworkSDRInterfacePlugin.NETWORK_SDR_NAME) send_dialog.device.set_client_port(3333) send_dialog.ui.spinBoxNRepeat.setValue(2) send_dialog.ui.btnStart.click() app.processEvents() QTest.qWait(self.SEND_RECV_TIMEOUT) self.assertEqual(receive_dialog.device.current_index, 2 * self.signal.num_samples) self.assertTrue( np.array_equal( receive_dialog.device. data[:receive_dialog.device.current_index // 2], self.signal.data)) self.assertEqual(send_dialog.send_indicator.rect().width(), self.signal.num_samples) self.assertFalse(send_dialog.ui.btnClear.isEnabled()) send_dialog.on_clear_clicked() self.assertEqual(send_dialog.send_indicator.rect().width(), 0) def test_sniff(self): # add a signal so we can use it tests.utils_testing.short_wait() self.form.add_signalfile(get_path_for_data_file("esaver.complex")) logger.debug("Added signalfile") app.processEvents() # Move with encoding to generator gframe = self.form.generator_tab_controller gframe.ui.cbViewType.setCurrentIndex(0) item = gframe.tree_model.rootItem.children[0].children[0] index = gframe.tree_model.createIndex(0, 0, item) rect = gframe.ui.treeProtocols.visualRect(index) QTest.mousePress(gframe.ui.treeProtocols.viewport(), Qt.LeftButton, pos=rect.center()) self.assertEqual(gframe.ui.treeProtocols.selectedIndexes()[0], index) mimedata = gframe.tree_model.mimeData( gframe.ui.treeProtocols.selectedIndexes()) gframe.table_model.dropMimeData(mimedata, 1, -1, -1, gframe.table_model.createIndex(0, 0)) self.assertEqual(gframe.table_model.rowCount(), 3) tests.utils_testing.short_wait() sniff_dialog = self.__get_sniff_dialog() sniff_dialog.ui.cbDevice.setCurrentText( NetworkSDRInterfacePlugin.NETWORK_SDR_NAME) self.assertEqual(sniff_dialog.device.name, NetworkSDRInterfacePlugin.NETWORK_SDR_NAME) sniff_dialog.device.set_server_port(4444) gframe.network_sdr_plugin.client_port = 4444 sniff_dialog.ui.btnStart.click() app.processEvents() gframe.ui.btnNetworkSDRSend.click() app.processEvents() QTest.qWait(self.SEND_RECV_TIMEOUT) received_msgs = sniff_dialog.ui.txtEd_sniff_Preview.toPlainText( ).split("\n") orig_msgs = gframe.table_model.protocol.plain_bits_str self.assertEqual(len(received_msgs), len(orig_msgs)) for received, orig in zip(received_msgs, orig_msgs): pad = 0 if len(orig) % 8 == 0 else 8 - len(orig) % 8 self.assertEqual(received, orig + "0" * pad) sniff_dialog.ui.btnStop.click() target_file = os.path.join(QDir.tempPath(), "sniff_file") self.assertFalse(os.path.isfile(target_file)) sniff_dialog.ui.btnClear.click() app.processEvents() sniff_dialog.ui.lineEdit_sniff_OutputFile.setText(target_file) sniff_dialog.ui.btnStart.click() app.processEvents() self.assertFalse(sniff_dialog.ui.btnAccept.isEnabled()) gframe.ui.btnNetworkSDRSend.click() app.processEvents() QTest.qWait(self.SEND_RECV_TIMEOUT) with open(target_file, "r") as f: for i, line in enumerate(f): pad = 0 if len( orig_msgs[i]) % 8 == 0 else 8 - len(orig_msgs[i]) % 8 self.assertEqual(line.strip(), orig_msgs[i] + "0" * pad) os.remove(target_file) def test_send_dialog_scene_zoom(self): send_dialog = self.__get_send_dialog() app.processEvents() self.assertEqual(send_dialog.graphics_view.sceneRect().width(), self.signal.num_samples) view_width = send_dialog.graphics_view.view_rect().width() send_dialog.graphics_view.zoom(1.1) app.processEvents() self.assertLess(send_dialog.graphics_view.view_rect().width(), view_width) send_dialog.graphics_view.zoom(0.8) app.processEvents() self.assertLessEqual(send_dialog.graphics_view.view_rect().width(), view_width) def test_send_dialog_delete(self): num_samples = self.signal.num_samples send_dialog = self.__get_send_dialog() self.assertEqual(num_samples, send_dialog.scene_manager.signal.num_samples) self.assertEqual(num_samples, len(send_dialog.device.samples_to_send)) send_dialog.graphics_view.set_selection_area(0, 1337) send_dialog.graphics_view.delete_action.trigger() self.assertEqual(send_dialog.scene_manager.signal.num_samples, num_samples - 1337) self.assertEqual(len(send_dialog.device.samples_to_send), num_samples - 1337) def test_send_dialog_y_slider(self): send_dialog = self.__get_send_dialog() app.processEvents() y, h = send_dialog.graphics_view.view_rect().y( ), send_dialog.graphics_view.view_rect().height() send_dialog.ui.sliderYscale.setValue( send_dialog.ui.sliderYscale.value() + send_dialog.ui.sliderYscale.singleStep()) self.assertNotEqual(y, send_dialog.graphics_view.view_rect().y()) self.assertNotEqual(h, send_dialog.graphics_view.view_rect().height()) def test_change_device_parameters(self): for dialog in self.__get_all_dialogs(): dialog.ui.cbDevice.setCurrentText("HackRF") self.assertEqual(dialog.device.name, "HackRF", msg=type(dialog)) dialog.ui.cbDevice.setCurrentText("USRP") self.assertEqual(dialog.device.name, "USRP", msg=type(dialog)) dialog.ui.lineEditIP.setText("1.3.3.7") dialog.ui.lineEditIP.editingFinished.emit() self.assertEqual(dialog.device.ip, "1.3.3.7", msg=type(dialog)) dialog.ui.spinBoxFreq.setValue(2e9) dialog.ui.spinBoxFreq.editingFinished.emit() self.assertEqual(dialog.ui.spinBoxFreq.text()[-1], "G") self.assertEqual(dialog.device.frequency, 2e9) dialog.ui.spinBoxSampleRate.setValue(10e6) dialog.ui.spinBoxSampleRate.editingFinished.emit() self.assertEqual(dialog.ui.spinBoxSampleRate.text()[-1], "M") self.assertEqual(dialog.device.sample_rate, 10e6) dialog.ui.spinBoxBandwidth.setValue(1e3) dialog.ui.spinBoxBandwidth.editingFinished.emit() self.assertEqual(dialog.ui.spinBoxBandwidth.text()[-1], "K") self.assertEqual(dialog.device.bandwidth, 1e3) dialog.ui.spinBoxGain.setValue(5) dialog.ui.spinBoxGain.editingFinished.emit() self.assertEqual(dialog.device.gain, 5) dialog.ui.spinBoxNRepeat.setValue(10) dialog.ui.spinBoxNRepeat.editingFinished.emit() if isinstance(dialog, SendDialogController): self.assertEqual(dialog.device.num_sending_repeats, 10) else: self.assertEqual(dialog.device.num_sending_repeats, None) app.processEvents()
class TestSignalTabGUI(unittest.TestCase): def setUp(self): constants.SETTINGS.setValue("not_show_save_dialog", True) logger.debug("Init Form") tests.utils_testing.short_wait() self.form = MainController() tests.utils_testing.short_wait() logger.debug("Add Signal") self.form.add_signalfile(get_path_for_data_file("esaver.complex")) logger.debug("Added Signal") app.processEvents() self.frame = self.form.signal_tab_controller.signal_frames[0] self.frame.signal.noise_threshold = 0.0023 self.frame.signal.qad_center = 0.3817 self.frame.signal.bit_len = 84 def tearDown(self): self.form.close_all() tests.utils_testing.short_wait() self.form.close() self.form.setParent(None) self.form.deleteLater() tests.utils_testing.short_wait() def test_close_all(self): logger.debug("Close all") self.form.close_all() tests.utils_testing.short_wait() logger.debug("Called close all") app.processEvents() self.assertEqual(self.form.signal_tab_controller.num_signals, 0) # Add a bunch of signals num_signals = 5 for _ in range(num_signals): tests.utils_testing.short_wait() self.form.add_signalfile(get_path_for_data_file("esaver.complex")) self.assertEqual(self.form.signal_tab_controller.num_signals, num_signals) self.form.close_all() tests.utils_testing.short_wait() self.form.add_signalfile(get_path_for_data_file("ask.complex")) self.assertEqual(self.form.signal_tab_controller.num_signals, 1) def test_zoom(self): tests.utils_testing.short_wait() x_zoom = self.frame.ui.spinBoxXZoom.value() self.assertEqual(x_zoom, 100) for _ in range(10): self.frame.ui.gvSignal.zoom(1.01) self.assertGreater(self.frame.ui.spinBoxXZoom.value(), x_zoom) x_zoom = self.frame.ui.spinBoxXZoom.value() for _ in range(10): self.frame.ui.gvSignal.zoom(0.99) self.assertLess(self.frame.ui.spinBoxXZoom.value(), x_zoom) x_zoom = self.frame.ui.spinBoxXZoom.value() samples_in_view = self.frame.ui.lSamplesInView.text() self.frame.ui.spinBoxXZoom.setValue( self.frame.ui.spinBoxXZoom.value() + 400) self.assertNotEqual(samples_in_view, self.frame.ui.lSamplesInView.text()) def test_load_proto(self): self.form.add_files([get_path_for_data_file("protocol.proto")]) self.assertEqual( self.form.signal_tab_controller.signal_frames[1].ui.lSignalTyp. text(), "Protocol (*.proto)") def test_graphic_view_selection(self): self.frame.ui.gvSignal.selection_area.start = 0 self.frame.ui.gvSignal.selection_area.end = 4000 self.frame.ui.gvSignal.sel_area_start_end_changed.emit(0, 4000) self.assertEqual(self.frame.ui.lNumSelectedSamples.text(), "4000") noise_val = self.frame.ui.spinBoxNoiseTreshold.value() self.frame.ui.gvSignal.set_noise_clicked.emit() self.assertNotEqual(noise_val, self.frame.ui.spinBoxNoiseTreshold.value()) self.frame.ui.spinBoxSelectionStart.setValue(300) self.assertEqual(self.frame.ui.gvSignal.selection_area.start, 300) self.frame.ui.spinBoxSelectionEnd.setValue(6000) self.assertEqual(self.frame.ui.gvSignal.selection_area.end, 6000) def test_graphic_view_zoom_to_selection(self): self.frame.ui.gvSignal.context_menu_position = QPoint(0, 0) menu = self.frame.ui.gvSignal.create_context_menu() self.assertTrue(self.frame.ui.gvSignal.selection_area.is_empty) self.assertIsNone( next((action for action in menu.actions() if action.text() == "Zoom selection"), None)) self.frame.ui.gvSignal.selection_area.start = 1337 self.frame.ui.gvSignal.selection_area.end = 4711 self.frame.ui.gvSignal.sel_area_start_end_changed.emit(1337, 4711) menu = self.frame.ui.gvSignal.create_context_menu() self.assertFalse(self.frame.ui.gvSignal.selection_area.is_empty) zoom_action = next(action for action in menu.actions() if action.text() == "Zoom selection") zoom_action.trigger() self.assertEqual(self.frame.ui.spinBoxSelectionStart.value(), 1337) self.assertEqual(self.frame.ui.spinBoxSelectionEnd.value(), 4711) def test_show_hide_start_end(self): self.assertEqual(self.frame.ui.btnShowHideStartEnd.text(), "+") self.frame.ui.btnShowHideStartEnd.click() self.assertEqual(self.frame.ui.btnShowHideStartEnd.text(), "-") def test_apply_to_all(self): logger.debug("Test apply to all") tests.utils_testing.short_wait() self.form.add_signalfile(get_path_for_data_file("ask.complex")) logger.debug("added new signal") frame2 = self.form.signal_tab_controller.signal_frames[1] self.frame.ui.spinBoxInfoLen.setValue(42) self.frame.ui.spinBoxInfoLen.editingFinished.emit() self.frame.ui.spinBoxCenterOffset.setValue(0.1) self.frame.ui.spinBoxCenterOffset.editingFinished.emit() self.frame.ui.spinBoxNoiseTreshold.setValue(0.5) self.frame.ui.spinBoxNoiseTreshold.editingFinished.emit() self.frame.ui.spinBoxTolerance.setValue(10) self.frame.ui.spinBoxTolerance.editingFinished.emit() self.frame.apply_to_all_clicked.emit(self.frame.signal) self.assertEqual(42, frame2.ui.spinBoxInfoLen.value()) self.assertEqual(0.1, frame2.ui.spinBoxCenterOffset.value()) self.assertEqual(0.5, frame2.ui.spinBoxNoiseTreshold.value()) self.assertEqual(10, frame2.ui.spinBoxTolerance.value()) def test_save_all(self): logger.debug("Test save all") tests.utils_testing.short_wait() self.form.add_signalfile(get_path_for_data_file("ask.complex")) frame2 = self.form.signal_tab_controller.signal_frames[1] self.frame.signal.changed = True self.frame.signal.filename = os.path.join(QDir.tempPath(), "sig1.complex") frame2.signal.changed = True frame2.signal.filename = os.path.join(QDir.tempPath(), "sig2.complex") self.assertFalse(os.path.isfile(self.frame.signal.filename)) self.assertFalse(os.path.isfile(frame2.signal.filename)) self.form.signal_tab_controller.save_all() self.assertTrue(os.path.isfile(self.frame.signal.filename)) self.assertTrue(os.path.isfile(frame2.signal.filename)) os.remove(self.frame.signal.filename) os.remove(frame2.signal.filename) def test_crop_and_save_signal(self): self.frame.ui.gvSignal.selection_area.end = 4000 self.frame.ui.gvSignal.selection_area.start = 1000 self.assertEqual(self.frame.ui.gvSignal.selection_area.end, 4000) self.assertEqual(self.frame.ui.gvSignal.selection_area.width, 3000) self.frame.ui.gvSignal.sel_area_start_end_changed.emit(1000, 4000) self.frame.ui.gvSignal.on_crop_action_triggered() self.assertEqual(self.frame.signal.num_samples, 3000) self.assertTrue(self.frame.signal.changed) self.frame.signal.filename = os.path.join(QDir.tempPath(), "sig.complex") if os.path.isfile(self.frame.signal.filename): os.remove(self.frame.signal.filename) self.assertFalse(os.path.isfile(self.frame.signal.filename)) self.frame.ui.btnSaveSignal.click() self.form.close_signal_frame(self.frame) tests.utils_testing.short_wait() self.form.add_signalfile(os.path.join(QDir.tempPath(), "sig.complex")) self.assertEqual( self.form.signal_tab_controller.signal_frames[0].signal. num_samples, 3000) os.remove(os.path.join(QDir.tempPath(), "sig.complex")) def test_selection_sync(self): self.frame.ui.gvSignal.selection_area.end = 128440 self.frame.ui.gvSignal.selection_area.start = 89383 self.frame.ui.gvSignal.sel_area_start_end_changed.emit(89383, 128440) tests.utils_testing.short_wait() self.assertEqual(self.frame.proto_analyzer.messages[0].plain_bits_str, self.frame.ui.txtEdProto.selected_text) self.frame.ui.txtEdProto.show_proto_clicked.emit() tests.utils_testing.short_wait() self.assertAlmostEqual( (128440 - 89383) / 1000000, (self.frame.ui.gvSignal.view_rect().width()) / 1000000, places=1) def test_show_demod_view(self): self.assertTrue(self.frame.ui.gvLegend.isHidden()) self.frame.ui.cbSignalView.setCurrentIndex(1) self.assertFalse(self.frame.ui.gvLegend.isHidden()) def test_auto_detect_button(self): self.assertTrue(self.frame.ui.btnAutoDetect.isChecked()) self.frame.ui.btnAutoDetect.click() self.assertFalse(self.frame.ui.btnAutoDetect.isChecked())