class QtTestCase(unittest.TestCase):
CLOSE_TIMEOUT = 10
WAIT_TIMEOUT_BEFORE_NEW = 10
SHOW = os.path.exists(os.path.join(os.path.dirname(os.path.realpath(__file__)), "show_gui"))
@classmethod
def setUpClass(cls):
write_settings()
cls.app = QApplication(sys.argv)
@classmethod
def tearDownClass(cls):
cls.app.quit()
sip.delete(cls.app)
cls.app = None
def setUp(self):
self.form = MainController()
if self.SHOW:
self.form.show()
def add_signal_to_form(self, filename: str):
QApplication.instance().processEvents()
QTest.qWait(self.WAIT_TIMEOUT_BEFORE_NEW)
self.form.add_signalfile(get_path_for_data_file(filename))
def tearDown(self):
if hasattr(self, "dialog"):
self.dialog.close()
if hasattr(self, "form"):
self.form.close_all()
if self.SHOW:
self.form.close()
QApplication.instance().processEvents()
QTest.qWait(self.CLOSE_TIMEOUT)
class QtTestCase(unittest.TestCase):
CLOSE_TIMEOUT = 10
WAIT_TIMEOUT_BEFORE_NEW = 10
SHOW = os.path.exists(
os.path.join(os.path.dirname(os.path.realpath(__file__)), "show_gui"))
@classmethod
def setUpClass(cls):
write_settings()
cls.app = QApplication(sys.argv)
@classmethod
def tearDownClass(cls):
cls.app.quit()
sip.delete(cls.app)
cls.app = None
def setUp(self):
self.form = MainController()
if self.SHOW:
self.form.show()
def wait_before_new_file(self):
QApplication.instance().processEvents()
QTest.qWait(self.WAIT_TIMEOUT_BEFORE_NEW)
def add_signal_to_form(self, filename: str):
self.wait_before_new_file()
self.form.add_signalfile(get_path_for_data_file(filename))
def get_path_for_filename(self, filename) -> str:
return get_path_for_data_file(filename)
def add_signal_to_generator(self, signal_index: int):
gframe = self.form.generator_tab_controller
item = gframe.tree_model.rootItem.children[0].children[signal_index]
index = gframe.tree_model.createIndex(signal_index, 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))
def tearDown(self):
if hasattr(self, "dialog"):
self.dialog.close()
if hasattr(self, "form"):
self.form.close_all()
if self.SHOW:
self.form.close()
QApplication.instance().processEvents()
QTest.qWait(self.CLOSE_TIMEOUT)
class TestDecodingGUI(unittest.TestCase):
def setUp(self):
short_wait()
self.form = MainController()
short_wait()
self.form.add_signalfile(get_path_for_data_file("esaver.complex"))
self.signal = self.form.signal_tab_controller.signal_frames[0].signal
self.dialog = DecoderWidgetController(decodings=self.form.compare_frame_controller.decodings,
signals=[self.signal], parent=self.form,
project_manager=self.form.project_manager)
def tearDown(self):
self.dialog.close()
self.dialog.setParent(None)
short_wait()
self.form.close()
short_wait(interval=10)
def test_edit_decoding(self):
self.dialog.ui.combobox_decodings.setCurrentIndex(1) # NRZI
self.assertEqual(self.dialog.ui.decoderchain.count(), 1) # One Invert
self.dialog.save_to_file()
def test_build_decoding(self):
self.dialog.ui.combobox_decodings.setCurrentIndex(4)
chain = [constants.DECODING_INVERT, constants.DECODING_ENOCEAN, constants.DECODING_DIFFERENTIAL,
constants.DECODING_REDUNDANCY,
constants.DECODING_CARRIER, constants.DECODING_BITORDER, constants.DECODING_EDGE,
constants.DECODING_DATAWHITENING,
constants.DECODING_SUBSTITUTION, constants.DECODING_EXTERNAL, constants.DECODING_CUT]
decoding = Encoder(chain=chain)
self.dialog.decodings[4] = decoding
self.dialog.set_e()
self.assertEqual(len(chain), self.dialog.ui.decoderchain.count())
for i in range(0, self.dialog.ui.decoderchain.count()):
self.dialog.ui.decoderchain.setCurrentRow(i)
self.dialog.set_information(2)
self.assertIn(chain[i], self.dialog.ui.info.text())
def test_set_signal(self):
self.dialog.ui.combobox_signals.currentIndexChanged.emit(0)
self.assertEqual(self.dialog.ui.inpt.text(), "10010110")
def test_select_items(self):
for i in range(0, self.dialog.ui.basefunctions.count()):
self.dialog.ui.basefunctions.setCurrentRow(i)
self.assertIn(self.dialog.ui.basefunctions.currentItem().text(), self.dialog.ui.info.text())
for i in range(0, self.dialog.ui.additionalfunctions.count()):
self.dialog.ui.additionalfunctions.setCurrentRow(i)
self.assertIn(self.dialog.ui.additionalfunctions.currentItem().text(), self.dialog.ui.info.text())
class QtTestCase(unittest.TestCase):
CLOSE_TIMEOUT = 10
WAIT_TIMEOUT_BEFORE_NEW = 10
SHOW = os.path.exists(os.path.join(os.path.dirname(os.path.realpath(__file__)), "show_gui"))
@classmethod
def setUpClass(cls):
write_settings()
cls.app = QApplication(sys.argv)
@classmethod
def tearDownClass(cls):
cls.app.quit()
sip.delete(cls.app)
cls.app = None
def setUp(self):
self.form = MainController()
if self.SHOW:
self.form.show()
def wait_before_new_file(self):
QApplication.instance().processEvents()
QTest.qWait(self.WAIT_TIMEOUT_BEFORE_NEW)
def add_signal_to_form(self, filename: str):
self.wait_before_new_file()
self.form.add_signalfile(get_path_for_data_file(filename))
def add_signal_to_generator(self, signal_index: int):
gframe = self.form.generator_tab_controller
item = gframe.tree_model.rootItem.children[0].children[signal_index]
index = gframe.tree_model.createIndex(signal_index, 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))
def tearDown(self):
if hasattr(self, "dialog"):
self.dialog.close()
if hasattr(self, "form"):
self.form.close_all()
if self.SHOW:
self.form.close()
QApplication.instance().processEvents()
QTest.qWait(self.CLOSE_TIMEOUT)
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 QtTestCase(unittest.TestCase):
CLOSE_TIMEOUT = 10
WAIT_TIMEOUT_BEFORE_NEW = 10
SHOW = os.path.exists(
os.path.join(os.path.dirname(os.path.realpath(__file__)), "show_gui"))
@classmethod
def setUpClass(cls):
import multiprocessing as mp
try:
mp.set_start_method("spawn")
except RuntimeError:
pass
assert mp.get_start_method() == "spawn"
write_settings()
cls.app = QApplication([cls.__name__])
@classmethod
def tearDownClass(cls):
cls.app.quit()
cls.app = None
QTest.qWait(10)
time.sleep(0.1)
def setUp(self):
ProtocolSniffer.BUFFER_SIZE_MB = 0.5
self.form = MainController()
self.form.ui.actionAuto_detect_new_signals.setChecked(False)
if self.SHOW:
self.form.show()
def tearDown(self):
if hasattr(self, "dialog"):
self.dialog.close()
try:
sip.delete(self.dialog)
except TypeError:
pass
self.dialog = None
if hasattr(self, "form"):
self.form.close_all_files()
self.form.close()
try:
sip.delete(self.form)
except TypeError:
pass
self.form = None
gc.collect()
def wait_before_new_file(self):
QApplication.instance().processEvents()
QTest.qWait(self.WAIT_TIMEOUT_BEFORE_NEW)
def add_signal_to_form(self, filename: str):
self.wait_before_new_file()
self.form.add_signalfile(get_path_for_data_file(filename))
def get_path_for_filename(self, filename) -> str:
return get_path_for_data_file(filename)
def add_signal_to_generator(self, signal_index: int):
gframe = self.form.generator_tab_controller
item = gframe.tree_model.rootItem.children[0].children[signal_index]
index = gframe.tree_model.createIndex(signal_index, 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))
def add_all_signals_to_simulator(self):
assert isinstance(self.form, MainController)
sim_frame = self.form.simulator_tab_controller
sim_frame.ui.treeProtocols.selectAll()
self.assertGreater(len(sim_frame.ui.treeProtocols.selectedIndexes()),
0)
mimedata = sim_frame.tree_model.mimeData(
sim_frame.ui.treeProtocols.selectedIndexes())
drop_event = QDropEvent(sim_frame.ui.gvSimulator.rect().center(),
Qt.CopyAction | Qt.MoveAction, mimedata,
Qt.LeftButton, Qt.NoModifier)
drop_event.acceptProposedAction()
sim_frame.ui.gvSimulator.dropEvent(drop_event)
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 TestSendRecvDialog(unittest.TestCase):
def setUp(self):
constants.SETTINGS.setValue("NetworkSDRInterface", True)
QTest.qWait(50)
logger.debug("init form")
self.form = MainController()
logger.debug("initalized form")
app.processEvents()
QTest.qWait(75)
logger.debug("Adding signalfile")
self.form.add_signalfile(get_path_for_data_file("esaver.complex"))
logger.debug("Added signalfile")
app.processEvents()
self.signal = self.form.signal_tab_controller.signal_frames[0].signal
self.gframe = self.form.generator_tab_controller
self.form.ui.tabWidget.setCurrentIndex(2)
app.processEvents()
QTest.qWait(100)
project_manager = self.form.project_manager
logger.debug("Creating Receive Dialog")
self.receive_dialog = ReceiveDialogController(project_manager,
testing_mode=True,
parent=self.form)
app.processEvents()
QTest.qWait(50)
logger.debug("Creating Send Dialog")
self.send_dialog = SendDialogController(
project_manager,
modulated_data=self.signal.data,
testing_mode=True,
parent=self.form)
self.send_dialog.graphics_view.show_full_scene(reinitialize=True)
app.processEvents()
QTest.qWait(50)
logger.debug("Creating Spectrum Dialog")
self.spectrum_dialog = SpectrumDialogController(project_manager,
testing_mode=True,
parent=self.form)
app.processEvents()
QTest.qWait(50)
logger.debug("Creating Sniff Dialog")
self.sniff_dialog = ProtocolSniffDialogController(
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)
app.processEvents()
QTest.qWait(50)
logger.debug("Created Dialogs")
self.dialogs = [
self.receive_dialog, self.send_dialog, self.spectrum_dialog,
self.sniff_dialog
]
def tearDown(self):
for dialog in self.dialogs:
dialog.close()
dialog.setParent(None)
dialog.deleteLater()
app.processEvents()
QTest.qWait(10)
app.processEvents()
QTest.qWait(10)
self.form.close()
self.form.setParent(None)
app.processEvents()
QTest.qWait(10)
def test_network_sdr_enabled(self):
for dialog in self.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)
def test_receive(self):
self.receive_dialog.ui.cbDevice.setCurrentText(
NetworkSDRInterfacePlugin.NETWORK_SDR_NAME)
self.receive_dialog.device.set_server_port(2222)
self.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(500)
self.assertEqual(self.receive_dialog.device.current_index, 3)
self.assertTrue(
np.array_equal(self.receive_dialog.device.data[:3], data))
self.receive_dialog.ui.btnStop.click()
self.receive_dialog.ui.btnClear.click()
self.assertEqual(self.receive_dialog.device.current_index, 0)
def test_send(self):
self.receive_dialog.ui.cbDevice.setCurrentText(
NetworkSDRInterfacePlugin.NETWORK_SDR_NAME)
self.receive_dialog.device.set_server_port(3333)
self.receive_dialog.ui.btnStart.click()
self.send_dialog.ui.cbDevice.setCurrentText(
NetworkSDRInterfacePlugin.NETWORK_SDR_NAME)
self.send_dialog.device.set_client_port(3333)
self.send_dialog.ui.spinBoxNRepeat.setValue(2)
self.send_dialog.ui.btnStart.click()
app.processEvents()
QTest.qWait(500)
self.assertEqual(self.receive_dialog.device.current_index,
2 * self.signal.num_samples)
self.assertTrue(
np.array_equal(
self.receive_dialog.device.
data[:self.receive_dialog.device.current_index // 2],
self.signal.data))
self.assertEqual(self.send_dialog.send_indicator.rect().width(),
self.signal.num_samples)
self.assertFalse(self.send_dialog.ui.btnClear.isEnabled())
self.send_dialog.on_clear_clicked()
self.assertEqual(self.send_dialog.send_indicator.rect().width(), 0)
def test_sniff(self):
# 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)
self.sniff_dialog.ui.cbDevice.setCurrentText(
NetworkSDRInterfacePlugin.NETWORK_SDR_NAME)
self.assertEqual(self.sniff_dialog.device.name,
NetworkSDRInterfacePlugin.NETWORK_SDR_NAME)
self.sniff_dialog.device.set_server_port(4444)
gframe.network_sdr_plugin.client_port = 4444
self.sniff_dialog.ui.btnStart.click()
app.processEvents()
gframe.ui.btnNetworkSDRSend.click()
app.processEvents()
QTest.qWait(500)
received_msgs = self.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)
self.sniff_dialog.ui.btnStop.click()
target_file = os.path.join(QDir.tempPath(), "sniff_file")
self.assertFalse(os.path.isfile(target_file))
self.sniff_dialog.ui.btnClear.click()
app.processEvents()
self.sniff_dialog.ui.lineEdit_sniff_OutputFile.setText(target_file)
self.sniff_dialog.ui.btnStart.click()
app.processEvents()
self.assertFalse(self.sniff_dialog.ui.btnAccept.isEnabled())
gframe.ui.btnNetworkSDRSend.click()
app.processEvents()
QTest.qWait(500)
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):
app.processEvents()
self.assertEqual(self.send_dialog.graphics_view.sceneRect().width(),
self.signal.num_samples)
view_width = self.send_dialog.graphics_view.view_rect().width()
self.send_dialog.graphics_view.zoom(1.1)
app.processEvents()
self.assertLess(self.send_dialog.graphics_view.view_rect().width(),
view_width)
self.send_dialog.graphics_view.zoom(0.8)
app.processEvents()
self.assertLessEqual(
self.send_dialog.graphics_view.view_rect().width(), view_width)
def test_send_dialog_delete(self):
num_samples = self.signal.num_samples
self.assertEqual(num_samples,
self.send_dialog.scene_manager.signal.num_samples)
self.assertEqual(num_samples,
len(self.send_dialog.device.samples_to_send))
self.send_dialog.graphics_view.set_selection_area(0, 1337)
self.send_dialog.graphics_view.delete_action.trigger()
self.assertEqual(self.send_dialog.scene_manager.signal.num_samples,
num_samples - 1337)
self.assertEqual(len(self.send_dialog.device.samples_to_send),
num_samples - 1337)
def test_send_dialog_y_slider(self):
y, h = self.send_dialog.graphics_view.view_rect().y(
), self.send_dialog.graphics_view.view_rect().height()
self.send_dialog.ui.sliderYscale.setValue(
self.send_dialog.ui.sliderYscale.value() +
self.send_dialog.ui.sliderYscale.singleStep())
self.assertNotEqual(y, self.send_dialog.graphics_view.view_rect().y())
self.assertNotEqual(
h,
self.send_dialog.graphics_view.view_rect().height())
def test_change_device_parameters(self):
for dialog in self.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 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 QtTestCase(unittest.TestCase):
CLOSE_TIMEOUT = 10
WAIT_TIMEOUT_BEFORE_NEW = 10
SHOW = os.path.exists(os.path.join(os.path.dirname(os.path.realpath(__file__)), "show_gui"))
@classmethod
def setUpClass(cls):
import multiprocessing as mp
try:
mp.set_start_method("spawn")
except RuntimeError:
pass
assert mp.get_start_method() == "spawn"
write_settings()
cls.app = QApplication([cls.__name__])
@classmethod
def tearDownClass(cls):
cls.app.quit()
cls.app = None
QTest.qWait(10)
time.sleep(0.1)
def setUp(self):
ProtocolSniffer.BUFFER_SIZE_MB = 0.5
self.form = MainController()
if self.SHOW:
self.form.show()
def tearDown(self):
if hasattr(self, "dialog"):
self.dialog.close()
sip.delete(self.dialog)
self.dialog = None
if hasattr(self, "form"):
self.form.close_all_files()
self.form.close()
sip.delete(self.form)
self.form = None
gc.collect()
def wait_before_new_file(self):
QApplication.instance().processEvents()
QTest.qWait(self.WAIT_TIMEOUT_BEFORE_NEW)
def add_signal_to_form(self, filename: str):
self.wait_before_new_file()
self.form.add_signalfile(get_path_for_data_file(filename))
def get_path_for_filename(self, filename) -> str:
return get_path_for_data_file(filename)
def add_signal_to_generator(self, signal_index: int):
gframe = self.form.generator_tab_controller
item = gframe.tree_model.rootItem.children[0].children[signal_index]
index = gframe.tree_model.createIndex(signal_index, 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))
def add_all_signals_to_simulator(self):
assert isinstance(self.form, MainController)
sim_frame = self.form.simulator_tab_controller
sim_frame.ui.treeProtocols.selectAll()
self.assertGreater(len(sim_frame.ui.treeProtocols.selectedIndexes()), 0)
mimedata = sim_frame.tree_model.mimeData(sim_frame.ui.treeProtocols.selectedIndexes())
drop_event = QDropEvent(sim_frame.ui.gvSimulator.rect().center(), Qt.CopyAction | Qt.MoveAction,
mimedata, Qt.LeftButton, Qt.NoModifier)
drop_event.acceptProposedAction()
sim_frame.ui.gvSimulator.dropEvent(drop_event)
def get_free_port(self):
import socket
s = socket.socket()
s.bind(("", 0))
port = s.getsockname()[1]
s.close()
return port
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())
