Mirrors/urh
2
0
Fork 0
mirror of https://github.com/jopohl/urh.git synced 2026-03-09 09:46:48 +01:00
Code Issues Packages Projects Releases Wiki Activity

add unittest and nonecheck for generator crash #1

Browse Source
This commit is contained in:
jopohl
2016-11-15 13:44:23 +01:00
parent 0012b07774
commit 5013a32a3c
2 changed files with 33 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/urh/models/ProtocolTreeModel.py
View File

@@ -108,7 +108,7 @@ class ProtocolTreeModel(QAbstractItemModel):
except AttributeError:
return QModelIndex()
if parentItem == self.rootItem:
if parentItem == self.rootItem or parentItem is None:
return QModelIndex()
return self.createIndex(parentItem.indexInParent(), 0, parentItem)

32
tests/test_generator.py
View File

@@ -85,6 +85,38 @@ class TestGenerator(unittest.TestCase):
gen_proto = gen_proto[:gen_proto.index(" ")]
self.assertTrue(proto.startswith(gen_proto))
def test_close_signal(self):
self.form.add_signalfile("./data/ask.complex")
sframe = self.form.signal_tab_controller.signal_frames[0]
sframe.ui.cbModulationType.setCurrentIndex(0) # ASK
sframe.ui.spinBoxInfoLen.setValue(295)
sframe.ui.spinBoxCenterOffset.setValue(-0.1667)
sframe.refresh()
# 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.row_count, self.form.compare_frame_controller.protocol_model.row_count)
self.form.ui.tabWidget.setCurrentIndex(0)
self.form.on_selected_tab_changed()
sframe.ui.btnCloseSignal.click()
QTest.qWait(1000)
self.form.ui.tabWidget.setCurrentIndex(1)
self.form.on_selected_tab_changed()
QTest.qWait(1000)
self.form.ui.tabWidget.setCurrentIndex(2)
self.form.on_selected_tab_changed()
QTest.qWait(1000)
self.assertEqual(1, 1)
def __is_inv_proto(self, proto1: str, proto2: str):
if len(proto1) != len(proto2):
return False