mirror of
https://github.com/jopohl/urh.git
synced 2026-03-07 16:56:48 +01:00
235 lines
10 KiB
Python
235 lines
10 KiB
Python
import os
|
|
|
|
from PyQt5.QtCore import QDir, QPoint
|
|
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
|
|
|
|
|
|
class TestSignalTabGUI(QtTestCase):
|
|
def test_close_all(self):
|
|
self.add_signal_to_form("esaver.complex")
|
|
self.assertEqual(self.form.signal_tab_controller.num_frames, 1)
|
|
self.form.close_all()
|
|
QApplication.instance().processEvents()
|
|
QTest.qWait(self.CLOSE_TIMEOUT)
|
|
self.assertEqual(self.form.signal_tab_controller.num_frames, 0)
|
|
|
|
# Add a bunch of signals
|
|
num_frames = 5
|
|
for _ in range(num_frames):
|
|
self.add_signal_to_form("esaver.complex")
|
|
|
|
self.assertEqual(self.form.signal_tab_controller.num_frames, num_frames)
|
|
|
|
self.form.close_all()
|
|
QApplication.instance().processEvents()
|
|
QTest.qWait(self.CLOSE_TIMEOUT)
|
|
|
|
self.add_signal_to_form("ask.complex")
|
|
self.assertEqual(self.form.signal_tab_controller.num_frames, 1)
|
|
|
|
def test_zoom(self):
|
|
self.add_signal_to_form("esaver.complex")
|
|
frame = self.form.signal_tab_controller.signal_frames[0]
|
|
QApplication.instance().processEvents()
|
|
x_zoom = frame.ui.spinBoxXZoom.value()
|
|
self.assertEqual(x_zoom, 100)
|
|
|
|
for _ in range(10):
|
|
frame.ui.gvSignal.zoom(1.01)
|
|
self.assertGreater(frame.ui.spinBoxXZoom.value(), x_zoom)
|
|
x_zoom = frame.ui.spinBoxXZoom.value()
|
|
|
|
for _ in range(10):
|
|
frame.ui.gvSignal.zoom(0.99)
|
|
self.assertLess(frame.ui.spinBoxXZoom.value(), x_zoom)
|
|
x_zoom = frame.ui.spinBoxXZoom.value()
|
|
|
|
samples_in_view = frame.ui.lSamplesInView.text()
|
|
frame.ui.spinBoxXZoom.setValue(frame.ui.spinBoxXZoom.value() + 400)
|
|
self.assertNotEqual(samples_in_view, frame.ui.lSamplesInView.text())
|
|
|
|
def test_load_proto(self):
|
|
QApplication.instance().processEvents()
|
|
QTest.qWait(self.WAIT_TIMEOUT_BEFORE_NEW)
|
|
self.form.add_files([get_path_for_data_file("protocol.proto")])
|
|
self.assertEqual(self.form.signal_tab_controller.signal_frames[0].ui.lSignalTyp.text(), "Protocol (*.proto)")
|
|
|
|
def test_graphic_view_selection(self):
|
|
self.add_signal_to_form("esaver.complex")
|
|
frame = self.form.signal_tab_controller.signal_frames[0]
|
|
frame.ui.gvSignal.selection_area.start = 0
|
|
frame.ui.gvSignal.selection_area.end = 4000
|
|
frame.ui.gvSignal.sel_area_start_end_changed.emit(0, 4000)
|
|
|
|
self.assertEqual(frame.ui.lNumSelectedSamples.text(), "4000")
|
|
|
|
noise_val = frame.ui.spinBoxNoiseTreshold.value()
|
|
frame.ui.gvSignal.set_noise_clicked.emit()
|
|
self.assertNotEqual(noise_val, frame.ui.spinBoxNoiseTreshold.value())
|
|
|
|
frame.ui.spinBoxSelectionStart.setValue(300)
|
|
self.assertEqual(frame.ui.gvSignal.selection_area.start, 300)
|
|
frame.ui.spinBoxSelectionEnd.setValue(6000)
|
|
self.assertEqual(frame.ui.gvSignal.selection_area.end, 6000)
|
|
|
|
def test_graphic_view_zoom_to_selection(self):
|
|
self.add_signal_to_form("esaver.complex")
|
|
frame = self.form.signal_tab_controller.signal_frames[0]
|
|
frame.ui.gvSignal.context_menu_position = QPoint(0, 0)
|
|
menu = frame.ui.gvSignal.create_context_menu()
|
|
self.assertTrue(frame.ui.gvSignal.selection_area.is_empty)
|
|
self.assertIsNone(next((action for action in menu.actions() if action.text() == "Zoom selection"), None))
|
|
|
|
frame.ui.gvSignal.selection_area.start = 1337
|
|
frame.ui.gvSignal.selection_area.end = 4711
|
|
frame.ui.gvSignal.sel_area_start_end_changed.emit(1337, 4711)
|
|
|
|
menu = frame.ui.gvSignal.create_context_menu()
|
|
self.assertFalse(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(frame.ui.spinBoxSelectionStart.value(), 1337)
|
|
self.assertEqual(frame.ui.spinBoxSelectionEnd.value(), 4711)
|
|
|
|
def test_show_hide_start_end(self):
|
|
self.add_signal_to_form("esaver.complex")
|
|
frame = self.form.signal_tab_controller.signal_frames[0]
|
|
self.assertEqual(frame.ui.btnShowHideStartEnd.text(), "+")
|
|
frame.ui.btnShowHideStartEnd.click()
|
|
self.assertEqual(frame.ui.btnShowHideStartEnd.text(), "-")
|
|
|
|
def test_apply_to_all(self):
|
|
self.add_signal_to_form("esaver.complex")
|
|
frame = self.form.signal_tab_controller.signal_frames[0]
|
|
self.add_signal_to_form("ask.complex")
|
|
frame2 = self.form.signal_tab_controller.signal_frames[1]
|
|
|
|
frame.ui.spinBoxInfoLen.setValue(42)
|
|
frame.ui.spinBoxInfoLen.editingFinished.emit()
|
|
|
|
frame.ui.spinBoxCenterOffset.setValue(0.1)
|
|
frame.ui.spinBoxCenterOffset.editingFinished.emit()
|
|
|
|
frame.ui.spinBoxNoiseTreshold.setValue(0.5)
|
|
frame.ui.spinBoxNoiseTreshold.editingFinished.emit()
|
|
|
|
frame.ui.spinBoxTolerance.setValue(10)
|
|
frame.ui.spinBoxTolerance.editingFinished.emit()
|
|
|
|
frame.apply_to_all_clicked.emit(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):
|
|
self.add_signal_to_form("esaver.complex")
|
|
frame = self.form.signal_tab_controller.signal_frames[0]
|
|
self.add_signal_to_form("ask.complex")
|
|
frame2 = self.form.signal_tab_controller.signal_frames[1]
|
|
|
|
frame.signal.changed = True
|
|
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(frame.signal.filename))
|
|
self.assertFalse(os.path.isfile(frame2.signal.filename))
|
|
|
|
self.form.signal_tab_controller.save_all()
|
|
|
|
self.assertTrue(os.path.isfile(frame.signal.filename))
|
|
self.assertTrue(os.path.isfile(frame2.signal.filename))
|
|
|
|
os.remove(frame.signal.filename)
|
|
os.remove(frame2.signal.filename)
|
|
|
|
def test_crop_and_save_signal(self):
|
|
self.add_signal_to_form("esaver.complex")
|
|
frame = self.form.signal_tab_controller.signal_frames[0]
|
|
frame.ui.gvSignal.selection_area.end = 4000
|
|
frame.ui.gvSignal.selection_area.start = 1000
|
|
|
|
self.assertEqual(frame.ui.gvSignal.selection_area.end, 4000)
|
|
self.assertEqual(frame.ui.gvSignal.selection_area.width, 3000)
|
|
frame.ui.gvSignal.sel_area_start_end_changed.emit(1000, 4000)
|
|
|
|
frame.ui.gvSignal.on_crop_action_triggered()
|
|
self.assertEqual(frame.signal.num_samples, 3000)
|
|
self.assertTrue(frame.signal.changed)
|
|
|
|
frame.signal.filename = os.path.join(QDir.tempPath(), "sig.complex")
|
|
if os.path.isfile(frame.signal.filename):
|
|
os.remove(frame.signal.filename)
|
|
|
|
self.assertFalse(os.path.isfile(frame.signal.filename))
|
|
frame.ui.btnSaveSignal.click()
|
|
self.form.close_signal_frame(frame)
|
|
self.add_signal_to_form(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.add_signal_to_form("esaver.complex")
|
|
frame = self.form.signal_tab_controller.signal_frames[0]
|
|
frame.ui.cbProtoView.setCurrentIndex(0) # set to bit view
|
|
frame.ui.gvSignal.selection_area.end = 128440
|
|
frame.ui.gvSignal.selection_area.start = 89383
|
|
frame.ui.gvSignal.sel_area_start_end_changed.emit(89383, 128440)
|
|
QApplication.instance().processEvents()
|
|
QTest.qWait(100)
|
|
self.assertEqual(frame.proto_analyzer.messages[0].plain_bits_str, frame.ui.txtEdProto.selected_text)
|
|
frame.ui.txtEdProto.show_proto_clicked.emit()
|
|
QApplication.instance().processEvents()
|
|
self.assertAlmostEqual((128440 - 89383) / 1000000,
|
|
(frame.ui.gvSignal.view_rect().width()) / 1000000, places=1)
|
|
|
|
def test_show_demod_view(self):
|
|
self.add_signal_to_form("esaver.complex")
|
|
frame = self.form.signal_tab_controller.signal_frames[0]
|
|
self.assertTrue(frame.ui.gvLegend.isHidden())
|
|
frame.ui.cbSignalView.setCurrentIndex(1)
|
|
self.assertFalse(frame.ui.gvLegend.isHidden())
|
|
|
|
def test_auto_detect_button(self):
|
|
self.add_signal_to_form("esaver.complex")
|
|
frame = self.form.signal_tab_controller.signal_frames[0]
|
|
self.assertTrue(frame.ui.btnAutoDetect.isChecked())
|
|
frame.ui.btnAutoDetect.click()
|
|
self.assertFalse(frame.ui.btnAutoDetect.isChecked())
|
|
|
|
def test_create_new_signal(self):
|
|
self.add_signal_to_form("esaver.complex")
|
|
frame = self.form.signal_tab_controller.signal_frames[0]
|
|
start, end = 400, 8568
|
|
frame.ui.gvSignal.selection_area.end = end
|
|
frame.ui.gvSignal.selection_area.start = start
|
|
|
|
self.assertEqual(frame.ui.gvSignal.selection_area.end, end)
|
|
self.assertEqual(frame.ui.gvSignal.selection_area.width, end - start)
|
|
frame.ui.gvSignal.sel_area_start_end_changed.emit(start, end)
|
|
|
|
self.assertEqual(self.form.signal_tab_controller.num_frames, 1)
|
|
QApplication.instance().processEvents()
|
|
QTest.qWait(self.WAIT_TIMEOUT_BEFORE_NEW)
|
|
frame.ui.gvSignal.on_create_action_triggered()
|
|
QApplication.instance().processEvents()
|
|
|
|
self.assertEqual(self.form.signal_tab_controller.num_frames, 2)
|
|
self.assertEqual(self.form.signal_tab_controller.signal_frames[1].signal.num_samples, end - start)
|
|
|
|
def test_demodulated_view(self):
|
|
self.add_signal_to_form("esaver.complex")
|
|
frame = self.form.signal_tab_controller.signal_frames[0]
|
|
frame.ui.cbSignalView.setCurrentIndex(1)
|
|
QApplication.instance().processEvents()
|
|
self.assertEqual(frame.ui.gvSignal.scene_type, 1)
|
|
if self.SHOW:
|
|
self.assertTrue(frame.ui.gvLegend.isVisible())
|