def test_set_db_mcl_number_bad_mcl(self, mock_msg):
# Arrange
mock_vlayer = MagicMock(spec=QgsVectorLayer)
auto_num_tool = mant.MclAutoNumberingTool(mock_vlayer, self.db,
self.iface)
auto_num_tool.dlg.ui.mclListWidget.addItems(['bad', 'items'])
# Act and assert
with self.assertRaises(rn_except.RampMclNumberingFailedPopupError):
auto_num_tool.number_mcls()
def test_warn_and_revert_selection(self):
# Arrange
self.load_mcl_layer()
auto_num_tool = mant.MclAutoNumberingTool(self.mcl, self.db,
self.iface)
expected = [11111]
# Act
auto_num_tool.warn_and_revert_selection(2, ['11111'])
# Assert
result = [f['mcl_ref'] for f in self.mcl.selectedFeatures()]
self.assertEqual(result, expected)
def test_update_current_mcls_add_one(self, mock_warn):
# Arrange
mock_vlayer = MagicMock(spec=QgsVectorLayer)
auto_num_tool = mant.MclAutoNumberingTool(mock_vlayer, self.db,
self.iface)
mcls = ['1', '2', '3']
auto_num_tool.current_mcls = mcls
selected = ['3', '2', '1', '4']
expected = ['1', '2', '3', '4']
# Act
result = auto_num_tool.update_current_mcls(selected)
# Assert
self.assertEqual(expected, result)
def test_apply(self):
# Arrange
self.load_mcl_layer()
auto_num_tool = mant.MclAutoNumberingTool(self.mcl, self.db,
self.iface)
auto_num_tool.dlg.ui.mclListWidget.addItems(['22222', '11111'])
auto_num_tool.dlg.ui.startValueSpinBox.setValue(3)
auto_num_tool.dlg.ui.incrementSpinBox.setValue(20)
ok_button = auto_num_tool.dlg.ui.buttonBox.button(QDialogButtonBox.Ok)
expected = [23, 3]
# Act
QTest.mouseClick(ok_button, Qt.LeftButton)
# Assert
mcl_data = get_lor_ref_2_values(self.db)
pprint.pprint(mcl_data)
self.assertEqual(mcl_data, expected)