HackRF-Treasure-Chest/Software/Universal Radio Hacker/tests/test_project_manager.py

252 lines
12 KiB
Python
Raw Permalink Normal View History

2022-09-22 22:46:47 +02:00
import os
import random
import tempfile
from PyQt5.QtCore import QDir, Qt
from PyQt5.QtTest import QTest
from PyQt5.QtWidgets import QApplication
from tests.QtTestCase import QtTestCase
from tests.utils_testing import get_path_for_data_file
from urh import settings
from urh.controller.dialogs.ProjectDialog import ProjectDialog
from urh.signalprocessing.FieldType import FieldType
from urh.signalprocessing.Modulator import Modulator
from urh.signalprocessing.Participant import Participant
class TestProjectManager(QtTestCase):
def setUp(self):
super().setUp()
if os.path.isfile(get_path_for_data_file("URHProject.xml")):
os.remove(get_path_for_data_file("URHProject.xml"))
self.form.project_manager.set_project_folder(get_path_for_data_file(""), ask_for_new_project=False)
self.gframe = self.form.generator_tab_controller
def test_load_protocol_file(self):
self.form.add_protocol_file(self.get_path_for_filename("protocol_wsp.proto.xml"))
self.assertEqual(len(self.form.compare_frame_controller.proto_analyzer.messages), 6)
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 = "FSK"
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.save_project()
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, "FSK")
self.assertEqual(loaded_mods[0].sample_rate, 10 ** 3)
self.gframe.modulators.clear()
self.assertEqual(len(self.gframe.modulators), 0)
self.form.project_manager.project_file = None # prevent saving of the zero modulators
self.form.project_manager.set_project_folder(self.form.project_manager.project_path, close_all=False)
self.assertEqual(len(self.gframe.modulators), 2)
def test_close_all(self):
self.form.close_project()
self.assertEqual(self.form.signal_tab_controller.num_frames, 0)
self.add_signal_to_form("ask.complex")
self.add_signal_to_form("fsk.complex")
self.assertEqual(self.form.signal_tab_controller.num_frames, 2)
self.form.close_project()
self.assertEqual(self.form.signal_tab_controller.num_frames, 0)
self.assertEqual(self.form.project_manager.project_file, None)
def test_save_and_load_participants(self):
target_dir = os.path.join(tempfile.gettempdir(), "urh", "multi_participant_test")
os.makedirs(target_dir, exist_ok=True)
if os.path.isfile(os.path.join(target_dir, settings.PROJECT_FILE)):
os.remove(os.path.join(target_dir, settings.PROJECT_FILE))
self.form.project_manager.set_project_folder(target_dir, ask_for_new_project=False)
self.form.project_manager.participants = [Participant("Alice", "A"), Participant("Bob", "B")]
self.add_signal_to_form("esaver.complex16s")
self.assertEqual(len(self.form.signal_tab_controller.signal_frames[0].proto_analyzer.messages), 3)
self.add_signal_to_form("two_participants.complex16s")
self.assertEqual(len(self.form.signal_tab_controller.signal_frames[1].proto_analyzer.messages), 18)
self.add_signal_to_form("fsk.complex")
self.assertEqual(len(self.form.signal_tab_controller.signal_frames[2].proto_analyzer.messages), 1)
self.assertEqual(self.form.compare_frame_controller.protocol_model.row_count, 22)
target = {0: "A", 1: "A", 2: "B", 3: "B", 4: "A", 5: "B", 6: "A", 7: "A", 8: "A", 9: "B", 10: "B",
11: "A", 12: "B", 13: "A", 14: "A", 15: "B", 16: "A", 17: "B", 18: "B", 19: "B", 20: "A", 21: "B"}
for row, shortname in target.items():
participant = next(p for p in self.form.project_manager.participants if p.shortname == shortname)
self.form.compare_frame_controller.proto_analyzer.messages[row].participant = participant
self.form.compare_frame_controller.proto_tree_model.rootItem.child(0).child(0).show = False
self.assertEqual(self.form.compare_frame_controller.protocol_model.row_count, 19)
for row, shortname in target.items():
row -= 3
if row >= 0:
self.assertEqual(self.form.compare_frame_controller.proto_analyzer.messages[row].participant.shortname,
shortname)
self.form.compare_frame_controller.refresh_assigned_participants_ui()
self.form.save_project()
self.form.close_all_files()
self.assertEqual(self.form.compare_frame_controller.protocol_model.row_count, 0)
self.form.project_manager.set_project_folder(target_dir, ask_for_new_project=False)
self.assertEqual(self.form.compare_frame_controller.protocol_model.row_count, 22)
for row, shortname in target.items():
self.assertEqual(self.form.compare_frame_controller.proto_analyzer.messages[row].participant.shortname,
shortname, msg=str(row))
def test_save_and_load_with_fieldtypes(self):
target_dir = os.path.join(tempfile.gettempdir(), "urh", "project_fieldtype_test")
os.makedirs(target_dir, exist_ok=True)
if os.path.isfile(os.path.join(target_dir, settings.PROJECT_FILE)):
os.remove(os.path.join(target_dir, settings.PROJECT_FILE))
self.form.project_manager.set_project_folder(target_dir, ask_for_new_project=False)
self.add_signal_to_form("esaver.complex16s")
self.assertEqual(len(self.form.signal_tab_controller.signal_frames[0].proto_analyzer.messages), 3)
preamble_field_type = next(ft for ft in self.form.compare_frame_controller.field_types
if ft.function == FieldType.Function.PREAMBLE) # type: FieldType
sync_field_type = next(ft for ft in self.form.compare_frame_controller.field_types
if ft.function == FieldType.Function.SYNC) # type: FieldType
checksum_field_type = next(ft for ft in self.form.compare_frame_controller.field_types
if ft.function == FieldType.Function.CHECKSUM) # type: FieldType
self.form.compare_frame_controller.ui.cbProtoView.setCurrentText("Hex")
self.form.compare_frame_controller.add_protocol_label(0, 9, 0, 1, False)
self.__set_label_name(0, preamble_field_type.caption)
self.form.compare_frame_controller.add_protocol_label(10, 13, 0, 1, False)
self.__set_label_name(1, sync_field_type.caption)
self.form.compare_frame_controller.add_protocol_label(14, 16, 0, 1, False)
self.__set_label_name(2, checksum_field_type.caption)
self.assertEqual(self.form.compare_frame_controller.active_message_type[0].field_type, preamble_field_type)
self.assertEqual(self.form.compare_frame_controller.active_message_type[1].field_type, sync_field_type)
self.assertEqual(self.form.compare_frame_controller.active_message_type[2].field_type, checksum_field_type)
self.form.close_project()
self.assertEqual(len(self.form.compare_frame_controller.active_message_type), 0)
self.form.project_manager.set_project_folder(target_dir, ask_for_new_project=False)
self.assertEqual(len(self.form.compare_frame_controller.active_message_type), 3)
preamble_field_type = next(ft for ft in self.form.compare_frame_controller.field_types
if ft.function == FieldType.Function.PREAMBLE) # type: FieldType
sync_field_type = next(ft for ft in self.form.compare_frame_controller.field_types
if ft.function == FieldType.Function.SYNC) # type: FieldType
checksum_field_type = next(ft for ft in self.form.compare_frame_controller.field_types
if ft.function == FieldType.Function.CHECKSUM) # type: FieldType
self.assertEqual(self.form.compare_frame_controller.active_message_type[0].field_type, preamble_field_type)
self.assertEqual(self.form.compare_frame_controller.active_message_type[1].field_type, sync_field_type)
self.assertEqual(self.form.compare_frame_controller.active_message_type[2].field_type, checksum_field_type)
def __set_label_name(self, index: int, name: str):
model = self.form.compare_frame_controller.ui.tblLabelValues.model()
model.setData(model.createIndex(index, 0), name, role=Qt.EditRole)
def test_project_dialog(self):
frequency = 1e9
sample_rate = 10e9
bandwidth = 10
gain = 42
descr = "URH rockz."
dialog = ProjectDialog(project_manager=self.form.project_manager, parent=self.form)
dialog.ui.spinBoxFreq.setValue(frequency)
self.assertEqual(dialog.freq, frequency)
dialog.ui.spinBoxSampleRate.setValue(sample_rate)
self.assertEqual(dialog.sample_rate, sample_rate)
dialog.ui.spinBoxBandwidth.setValue(bandwidth)
self.assertEqual(dialog.bandwidth, bandwidth)
dialog.ui.spinBoxGain.setValue(gain)
self.assertEqual(dialog.gain, gain)
dialog.ui.txtEdDescription.setPlainText(descr)
self.assertEqual(dialog.description, descr)
dialog.ui.lineEditBroadcastAddress.setText("abcd")
dialog.ui.lineEditBroadcastAddress.textEdited.emit("abcd")
self.assertEqual(dialog.broadcast_address_hex, "abcd")
if len(dialog.participants) == 0:
dialog.ui.btnAddParticipant.click()
self.assertEqual(len(dialog.participants), 1)
model = 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 = 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(dialog.participants)
dialog.ui.btnAddParticipant.click()
dialog.ui.btnAddParticipant.click()
dialog.ui.btnAddParticipant.click()
self.assertEqual(len(dialog.participants), num_participants + 3)
dialog.ui.btnRemoveParticipant.click()
dialog.ui.btnRemoveParticipant.click()
dialog.ui.btnRemoveParticipant.click()
self.assertEqual(len(dialog.participants), num_participants)
test_path = os.path.join(QDir.tempPath(), "urh_test")
dialog.ui.lineEdit_Path.setText(test_path)
dialog.ui.lineEdit_Path.textEdited.emit(test_path)
self.assertEqual(dialog.path, test_path)
dialog.on_button_box_accepted()
self.form.ui.tabWidget.setCurrentWidget(self.form.ui.tab_protocol)
self.form.compare_frame_controller.ui.tabWidget.setCurrentWidget(
self.form.compare_frame_controller.ui.tab_participants)
self.assertGreater(self.form.compare_frame_controller.participant_list_model.rowCount(), 0)
self.assertTrue(os.path.isdir(test_path))
self.form.project_manager.from_dialog(dialog)
dialog = ProjectDialog(project_manager=self.form.project_manager, parent=self.form, new_project=False)
self.assertEqual(dialog.ui.spinBoxFreq.value(), frequency)
self.assertEqual(dialog.ui.spinBoxSampleRate.value(), sample_rate)
self.assertEqual(dialog.ui.spinBoxBandwidth.value(), bandwidth)
self.assertEqual(dialog.ui.spinBoxGain.value(), gain)
self.assertEqual(dialog.ui.txtEdDescription.toPlainText(), descr)
self.assertFalse(dialog.ui.lineEdit_Path.isEnabled())