mirror of
https://github.com/jopohl/urh.git
synced 2026-03-03 06:54:00 +01:00
102 lines
3.2 KiB
Python
102 lines
3.2 KiB
Python
import faulthandler
|
|
import gc
|
|
import os
|
|
import unittest
|
|
|
|
from PyQt6.QtCore import Qt
|
|
from PyQt6.QtGui import QDropEvent
|
|
from PyQt6.QtTest import QTest
|
|
from PyQt6.QtWidgets import QApplication
|
|
|
|
from tests.utils_testing import write_settings, get_path_for_data_file
|
|
from urh.controller.MainController import MainController
|
|
from urh.signalprocessing.ProtocolSniffer import ProtocolSniffer
|
|
|
|
# noinspection PyUnresolvedReferences
|
|
import urh.ui.urh_rc
|
|
|
|
faulthandler.enable()
|
|
|
|
|
|
class QtTestCase(unittest.TestCase):
|
|
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()
|
|
del cls.app
|
|
|
|
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()
|
|
del self.dialog
|
|
|
|
if hasattr(self, "form"):
|
|
self.form.close_all_files()
|
|
self.form.close()
|
|
del self.form
|
|
|
|
gc.collect()
|
|
|
|
def add_signal_to_form(self, filename: str):
|
|
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.MouseButton.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().toPointF(),
|
|
Qt.DropAction.CopyAction | Qt.DropAction.MoveAction,
|
|
mimedata,
|
|
Qt.MouseButton.LeftButton,
|
|
Qt.KeyboardModifier.NoModifier,
|
|
)
|
|
drop_event.acceptProposedAction()
|
|
sim_frame.ui.gvSimulator.dropEvent(drop_event)
|