mirror of
https://github.com/jopohl/urh.git
synced 2026-03-04 23:44:02 +01:00
164 lines
7.0 KiB
Python
164 lines
7.0 KiB
Python
from array import array
|
|
|
|
from tests.QtTestCase import QtTestCase
|
|
from urh.controller.dialogs.SimulatorDialog import SimulatorDialog
|
|
from urh.dev.BackendHandler import BackendContainer, Backends
|
|
from urh.signalprocessing.Participant import Participant
|
|
from urh.simulator.SimulatorMessage import SimulatorMessage
|
|
|
|
|
|
class TestSimulatorDialog(QtTestCase):
|
|
def setUp(self):
|
|
super().setUp()
|
|
alice = Participant("Alice", "A")
|
|
bob = Participant("Bob", "B")
|
|
alice.simulate = True
|
|
bob.simulate = True
|
|
self.form.project_manager.participants.append(alice)
|
|
self.form.project_manager.participants.append(bob)
|
|
self.form.project_manager.project_updated.emit()
|
|
|
|
mt = self.form.compare_frame_controller.proto_analyzer.default_message_type
|
|
msg1 = SimulatorMessage(
|
|
source=bob,
|
|
destination=alice,
|
|
plain_bits=array("B", [1, 0, 1, 1]),
|
|
pause=100,
|
|
message_type=mt,
|
|
)
|
|
msg2 = SimulatorMessage(
|
|
source=alice,
|
|
destination=bob,
|
|
plain_bits=array("B", [1, 0, 1, 1]),
|
|
pause=100,
|
|
message_type=mt,
|
|
)
|
|
|
|
simulator_manager = self.form.simulator_tab_controller.simulator_config
|
|
simulator_manager.add_items([msg1, msg2], 0, simulator_manager.rootItem)
|
|
simulator_manager.add_label(
|
|
5, 15, "test", parent_item=simulator_manager.rootItem.children[0]
|
|
)
|
|
|
|
self.dialog = SimulatorDialog(
|
|
self.form.simulator_tab_controller.simulator_config,
|
|
self.form.generator_tab_controller.modulators,
|
|
self.form.simulator_tab_controller.sim_expression_parser,
|
|
self.form.project_manager,
|
|
)
|
|
|
|
if self.SHOW:
|
|
self.dialog.show()
|
|
|
|
def test_set_rx_parameters(self):
|
|
rx_settings_widget = self.dialog.device_settings_rx_widget
|
|
bh = BackendContainer("test", {Backends.native}, True, True)
|
|
self.assertTrue(bh.is_enabled)
|
|
rx_settings_widget.backend_handler.device_backends["test"] = bh
|
|
rx_settings_widget.ui.cbDevice.addItem("test")
|
|
rx_settings_widget.ui.cbDevice.setCurrentText("test")
|
|
self.assertEqual(rx_settings_widget.device.name, "test")
|
|
self.assertEqual(rx_settings_widget.device.backend, Backends.native)
|
|
|
|
simulator = self.dialog.simulator
|
|
self.__edit_spinbox_value(rx_settings_widget.ui.spinBoxFreq, 500e6)
|
|
self.assertEqual(simulator.sniffer.rcv_device.frequency, 500e6)
|
|
|
|
self.__edit_spinbox_value(rx_settings_widget.ui.spinBoxSampleRate, 4e6)
|
|
self.assertEqual(simulator.sniffer.rcv_device.sample_rate, 4e6)
|
|
|
|
self.__edit_spinbox_value(rx_settings_widget.ui.spinBoxBandwidth, 5e6)
|
|
self.assertEqual(simulator.sniffer.rcv_device.bandwidth, 5e6)
|
|
|
|
self.__edit_spinbox_value(rx_settings_widget.ui.spinBoxGain, 15)
|
|
self.assertEqual(simulator.sniffer.rcv_device.gain, 15)
|
|
|
|
self.__edit_spinbox_value(rx_settings_widget.ui.spinBoxIFGain, 10)
|
|
self.assertEqual(simulator.sniffer.rcv_device.if_gain, 10)
|
|
|
|
self.__edit_spinbox_value(rx_settings_widget.ui.spinBoxBasebandGain, 11)
|
|
self.assertEqual(simulator.sniffer.rcv_device.baseband_gain, 11)
|
|
|
|
self.__edit_spinbox_value(rx_settings_widget.ui.spinBoxFreqCorrection, 22)
|
|
self.assertEqual(simulator.sniffer.rcv_device.freq_correction, 22)
|
|
|
|
rx_settings_widget.ui.lineEditIP.setText("4.4.4.4")
|
|
rx_settings_widget.ui.lineEditIP.editingFinished.emit()
|
|
self.assertEqual(simulator.sniffer.rcv_device.ip, "4.4.4.4")
|
|
|
|
def test_set_sniff_parameters(self):
|
|
sniff_settings_widget = self.dialog.sniff_settings_widget
|
|
simulator = self.dialog.simulator
|
|
self.__edit_spinbox_value(
|
|
sniff_settings_widget.ui.spinbox_sniff_SamplesPerSymbol, 111
|
|
)
|
|
self.assertEqual(simulator.sniffer.signal.samples_per_symbol, 111)
|
|
|
|
self.__edit_spinbox_value(sniff_settings_widget.ui.spinbox_sniff_Center, 0.1337)
|
|
self.assertEqual(simulator.sniffer.signal.center, 0.1337)
|
|
|
|
self.__edit_spinbox_value(sniff_settings_widget.ui.spinBoxCenterSpacing, 0.4)
|
|
self.assertEqual(simulator.sniffer.signal.center_spacing, 0.4)
|
|
|
|
self.__edit_spinbox_value(
|
|
sniff_settings_widget.ui.spinbox_sniff_ErrorTolerance, 13
|
|
)
|
|
self.assertEqual(simulator.sniffer.signal.tolerance, 13)
|
|
|
|
self.__edit_spinbox_value(sniff_settings_widget.ui.spinbox_sniff_Noise, 0.1234)
|
|
self.assertEqual(simulator.sniffer.signal.noise_threshold_relative, 0.1234)
|
|
|
|
sniff_settings_widget.ui.combox_sniff_Modulation.setCurrentText("PSK")
|
|
self.assertEqual(simulator.sniffer.signal.modulation_type, "PSK")
|
|
|
|
self.__edit_spinbox_value(sniff_settings_widget.ui.spinBoxBitsPerSymbol, 5)
|
|
self.assertEqual(simulator.sniffer.signal.bits_per_symbol, 5)
|
|
|
|
decodings = [
|
|
sniff_settings_widget.ui.comboBox_sniff_encoding.itemText(i)
|
|
for i in range(sniff_settings_widget.ui.comboBox_sniff_encoding.count())
|
|
]
|
|
sniff_settings_widget.ui.comboBox_sniff_encoding.setCurrentIndex(2)
|
|
self.assertEqual(simulator.sniffer.decoder.name, decodings[2])
|
|
|
|
def test_set_tx_parameters(self):
|
|
tx_settings_widget = self.dialog.device_settings_tx_widget
|
|
simulator = self.dialog.simulator
|
|
|
|
bh = BackendContainer("test", {Backends.native}, True, True)
|
|
self.assertTrue(bh.is_enabled)
|
|
tx_settings_widget.backend_handler.device_backends["test"] = bh
|
|
tx_settings_widget.ui.cbDevice.addItem("test")
|
|
tx_settings_widget.ui.cbDevice.setCurrentText("test")
|
|
self.assertEqual(tx_settings_widget.device.name, "test")
|
|
self.assertEqual(tx_settings_widget.device.backend, Backends.native)
|
|
|
|
self.__edit_spinbox_value(tx_settings_widget.ui.spinBoxFreq, 300e6)
|
|
self.assertEqual(simulator.sender.device.frequency, 300e6)
|
|
|
|
self.__edit_spinbox_value(tx_settings_widget.ui.spinBoxSampleRate, 5e6)
|
|
self.assertEqual(simulator.sender.device.sample_rate, 5e6)
|
|
|
|
self.__edit_spinbox_value(tx_settings_widget.ui.spinBoxBandwidth, 3e6)
|
|
self.assertEqual(simulator.sender.device.bandwidth, 3e6)
|
|
|
|
self.__edit_spinbox_value(tx_settings_widget.ui.spinBoxGain, 16)
|
|
self.assertEqual(simulator.sender.device.gain, 16)
|
|
|
|
self.__edit_spinbox_value(tx_settings_widget.ui.spinBoxIFGain, 13)
|
|
self.assertEqual(simulator.sender.device.if_gain, 13)
|
|
|
|
self.__edit_spinbox_value(tx_settings_widget.ui.spinBoxBasebandGain, 10)
|
|
self.assertEqual(simulator.sender.device.baseband_gain, 10)
|
|
|
|
self.__edit_spinbox_value(tx_settings_widget.ui.spinBoxFreqCorrection, 33)
|
|
self.assertEqual(simulator.sender.device.freq_correction, 33)
|
|
|
|
tx_settings_widget.ui.lineEditIP.setText("1.2.6.2")
|
|
tx_settings_widget.ui.lineEditIP.editingFinished.emit()
|
|
self.assertEqual(simulator.sender.device.ip, "1.2.6.2")
|
|
|
|
def __edit_spinbox_value(self, spinbox, value):
|
|
spinbox.setValue(value)
|
|
spinbox.editingFinished.emit()
|