tests: changing netvm and template via right click
This commit is contained in:
Marek Marczykowski-Górecki
parent
66e478d968
commit
465d66189a
@ -733,6 +733,241 @@ class QubeManagerTest(unittest.TestCase):
|
|||||||
self.assertEqual(expected_number, actual_number,
|
self.assertEqual(expected_number, actual_number,
|
||||||
"Incorrect number of vms shown for cleared search box")
|
"Incorrect number of vms shown for cleared search box")
|
||||||
|
|
||||||
|
@unittest.mock.patch('PyQt5.QtWidgets.QMessageBox.question')
|
||||||
|
@listen_for_events
|
||||||
|
def test_240_network_menu_single(self, mock_question):
|
||||||
|
mock_question.return_value = QtWidgets.QMessageBox.Yes
|
||||||
|
target_vm_name = 'work'
|
||||||
|
|
||||||
|
self._run_command_and_process_events(
|
||||||
|
['qvm-prefs', '-D', target_vm_name, 'netvm'], timeout=20)
|
||||||
|
self._select_vms(['work'])
|
||||||
|
selected_vm = self.qapp.domains[target_vm_name]
|
||||||
|
# reset to default even in case of failure
|
||||||
|
self.addCleanup(functools.partial(delattr, selected_vm, 'netvm'))
|
||||||
|
|
||||||
|
# this is the method to get '==' operator working on icons...
|
||||||
|
on_icon = QIcon(":/on.png").pixmap(64).toImage()
|
||||||
|
off_icon = QIcon().pixmap(64).toImage()
|
||||||
|
for action in self.dialog.network_menu.actions():
|
||||||
|
if action.text().startswith('default '):
|
||||||
|
self.assertEqual(action.icon().pixmap(64).toImage(), on_icon)
|
||||||
|
break
|
||||||
|
else:
|
||||||
|
self.fail('default netvm not found')
|
||||||
|
|
||||||
|
# change to specific value
|
||||||
|
for action in self.dialog.network_menu.actions():
|
||||||
|
if action.text() == 'sys-net':
|
||||||
|
self.assertEqual(action.icon().pixmap(64).toImage(), off_icon)
|
||||||
|
action.trigger()
|
||||||
|
break
|
||||||
|
else:
|
||||||
|
self.fail('sys-net netvm not found')
|
||||||
|
# process events
|
||||||
|
self.loop.run_until_complete(asyncio.sleep(0))
|
||||||
|
mock_question.assert_called()
|
||||||
|
self.assertEqual(str(selected_vm.netvm), 'sys-net')
|
||||||
|
mock_question.reset_mock()
|
||||||
|
|
||||||
|
# change to none
|
||||||
|
for action in self.dialog.network_menu.actions():
|
||||||
|
if action.text() == 'None':
|
||||||
|
self.assertEqual(action.icon().pixmap(64).toImage(), off_icon)
|
||||||
|
action.trigger()
|
||||||
|
break
|
||||||
|
else:
|
||||||
|
self.fail('"none" netvm not found')
|
||||||
|
# process events
|
||||||
|
self.loop.run_until_complete(asyncio.sleep(0))
|
||||||
|
mock_question.assert_called()
|
||||||
|
self.assertIsNone(selected_vm.netvm)
|
||||||
|
mock_question.reset_mock()
|
||||||
|
|
||||||
|
# then go back to the default
|
||||||
|
for action in self.dialog.network_menu.actions():
|
||||||
|
if action.text().startswith('default '):
|
||||||
|
self.assertEqual(action.icon().pixmap(64).toImage(), off_icon)
|
||||||
|
action.trigger()
|
||||||
|
break
|
||||||
|
# process events
|
||||||
|
self.loop.run_until_complete(asyncio.sleep(0))
|
||||||
|
|
||||||
|
mock_question.assert_called()
|
||||||
|
self.assertTrue(selected_vm.property_is_default('netvm'))
|
||||||
|
|
||||||
|
@unittest.mock.patch('PyQt5.QtWidgets.QMessageBox.question')
|
||||||
|
@listen_for_events
|
||||||
|
def test_241_network_menu_multiple(self, mock_question):
|
||||||
|
mock_question.return_value = QtWidgets.QMessageBox.Yes
|
||||||
|
target_vm_names = ['work', 'personal', 'vault']
|
||||||
|
work = self.qapp.domains['work']
|
||||||
|
personal = self.qapp.domains['personal']
|
||||||
|
vault = self.qapp.domains['vault']
|
||||||
|
# reset to default even in case of failure
|
||||||
|
self.addCleanup(functools.partial(delattr, work, 'netvm'))
|
||||||
|
self.addCleanup(functools.partial(delattr, personal, 'netvm'))
|
||||||
|
self.addCleanup(functools.partial(setattr, vault, 'netvm', None))
|
||||||
|
|
||||||
|
self._run_command_and_process_events(
|
||||||
|
['qvm-prefs', '-D', 'work', 'netvm'], timeout=5)
|
||||||
|
self._run_command_and_process_events(
|
||||||
|
['qvm-prefs', '-D', 'personal', 'netvm'], timeout=5)
|
||||||
|
self._run_command_and_process_events(
|
||||||
|
['qvm-prefs', 'vault', ''], timeout=5)
|
||||||
|
self._select_vms(target_vm_names)
|
||||||
|
|
||||||
|
# this is the method to get '==' operator working on icons...
|
||||||
|
on_icon = QIcon(":/on.png").pixmap(64).toImage()
|
||||||
|
transient_icon = QIcon(":/transient.png").pixmap(64).toImage()
|
||||||
|
off_icon = QIcon().pixmap(64).toImage()
|
||||||
|
for action in self.dialog.network_menu.actions():
|
||||||
|
if action.text().startswith('default '):
|
||||||
|
# work, personal
|
||||||
|
self.assertEqual(action.icon().pixmap(64).toImage(), transient_icon)
|
||||||
|
elif action.text() == 'None':
|
||||||
|
# vault
|
||||||
|
self.assertEqual(action.icon().pixmap(64).toImage(), transient_icon)
|
||||||
|
else:
|
||||||
|
self.assertEqual(action.icon().pixmap(64).toImage(), off_icon)
|
||||||
|
|
||||||
|
# change to specific value
|
||||||
|
for action in self.dialog.network_menu.actions():
|
||||||
|
if action.text() == 'sys-net':
|
||||||
|
self.assertEqual(action.icon().pixmap(64).toImage(), off_icon)
|
||||||
|
action.trigger()
|
||||||
|
break
|
||||||
|
else:
|
||||||
|
self.fail('sys-net netvm not found')
|
||||||
|
# process events
|
||||||
|
self.loop.run_until_complete(asyncio.sleep(0))
|
||||||
|
mock_question.assert_called()
|
||||||
|
self.assertEqual(str(work.netvm), 'sys-net')
|
||||||
|
self.assertEqual(str(personal.netvm), 'sys-net')
|
||||||
|
self.assertEqual(str(vault.netvm), 'sys-net')
|
||||||
|
mock_question.reset_mock()
|
||||||
|
|
||||||
|
@unittest.mock.patch('PyQt5.QtWidgets.QMessageBox.question')
|
||||||
|
@listen_for_events
|
||||||
|
def test_250_template_menu_single(self, mock_question):
|
||||||
|
mock_question.return_value = QtWidgets.QMessageBox.Yes
|
||||||
|
target_vm_name = 'work'
|
||||||
|
selected_vm = self.qapp.domains[target_vm_name]
|
||||||
|
if selected_vm.is_running():
|
||||||
|
self.skipTest(
|
||||||
|
'VM {!s} is running, please stop it first'.format(selected_vm))
|
||||||
|
current_template = selected_vm.template
|
||||||
|
new_template = self._select_templatevm(
|
||||||
|
different_than=[str(current_template)])
|
||||||
|
|
||||||
|
self._select_vms(['work'])
|
||||||
|
|
||||||
|
# reset to previous value even in case of failure
|
||||||
|
self.addCleanup(functools.partial(
|
||||||
|
setattr, selected_vm, 'template', str(current_template)))
|
||||||
|
|
||||||
|
# this is the method to get '==' operator working on icons...
|
||||||
|
on_icon = QIcon(":/on.png").pixmap(64).toImage()
|
||||||
|
off_icon = QIcon().pixmap(64).toImage()
|
||||||
|
found = False
|
||||||
|
for action in self.dialog.template_menu.actions():
|
||||||
|
if action.text() == str(current_template):
|
||||||
|
self.assertEqual(action.icon().pixmap(64).toImage(), on_icon)
|
||||||
|
found = True
|
||||||
|
else:
|
||||||
|
self.assertEqual(action.icon().pixmap(64).toImage(), off_icon)
|
||||||
|
|
||||||
|
if not found:
|
||||||
|
self.fail(
|
||||||
|
'current template value ({!s}) not found in the menu'.format(
|
||||||
|
current_template))
|
||||||
|
|
||||||
|
# change to specific value
|
||||||
|
for action in self.dialog.template_menu.actions():
|
||||||
|
if action.text() == str(new_template):
|
||||||
|
self.assertEqual(action.icon().pixmap(64).toImage(), off_icon)
|
||||||
|
action.trigger()
|
||||||
|
break
|
||||||
|
else:
|
||||||
|
self.fail('template {!s} not found in the menu'.format(new_template))
|
||||||
|
# process events
|
||||||
|
self.loop.run_until_complete(asyncio.sleep(0))
|
||||||
|
mock_question.assert_called()
|
||||||
|
# compare str(), to have better error message on mismatch
|
||||||
|
self.assertEqual(str(selected_vm.template), str(new_template))
|
||||||
|
mock_question.reset_mock()
|
||||||
|
|
||||||
|
@unittest.mock.patch('PyQt5.QtWidgets.QMessageBox.question')
|
||||||
|
@listen_for_events
|
||||||
|
def test_251_template_menu_multiple(self, mock_question):
|
||||||
|
mock_question.return_value = QtWidgets.QMessageBox.Yes
|
||||||
|
target_vm_names = ['work', 'personal', 'untrusted']
|
||||||
|
work = self.qapp.domains['work']
|
||||||
|
personal = self.qapp.domains['personal']
|
||||||
|
untrusted = self.qapp.domains['untrusted']
|
||||||
|
if any(vm.is_running() for vm in [work, personal, untrusted]):
|
||||||
|
self.skipTest('Any of work, personal, untrusted VM is running')
|
||||||
|
|
||||||
|
old_template = work.template
|
||||||
|
new_template = self._select_templatevm(
|
||||||
|
different_than=[str(work.template),
|
||||||
|
str(personal.template),
|
||||||
|
str(untrusted.template)])
|
||||||
|
# reset to previous value even in case of failure
|
||||||
|
self.addCleanup(functools.partial(
|
||||||
|
setattr, work, 'template', str(work.template)))
|
||||||
|
self.addCleanup(functools.partial(
|
||||||
|
setattr, personal, 'template', str(personal.template)))
|
||||||
|
self.addCleanup(functools.partial(
|
||||||
|
setattr, untrusted, 'template', str(untrusted.template)))
|
||||||
|
|
||||||
|
# set all to the same value
|
||||||
|
self._run_command_and_process_events(
|
||||||
|
['qvm-prefs', 'personal', 'template', str(work.template)], timeout=5)
|
||||||
|
self._run_command_and_process_events(
|
||||||
|
['qvm-prefs', 'untrusted', 'template', str(work.template)], timeout=5)
|
||||||
|
|
||||||
|
self._select_vms(target_vm_names)
|
||||||
|
|
||||||
|
# this is the method to get '==' operator working on icons...
|
||||||
|
on_icon = QIcon(":/on.png").pixmap(64).toImage()
|
||||||
|
transient_icon = QIcon(":/transient.png").pixmap(64).toImage()
|
||||||
|
off_icon = QIcon().pixmap(64).toImage()
|
||||||
|
for action in self.dialog.template_menu.actions():
|
||||||
|
if action.text() == str(old_template):
|
||||||
|
self.assertIn(
|
||||||
|
action.icon().pixmap(64).toImage(),
|
||||||
|
(on_icon, transient_icon))
|
||||||
|
else:
|
||||||
|
self.assertEqual(action.icon().pixmap(64).toImage(), off_icon)
|
||||||
|
|
||||||
|
# make one different
|
||||||
|
self._run_command_and_process_events(
|
||||||
|
['qvm-prefs', 'work', 'template', str(new_template)], timeout=5)
|
||||||
|
|
||||||
|
for action in self.dialog.template_menu.actions():
|
||||||
|
if action.text() == str(old_template):
|
||||||
|
self.assertEqual(action.icon().pixmap(64).toImage(), transient_icon)
|
||||||
|
elif action.text() == str(new_template):
|
||||||
|
self.assertEqual(action.icon().pixmap(64).toImage(), transient_icon)
|
||||||
|
else:
|
||||||
|
self.assertEqual(action.icon().pixmap(64).toImage(), off_icon)
|
||||||
|
|
||||||
|
# change all to the same value
|
||||||
|
for action in self.dialog.template_menu.actions():
|
||||||
|
if action.text() == str(new_template):
|
||||||
|
action.trigger()
|
||||||
|
break
|
||||||
|
else:
|
||||||
|
self.fail('{!s} template not found'.format(new_template))
|
||||||
|
# process events
|
||||||
|
self.loop.run_until_complete(asyncio.sleep(0))
|
||||||
|
mock_question.assert_called()
|
||||||
|
self.assertEqual(str(work.template), str(new_template))
|
||||||
|
self.assertEqual(str(personal.template), str(new_template))
|
||||||
|
self.assertEqual(str(untrusted.template), str(new_template))
|
||||||
|
|
||||||
|
|
||||||
@unittest.mock.patch('PyQt5.QtWidgets.QMessageBox.information')
|
@unittest.mock.patch('PyQt5.QtWidgets.QMessageBox.information')
|
||||||
@unittest.mock.patch('PyQt5.QtWidgets.QMessageBox.warning')
|
@unittest.mock.patch('PyQt5.QtWidgets.QMessageBox.warning')
|
||||||
def test_300_clear_threads(self, mock_warning, mock_info):
|
def test_300_clear_threads(self, mock_warning, mock_info):
|
||||||
@ -1258,19 +1493,28 @@ class QubeManagerTest(unittest.TestCase):
|
|||||||
return vm
|
return vm
|
||||||
return None
|
return None
|
||||||
|
|
||||||
def _select_templatevm(self, running=None):
|
def _select_templatevm(self, running=None, different_than=()):
|
||||||
for row in range(self.dialog.table.model().rowCount()):
|
for row in range(self.dialog.table.model().rowCount()):
|
||||||
template = self._get_table_item(row, "Template")
|
template = self._get_table_item(row, "Template")
|
||||||
vm = self._get_table_vm(row)
|
vm = self._get_table_vm(row)
|
||||||
if template == 'TemplateVM' and \
|
if template == 'TemplateVM' and \
|
||||||
|
(template not in different_than) and \
|
||||||
(running is None
|
(running is None
|
||||||
or (running and vm.is_running())
|
or (bool(running) == bool(vm.is_running()))):
|
||||||
or (not running and not vm.is_running())):
|
|
||||||
index = self.dialog.table.model().index(row, 0)
|
index = self.dialog.table.model().index(row, 0)
|
||||||
self.dialog.table.setCurrentIndex(index)
|
self.dialog.table.setCurrentIndex(index)
|
||||||
return vm
|
return vm
|
||||||
return None
|
return None
|
||||||
|
|
||||||
|
def _select_vms(self, vms: list):
|
||||||
|
self.dialog.table.selectionModel().clear()
|
||||||
|
mode = QtCore.QItemSelectionModel.Select | QtCore.QItemSelectionModel.Rows
|
||||||
|
for row in range(self.dialog.table.model().rowCount()):
|
||||||
|
vm = self._get_table_vm(row)
|
||||||
|
if str(vm) in vms:
|
||||||
|
index = self.dialog.table.model().index(row, 0)
|
||||||
|
self.dialog.table.selectionModel().select(index, mode)
|
||||||
|
|
||||||
def __check_sorting(self, column_name):
|
def __check_sorting(self, column_name):
|
||||||
last_text = None
|
last_text = None
|
||||||
last_vm = None
|
last_vm = None
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user