tests: add tests for qvm-template remove
This commit is contained in:
WillyPillow
parent
e00f35b9c3
commit
ed3e368673
@ -3892,3 +3892,205 @@ test-vm : Qubes template for fedora-31
|
|||||||
mock.call(args.cachedir, exist_ok=True)
|
mock.call(args.cachedir, exist_ok=True)
|
||||||
])
|
])
|
||||||
self.assertAllCalled()
|
self.assertAllCalled()
|
||||||
|
|
||||||
|
@mock.patch('qubesadmin.tools.qvm_remove.main')
|
||||||
|
@mock.patch('qubesadmin.tools.qvm_template.confirm_action')
|
||||||
|
def test_210_remove_success(self, mock_confirm, mock_remove):
|
||||||
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = (
|
||||||
|
b'0\x00vm1 class=TemplateVM state=Halted\n'
|
||||||
|
b'vm2 class=TemplateVM state=Halted\n'
|
||||||
|
)
|
||||||
|
args = argparse.Namespace(
|
||||||
|
templates=['vm1', 'vm2'],
|
||||||
|
yes=False
|
||||||
|
)
|
||||||
|
qubesadmin.tools.qvm_template.remove(args, self.app)
|
||||||
|
self.assertEqual(mock_confirm.mock_calls,
|
||||||
|
[mock.call(re_str(r'.*completely remove.*'), ['vm1', 'vm2'])])
|
||||||
|
self.assertEqual(mock_remove.mock_calls, [
|
||||||
|
mock.call(['--force', '--', 'vm1', 'vm2'], self.app)
|
||||||
|
])
|
||||||
|
self.assertAllCalled()
|
||||||
|
|
||||||
|
@mock.patch('qubesadmin.tools.qvm_kill.main')
|
||||||
|
@mock.patch('qubesadmin.tools.qvm_remove.main')
|
||||||
|
@mock.patch('qubesadmin.tools.qvm_template.confirm_action')
|
||||||
|
def test_211_remove_purge_disassoc_success(
|
||||||
|
self,
|
||||||
|
mock_confirm,
|
||||||
|
mock_remove,
|
||||||
|
mock_kill):
|
||||||
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = (
|
||||||
|
b'0\x00vm1 class=TemplateVM state=Halted\n'
|
||||||
|
b'vm2 class=TemplateVM state=Halted\n'
|
||||||
|
b'vm3 class=TemplateVM state=Halted\n'
|
||||||
|
b'vm4 class=TemplateVM state=Halted\n'
|
||||||
|
b'dummy class=TemplateVM state=Halted\n'
|
||||||
|
b'dummy-1 class=TemplateVM state=Halted\n'
|
||||||
|
)
|
||||||
|
self.app.expected_calls[
|
||||||
|
('dummy', 'admin.vm.feature.Get', 'template-dummy', None)] = \
|
||||||
|
b'0\x000'
|
||||||
|
self.app.expected_calls[
|
||||||
|
('dummy-1', 'admin.vm.feature.Get',
|
||||||
|
'template-dummy', None)] = \
|
||||||
|
b'0\x001'
|
||||||
|
self.app.expected_calls[
|
||||||
|
('vm2', 'admin.vm.property.Set',
|
||||||
|
'default_template', b'dummy-1')] = \
|
||||||
|
b'0\x00'
|
||||||
|
self.app.expected_calls[
|
||||||
|
('vm2', 'admin.vm.property.Set', 'template', b'dummy-1')] = \
|
||||||
|
b'0\x00'
|
||||||
|
self.app.expected_calls[
|
||||||
|
('vm3', 'admin.vm.property.Set', 'netvm', b'dummy-1')] = \
|
||||||
|
b'0\x00'
|
||||||
|
self.app.expected_calls[
|
||||||
|
('vm3', 'admin.vm.property.Set', 'template', b'dummy-1')] = \
|
||||||
|
b'0\x00'
|
||||||
|
self.app.expected_calls[
|
||||||
|
('vm4', 'admin.vm.property.Set', 'netvm', b'dummy-1')] = \
|
||||||
|
b'0\x00'
|
||||||
|
self.app.expected_calls[
|
||||||
|
('vm4', 'admin.vm.property.Set', 'template', b'dummy-1')] = \
|
||||||
|
b'0\x00'
|
||||||
|
self.app.expected_calls[
|
||||||
|
('dom0', 'admin.property.Set', 'updatevm', b'')] = \
|
||||||
|
b'0\x00'
|
||||||
|
args = argparse.Namespace(
|
||||||
|
templates=['vm1'],
|
||||||
|
yes=False
|
||||||
|
)
|
||||||
|
def deps(app, vm):
|
||||||
|
if vm == 'vm1':
|
||||||
|
return [(self.app.domains['vm2'], 'default_template'),
|
||||||
|
(self.app.domains['vm3'], 'netvm')]
|
||||||
|
if vm == 'vm2' or vm == 'vm3':
|
||||||
|
return [(self.app.domains['vm4'], 'netvm')]
|
||||||
|
if vm == 'vm4':
|
||||||
|
return [(None, 'updatevm')]
|
||||||
|
return []
|
||||||
|
with mock.patch('qubesadmin.utils.vm_dependencies') as mock_deps:
|
||||||
|
mock_deps.side_effect = deps
|
||||||
|
qubesadmin.tools.qvm_template.remove(args, self.app, purge=True)
|
||||||
|
# Once for purge (dependency detection) and
|
||||||
|
# one for disassoc (actually disassociating the dependencies
|
||||||
|
self.assertEqual(mock_deps.mock_calls, [
|
||||||
|
mock.call(self.app, self.app.domains['vm1']),
|
||||||
|
mock.call(self.app, self.app.domains['vm2']),
|
||||||
|
mock.call(self.app, self.app.domains['vm3']),
|
||||||
|
mock.call(self.app, self.app.domains['vm4']),
|
||||||
|
mock.call(self.app, self.app.domains['vm1']),
|
||||||
|
mock.call(self.app, self.app.domains['vm2']),
|
||||||
|
mock.call(self.app, self.app.domains['vm3']),
|
||||||
|
mock.call(self.app, self.app.domains['vm4'])
|
||||||
|
])
|
||||||
|
self.assertEqual(mock_confirm.mock_calls, [
|
||||||
|
mock.call(re_str(r'.*completely remove.*'),
|
||||||
|
['vm1', 'vm2', 'vm3', 'vm4']),
|
||||||
|
mock.call(re_str(r'.*completely remove.*'),
|
||||||
|
['vm1', 'vm2', 'vm3', 'vm4']),
|
||||||
|
mock.call(re_str(r'.*completely remove.*'),
|
||||||
|
['vm1', 'vm2', 'vm3', 'vm4'])
|
||||||
|
])
|
||||||
|
self.assertEqual(mock_remove.mock_calls, [
|
||||||
|
mock.call(['--force', '--', 'vm1', 'vm2', 'vm3', 'vm4', 'dummy-1'],
|
||||||
|
self.app)
|
||||||
|
])
|
||||||
|
self.assertEqual(mock_kill.mock_calls, [
|
||||||
|
mock.call(['--', 'vm1', 'vm2', 'vm3', 'vm4', 'dummy-1'], self.app)
|
||||||
|
])
|
||||||
|
self.assertAllCalled()
|
||||||
|
|
||||||
|
@mock.patch('qubesadmin.tools.qvm_kill.main')
|
||||||
|
@mock.patch('qubesadmin.tools.qvm_remove.main')
|
||||||
|
@mock.patch('qubesadmin.tools.qvm_template.confirm_action')
|
||||||
|
def test_212_remove_disassoc_success(
|
||||||
|
self,
|
||||||
|
mock_confirm,
|
||||||
|
mock_remove,
|
||||||
|
mock_kill):
|
||||||
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = (
|
||||||
|
b'0\x00vm1 class=TemplateVM state=Halted\n'
|
||||||
|
b'vm2 class=TemplateVM state=Halted\n'
|
||||||
|
b'vm3 class=TemplateVM state=Halted\n'
|
||||||
|
b'vm4 class=TemplateVM state=Halted\n'
|
||||||
|
b'dummy class=TemplateVM state=Halted\n'
|
||||||
|
b'dummy-1 class=TemplateVM state=Halted\n'
|
||||||
|
)
|
||||||
|
self.app.expected_calls[
|
||||||
|
('dummy', 'admin.vm.feature.Get', 'template-dummy', None)] = \
|
||||||
|
b'0\x000'
|
||||||
|
self.app.expected_calls[
|
||||||
|
('dummy-1', 'admin.vm.feature.Get',
|
||||||
|
'template-dummy', None)] = \
|
||||||
|
b'0\x001'
|
||||||
|
self.app.expected_calls[
|
||||||
|
('vm2', 'admin.vm.property.Set',
|
||||||
|
'default_template', b'dummy-1')] = \
|
||||||
|
b'0\x00'
|
||||||
|
self.app.expected_calls[
|
||||||
|
('vm2', 'admin.vm.property.Set', 'template', b'dummy-1')] = \
|
||||||
|
b'0\x00'
|
||||||
|
self.app.expected_calls[
|
||||||
|
('vm3', 'admin.vm.property.Set', 'netvm', b'dummy-1')] = \
|
||||||
|
b'0\x00'
|
||||||
|
self.app.expected_calls[
|
||||||
|
('vm3', 'admin.vm.property.Set', 'template', b'dummy-1')] = \
|
||||||
|
b'0\x00'
|
||||||
|
self.app.expected_calls[
|
||||||
|
('vm4', 'admin.vm.property.Set', 'netvm', b'dummy-1')] = \
|
||||||
|
b'0\x00'
|
||||||
|
self.app.expected_calls[
|
||||||
|
('vm4', 'admin.vm.property.Set', 'template', b'dummy-1')] = \
|
||||||
|
b'0\x00'
|
||||||
|
self.app.expected_calls[
|
||||||
|
('dom0', 'admin.property.Set', 'updatevm', b'')] = \
|
||||||
|
b'0\x00'
|
||||||
|
args = argparse.Namespace(
|
||||||
|
templates=['vm1', 'vm2', 'vm3', 'vm4'],
|
||||||
|
yes=False
|
||||||
|
)
|
||||||
|
def deps(app, vm):
|
||||||
|
if vm == 'vm1':
|
||||||
|
return [(self.app.domains['vm2'], 'default_template'),
|
||||||
|
(self.app.domains['vm3'], 'netvm')]
|
||||||
|
if vm == 'vm2' or vm == 'vm3':
|
||||||
|
return [(self.app.domains['vm4'], 'netvm')]
|
||||||
|
if vm == 'vm4':
|
||||||
|
return [(None, 'updatevm')]
|
||||||
|
return []
|
||||||
|
with mock.patch('qubesadmin.utils.vm_dependencies') as mock_deps:
|
||||||
|
mock_deps.side_effect = deps
|
||||||
|
qubesadmin.tools.qvm_template.remove(args, self.app, disassoc=True)
|
||||||
|
self.assertEqual(mock_deps.mock_calls, [
|
||||||
|
mock.call(self.app, self.app.domains['vm1']),
|
||||||
|
mock.call(self.app, self.app.domains['vm2']),
|
||||||
|
mock.call(self.app, self.app.domains['vm3']),
|
||||||
|
mock.call(self.app, self.app.domains['vm4'])
|
||||||
|
])
|
||||||
|
self.assertEqual(mock_confirm.mock_calls, [
|
||||||
|
mock.call(re_str(r'.*completely remove.*'),
|
||||||
|
['vm1', 'vm2', 'vm3', 'vm4'])
|
||||||
|
])
|
||||||
|
self.assertEqual(mock_remove.mock_calls, [
|
||||||
|
mock.call(['--force', '--', 'vm1', 'vm2', 'vm3', 'vm4'],
|
||||||
|
self.app)
|
||||||
|
])
|
||||||
|
self.assertEqual(mock_kill.mock_calls, [
|
||||||
|
mock.call(['--', 'vm1', 'vm2', 'vm3', 'vm4'], self.app)
|
||||||
|
])
|
||||||
|
self.assertAllCalled()
|
||||||
|
|
||||||
|
def test_213_remove_fail_nodomain(self):
|
||||||
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
||||||
|
b'0\x00vm1 class=TemplateVM state=Halted\n'
|
||||||
|
args = argparse.Namespace(
|
||||||
|
templates=['vm0'],
|
||||||
|
yes=False
|
||||||
|
)
|
||||||
|
with mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
||||||
|
with self.assertRaises(SystemExit):
|
||||||
|
qubesadmin.tools.qvm_template.remove(args, self.app)
|
||||||
|
self.assertTrue('no such domain:' in mock_err.getvalue())
|
||||||
|
self.assertAllCalled()
|
||||||
|
|||||||
@ -1360,13 +1360,16 @@ def remove(
|
|||||||
|
|
||||||
if not args.yes:
|
if not args.yes:
|
||||||
repeat = 3 if purge else 1
|
repeat = 3 if purge else 1
|
||||||
|
# XXX: Mutating the list later seems to break the tests...
|
||||||
|
remove_list_copy = remove_list.copy()
|
||||||
for _ in range(repeat):
|
for _ in range(repeat):
|
||||||
confirm_action(
|
confirm_action(
|
||||||
'This will completely remove the selected VM(s)...',
|
'This will completely remove the selected VM(s)...',
|
||||||
remove_list)
|
remove_list_copy)
|
||||||
|
|
||||||
if disassoc:
|
if disassoc:
|
||||||
# Remove the dummy afterwards if we're purging
|
# Remove the dummy afterwards if we're purging
|
||||||
|
# as nothing should depend on it in the end
|
||||||
remove_dummy = purge
|
remove_dummy = purge
|
||||||
# Create dummy template; handle name collisions
|
# Create dummy template; handle name collisions
|
||||||
orig_dummy = dummy
|
orig_dummy = dummy
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user