5279 lines
214 KiB
Python
5279 lines
214 KiB
Python
import re
|
|
from unittest import mock
|
|
import argparse
|
|
import asyncio
|
|
import datetime
|
|
import io
|
|
import os
|
|
import pathlib
|
|
import subprocess
|
|
import tempfile
|
|
|
|
import fcntl
|
|
import rpm
|
|
|
|
import qubesadmin.tests
|
|
import qubesadmin.tools.qvm_template
|
|
|
|
class re_str(str):
|
|
def __eq__(self, other):
|
|
return bool(re.match(self, other))
|
|
|
|
def __hash__(self):
|
|
return super().__hash__()
|
|
|
|
class TC_00_qvm_template(qubesadmin.tests.QubesTestCase):
|
|
def setUp(self):
|
|
# Print str(list) directly so that the output is consistent no matter
|
|
# which implementation of `column` we use
|
|
self.mock_table = mock.patch('qubesadmin.tools.print_table')
|
|
mock_table = self.mock_table.start()
|
|
def print_table(table, *args):
|
|
print(str(table))
|
|
mock_table.side_effect = print_table
|
|
|
|
super().setUp()
|
|
|
|
def tearDown(self):
|
|
self.mock_table.stop()
|
|
super().tearDown()
|
|
|
|
@mock.patch('rpm.TransactionSet')
|
|
@mock.patch('subprocess.check_call')
|
|
@mock.patch('subprocess.check_output')
|
|
def test_000_verify_rpm_success(self, mock_proc, mock_call, mock_ts):
|
|
# Just return a dict instead of rpm.hdr
|
|
hdr = {
|
|
rpm.RPMTAG_SIGPGP: 'xxx', # non-empty
|
|
rpm.RPMTAG_SIGGPG: 'xxx', # non-empty
|
|
rpm.RPMTAG_NAME: 'qubes-template-test-vm',
|
|
}
|
|
mock_ts.return_value.hdrFromFdno.return_value = hdr
|
|
mock_proc.return_value = b'dummy.rpm: digests signatures OK\n'
|
|
ret = qubesadmin.tools.qvm_template.verify_rpm('/dev/null',
|
|
'/path/to/key', template_name='test-vm')
|
|
mock_call.assert_called_once()
|
|
mock_proc.assert_called_once()
|
|
self.assertEqual(hdr, ret)
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('rpm.TransactionSet')
|
|
@mock.patch('subprocess.check_call')
|
|
@mock.patch('subprocess.check_output')
|
|
def test_001_verify_rpm_nosig_fail(self, mock_proc, mock_call, mock_ts):
|
|
# Just return a dict instead of rpm.hdr
|
|
hdr = {
|
|
rpm.RPMTAG_SIGPGP: None, # empty
|
|
rpm.RPMTAG_SIGGPG: None, # empty
|
|
}
|
|
mock_ts.return_value.hdrFromFdno.return_value = hdr
|
|
mock_proc.return_value = b'dummy.rpm: digests OK\n'
|
|
with self.assertRaises(Exception) as e:
|
|
qubesadmin.tools.qvm_template.verify_rpm('/dev/null',
|
|
'/path/to/key')
|
|
mock_call.assert_called_once()
|
|
mock_proc.assert_called_once()
|
|
self.assertIn('Signature verification failed', e.exception.args[0])
|
|
mock_ts.assert_not_called()
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('rpm.TransactionSet')
|
|
@mock.patch('subprocess.check_call')
|
|
@mock.patch('subprocess.check_output')
|
|
def test_002_verify_rpm_nosig_success(self, mock_proc, mock_call, mock_ts):
|
|
# Just return a dict instead of rpm.hdr
|
|
hdr = {
|
|
rpm.RPMTAG_SIGPGP: None, # empty
|
|
rpm.RPMTAG_SIGGPG: None, # empty
|
|
}
|
|
mock_ts.return_value.hdrFromFdno.return_value = hdr
|
|
mock_proc.return_value = b'dummy.rpm: digests OK\n'
|
|
ret = qubesadmin.tools.qvm_template.verify_rpm('/dev/null',
|
|
'/path/to/key', True)
|
|
mock_proc.assert_not_called()
|
|
mock_call.assert_not_called()
|
|
self.assertEqual(ret, hdr)
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('rpm.TransactionSet')
|
|
@mock.patch('subprocess.check_call')
|
|
@mock.patch('subprocess.check_output')
|
|
def test_003_verify_rpm_badsig_fail(self, mock_proc, mock_call, mock_ts):
|
|
mock_proc.side_effect = subprocess.CalledProcessError(1,
|
|
['rpmkeys', '--checksig'], b'/dev/null: digests SIGNATURES NOT OK\n')
|
|
with self.assertRaises(Exception) as e:
|
|
qubesadmin.tools.qvm_template.verify_rpm('/dev/null',
|
|
'/path/to/key')
|
|
mock_call.assert_called_once()
|
|
mock_proc.assert_called_once()
|
|
self.assertIn('Signature verification failed', e.exception.args[0])
|
|
mock_ts.assert_not_called()
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('rpm.TransactionSet')
|
|
@mock.patch('subprocess.check_call')
|
|
@mock.patch('subprocess.check_output')
|
|
def test_004_verify_rpm_badname(self, mock_proc, mock_call, mock_ts):
|
|
mock_proc.return_value = b'/dev/null: digests signatures OK\n'
|
|
hdr = {
|
|
rpm.RPMTAG_SIGPGP: 'xxx', # non-empty
|
|
rpm.RPMTAG_SIGGPG: 'xxx', # non-empty
|
|
rpm.RPMTAG_NAME: 'qubes-template-unexpected',
|
|
}
|
|
mock_ts.return_value.hdrFromFdno.return_value = hdr
|
|
with self.assertRaises(
|
|
qubesadmin.tools.qvm_template.SignatureVerificationError) as e:
|
|
qubesadmin.tools.qvm_template.verify_rpm('/dev/null',
|
|
'/path/to/key', template_name='test-vm')
|
|
mock_call.assert_called_once()
|
|
mock_proc.assert_called_once()
|
|
self.assertIn('package does not match expected template name',
|
|
e.exception.args[0])
|
|
mock_ts.assert_called_once()
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('subprocess.Popen')
|
|
def test_010_extract_rpm_success(self, mock_popen):
|
|
pipe = mock.Mock()
|
|
mock_popen.return_value.stdout = pipe
|
|
mock_popen.return_value.wait.return_value = 0
|
|
with tempfile.NamedTemporaryFile() as fd, \
|
|
tempfile.TemporaryDirectory() as dir:
|
|
path = fd.name
|
|
dirpath = dir
|
|
ret = qubesadmin.tools.qvm_template.extract_rpm(
|
|
'test-vm', path, dirpath)
|
|
self.assertEqual(ret, True)
|
|
self.assertEqual(mock_popen.mock_calls, [
|
|
mock.call(['rpm2cpio', path], stdout=subprocess.PIPE),
|
|
mock.call([
|
|
'cpio',
|
|
'-idm',
|
|
'-D',
|
|
dirpath,
|
|
'./var/lib/qubes/vm-templates/test-vm/*'
|
|
], stdin=pipe, stdout=subprocess.DEVNULL),
|
|
mock.call().wait(),
|
|
mock.call().wait()
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('subprocess.Popen')
|
|
def test_011_extract_rpm_fail(self, mock_popen):
|
|
pipe = mock.Mock()
|
|
mock_popen.return_value.stdout = pipe
|
|
mock_popen.return_value.wait.return_value = 1
|
|
with tempfile.NamedTemporaryFile() as fd, \
|
|
tempfile.TemporaryDirectory() as dir:
|
|
path = fd.name
|
|
dirpath = dir
|
|
ret = qubesadmin.tools.qvm_template.extract_rpm(
|
|
'test-vm', path, dirpath)
|
|
self.assertEqual(ret, False)
|
|
self.assertEqual(mock_popen.mock_calls, [
|
|
mock.call(['rpm2cpio', path], stdout=subprocess.PIPE),
|
|
mock.call([
|
|
'cpio',
|
|
'-idm',
|
|
'-D',
|
|
dirpath,
|
|
'./var/lib/qubes/vm-templates/test-vm/*'
|
|
], stdin=pipe, stdout=subprocess.DEVNULL),
|
|
mock.call().wait()
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.get_keys_for_repos')
|
|
def test_090_install_lock(self, mock_get_keys):
|
|
class SuccessError(Exception):
|
|
pass
|
|
mock_get_keys.side_effect = SuccessError
|
|
with mock.patch('qubesadmin.tools.qvm_template.LOCK_FILE', '/tmp/test.lock'):
|
|
with self.subTest('not locked'):
|
|
with self.assertRaises(SuccessError):
|
|
# args don't matter
|
|
qubesadmin.tools.qvm_template.install(mock.MagicMock(), None)
|
|
self.assertFalse(os.path.exists('/tmp/test.lock'))
|
|
|
|
with self.subTest('lock exists but unlocked'):
|
|
with open('/tmp/test.lock', 'w') as f:
|
|
with self.assertRaises(SuccessError):
|
|
# args don't matter
|
|
qubesadmin.tools.qvm_template.install(mock.MagicMock(), None)
|
|
self.assertFalse(os.path.exists('/tmp/test.lock'))
|
|
with self.subTest('locked'):
|
|
with open('/tmp/test.lock', 'w') as f:
|
|
fcntl.flock(f, fcntl.LOCK_EX)
|
|
with self.assertRaises(
|
|
qubesadmin.tools.qvm_template.AlreadyRunning):
|
|
# args don't matter
|
|
qubesadmin.tools.qvm_template.install(mock.MagicMock(), None)
|
|
# and not cleaned up then
|
|
self.assertTrue(os.path.exists('/tmp/test.lock'))
|
|
|
|
def add_new_vm_side_effect(self, *args, **kwargs):
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\0test-vm class=TemplateVM state=Halted\n'
|
|
self.app.domains.clear_cache()
|
|
return self.app.domains['test-vm']
|
|
|
|
@mock.patch('os.rename')
|
|
@mock.patch('os.makedirs')
|
|
@mock.patch('subprocess.check_call')
|
|
@mock.patch('qubesadmin.tools.qvm_template.confirm_action')
|
|
@mock.patch('qubesadmin.tools.qvm_template.extract_rpm')
|
|
@mock.patch('qubesadmin.tools.qvm_template.download')
|
|
@mock.patch('qubesadmin.tools.qvm_template.get_dl_list')
|
|
@mock.patch('qubesadmin.tools.qvm_template.verify_rpm')
|
|
def test_100_install_local_success(
|
|
self,
|
|
mock_verify,
|
|
mock_dl_list,
|
|
mock_dl,
|
|
mock_extract,
|
|
mock_confirm,
|
|
mock_call,
|
|
mock_mkdirs,
|
|
mock_rename):
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = b'0\0'
|
|
build_time = '2020-09-01 14:30:00' # 1598970600
|
|
install_time = '2020-09-01 15:30:00'
|
|
for key, val in [
|
|
('name', 'test-vm'),
|
|
('epoch', '2'),
|
|
('version', '4.1'),
|
|
('release', '2020'),
|
|
('reponame', '@commandline'),
|
|
('buildtime', build_time),
|
|
('installtime', install_time),
|
|
('license', 'GPL'),
|
|
('url', 'https://qubes-os.org'),
|
|
('summary', 'Summary'),
|
|
('description', 'Desc|desc')]:
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Set',
|
|
f'template-{key}',
|
|
val.encode())] = b'0\0'
|
|
mock_verify.return_value = {
|
|
rpm.RPMTAG_NAME : 'qubes-template-test-vm',
|
|
rpm.RPMTAG_BUILDTIME : 1598970600,
|
|
rpm.RPMTAG_DESCRIPTION : 'Desc\ndesc',
|
|
rpm.RPMTAG_EPOCHNUM : 2,
|
|
rpm.RPMTAG_LICENSE : 'GPL',
|
|
rpm.RPMTAG_RELEASE : '2020',
|
|
rpm.RPMTAG_SUMMARY : 'Summary',
|
|
rpm.RPMTAG_URL : 'https://qubes-os.org',
|
|
rpm.RPMTAG_VERSION : '4.1'
|
|
}
|
|
mock_dl_list.return_value = {}
|
|
mock_call.side_effect = self.add_new_vm_side_effect
|
|
mock_time = mock.Mock(wraps=datetime.datetime)
|
|
mock_time.now.return_value = \
|
|
datetime.datetime(2020, 9, 1, 15, 30, tzinfo=datetime.timezone.utc)
|
|
with mock.patch('qubesadmin.tools.qvm_template.LOCK_FILE', '/tmp/test.lock'), \
|
|
mock.patch('datetime.datetime', new=mock_time), \
|
|
mock.patch('tempfile.TemporaryDirectory') as mock_tmpdir, \
|
|
mock.patch('sys.stderr', new=io.StringIO()) as mock_err, \
|
|
tempfile.NamedTemporaryFile(suffix='.rpm') as template_file:
|
|
path = template_file.name
|
|
args = argparse.Namespace(
|
|
templates=[path],
|
|
keyring='/tmp/keyring.gpg',
|
|
nogpgcheck=False,
|
|
cachedir='/var/cache/qvm-template',
|
|
repo_files=[],
|
|
releasever='4.1',
|
|
yes=False,
|
|
allow_pv=False,
|
|
pool=None
|
|
)
|
|
mock_tmpdir.return_value.__enter__.return_value = \
|
|
'/var/tmp/qvm-template-tmpdir'
|
|
qubesadmin.tools.qvm_template.install(args, self.app)
|
|
# Downloaded package should not be removed
|
|
self.assertTrue(os.path.exists(path))
|
|
# Attempt to get download list
|
|
selector = qubesadmin.tools.qvm_template.VersionSelector.LATEST
|
|
self.assertEqual(mock_dl_list.mock_calls, [
|
|
mock.call(args, self.app, version_selector=selector)
|
|
])
|
|
# Nothing downloaded
|
|
mock_dl.assert_called_with(args, self.app,
|
|
dl_list={}, version_selector=selector)
|
|
mock_verify.assert_called_once_with(template_file.name, '/tmp/keyring.gpg',
|
|
nogpgcheck=False)
|
|
# Package is extracted
|
|
mock_extract.assert_called_with('test-vm', path,
|
|
'/var/tmp/qvm-template-tmpdir')
|
|
# No packages overwritten, so no confirm needed
|
|
self.assertEqual(mock_confirm.mock_calls, [])
|
|
# qvm-template-postprocess is called
|
|
self.assertEqual(mock_call.mock_calls, [
|
|
mock.call([
|
|
'qvm-template-postprocess',
|
|
'--really',
|
|
'--no-installed-by-rpm',
|
|
'post-install',
|
|
'test-vm',
|
|
'/var/tmp/qvm-template-tmpdir'
|
|
'/var/lib/qubes/vm-templates/test-vm'
|
|
])
|
|
])
|
|
# Cache directory created
|
|
self.assertEqual(mock_mkdirs.mock_calls, [
|
|
mock.call(args.cachedir, exist_ok=True)
|
|
])
|
|
# No templates downloaded, thus no renames needed
|
|
self.assertEqual(mock_rename.mock_calls, [])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('os.rename')
|
|
@mock.patch('os.makedirs')
|
|
@mock.patch('subprocess.check_call')
|
|
@mock.patch('qubesadmin.tools.qvm_template.confirm_action')
|
|
@mock.patch('qubesadmin.tools.qvm_template.extract_rpm')
|
|
@mock.patch('qubesadmin.tools.qvm_template.download')
|
|
@mock.patch('qubesadmin.tools.qvm_template.get_dl_list')
|
|
@mock.patch('qubesadmin.tools.qvm_template.verify_rpm')
|
|
def test_101_install_local_postprocargs_success(
|
|
self,
|
|
mock_verify,
|
|
mock_dl_list,
|
|
mock_dl,
|
|
mock_extract,
|
|
mock_confirm,
|
|
mock_call,
|
|
mock_mkdirs,
|
|
mock_rename):
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = b'0\0'
|
|
build_time = '2020-09-01 14:30:00' # 1598970600
|
|
install_time = '2020-09-01 15:30:00'
|
|
for key, val in [
|
|
('name', 'test-vm'),
|
|
('epoch', '2'),
|
|
('version', '4.1'),
|
|
('release', '2020'),
|
|
('reponame', '@commandline'),
|
|
('buildtime', build_time),
|
|
('installtime', install_time),
|
|
('license', 'GPL'),
|
|
('url', 'https://qubes-os.org'),
|
|
('summary', 'Summary'),
|
|
('description', 'Desc|desc')]:
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Set',
|
|
f'template-{key}',
|
|
val.encode())] = b'0\0'
|
|
mock_verify.return_value = {
|
|
rpm.RPMTAG_NAME : 'qubes-template-test-vm',
|
|
rpm.RPMTAG_BUILDTIME : 1598970600,
|
|
rpm.RPMTAG_DESCRIPTION : 'Desc\ndesc',
|
|
rpm.RPMTAG_EPOCHNUM : 2,
|
|
rpm.RPMTAG_LICENSE : 'GPL',
|
|
rpm.RPMTAG_RELEASE : '2020',
|
|
rpm.RPMTAG_SUMMARY : 'Summary',
|
|
rpm.RPMTAG_URL : 'https://qubes-os.org',
|
|
rpm.RPMTAG_VERSION : '4.1'
|
|
}
|
|
mock_dl_list.return_value = {}
|
|
mock_call.side_effect = self.add_new_vm_side_effect
|
|
mock_time = mock.Mock(wraps=datetime.datetime)
|
|
mock_time.now.return_value = \
|
|
datetime.datetime(2020, 9, 1, 15, 30, tzinfo=datetime.timezone.utc)
|
|
with mock.patch('qubesadmin.tools.qvm_template.LOCK_FILE', '/tmp/test.lock'), \
|
|
mock.patch('datetime.datetime', new=mock_time), \
|
|
mock.patch('tempfile.TemporaryDirectory') as mock_tmpdir, \
|
|
mock.patch('sys.stderr', new=io.StringIO()) as mock_err, \
|
|
tempfile.NamedTemporaryFile(suffix='.rpm') as template_file:
|
|
path = template_file.name
|
|
args = argparse.Namespace(
|
|
templates=[path],
|
|
keyring='/tmp',
|
|
nogpgcheck=False,
|
|
cachedir='/var/cache/qvm-template',
|
|
repo_files=[],
|
|
releasever='4.1',
|
|
yes=False,
|
|
allow_pv=True,
|
|
pool='my-pool'
|
|
)
|
|
mock_tmpdir.return_value.__enter__.return_value = \
|
|
'/var/tmp/qvm-template-tmpdir'
|
|
qubesadmin.tools.qvm_template.install(args, self.app)
|
|
# Attempt to get download list
|
|
selector = qubesadmin.tools.qvm_template.VersionSelector.LATEST
|
|
self.assertEqual(mock_dl_list.mock_calls, [
|
|
mock.call(args, self.app, version_selector=selector)
|
|
])
|
|
# Nothing downloaded
|
|
mock_dl.assert_called_with(args, self.app,
|
|
dl_list={}, version_selector=selector)
|
|
# Package is extracted
|
|
mock_extract.assert_called_with('test-vm', path,
|
|
'/var/tmp/qvm-template-tmpdir')
|
|
# No packages overwritten, so no confirm needed
|
|
self.assertEqual(mock_confirm.mock_calls, [])
|
|
# qvm-template-postprocess is called
|
|
self.assertEqual(mock_call.mock_calls, [
|
|
mock.call([
|
|
'qvm-template-postprocess',
|
|
'--really',
|
|
'--no-installed-by-rpm',
|
|
'--allow-pv',
|
|
'--pool',
|
|
'my-pool',
|
|
'post-install',
|
|
'test-vm',
|
|
'/var/tmp/qvm-template-tmpdir'
|
|
'/var/lib/qubes/vm-templates/test-vm'
|
|
])
|
|
])
|
|
# Cache directory created
|
|
self.assertEqual(mock_mkdirs.mock_calls, [
|
|
mock.call(args.cachedir, exist_ok=True)
|
|
])
|
|
# No templates downloaded, thus no renames needed
|
|
self.assertEqual(mock_rename.mock_calls, [])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('os.rename')
|
|
@mock.patch('os.makedirs')
|
|
@mock.patch('subprocess.check_call')
|
|
@mock.patch('qubesadmin.tools.qvm_template.confirm_action')
|
|
@mock.patch('qubesadmin.tools.qvm_template.extract_rpm')
|
|
@mock.patch('qubesadmin.tools.qvm_template.download')
|
|
@mock.patch('qubesadmin.tools.qvm_template.get_dl_list')
|
|
@mock.patch('qubesadmin.tools.qvm_template.verify_rpm')
|
|
def test_102_install_local_badsig_fail(
|
|
self,
|
|
mock_verify,
|
|
mock_dl_list,
|
|
mock_dl,
|
|
mock_extract,
|
|
mock_confirm,
|
|
mock_call,
|
|
mock_mkdirs,
|
|
mock_rename):
|
|
mock_verify.return_value = None
|
|
mock_time = mock.Mock(wraps=datetime.datetime)
|
|
with mock.patch('qubesadmin.tools.qvm_template.LOCK_FILE', '/tmp/test.lock'), \
|
|
mock.patch('datetime.datetime', new=mock_time), \
|
|
mock.patch('tempfile.TemporaryDirectory') as mock_tmpdir, \
|
|
mock.patch('sys.stderr', new=io.StringIO()) as mock_err, \
|
|
tempfile.NamedTemporaryFile(suffix='.rpm') as template_file:
|
|
path = template_file.name
|
|
args = argparse.Namespace(
|
|
templates=[path],
|
|
keyring='/tmp',
|
|
nogpgcheck=False,
|
|
cachedir='/var/cache/qvm-template',
|
|
repo_files=[],
|
|
releasever='4.1',
|
|
yes=False,
|
|
allow_pv=False,
|
|
pool=None
|
|
)
|
|
mock_tmpdir.return_value.__enter__.return_value = \
|
|
'/var/tmp/qvm-template-tmpdir'
|
|
# Should raise parser.error
|
|
with self.assertRaises(SystemExit):
|
|
qubesadmin.tools.qvm_template.install(args, self.app)
|
|
# Check error message
|
|
self.assertTrue('verification failed' in mock_err.getvalue())
|
|
# Should not be executed:
|
|
self.assertEqual(mock_dl_list.mock_calls, [])
|
|
self.assertEqual(mock_dl.mock_calls, [])
|
|
self.assertEqual(mock_extract.mock_calls, [])
|
|
self.assertEqual(mock_confirm.mock_calls, [])
|
|
self.assertEqual(mock_call.mock_calls, [])
|
|
self.assertEqual(mock_rename.mock_calls, [])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('os.rename')
|
|
@mock.patch('os.makedirs')
|
|
@mock.patch('subprocess.check_call')
|
|
@mock.patch('qubesadmin.tools.qvm_template.confirm_action')
|
|
@mock.patch('qubesadmin.tools.qvm_template.extract_rpm')
|
|
@mock.patch('qubesadmin.tools.qvm_template.download')
|
|
@mock.patch('qubesadmin.tools.qvm_template.get_dl_list')
|
|
@mock.patch('qubesadmin.tools.qvm_template.verify_rpm')
|
|
def test_103_install_local_exists_fail(
|
|
self,
|
|
mock_verify,
|
|
mock_dl_list,
|
|
mock_dl,
|
|
mock_extract,
|
|
mock_confirm,
|
|
mock_call,
|
|
mock_mkdirs,
|
|
mock_rename):
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\0test-vm class=TemplateVM state=Halted\n'
|
|
mock_verify.return_value = {
|
|
rpm.RPMTAG_NAME : 'qubes-template-test-vm',
|
|
rpm.RPMTAG_BUILDTIME : 1598970600,
|
|
rpm.RPMTAG_DESCRIPTION : 'Desc\ndesc',
|
|
rpm.RPMTAG_EPOCHNUM : 2,
|
|
rpm.RPMTAG_LICENSE : 'GPL',
|
|
rpm.RPMTAG_RELEASE : '2020',
|
|
rpm.RPMTAG_SUMMARY : 'Summary',
|
|
rpm.RPMTAG_URL : 'https://qubes-os.org',
|
|
rpm.RPMTAG_VERSION : '4.1'
|
|
}
|
|
mock_dl_list.return_value = {}
|
|
mock_time = mock.Mock(wraps=datetime.datetime)
|
|
with mock.patch('qubesadmin.tools.qvm_template.LOCK_FILE', '/tmp/test.lock'), \
|
|
mock.patch('datetime.datetime', new=mock_time), \
|
|
mock.patch('tempfile.TemporaryDirectory') as mock_tmpdir, \
|
|
mock.patch('sys.stderr', new=io.StringIO()) as mock_err, \
|
|
tempfile.NamedTemporaryFile(suffix='.rpm') as template_file:
|
|
path = template_file.name
|
|
args = argparse.Namespace(
|
|
templates=[path],
|
|
keyring='/tmp',
|
|
nogpgcheck=False,
|
|
cachedir='/var/cache/qvm-template',
|
|
repo_files=[],
|
|
releasever='4.1',
|
|
yes=False,
|
|
allow_pv=False,
|
|
pool=None
|
|
)
|
|
mock_tmpdir.return_value.__enter__.return_value = \
|
|
'/var/tmp/qvm-template-tmpdir'
|
|
qubesadmin.tools.qvm_template.install(args, self.app)
|
|
# Check warning message
|
|
self.assertTrue('already installed' in mock_err.getvalue())
|
|
# Attempt to get download list
|
|
selector = qubesadmin.tools.qvm_template.VersionSelector.LATEST
|
|
self.assertEqual(mock_dl_list.mock_calls, [
|
|
mock.call(args, self.app, version_selector=selector)
|
|
])
|
|
# Nothing downloaded
|
|
self.assertEqual(mock_dl.mock_calls, [
|
|
mock.call(args, self.app,
|
|
dl_list={}, version_selector=selector)
|
|
])
|
|
# Should not be executed:
|
|
self.assertEqual(mock_extract.mock_calls, [])
|
|
self.assertEqual(mock_confirm.mock_calls, [])
|
|
self.assertEqual(mock_call.mock_calls, [])
|
|
self.assertEqual(mock_rename.mock_calls, [])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('os.rename')
|
|
@mock.patch('os.makedirs')
|
|
@mock.patch('subprocess.check_call')
|
|
@mock.patch('qubesadmin.tools.qvm_template.confirm_action')
|
|
@mock.patch('qubesadmin.tools.qvm_template.extract_rpm')
|
|
@mock.patch('qubesadmin.tools.qvm_template.download')
|
|
@mock.patch('qubesadmin.tools.qvm_template.get_dl_list')
|
|
@mock.patch('qubesadmin.tools.qvm_template.verify_rpm')
|
|
def test_104_install_local_badpkgname_fail(
|
|
self,
|
|
mock_verify,
|
|
mock_dl_list,
|
|
mock_dl,
|
|
mock_extract,
|
|
mock_confirm,
|
|
mock_call,
|
|
mock_mkdirs,
|
|
mock_rename):
|
|
mock_verify.return_value = {
|
|
rpm.RPMTAG_NAME : 'Xqubes-template-test-vm',
|
|
rpm.RPMTAG_BUILDTIME : 1598970600,
|
|
rpm.RPMTAG_DESCRIPTION : 'Desc\ndesc',
|
|
rpm.RPMTAG_EPOCHNUM : 2,
|
|
rpm.RPMTAG_LICENSE : 'GPL',
|
|
rpm.RPMTAG_RELEASE : '2020',
|
|
rpm.RPMTAG_SUMMARY : 'Summary',
|
|
rpm.RPMTAG_URL : 'https://qubes-os.org',
|
|
rpm.RPMTAG_VERSION : '4.1'
|
|
}
|
|
mock_time = mock.Mock(wraps=datetime.datetime)
|
|
with mock.patch('qubesadmin.tools.qvm_template.LOCK_FILE', '/tmp/test.lock'), \
|
|
mock.patch('datetime.datetime', new=mock_time), \
|
|
mock.patch('tempfile.TemporaryDirectory') as mock_tmpdir, \
|
|
mock.patch('sys.stderr', new=io.StringIO()) as mock_err, \
|
|
tempfile.NamedTemporaryFile(suffix='.rpm') as template_file:
|
|
path = template_file.name
|
|
args = argparse.Namespace(
|
|
templates=[path],
|
|
keyring='/tmp',
|
|
nogpgcheck=False,
|
|
cachedir='/var/cache/qvm-template',
|
|
repo_files=[],
|
|
releasever='4.1',
|
|
yes=False,
|
|
allow_pv=False,
|
|
pool=None
|
|
)
|
|
mock_tmpdir.return_value.__enter__.return_value = \
|
|
'/var/tmp/qvm-template-tmpdir'
|
|
with self.assertRaises(SystemExit):
|
|
qubesadmin.tools.qvm_template.install(args, self.app)
|
|
# Check error message
|
|
self.assertTrue('Illegal package name' in mock_err.getvalue())
|
|
# Should not be executed:
|
|
self.assertEqual(mock_dl_list.mock_calls, [])
|
|
self.assertEqual(mock_dl.mock_calls, [])
|
|
self.assertEqual(mock_extract.mock_calls, [])
|
|
self.assertEqual(mock_confirm.mock_calls, [])
|
|
self.assertEqual(mock_call.mock_calls, [])
|
|
self.assertEqual(mock_rename.mock_calls, [])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('os.rename')
|
|
@mock.patch('os.makedirs')
|
|
@mock.patch('subprocess.check_call')
|
|
@mock.patch('qubesadmin.tools.qvm_template.confirm_action')
|
|
@mock.patch('qubesadmin.tools.qvm_template.extract_rpm')
|
|
@mock.patch('qubesadmin.tools.qvm_template.download')
|
|
@mock.patch('qubesadmin.tools.qvm_template.get_dl_list')
|
|
@mock.patch('qubesadmin.tools.qvm_template.verify_rpm')
|
|
def test_106_install_local_badpath_fail(
|
|
self,
|
|
mock_verify,
|
|
mock_dl_list,
|
|
mock_dl,
|
|
mock_extract,
|
|
mock_confirm,
|
|
mock_call,
|
|
mock_mkdirs,
|
|
mock_rename):
|
|
mock_time = mock.Mock(wraps=datetime.datetime)
|
|
with mock.patch('qubesadmin.tools.qvm_template.LOCK_FILE', '/tmp/test.lock'), \
|
|
mock.patch('datetime.datetime', new=mock_time), \
|
|
mock.patch('tempfile.TemporaryDirectory') as mock_tmpdir, \
|
|
mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
|
path = '/var/tmp/ShOulD-NoT-ExIsT.rpm'
|
|
args = argparse.Namespace(
|
|
templates=[path],
|
|
keyring='/tmp',
|
|
nogpgcheck=False,
|
|
cachedir='/var/cache/qvm-template',
|
|
repo_files=[],
|
|
releasever='4.1',
|
|
yes=False,
|
|
allow_pv=False,
|
|
pool=None
|
|
)
|
|
mock_tmpdir.return_value.__enter__.return_value = \
|
|
'/var/tmp/qvm-template-tmpdir'
|
|
with self.assertRaises(SystemExit):
|
|
qubesadmin.tools.qvm_template.install(args, self.app)
|
|
# Check error message
|
|
self.assertTrue(f"RPM file '{path}' not found" \
|
|
in mock_err.getvalue())
|
|
# Should not be executed:
|
|
self.assertEqual(mock_verify.mock_calls, [])
|
|
self.assertEqual(mock_dl_list.mock_calls, [])
|
|
self.assertEqual(mock_dl.mock_calls, [])
|
|
self.assertEqual(mock_extract.mock_calls, [])
|
|
self.assertEqual(mock_confirm.mock_calls, [])
|
|
self.assertEqual(mock_call.mock_calls, [])
|
|
self.assertEqual(mock_rename.mock_calls, [])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('os.remove')
|
|
@mock.patch('os.rename')
|
|
@mock.patch('os.makedirs')
|
|
@mock.patch('subprocess.check_call')
|
|
@mock.patch('qubesadmin.tools.qvm_template.confirm_action')
|
|
@mock.patch('qubesadmin.tools.qvm_template.extract_rpm')
|
|
@mock.patch('qubesadmin.tools.qvm_template.download')
|
|
@mock.patch('qubesadmin.tools.qvm_template.get_dl_list')
|
|
@mock.patch('qubesadmin.tools.qvm_template.verify_rpm')
|
|
def test_107_install_download_success(
|
|
self,
|
|
mock_verify,
|
|
mock_dl_list,
|
|
mock_dl,
|
|
mock_extract,
|
|
mock_confirm,
|
|
mock_call,
|
|
mock_mkdirs,
|
|
mock_rename,
|
|
mock_remove):
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = b'0\0'
|
|
build_time = '2020-09-01 14:30:00' # 1598970600
|
|
install_time = '2020-09-01 15:30:00'
|
|
for key, val in [
|
|
('name', 'test-vm'),
|
|
('epoch', '2'),
|
|
('version', '4.1'),
|
|
('release', '2020'),
|
|
('reponame', 'qubes-templates-itl'),
|
|
('buildtime', build_time),
|
|
('installtime', install_time),
|
|
('license', 'GPL'),
|
|
('url', 'https://qubes-os.org'),
|
|
('summary', 'Summary'),
|
|
('description', 'Desc|desc')]:
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Set',
|
|
f'template-{key}',
|
|
val.encode())] = b'0\0'
|
|
mock_dl.return_value = {'test-vm': {
|
|
rpm.RPMTAG_NAME : 'qubes-template-test-vm',
|
|
rpm.RPMTAG_BUILDTIME : 1598970600,
|
|
rpm.RPMTAG_DESCRIPTION : 'Desc\ndesc',
|
|
rpm.RPMTAG_EPOCHNUM : 2,
|
|
rpm.RPMTAG_LICENSE : 'GPL',
|
|
rpm.RPMTAG_RELEASE : '2020',
|
|
rpm.RPMTAG_SUMMARY : 'Summary',
|
|
rpm.RPMTAG_URL : 'https://qubes-os.org',
|
|
rpm.RPMTAG_VERSION : '4.1'
|
|
}}
|
|
dl_list = {
|
|
'test-vm': qubesadmin.tools.qvm_template.DlEntry(
|
|
('1', '4.1', '20200101'), 'qubes-templates-itl', 1048576)
|
|
}
|
|
mock_dl_list.return_value = dl_list
|
|
mock_call.side_effect = self.add_new_vm_side_effect
|
|
mock_time = mock.Mock(wraps=datetime.datetime)
|
|
mock_time.now.return_value = \
|
|
datetime.datetime(2020, 9, 1, 15, 30, tzinfo=datetime.timezone.utc)
|
|
with mock.patch('qubesadmin.tools.qvm_template.LOCK_FILE', '/tmp/test.lock'), \
|
|
mock.patch('datetime.datetime', new=mock_time), \
|
|
mock.patch('tempfile.TemporaryDirectory') as mock_tmpdir, \
|
|
mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
|
args = argparse.Namespace(
|
|
templates='test-vm',
|
|
keyring='/tmp/keyring.gpg',
|
|
nogpgcheck=False,
|
|
cachedir='/var/cache/qvm-template',
|
|
repo_files=[],
|
|
releasever='4.1',
|
|
yes=False,
|
|
keep_cache=False,
|
|
allow_pv=False,
|
|
pool=None
|
|
)
|
|
mock_tmpdir.return_value.__enter__.return_value = \
|
|
'/var/tmp/qvm-template-tmpdir'
|
|
qubesadmin.tools.qvm_template.install(args, self.app)
|
|
# Attempt to get download list
|
|
selector = qubesadmin.tools.qvm_template.VersionSelector.LATEST
|
|
self.assertEqual(mock_dl_list.mock_calls, [
|
|
mock.call(args, self.app, version_selector=selector)
|
|
])
|
|
mock_dl.assert_called_with(args, self.app,
|
|
dl_list=dl_list, version_selector=selector)
|
|
# download already verify the package internally
|
|
self.assertEqual(mock_verify.mock_calls, [])
|
|
# Package is extracted
|
|
mock_extract.assert_called_with('test-vm',
|
|
'/var/cache/qvm-template/qubes-template-test-vm-1:4.1-20200101.rpm',
|
|
'/var/tmp/qvm-template-tmpdir')
|
|
# No packages overwritten, so no confirm needed
|
|
self.assertEqual(mock_confirm.mock_calls, [])
|
|
# qvm-template-postprocess is called
|
|
self.assertEqual(mock_call.mock_calls, [
|
|
mock.call([
|
|
'qvm-template-postprocess',
|
|
'--really',
|
|
'--no-installed-by-rpm',
|
|
'post-install',
|
|
'test-vm',
|
|
'/var/tmp/qvm-template-tmpdir'
|
|
'/var/lib/qubes/vm-templates/test-vm'
|
|
])
|
|
])
|
|
# Cache directory created
|
|
self.assertEqual(mock_mkdirs.mock_calls, [
|
|
mock.call(args.cachedir, exist_ok=True)
|
|
])
|
|
# No templates downloaded, thus no renames needed
|
|
self.assertEqual(mock_rename.mock_calls, [])
|
|
# Downloaded template is removed
|
|
self.assertEqual(mock_remove.mock_calls, [
|
|
mock.call('/var/cache/qvm-template/' \
|
|
'qubes-template-test-vm-1:4.1-20200101.rpm'),
|
|
mock.call('/tmp/test.lock')
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('os.rename')
|
|
@mock.patch('os.makedirs')
|
|
@mock.patch('subprocess.check_call')
|
|
@mock.patch('qubesadmin.tools.qvm_template.confirm_action')
|
|
@mock.patch('qubesadmin.tools.qvm_template.extract_rpm')
|
|
@mock.patch('qubesadmin.tools.qvm_template.download')
|
|
@mock.patch('qubesadmin.tools.qvm_template.get_dl_list')
|
|
@mock.patch('qubesadmin.tools.qvm_template.verify_rpm')
|
|
def test_108_install_download_fail_exists(
|
|
self,
|
|
mock_verify,
|
|
mock_dl_list,
|
|
mock_dl,
|
|
mock_extract,
|
|
mock_confirm,
|
|
mock_call,
|
|
mock_mkdirs,
|
|
mock_rename):
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
mock_dl.return_value = {'test-vm': {
|
|
rpm.RPMTAG_NAME : 'qubes-template-test-vm',
|
|
rpm.RPMTAG_BUILDTIME : 1598970600,
|
|
rpm.RPMTAG_DESCRIPTION : 'Desc\ndesc',
|
|
rpm.RPMTAG_EPOCHNUM : 2,
|
|
rpm.RPMTAG_LICENSE : 'GPL',
|
|
rpm.RPMTAG_RELEASE : '2020',
|
|
rpm.RPMTAG_SUMMARY : 'Summary',
|
|
rpm.RPMTAG_URL : 'https://qubes-os.org',
|
|
rpm.RPMTAG_VERSION : '4.1'
|
|
}}
|
|
dl_list = {
|
|
'test-vm': qubesadmin.tools.qvm_template.DlEntry(
|
|
('1', '4.1', '20200101'), 'qubes-templates-itl', 1048576)
|
|
}
|
|
mock_dl_list.return_value = dl_list
|
|
mock_time = mock.Mock(wraps=datetime.datetime)
|
|
mock_time.now.return_value = \
|
|
datetime.datetime(2020, 9, 1, 15, 30, tzinfo=datetime.timezone.utc)
|
|
with mock.patch('qubesadmin.tools.qvm_template.LOCK_FILE', '/tmp/test.lock'), \
|
|
mock.patch('datetime.datetime', new=mock_time), \
|
|
mock.patch('tempfile.TemporaryDirectory') as mock_tmpdir, \
|
|
mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
|
args = argparse.Namespace(
|
|
templates='test-vm',
|
|
keyring='/tmp/keyring.gpg',
|
|
nogpgcheck=False,
|
|
cachedir='/var/cache/qvm-template',
|
|
repo_files=[],
|
|
releasever='4.1',
|
|
yes=False,
|
|
keep_cache=True,
|
|
allow_pv=False,
|
|
pool=None
|
|
)
|
|
mock_tmpdir.return_value.__enter__.return_value = \
|
|
'/var/tmp/qvm-template-tmpdir'
|
|
qubesadmin.tools.qvm_template.install(args, self.app)
|
|
self.assertIn('already installed, skipping', mock_err.getvalue())
|
|
# Attempt to get download list
|
|
selector = qubesadmin.tools.qvm_template.VersionSelector.LATEST
|
|
self.assertEqual(mock_dl_list.mock_calls, [
|
|
mock.call(args, self.app, version_selector=selector)
|
|
])
|
|
# Nothing downloaded nor installed
|
|
mock_dl.assert_called_with(args, self.app,
|
|
dl_list={}, version_selector=selector)
|
|
mock_verify.assert_not_called()
|
|
mock_extract.assert_not_called()
|
|
mock_confirm.assert_not_called()
|
|
mock_call.assert_not_called()
|
|
# Cache directory created
|
|
self.assertEqual(mock_mkdirs.mock_calls, [
|
|
mock.call(args.cachedir, exist_ok=True)
|
|
])
|
|
# No templates downloaded, thus no renames needed
|
|
self.assertEqual(mock_rename.mock_calls, [])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('os.rename')
|
|
@mock.patch('os.makedirs')
|
|
@mock.patch('subprocess.check_call')
|
|
@mock.patch('qubesadmin.tools.qvm_template.confirm_action')
|
|
@mock.patch('qubesadmin.tools.qvm_template.extract_rpm')
|
|
@mock.patch('qubesadmin.tools.qvm_template.download')
|
|
@mock.patch('qubesadmin.tools.qvm_template.get_dl_list')
|
|
@mock.patch('qubesadmin.tools.qvm_template.verify_rpm')
|
|
def test_109_install_fail_extract(
|
|
self,
|
|
mock_verify,
|
|
mock_dl_list,
|
|
mock_dl,
|
|
mock_extract,
|
|
mock_confirm,
|
|
mock_call,
|
|
mock_mkdirs,
|
|
mock_rename):
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = b'0\0'
|
|
mock_verify.return_value = {
|
|
rpm.RPMTAG_NAME : 'qubes-template-test-vm',
|
|
rpm.RPMTAG_BUILDTIME : 1598970600,
|
|
rpm.RPMTAG_DESCRIPTION : 'Desc\ndesc',
|
|
rpm.RPMTAG_EPOCHNUM : 2,
|
|
rpm.RPMTAG_LICENSE : 'GPL',
|
|
rpm.RPMTAG_RELEASE : '2020',
|
|
rpm.RPMTAG_SUMMARY : 'Summary',
|
|
rpm.RPMTAG_URL : 'https://qubes-os.org',
|
|
rpm.RPMTAG_VERSION : '4.1'
|
|
}
|
|
mock_dl_list.return_value = {}
|
|
mock_call.side_effect = self.add_new_vm_side_effect
|
|
mock_time = mock.Mock(wraps=datetime.datetime)
|
|
mock_time.now.return_value = \
|
|
datetime.datetime(2020, 9, 1, 15, 30, tzinfo=datetime.timezone.utc)
|
|
# Extraction error
|
|
mock_extract.return_value = False
|
|
with mock.patch('qubesadmin.tools.qvm_template.LOCK_FILE', '/tmp/test.lock'), \
|
|
mock.patch('datetime.datetime', new=mock_time), \
|
|
mock.patch('tempfile.TemporaryDirectory') as mock_tmpdir, \
|
|
mock.patch('sys.stderr', new=io.StringIO()) as mock_err, \
|
|
tempfile.NamedTemporaryFile(suffix='.rpm') as template_file:
|
|
path = template_file.name
|
|
args = argparse.Namespace(
|
|
templates=[path],
|
|
keyring='/tmp/keyring.gpg',
|
|
nogpgcheck=False,
|
|
cachedir='/var/cache/qvm-template',
|
|
repo_files=[],
|
|
releasever='4.1',
|
|
yes=False,
|
|
allow_pv=False,
|
|
pool=None
|
|
)
|
|
mock_tmpdir.return_value.__enter__.return_value = \
|
|
'/var/tmp/qvm-template-tmpdir'
|
|
with self.assertRaises(Exception) as e:
|
|
qubesadmin.tools.qvm_template.install(args, self.app)
|
|
self.assertIn('Failed to extract', e.exception.args[0])
|
|
|
|
# Attempt to get download list
|
|
selector = qubesadmin.tools.qvm_template.VersionSelector.LATEST
|
|
self.assertEqual(mock_dl_list.mock_calls, [
|
|
mock.call(args, self.app, version_selector=selector)
|
|
])
|
|
# Nothing downloaded
|
|
mock_dl.assert_called_with(args, self.app,
|
|
dl_list={}, version_selector=selector)
|
|
mock_verify.assert_called_once_with(template_file.name,
|
|
'/tmp/keyring.gpg',
|
|
nogpgcheck=False)
|
|
# Package is (attempted to be) extracted
|
|
mock_extract.assert_called_with('test-vm', path,
|
|
'/var/tmp/qvm-template-tmpdir')
|
|
# No packages overwritten, so no confirm needed
|
|
self.assertEqual(mock_confirm.mock_calls, [])
|
|
# No VM created
|
|
mock_call.assert_not_called()
|
|
# Cache directory created
|
|
self.assertEqual(mock_mkdirs.mock_calls, [
|
|
mock.call(args.cachedir, exist_ok=True)
|
|
])
|
|
# No templates downloaded, thus no renames needed
|
|
self.assertEqual(mock_rename.mock_calls, [])
|
|
self.assertAllCalled()
|
|
|
|
def test_110_qrexec_payload_refresh_success(self):
|
|
with tempfile.NamedTemporaryFile() as repo_conf1, \
|
|
tempfile.NamedTemporaryFile() as repo_conf2:
|
|
repo_str1 = \
|
|
'''[qubes-templates-itl]
|
|
name = Qubes Templates repository
|
|
#baseurl = https://yum.qubes-os.org/r$releasever/templates-itl
|
|
#baseurl = http://yum.qubesosfasa4zl44o4tws22di6kepyzfeqv3tg4e3ztknltfxqrymdad.onion/r$releasever/templates-itl
|
|
metalink = https://yum.qubes-os.org/r$releasever/templates-itl/repodata/repomd.xml.metalink
|
|
enabled = 1
|
|
fastestmirror = 1
|
|
metadata_expire = 7d
|
|
gpgcheck = 1
|
|
gpgkey = file:///etc/qubes/repo-templates/keys/RPM-GPG-KEY-qubes-$releasever-primary
|
|
'''
|
|
repo_str2 = \
|
|
'''[qubes-templates-itl-testing]
|
|
name = Qubes Templates repository
|
|
#baseurl = https://yum.qubes-os.org/r$releasever/templates-itl-testing
|
|
#baseurl = http://yum.qubesosfasa4zl44o4tws22di6kepyzfeqv3tg4e3ztknltfxqrymdad.onion/r$releasever/templates-itl-testing
|
|
metalink = https://yum.qubes-os.org/r$releasever/templates-itl-testing/repodata/repomd.xml.metalink
|
|
enabled = 0
|
|
fastestmirror = 1
|
|
gpgcheck = 1
|
|
gpgkey = file:///etc/qubes/repo-templates/keys/RPM-GPG-KEY-qubes-$releasever-primary
|
|
'''
|
|
repo_conf1.write(repo_str1.encode())
|
|
repo_conf1.flush()
|
|
repo_conf2.write(repo_str2.encode())
|
|
repo_conf2.flush()
|
|
args = argparse.Namespace(
|
|
enablerepo=['repo1', 'repo2'],
|
|
disablerepo=['repo3', 'repo4', 'repo5'],
|
|
repoid=[],
|
|
releasever='4.1',
|
|
repo_files=[repo_conf1.name, repo_conf2.name]
|
|
)
|
|
res = qubesadmin.tools.qvm_template.qrexec_payload(args, self.app,
|
|
'qubes-template-fedora-32', True)
|
|
self.assertEqual(res,
|
|
'''--enablerepo=repo1
|
|
--enablerepo=repo2
|
|
--disablerepo=repo3
|
|
--disablerepo=repo4
|
|
--disablerepo=repo5
|
|
--refresh
|
|
--releasever=4.1
|
|
qubes-template-fedora-32
|
|
---
|
|
''' + repo_str1 + '\n' + repo_str2 + '\n')
|
|
self.assertAllCalled()
|
|
|
|
def test_111_qrexec_payload_norefresh_success(self):
|
|
with tempfile.NamedTemporaryFile() as repo_conf1:
|
|
repo_str1 = \
|
|
'''[qubes-templates-itl]
|
|
name = Qubes Templates repository
|
|
#baseurl = https://yum.qubes-os.org/r$releasever/templates-itl
|
|
#baseurl = http://yum.qubesosfasa4zl44o4tws22di6kepyzfeqv3tg4e3ztknltfxqrymdad.onion/r$releasever/templates-itl
|
|
metalink = https://yum.qubes-os.org/r$releasever/templates-itl/repodata/repomd.xml.metalink
|
|
enabled = 1
|
|
fastestmirror = 1
|
|
metadata_expire = 7d
|
|
gpgcheck = 1
|
|
gpgkey = file:///etc/qubes/repo-templates/keys/RPM-GPG-KEY-qubes-$releasever-primary
|
|
'''
|
|
repo_conf1.write(repo_str1.encode())
|
|
repo_conf1.flush()
|
|
args = argparse.Namespace(
|
|
enablerepo=[],
|
|
disablerepo=[],
|
|
repoid=['repo1', 'repo2'],
|
|
releasever='4.1',
|
|
repo_files=[repo_conf1.name]
|
|
)
|
|
res = qubesadmin.tools.qvm_template.qrexec_payload(args, self.app,
|
|
'qubes-template-fedora-32', False)
|
|
self.assertEqual(res,
|
|
'''--repoid=repo1
|
|
--repoid=repo2
|
|
--releasever=4.1
|
|
qubes-template-fedora-32
|
|
---
|
|
''' + repo_str1 + '\n')
|
|
self.assertAllCalled()
|
|
|
|
def test_112_qrexec_payload_specnewline_fail(self):
|
|
with tempfile.NamedTemporaryFile() as repo_conf1:
|
|
repo_str1 = \
|
|
'''[qubes-templates-itl]
|
|
name = Qubes Templates repository
|
|
#baseurl = https://yum.qubes-os.org/r$releasever/templates-itl
|
|
#baseurl = http://yum.qubesosfasa4zl44o4tws22di6kepyzfeqv3tg4e3ztknltfxqrymdad.onion/r$releasever/templates-itl
|
|
metalink = https://yum.qubes-os.org/r$releasever/templates-itl/repodata/repomd.xml.metalink
|
|
enabled = 1
|
|
fastestmirror = 1
|
|
metadata_expire = 7d
|
|
gpgcheck = 1
|
|
gpgkey = file:///etc/qubes/repo-templates/keys/RPM-GPG-KEY-qubes-$releasever-primary
|
|
'''
|
|
repo_conf1.write(repo_str1.encode())
|
|
repo_conf1.flush()
|
|
args = argparse.Namespace(
|
|
enablerepo=[],
|
|
disablerepo=[],
|
|
repoid=['repo1', 'repo2'],
|
|
releasever='4.1',
|
|
repo_files=[repo_conf1.name]
|
|
)
|
|
with mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
|
with self.assertRaises(SystemExit):
|
|
qubesadmin.tools.qvm_template.qrexec_payload(args,
|
|
self.app, 'qubes-template-fedora\n-32', False)
|
|
# Check error message
|
|
self.assertTrue('Malformed template name'
|
|
in mock_err.getvalue())
|
|
self.assertTrue("argument should not contain '\\n'"
|
|
in mock_err.getvalue())
|
|
self.assertAllCalled()
|
|
|
|
def test_113_qrexec_payload_enablereponewline_fail(self):
|
|
with tempfile.NamedTemporaryFile() as repo_conf1:
|
|
repo_str1 = \
|
|
'''[qubes-templates-itl]
|
|
name = Qubes Templates repository
|
|
#baseurl = https://yum.qubes-os.org/r$releasever/templates-itl
|
|
#baseurl = http://yum.qubesosfasa4zl44o4tws22di6kepyzfeqv3tg4e3ztknltfxqrymdad.onion/r$releasever/templates-itl
|
|
metalink = https://yum.qubes-os.org/r$releasever/templates-itl/repodata/repomd.xml.metalink
|
|
enabled = 1
|
|
fastestmirror = 1
|
|
metadata_expire = 7d
|
|
gpgcheck = 1
|
|
gpgkey = file:///etc/qubes/repo-templates/keys/RPM-GPG-KEY-qubes-$releasever-primary
|
|
'''
|
|
repo_conf1.write(repo_str1.encode())
|
|
repo_conf1.flush()
|
|
args = argparse.Namespace(
|
|
enablerepo=['repo\n0'],
|
|
disablerepo=[],
|
|
repoid=['repo1', 'repo2'],
|
|
releasever='4.1',
|
|
repo_files=[repo_conf1.name]
|
|
)
|
|
with mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
|
with self.assertRaises(SystemExit):
|
|
qubesadmin.tools.qvm_template.qrexec_payload(args,
|
|
self.app, 'qubes-template-fedora-32', False)
|
|
# Check error message
|
|
self.assertTrue('Malformed --enablerepo'
|
|
in mock_err.getvalue())
|
|
self.assertTrue("argument should not contain '\\n'"
|
|
in mock_err.getvalue())
|
|
self.assertAllCalled()
|
|
|
|
def test_114_qrexec_payload_disablereponewline_fail(self):
|
|
with tempfile.NamedTemporaryFile() as repo_conf1:
|
|
repo_str1 = \
|
|
'''[qubes-templates-itl]
|
|
name = Qubes Templates repository
|
|
#baseurl = https://yum.qubes-os.org/r$releasever/templates-itl
|
|
#baseurl = http://yum.qubesosfasa4zl44o4tws22di6kepyzfeqv3tg4e3ztknltfxqrymdad.onion/r$releasever/templates-itl
|
|
metalink = https://yum.qubes-os.org/r$releasever/templates-itl/repodata/repomd.xml.metalink
|
|
enabled = 1
|
|
fastestmirror = 1
|
|
metadata_expire = 7d
|
|
gpgcheck = 1
|
|
gpgkey = file:///etc/qubes/repo-templates/keys/RPM-GPG-KEY-qubes-$releasever-primary
|
|
'''
|
|
repo_conf1.write(repo_str1.encode())
|
|
repo_conf1.flush()
|
|
args = argparse.Namespace(
|
|
enablerepo=[],
|
|
disablerepo=['repo\n0'],
|
|
repoid=['repo1', 'repo2'],
|
|
releasever='4.1',
|
|
repo_files=[repo_conf1.name]
|
|
)
|
|
with mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
|
with self.assertRaises(SystemExit):
|
|
qubesadmin.tools.qvm_template.qrexec_payload(args,
|
|
self.app, 'qubes-template-fedora-32', False)
|
|
# Check error message
|
|
self.assertTrue('Malformed --disablerepo'
|
|
in mock_err.getvalue())
|
|
self.assertTrue("argument should not contain '\\n'"
|
|
in mock_err.getvalue())
|
|
self.assertAllCalled()
|
|
|
|
def test_115_qrexec_payload_repoidnewline_fail(self):
|
|
with tempfile.NamedTemporaryFile() as repo_conf1:
|
|
repo_str1 = \
|
|
'''[qubes-templates-itl]
|
|
name = Qubes Templates repository
|
|
#baseurl = https://yum.qubes-os.org/r$releasever/templates-itl
|
|
#baseurl = http://yum.qubesosfasa4zl44o4tws22di6kepyzfeqv3tg4e3ztknltfxqrymdad.onion/r$releasever/templates-itl
|
|
metalink = https://yum.qubes-os.org/r$releasever/templates-itl/repodata/repomd.xml.metalink
|
|
enabled = 1
|
|
fastestmirror = 1
|
|
metadata_expire = 7d
|
|
gpgcheck = 1
|
|
gpgkey = file:///etc/qubes/repo-templates/keys/RPM-GPG-KEY-qubes-$releasever-primary
|
|
'''
|
|
repo_conf1.write(repo_str1.encode())
|
|
repo_conf1.flush()
|
|
args = argparse.Namespace(
|
|
enablerepo=[],
|
|
disablerepo=[],
|
|
repoid=['repo\n1', 'repo2'],
|
|
releasever='4.1',
|
|
repo_files=[repo_conf1.name]
|
|
)
|
|
with mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
|
with self.assertRaises(SystemExit):
|
|
qubesadmin.tools.qvm_template.qrexec_payload(args,
|
|
self.app, 'qubes-template-fedora-32', False)
|
|
# Check error message
|
|
self.assertTrue('Malformed --repoid'
|
|
in mock_err.getvalue())
|
|
self.assertTrue("argument should not contain '\\n'"
|
|
in mock_err.getvalue())
|
|
self.assertAllCalled()
|
|
|
|
def test_116_qrexec_payload_releasevernewline_fail(self):
|
|
with tempfile.NamedTemporaryFile() as repo_conf1:
|
|
repo_str1 = \
|
|
'''[qubes-templates-itl]
|
|
name = Qubes Templates repository
|
|
#baseurl = https://yum.qubes-os.org/r$releasever/templates-itl
|
|
#baseurl = http://yum.qubesosfasa4zl44o4tws22di6kepyzfeqv3tg4e3ztknltfxqrymdad.onion/r$releasever/templates-itl
|
|
metalink = https://yum.qubes-os.org/r$releasever/templates-itl/repodata/repomd.xml.metalink
|
|
enabled = 1
|
|
fastestmirror = 1
|
|
metadata_expire = 7d
|
|
gpgcheck = 1
|
|
gpgkey = file:///etc/qubes/repo-templates/keys/RPM-GPG-KEY-qubes-$releasever-primary
|
|
'''
|
|
repo_conf1.write(repo_str1.encode())
|
|
repo_conf1.flush()
|
|
args = argparse.Namespace(
|
|
enablerepo=[],
|
|
disablerepo=[],
|
|
repoid=['repo1', 'repo2'],
|
|
releasever='4\n.1',
|
|
repo_files=[repo_conf1.name]
|
|
)
|
|
with mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
|
with self.assertRaises(SystemExit):
|
|
qubesadmin.tools.qvm_template.qrexec_payload(args,
|
|
self.app, 'qubes-template-fedora-32', False)
|
|
# Check error message
|
|
self.assertTrue('Malformed --releasever'
|
|
in mock_err.getvalue())
|
|
self.assertTrue("argument should not contain '\\n'"
|
|
in mock_err.getvalue())
|
|
self.assertAllCalled()
|
|
|
|
def test_117_qrexec_payload_specdash_fail(self):
|
|
with tempfile.NamedTemporaryFile() as repo_conf1:
|
|
repo_str1 = \
|
|
'''[qubes-templates-itl]
|
|
name = Qubes Templates repository
|
|
#baseurl = https://yum.qubes-os.org/r$releasever/templates-itl
|
|
#baseurl = http://yum.qubesosfasa4zl44o4tws22di6kepyzfeqv3tg4e3ztknltfxqrymdad.onion/r$releasever/templates-itl
|
|
metalink = https://yum.qubes-os.org/r$releasever/templates-itl/repodata/repomd.xml.metalink
|
|
enabled = 1
|
|
fastestmirror = 1
|
|
metadata_expire = 7d
|
|
gpgcheck = 1
|
|
gpgkey = file:///etc/qubes/repo-templates/keys/RPM-GPG-KEY-qubes-$releasever-primary
|
|
'''
|
|
repo_conf1.write(repo_str1.encode())
|
|
repo_conf1.flush()
|
|
args = argparse.Namespace(
|
|
enablerepo=[],
|
|
disablerepo=[],
|
|
repoid=['repo1', 'repo2'],
|
|
releasever='4.1',
|
|
repo_files=[repo_conf1.name]
|
|
)
|
|
with mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
|
with self.assertRaises(SystemExit):
|
|
qubesadmin.tools.qvm_template.qrexec_payload(args,
|
|
self.app, '---', False)
|
|
# Check error message
|
|
self.assertTrue('Malformed template name'
|
|
in mock_err.getvalue())
|
|
self.assertTrue("argument should not be '---'"
|
|
in mock_err.getvalue())
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_payload')
|
|
def test_120_qrexec_repoquery_success(self, mock_payload):
|
|
args = argparse.Namespace(updatevm='test-vm')
|
|
mock_payload.return_value = 'str1\nstr2'
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
self.app.expected_service_calls[
|
|
('test-vm', 'qubes.TemplateSearch')] = \
|
|
b'''qubes-template-fedora-32|0|4.1|20200101|qubes-templates-itl|1048576|2020-01-23 04:56|GPL|https://qubes-os.org|Qubes template for fedora-32|Qubes template\n for fedora-32\n|
|
|
qubes-template-fedora-32|1|4.2|20200201|qubes-templates-itl-testing|2048576|2020-02-23 04:56|GPLv2|https://qubes-os.org/?|Qubes template for fedora-32 v2|Qubes template\n for fedora-32 v2\n|
|
|
'''
|
|
res = qubesadmin.tools.qvm_template.qrexec_repoquery(args, self.app,
|
|
'qubes-template-fedora-32')
|
|
self.assertEqual(res, [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'0',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-32',
|
|
'Qubes template\n for fedora-32\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'1',
|
|
'4.2',
|
|
'20200201',
|
|
'qubes-templates-itl-testing',
|
|
2048576,
|
|
datetime.datetime(2020, 2, 23, 4, 56),
|
|
'GPLv2',
|
|
'https://qubes-os.org/?',
|
|
'Qubes template for fedora-32 v2',
|
|
'Qubes template\n for fedora-32 v2\n'
|
|
)
|
|
])
|
|
self.assertEqual(self.app.service_calls, [
|
|
('test-vm', 'qubes.TemplateSearch',
|
|
{'filter_esc': True, 'stdout': subprocess.PIPE}),
|
|
('test-vm', 'qubes.TemplateSearch', b'str1\nstr2')
|
|
])
|
|
self.assertEqual(mock_payload.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-fedora-32', False)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_payload')
|
|
def test_121_qrexec_repoquery_refresh_success(self, mock_payload):
|
|
args = argparse.Namespace(updatevm='test-vm')
|
|
mock_payload.return_value = 'str1\nstr2'
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
self.app.expected_service_calls[
|
|
('test-vm', 'qubes.TemplateSearch')] = \
|
|
b'''qubes-template-fedora-32|0|4.1|20200101|qubes-templates-itl|1048576|2020-01-23 04:56|GPL|https://qubes-os.org|Qubes template for fedora-32|Qubes template\n for fedora-32\n|
|
|
qubes-template-fedora-32|1|4.2|20200201|qubes-templates-itl-testing|2048576|2020-02-23 04:56|GPLv2|https://qubes-os.org/?|Qubes template for fedora-32 v2|Qubes template\n for fedora-32 v2\n|
|
|
'''
|
|
res = qubesadmin.tools.qvm_template.qrexec_repoquery(args, self.app,
|
|
'qubes-template-fedora-32', True)
|
|
self.assertEqual(res, [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'0',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-32',
|
|
'Qubes template\n for fedora-32\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'1',
|
|
'4.2',
|
|
'20200201',
|
|
'qubes-templates-itl-testing',
|
|
2048576,
|
|
datetime.datetime(2020, 2, 23, 4, 56),
|
|
'GPLv2',
|
|
'https://qubes-os.org/?',
|
|
'Qubes template for fedora-32 v2',
|
|
'Qubes template\n for fedora-32 v2\n'
|
|
)
|
|
])
|
|
self.assertEqual(self.app.service_calls, [
|
|
('test-vm', 'qubes.TemplateSearch',
|
|
{'filter_esc': True, 'stdout': subprocess.PIPE}),
|
|
('test-vm', 'qubes.TemplateSearch', b'str1\nstr2')
|
|
])
|
|
self.assertEqual(mock_payload.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-fedora-32', True)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_payload')
|
|
def test_122_qrexec_repoquery_ignorenonspec_success(self, mock_payload):
|
|
args = argparse.Namespace(updatevm='test-vm')
|
|
mock_payload.return_value = 'str1\nstr2'
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
self.app.expected_service_calls[
|
|
('test-vm', 'qubes.TemplateSearch')] = \
|
|
b'''qubes-template-debian-10|1|4.2|20200201|qubes-templates-itl-testing|2048576|2020-02-23 04:56|GPLv2|https://qubes-os.org/?|Qubes template for debian-10|Qubes template for debian-10\n|
|
|
qubes-template-fedora-32|0|4.1|20200101|qubes-templates-itl|1048576|2020-01-23 04:56|GPL|https://qubes-os.org|Qubes template for fedora-32|Qubes template for fedora-32\n|
|
|
'''
|
|
res = qubesadmin.tools.qvm_template.qrexec_repoquery(args, self.app,
|
|
'qubes-template-fedora-32')
|
|
self.assertEqual(res, [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'0',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-32',
|
|
'Qubes template for fedora-32\n'
|
|
)
|
|
])
|
|
self.assertEqual(self.app.service_calls, [
|
|
('test-vm', 'qubes.TemplateSearch',
|
|
{'filter_esc': True, 'stdout': subprocess.PIPE}),
|
|
('test-vm', 'qubes.TemplateSearch', b'str1\nstr2')
|
|
])
|
|
self.assertEqual(mock_payload.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-fedora-32', False)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_payload')
|
|
def test_123_qrexec_repoquery_ignorebadname_success(self, mock_payload):
|
|
args = argparse.Namespace(updatevm='test-vm')
|
|
mock_payload.return_value = 'str1\nstr2'
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
self.app.expected_service_calls[
|
|
('test-vm', 'qubes.TemplateSearch')] = \
|
|
b'''template-fedora-32|1|4.2|20200201|qubes-templates-itl-testing|2048576|2020-02-23 04:56|GPLv2|https://qubes-os.org/?|Qubes template for fedora-32 v2|Qubes template\n for fedora-32 v2\n|
|
|
qubes-template-fedora-32|0|4.1|20200101|qubes-templates-itl|1048576|2020-01-23 04:56|GPL|https://qubes-os.org|Qubes template for fedora-32|Qubes template for fedora-32\n|
|
|
'''
|
|
res = qubesadmin.tools.qvm_template.qrexec_repoquery(args, self.app,
|
|
'qubes-template-fedora-32')
|
|
self.assertEqual(res, [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'0',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-32',
|
|
'Qubes template for fedora-32\n'
|
|
)
|
|
])
|
|
self.assertEqual(self.app.service_calls, [
|
|
('test-vm', 'qubes.TemplateSearch',
|
|
{'filter_esc': True, 'stdout': subprocess.PIPE}),
|
|
('test-vm', 'qubes.TemplateSearch', b'str1\nstr2')
|
|
])
|
|
self.assertEqual(mock_payload.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-fedora-32', False)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_payload')
|
|
def test_124_qrexec_repoquery_searchfail_fail(self, mock_payload):
|
|
args = argparse.Namespace(updatevm='test-vm')
|
|
mock_payload.return_value = 'str1\nstr2'
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
with mock.patch('qubesadmin.tests.TestProcess.wait') \
|
|
as mock_wait:
|
|
mock_wait.return_value = 1
|
|
with self.assertRaises(ConnectionError):
|
|
qubesadmin.tools.qvm_template.qrexec_repoquery(args, self.app,
|
|
'qubes-template-fedora-32')
|
|
self.assertEqual(self.app.service_calls, [
|
|
('test-vm', 'qubes.TemplateSearch',
|
|
{'filter_esc': True, 'stdout': subprocess.PIPE}),
|
|
('test-vm', 'qubes.TemplateSearch', b'str1\nstr2')
|
|
])
|
|
self.assertEqual(mock_payload.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-fedora-32', False)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_payload')
|
|
def test_125_qrexec_repoquery_extrafield_fail(self, mock_payload):
|
|
args = argparse.Namespace(updatevm='test-vm')
|
|
mock_payload.return_value = 'str1\nstr2'
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
self.app.expected_service_calls[
|
|
('test-vm', 'qubes.TemplateSearch')] = \
|
|
b'''qubes-template-fedora-32|1|4.2|20200201|qubes-templates-itl-testing|2048576|2020-02-23 04:56|GPLv2|https://qubes-os.org/?|Qubes template for fedora-32 v2|Extra field|Qubes template\n for fedora-32 v2\n|
|
|
qubes-template-fedora-32|0|4.1|20200101|qubes-templates-itl|1048576|2020-01-23 04:56|GPL|https://qubes-os.org|Qubes template for fedora-32|Qubes template for fedora-32\n|
|
|
'''
|
|
with self.assertRaisesRegex(ConnectionError,
|
|
"unexpected data format"):
|
|
qubesadmin.tools.qvm_template.qrexec_repoquery(args, self.app,
|
|
'qubes-template-fedora-32')
|
|
self.assertEqual(self.app.service_calls, [
|
|
('test-vm', 'qubes.TemplateSearch',
|
|
{'filter_esc': True, 'stdout': subprocess.PIPE}),
|
|
('test-vm', 'qubes.TemplateSearch', b'str1\nstr2')
|
|
])
|
|
self.assertEqual(mock_payload.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-fedora-32', False)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_payload')
|
|
def test_125_qrexec_repoquery_missingfield_fail(self, mock_payload):
|
|
args = argparse.Namespace(updatevm='test-vm')
|
|
mock_payload.return_value = 'str1\nstr2'
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
self.app.expected_service_calls[
|
|
('test-vm', 'qubes.TemplateSearch')] = \
|
|
b'''qubes-template-fedora-32|1|4.2|20200201|qubes-templates-itl-testing|2048576|2020-02-23 04:56|GPLv2|Qubes template for fedora-32 v2|Qubes template\n for fedora-32 v2\n|
|
|
qubes-template-fedora-32|0|4.1|20200101|qubes-templates-itl|1048576|2020-01-23 04:56|GPL|https://qubes-os.org|Qubes template for fedora-32|Qubes template for fedora-32\n|
|
|
'''
|
|
with self.assertRaisesRegex(ConnectionError,
|
|
"unexpected data format"):
|
|
qubesadmin.tools.qvm_template.qrexec_repoquery(args, self.app,
|
|
'qubes-template-fedora-32')
|
|
self.assertEqual(self.app.service_calls, [
|
|
('test-vm', 'qubes.TemplateSearch',
|
|
{'filter_esc': True, 'stdout': subprocess.PIPE}),
|
|
('test-vm', 'qubes.TemplateSearch', b'str1\nstr2')
|
|
])
|
|
self.assertEqual(mock_payload.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-fedora-32', False)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_payload')
|
|
def test_126_qrexec_repoquery_badfieldname_fail(self, mock_payload):
|
|
args = argparse.Namespace(updatevm='test-vm')
|
|
mock_payload.return_value = 'str1\nstr2'
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
self.app.expected_service_calls[
|
|
('test-vm', 'qubes.TemplateSearch')] = \
|
|
b'''qubes-template-fedora-(32)|1|4.2|20200201|qubes-templates-itl-testing|2048576|2020-02-23 04:56|GPLv2|https://qubes-os.org/?|Qubes template for fedora-32 v2|Qubes template\n for fedora-32 v2\n|
|
|
qubes-template-fedora-32|0|4.1|20200101|qubes-templates-itl|1048576|2020-01-23 04:56|GPL|https://qubes-os.org|Qubes template for fedora-32|Qubes template for fedora-32\n|
|
|
'''
|
|
with self.assertRaisesRegex(ConnectionError,
|
|
"unexpected data format"):
|
|
qubesadmin.tools.qvm_template.qrexec_repoquery(args, self.app,
|
|
'qubes-template-fedora-32')
|
|
self.assertEqual(self.app.service_calls, [
|
|
('test-vm', 'qubes.TemplateSearch',
|
|
{'filter_esc': True, 'stdout': subprocess.PIPE}),
|
|
('test-vm', 'qubes.TemplateSearch', b'str1\nstr2')
|
|
])
|
|
self.assertEqual(mock_payload.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-fedora-32', False)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_payload')
|
|
def test_126_qrexec_repoquery_badfieldepoch_fail(self, mock_payload):
|
|
args = argparse.Namespace(updatevm='test-vm')
|
|
mock_payload.return_value = 'str1\nstr2'
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
self.app.expected_service_calls[
|
|
('test-vm', 'qubes.TemplateSearch')] = \
|
|
b'''qubes-template-fedora-32|!1|4.2|20200201|qubes-templates-itl-testing|2048576|2020-02-23 04:56|GPLv2|https://qubes-os.org/?|Qubes template for fedora-32 v2|Qubes template\n for fedora-32 v2\n|
|
|
qubes-template-fedora-32|0|4.1|20200101|qubes-templates-itl|1048576|2020-01-23 04:56|GPL|https://qubes-os.org|Qubes template for fedora-32|Qubes template for fedora-32\n|
|
|
'''
|
|
with self.assertRaisesRegex(ConnectionError,
|
|
"unexpected data format"):
|
|
qubesadmin.tools.qvm_template.qrexec_repoquery(args, self.app,
|
|
'qubes-template-fedora-32')
|
|
self.assertEqual(self.app.service_calls, [
|
|
('test-vm', 'qubes.TemplateSearch',
|
|
{'filter_esc': True, 'stdout': subprocess.PIPE}),
|
|
('test-vm', 'qubes.TemplateSearch', b'str1\nstr2')
|
|
])
|
|
self.assertEqual(mock_payload.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-fedora-32', False)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_payload')
|
|
def test_126_qrexec_repoquery_badfieldreponame_fail(self, mock_payload):
|
|
args = argparse.Namespace(updatevm='test-vm')
|
|
mock_payload.return_value = 'str1\nstr2'
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
self.app.expected_service_calls[
|
|
('test-vm', 'qubes.TemplateSearch')] = \
|
|
b'''qubes-template-fedora-32|1|4.2|20200201|qubes-templates-itl-<testing>|2048576|2020-02-23 04:56|GPLv2|https://qubes-os.org/?|Qubes template for fedora-32 v2|Qubes template\n for fedora-32 v2\n|
|
|
qubes-template-fedora-32|0|4.1|20200101|qubes-templates-itl|1048576|2020-01-23 04:56|GPL|https://qubes-os.org|Qubes template for fedora-32|Qubes template for fedora-32\n|
|
|
'''
|
|
with self.assertRaisesRegex(ConnectionError,
|
|
"unexpected data format"):
|
|
qubesadmin.tools.qvm_template.qrexec_repoquery(args, self.app,
|
|
'qubes-template-fedora-32')
|
|
self.assertEqual(self.app.service_calls, [
|
|
('test-vm', 'qubes.TemplateSearch',
|
|
{'filter_esc': True, 'stdout': subprocess.PIPE}),
|
|
('test-vm', 'qubes.TemplateSearch', b'str1\nstr2')
|
|
])
|
|
self.assertEqual(mock_payload.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-fedora-32', False)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_payload')
|
|
def test_126_qrexec_repoquery_badfielddlsize_fail(self, mock_payload):
|
|
args = argparse.Namespace(updatevm='test-vm')
|
|
mock_payload.return_value = 'str1\nstr2'
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
self.app.expected_service_calls[
|
|
('test-vm', 'qubes.TemplateSearch')] = \
|
|
b'''qubes-template-fedora-32|1|4.2|20200201|qubes-templates-itl-testing|2048a576|2020-02-23 04:56|GPLv2|https://qubes-os.org/?|Qubes template for fedora-32 v2|Qubes template\n for fedora-32 v2\n|
|
|
qubes-template-fedora-32|0|4.1|20200101|qubes-templates-itl|1048576|2020-01-23 04:56|GPL|https://qubes-os.org|Qubes template for fedora-32|Qubes template for fedora-32\n|
|
|
'''
|
|
with self.assertRaisesRegex(ConnectionError,
|
|
"unexpected data format"):
|
|
qubesadmin.tools.qvm_template.qrexec_repoquery(args, self.app,
|
|
'qubes-template-fedora-32')
|
|
self.assertEqual(self.app.service_calls, [
|
|
('test-vm', 'qubes.TemplateSearch',
|
|
{'filter_esc': True, 'stdout': subprocess.PIPE}),
|
|
('test-vm', 'qubes.TemplateSearch', b'str1\nstr2')
|
|
])
|
|
self.assertEqual(mock_payload.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-fedora-32', False)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_payload')
|
|
def test_126_qrexec_repoquery_badfielddate_fail(self, mock_payload):
|
|
args = argparse.Namespace(updatevm='test-vm')
|
|
mock_payload.return_value = 'str1\nstr2'
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
self.app.expected_service_calls[
|
|
('test-vm', 'qubes.TemplateSearch')] = \
|
|
b'''qubes-template-fedora-32|1|4.2|20200201|qubes-templates-itl-testing|2048576|2020-02-23|GPLv2|https://qubes-os.org/?|Qubes template for fedora-32 v2|Qubes template\n for fedora-32 v2\n|
|
|
qubes-template-fedora-32|0|4.1|20200101|qubes-templates-itl|1048576|2020-01-23 04:56|GPL|https://qubes-os.org|Qubes template for fedora-32|Qubes template for fedora-32\n|
|
|
'''
|
|
with self.assertRaisesRegex(ConnectionError,
|
|
"unexpected data format"):
|
|
qubesadmin.tools.qvm_template.qrexec_repoquery(args, self.app,
|
|
'qubes-template-fedora-32')
|
|
self.assertEqual(self.app.service_calls, [
|
|
('test-vm', 'qubes.TemplateSearch',
|
|
{'filter_esc': True, 'stdout': subprocess.PIPE}),
|
|
('test-vm', 'qubes.TemplateSearch', b'str1\nstr2')
|
|
])
|
|
self.assertEqual(mock_payload.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-fedora-32', False)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_payload')
|
|
def test_126_qrexec_repoquery_license_fail(self, mock_payload):
|
|
args = argparse.Namespace(updatevm='test-vm')
|
|
mock_payload.return_value = 'str1\nstr2'
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
self.app.expected_service_calls[
|
|
('test-vm', 'qubes.TemplateSearch')] = \
|
|
b'''qubes-template-fedora-32|1|4.2|20200201|qubes-templates-itl-testing|2048576|2020-02-23 04:56|GPLv2:)|https://qubes-os.org/?|Qubes template for fedora-32 v2|Qubes template\n for fedora-32 v2\n|
|
|
qubes-template-fedora-32|0|4.1|20200101|qubes-templates-itl|1048576|2020-01-23 04:56|GPL|https://qubes-os.org|Qubes template for fedora-32|Qubes template for fedora-32\n|
|
|
'''
|
|
with self.assertRaisesRegex(ConnectionError,
|
|
"unexpected data format"):
|
|
qubesadmin.tools.qvm_template.qrexec_repoquery(args, self.app,
|
|
'qubes-template-fedora-32')
|
|
self.assertEqual(self.app.service_calls, [
|
|
('test-vm', 'qubes.TemplateSearch',
|
|
{'filter_esc': True, 'stdout': subprocess.PIPE}),
|
|
('test-vm', 'qubes.TemplateSearch', b'str1\nstr2')
|
|
])
|
|
self.assertEqual(mock_payload.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-fedora-32', False)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def test_130_get_dl_list_latest_success(self, mock_query):
|
|
mock_query.return_value = [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'1',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-32',
|
|
'Qubes template\n for fedora-32\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'0',
|
|
'4.2',
|
|
'20200201',
|
|
'qubes-templates-itl-testing',
|
|
2048576,
|
|
datetime.datetime(2020, 2, 23, 4, 56),
|
|
'GPLv2',
|
|
'https://qubes-os.org/?',
|
|
'Qubes template for fedora-32 v2',
|
|
'Qubes template\n for fedora-32 v2\n'
|
|
)
|
|
]
|
|
args = argparse.Namespace(
|
|
templates=['some.local.file.rpm', 'fedora-32']
|
|
)
|
|
ret = qubesadmin.tools.qvm_template.get_dl_list(args, self.app)
|
|
self.assertEqual(ret, {
|
|
'fedora-32': qubesadmin.tools.qvm_template.DlEntry(
|
|
('1', '4.1', '20200101'), 'qubes-templates-itl', 1048576)
|
|
})
|
|
self.assertEqual(mock_query.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-fedora-32')
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def test_131_get_dl_list_latest_notfound_fail(self, mock_query):
|
|
mock_query.return_value = []
|
|
args = argparse.Namespace(
|
|
templates=['some.local.file.rpm', 'fedora-31']
|
|
)
|
|
with mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
|
with self.assertRaises(SystemExit):
|
|
qubesadmin.tools.qvm_template.get_dl_list(args, self.app)
|
|
self.assertTrue('not found' in mock_err.getvalue())
|
|
self.assertEqual(mock_query.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-fedora-31')
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def test_132_get_dl_list_multimerge0_success(self, mock_query):
|
|
counter = 0
|
|
def f(*args):
|
|
nonlocal counter
|
|
counter += 1
|
|
if counter == 1:
|
|
return [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'0',
|
|
'4.2',
|
|
'20200201',
|
|
'qubes-templates-itl-testing',
|
|
2048576,
|
|
datetime.datetime(2020, 2, 23, 4, 56),
|
|
'GPLv2',
|
|
'https://qubes-os.org/?',
|
|
'Qubes template for fedora-32 v2',
|
|
'Qubes template\n for fedora-32 v2\n'
|
|
)
|
|
]
|
|
return [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'1',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-32',
|
|
'Qubes template\n for fedora-32\n'
|
|
)
|
|
]
|
|
mock_query.side_effect = f
|
|
args = argparse.Namespace(
|
|
templates=['some.local.file.rpm', 'fedora-32:0', 'fedora-32:1']
|
|
)
|
|
ret = qubesadmin.tools.qvm_template.get_dl_list(args, self.app)
|
|
self.assertEqual(ret, {
|
|
'fedora-32': qubesadmin.tools.qvm_template.DlEntry(
|
|
('1', '4.1', '20200101'), 'qubes-templates-itl', 1048576)
|
|
})
|
|
self.assertEqual(mock_query.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-fedora-32:0'),
|
|
mock.call(args, self.app, 'qubes-template-fedora-32:1')
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def test_132_get_dl_list_multimerge1_success(self, mock_query):
|
|
counter = 0
|
|
def f(*args):
|
|
nonlocal counter
|
|
counter += 1
|
|
if counter == 1:
|
|
return [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'2',
|
|
'4.2',
|
|
'20200201',
|
|
'qubes-templates-itl-testing',
|
|
2048576,
|
|
datetime.datetime(2020, 2, 23, 4, 56),
|
|
'GPLv2',
|
|
'https://qubes-os.org/?',
|
|
'Qubes template for fedora-32 v2',
|
|
'Qubes template\n for fedora-32 v2\n'
|
|
)
|
|
]
|
|
return [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'1',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-32',
|
|
'Qubes template\n for fedora-32\n'
|
|
)
|
|
]
|
|
mock_query.side_effect = f
|
|
args = argparse.Namespace(
|
|
templates=['some.local.file.rpm', 'fedora-32:2', 'fedora-32:1']
|
|
)
|
|
ret = qubesadmin.tools.qvm_template.get_dl_list(args, self.app)
|
|
self.assertEqual(ret, {
|
|
'fedora-32': qubesadmin.tools.qvm_template.DlEntry(
|
|
('2', '4.2', '20200201'),
|
|
'qubes-templates-itl-testing',
|
|
2048576)
|
|
})
|
|
self.assertEqual(mock_query.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-fedora-32:2'),
|
|
mock.call(args, self.app, 'qubes-template-fedora-32:1')
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def test_133_get_dl_list_reinstall_success(self, mock_query):
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-name',
|
|
None)] = b'0\0test-vm'
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-epoch',
|
|
None)] = b'0\x000'
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-version',
|
|
None)] = b'0\x004.2'
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-release',
|
|
None)] = b'0\x0020200201'
|
|
mock_query.return_value = [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'1',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for test-vm',
|
|
'Qubes template\n for test-vm\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'0',
|
|
'4.2',
|
|
'20200201',
|
|
'qubes-templates-itl-testing',
|
|
2048576,
|
|
datetime.datetime(2020, 2, 23, 4, 56),
|
|
'GPLv2',
|
|
'https://qubes-os.org/?',
|
|
'Qubes template for test-vm v2',
|
|
'Qubes template\n for test-vm v2\n'
|
|
)
|
|
]
|
|
args = argparse.Namespace(
|
|
templates=['some.local.file.rpm', 'test-vm']
|
|
)
|
|
ret = qubesadmin.tools.qvm_template.get_dl_list(args, self.app,
|
|
qubesadmin.tools.qvm_template.VersionSelector.REINSTALL)
|
|
self.assertEqual(ret, {
|
|
'test-vm': qubesadmin.tools.qvm_template.DlEntry(
|
|
('0', '4.2', '20200201'),
|
|
'qubes-templates-itl-testing',
|
|
2048576
|
|
)
|
|
})
|
|
self.assertEqual(mock_query.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-test-vm')
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def test_134_get_dl_list_reinstall_nolocal_fail(self, mock_query):
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00'
|
|
mock_query.return_value = [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'1',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for test-vm',
|
|
'Qubes template\n for test-vm\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'0',
|
|
'4.2',
|
|
'20200201',
|
|
'qubes-templates-itl-testing',
|
|
2048576,
|
|
datetime.datetime(2020, 2, 23, 4, 56),
|
|
'GPLv2',
|
|
'https://qubes-os.org/?',
|
|
'Qubes template for test-vm v2',
|
|
'Qubes template\n for test-vm v2\n'
|
|
)
|
|
]
|
|
args = argparse.Namespace(templates=['test-vm'])
|
|
with mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
|
with self.assertRaises(SystemExit):
|
|
qubesadmin.tools.qvm_template.get_dl_list(args, self.app,
|
|
qubesadmin.tools.qvm_template.VersionSelector.REINSTALL)
|
|
self.assertTrue('not already installed' in mock_err.getvalue())
|
|
self.assertEqual(mock_query.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-test-vm')
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def test_135_get_dl_list_reinstall_nonmanaged_fail(self, mock_query):
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
mock_query.return_value = [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'1',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for test-vm',
|
|
'Qubes template\n for test-vm\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'0',
|
|
'4.2',
|
|
'20200201',
|
|
'qubes-templates-itl-testing',
|
|
2048576,
|
|
datetime.datetime(2020, 2, 23, 4, 56),
|
|
'GPLv2',
|
|
'https://qubes-os.org/?',
|
|
'Qubes template for test-vm v2',
|
|
'Qubes template\n for test-vm v2\n'
|
|
)
|
|
]
|
|
args = argparse.Namespace(templates=['test-vm'])
|
|
def qubesd_call(dest, method,
|
|
arg=None, payload=None, payload_stream=None,
|
|
orig_func=self.app.qubesd_call):
|
|
if method == 'admin.vm.feature.Get':
|
|
raise KeyError
|
|
return orig_func(dest, method, arg, payload, payload_stream)
|
|
with mock.patch('sys.stderr', new=io.StringIO()) as mock_err, \
|
|
mock.patch.object(self.app, 'qubesd_call') as mock_call:
|
|
mock_call.side_effect = qubesd_call
|
|
with self.assertRaises(SystemExit):
|
|
qubesadmin.tools.qvm_template.get_dl_list(args, self.app,
|
|
qubesadmin.tools.qvm_template.VersionSelector.REINSTALL)
|
|
self.assertTrue('not managed' in mock_err.getvalue())
|
|
self.assertEqual(mock_query.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-test-vm')
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def test_135_get_dl_list_reinstall_nonmanagednoname_fail(self, mock_query):
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-name',
|
|
None)] = b'0\0test-vm-2'
|
|
mock_query.return_value = [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'1',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for test-vm',
|
|
'Qubes template\n for test-vm\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'0',
|
|
'4.2',
|
|
'20200201',
|
|
'qubes-templates-itl-testing',
|
|
2048576,
|
|
datetime.datetime(2020, 2, 23, 4, 56),
|
|
'GPLv2',
|
|
'https://qubes-os.org/?',
|
|
'Qubes template for test-vm v2',
|
|
'Qubes template\n for test-vm v2\n'
|
|
)
|
|
]
|
|
args = argparse.Namespace(templates=['test-vm'])
|
|
with mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
|
with self.assertRaises(SystemExit):
|
|
qubesadmin.tools.qvm_template.get_dl_list(args, self.app,
|
|
qubesadmin.tools.qvm_template.VersionSelector.REINSTALL)
|
|
self.assertTrue('not managed' in mock_err.getvalue())
|
|
self.assertEqual(mock_query.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-test-vm')
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def test_136_get_dl_list_downgrade_success(self, mock_query):
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-name',
|
|
None)] = b'0\0test-vm'
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-epoch',
|
|
None)] = b'0\x000'
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-version',
|
|
None)] = b'0\x004.3'
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-release',
|
|
None)] = b'0\x0020200201'
|
|
mock_query.return_value = [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'0',
|
|
'4.2',
|
|
'20200201',
|
|
'qubes-templates-itl-testing',
|
|
2048576,
|
|
datetime.datetime(2020, 2, 23, 4, 56),
|
|
'GPLv2',
|
|
'https://qubes-os.org/?',
|
|
'Qubes template for test-vm v2',
|
|
'Qubes template\n for test-vm v2\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'0',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for test-vm',
|
|
'Qubes template\n for test-vm\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'1',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for test-vm',
|
|
'Qubes template\n for test-vm\n'
|
|
)
|
|
]
|
|
args = argparse.Namespace(templates=['test-vm'])
|
|
ret = qubesadmin.tools.qvm_template.get_dl_list(args, self.app,
|
|
qubesadmin.tools.qvm_template.VersionSelector.LATEST_LOWER)
|
|
self.assertEqual(ret, {
|
|
'test-vm': qubesadmin.tools.qvm_template.DlEntry(
|
|
('0', '4.2', '20200201'),
|
|
'qubes-templates-itl-testing',
|
|
2048576
|
|
)
|
|
})
|
|
self.assertEqual(mock_query.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-test-vm')
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def test_137_get_dl_list_downgrade_nonmanaged_fail(self, mock_query):
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-name',
|
|
None)] = b'0\0test-vm-2'
|
|
mock_query.return_value = [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'0',
|
|
'4.2',
|
|
'20200201',
|
|
'qubes-templates-itl-testing',
|
|
2048576,
|
|
datetime.datetime(2020, 2, 23, 4, 56),
|
|
'GPLv2',
|
|
'https://qubes-os.org/?',
|
|
'Qubes template for test-vm v2',
|
|
'Qubes template\n for test-vm v2\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'0',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for test-vm',
|
|
'Qubes template\n for test-vm\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'1',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for test-vm',
|
|
'Qubes template\n for test-vm\n'
|
|
)
|
|
]
|
|
args = argparse.Namespace(templates=['test-vm'])
|
|
with mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
|
with self.assertRaises(SystemExit):
|
|
qubesadmin.tools.qvm_template.get_dl_list(args, self.app,
|
|
qubesadmin.tools.qvm_template.VersionSelector.REINSTALL)
|
|
self.assertTrue('not managed' in mock_err.getvalue())
|
|
self.assertEqual(mock_query.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-test-vm')
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def test_138_get_dl_list_downgrade_notfound_skip(self, mock_query):
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-name',
|
|
None)] = b'0\0test-vm'
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-epoch',
|
|
None)] = b'0\x000'
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-version',
|
|
None)] = b'0\x004.3'
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-release',
|
|
None)] = b'0\x0020200201'
|
|
mock_query.return_value = [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'1',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for test-vm',
|
|
'Qubes template\n for test-vm\n'
|
|
)
|
|
]
|
|
args = argparse.Namespace(templates=['test-vm'])
|
|
with mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
|
ret = qubesadmin.tools.qvm_template.get_dl_list(args, self.app,
|
|
qubesadmin.tools.qvm_template.VersionSelector.LATEST_LOWER)
|
|
self.assertTrue('lowest version' in mock_err.getvalue())
|
|
self.assertEqual(ret, {})
|
|
self.assertEqual(mock_query.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-test-vm')
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def test_139_get_dl_list_upgrade_success(self, mock_query):
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-name',
|
|
None)] = b'0\0test-vm'
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-epoch',
|
|
None)] = b'0\x000'
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-version',
|
|
None)] = b'0\x004.3'
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-release',
|
|
None)] = b'0\x0020200201'
|
|
mock_query.return_value = [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'1',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for test-vm',
|
|
'Qubes template\n for test-vm\n'
|
|
)
|
|
]
|
|
args = argparse.Namespace(templates=['test-vm'])
|
|
ret = qubesadmin.tools.qvm_template.get_dl_list(args, self.app,
|
|
qubesadmin.tools.qvm_template.VersionSelector.LATEST_HIGHER)
|
|
self.assertEqual(ret, {
|
|
'test-vm': qubesadmin.tools.qvm_template.DlEntry(
|
|
('1', '4.1', '20200101'), 'qubes-templates-itl', 1048576
|
|
)
|
|
})
|
|
self.assertEqual(mock_query.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-test-vm')
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def test_140_get_dl_list_downgrade_notfound_skip(self, mock_query):
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-name',
|
|
None)] = b'0\0test-vm'
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-epoch',
|
|
None)] = b'0\x000'
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-version',
|
|
None)] = b'0\x004.3'
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-release',
|
|
None)] = b'0\x0020200201'
|
|
mock_query.return_value = [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'0',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for test-vm',
|
|
'Qubes template\n for test-vm\n'
|
|
)
|
|
]
|
|
args = argparse.Namespace(templates=['test-vm'])
|
|
with mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
|
ret = qubesadmin.tools.qvm_template.get_dl_list(args, self.app,
|
|
qubesadmin.tools.qvm_template.VersionSelector.LATEST_HIGHER)
|
|
self.assertTrue('highest version' in mock_err.getvalue())
|
|
self.assertEqual(ret, {})
|
|
self.assertEqual(mock_query.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-test-vm')
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def test_141_get_dl_list_reinstall_notfound_fail(self, mock_query):
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-name',
|
|
None)] = b'0\0test-vm'
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-epoch',
|
|
None)] = b'0\x000'
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-version',
|
|
None)] = b'0\x004.3'
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-release',
|
|
None)] = b'0\x0020200201'
|
|
mock_query.return_value = [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'0',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for test-vm',
|
|
'Qubes template\n for test-vm\n'
|
|
)
|
|
]
|
|
args = argparse.Namespace(templates=['test-vm'])
|
|
with mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
|
with self.assertRaises(SystemExit):
|
|
qubesadmin.tools.qvm_template.get_dl_list(args, self.app,
|
|
qubesadmin.tools.qvm_template.VersionSelector.REINSTALL)
|
|
self.assertTrue('Same version' in mock_err.getvalue())
|
|
self.assertTrue('not found' in mock_err.getvalue())
|
|
self.assertEqual(mock_query.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-test-vm')
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def test_150_list_templates_installed_success(self, mock_query):
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n' \
|
|
b'test-vm-2 class=TemplateVM state=Halted\n' \
|
|
b'non-spec class=TemplateVM state=Halted\n'
|
|
build_time = '2020-09-01 14:30:00' # 1598970600
|
|
install_time = '2020-09-01 15:30:00'
|
|
for key, val in [
|
|
('name', 'test-vm'),
|
|
('epoch', '2'),
|
|
('version', '4.1'),
|
|
('release', '2020'),
|
|
('reponame', '@commandline'),
|
|
('buildtime', build_time),
|
|
('installtime', install_time),
|
|
('license', 'GPL'),
|
|
('url', 'https://qubes-os.org'),
|
|
('summary', 'Summary'),
|
|
('description', 'Desc|desc')]:
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-{key}',
|
|
None)] = b'0\0' + val.encode()
|
|
for key, val in [('name', 'test-vm-2-not-managed')]:
|
|
self.app.expected_calls[(
|
|
'test-vm-2',
|
|
'admin.vm.feature.Get',
|
|
f'template-{key}',
|
|
None)] = b'0\0' + val.encode()
|
|
for key, val in [
|
|
('name', 'non-spec'),
|
|
('epoch', '0'),
|
|
('version', '4.3'),
|
|
('release', '20200201')]:
|
|
self.app.expected_calls[(
|
|
'non-spec',
|
|
'admin.vm.feature.Get',
|
|
f'template-{key}',
|
|
None)] = b'0\0' + val.encode()
|
|
args = argparse.Namespace(
|
|
all=False,
|
|
installed=True,
|
|
available=False,
|
|
extras=False,
|
|
upgrades=False,
|
|
all_versions=True,
|
|
machine_readable=False,
|
|
machine_readable_json=False,
|
|
templates=['test-vm*']
|
|
)
|
|
with mock.patch('sys.stdout', new=io.StringIO()) as mock_out, \
|
|
mock.patch.object(self.app.domains['test-vm'],
|
|
'get_disk_utilization') as mock_disk:
|
|
mock_disk.return_value = 1234321
|
|
qubesadmin.tools.qvm_template.list_templates(
|
|
args, self.app, 'list')
|
|
self.assertEqual(mock_out.getvalue(),
|
|
'''Installed Templates
|
|
[('test-vm', '2:4.1-2020', '@commandline')]
|
|
''')
|
|
self.assertEqual(mock_disk.mock_calls, [mock.call()])
|
|
self.assertEqual(mock_query.mock_calls, [])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def test_151_list_templates_available_success(self, mock_query):
|
|
counter = 0
|
|
def f(*args):
|
|
nonlocal counter
|
|
counter += 1
|
|
if counter == 1:
|
|
return [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'0',
|
|
'4.2',
|
|
'20200201',
|
|
'qubes-templates-itl-testing',
|
|
2048576,
|
|
datetime.datetime(2020, 2, 23, 4, 56),
|
|
'GPLv2',
|
|
'https://qubes-os.org/?',
|
|
'Qubes template for fedora-32 v2',
|
|
'Qubes template\n for fedora-32 v2\n'
|
|
)
|
|
]
|
|
return [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-31',
|
|
'1',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-31',
|
|
'Qubes template\n for fedora-31\n'
|
|
)
|
|
]
|
|
mock_query.side_effect = f
|
|
args = argparse.Namespace(
|
|
all=False,
|
|
installed=False,
|
|
available=True,
|
|
extras=False,
|
|
upgrades=False,
|
|
all_versions=True,
|
|
machine_readable=False,
|
|
machine_readable_json=False,
|
|
templates=['fedora-32', 'fedora-31']
|
|
)
|
|
with mock.patch('sys.stdout', new=io.StringIO()) as mock_out:
|
|
qubesadmin.tools.qvm_template.list_templates(
|
|
args, self.app, 'list')
|
|
# Order not determinstic because of sets
|
|
expected = [
|
|
('fedora-31', '1:4.1-20200101', 'qubes-templates-itl'),
|
|
('fedora-32', '0:4.2-20200201', 'qubes-templates-itl-testing')
|
|
]
|
|
self.assertTrue(mock_out.getvalue() == \
|
|
f'''Available Templates
|
|
{str([expected[1], expected[0]])}
|
|
''' \
|
|
or mock_out.getvalue() == \
|
|
f'''Available Templates
|
|
{str([expected[0], expected[1]])}
|
|
''')
|
|
self.assertEqual(mock_query.mock_calls, [
|
|
mock.call(args, self.app, 'fedora-32'),
|
|
mock.call(args, self.app, 'fedora-31')
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def test_151_list_templates_available_all_success(self, mock_query):
|
|
mock_query.return_value = [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-31',
|
|
'1',
|
|
'4.1',
|
|
'20190101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2019, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-31',
|
|
'Qubes template\n for fedora-31\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-31',
|
|
'1',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-31',
|
|
'Qubes template\n for fedora-31\n'
|
|
),
|
|
]
|
|
args = argparse.Namespace(
|
|
all=False,
|
|
installed=False,
|
|
available=True,
|
|
extras=False,
|
|
upgrades=False,
|
|
all_versions=True,
|
|
machine_readable=False,
|
|
machine_readable_json=False,
|
|
templates=[]
|
|
)
|
|
with mock.patch('sys.stdout', new=io.StringIO()) as mock_out:
|
|
qubesadmin.tools.qvm_template.list_templates(
|
|
args, self.app, 'list')
|
|
self.assertEqual(mock_out.getvalue(),
|
|
'''Available Templates
|
|
[('fedora-31', '1:4.1-20190101', 'qubes-templates-itl'), ('fedora-31', '1:4.1-20200101', 'qubes-templates-itl')]
|
|
''')
|
|
self.assertEqual(mock_query.mock_calls, [
|
|
mock.call(args, self.app)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def test_151_list_templates_available_only_latest_success(self, mock_query):
|
|
mock_query.return_value = [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-31',
|
|
'1',
|
|
'4.1',
|
|
'20190101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2019, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-31',
|
|
'Qubes template\n for fedora-31\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-31',
|
|
'1',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-31',
|
|
'Qubes template\n for fedora-31\n'
|
|
),
|
|
]
|
|
args = argparse.Namespace(
|
|
all=False,
|
|
installed=False,
|
|
available=True,
|
|
extras=False,
|
|
upgrades=False,
|
|
all_versions=False,
|
|
machine_readable=False,
|
|
machine_readable_json=False,
|
|
templates=[]
|
|
)
|
|
with mock.patch('sys.stdout', new=io.StringIO()) as mock_out:
|
|
qubesadmin.tools.qvm_template.list_templates(
|
|
args, self.app, 'list')
|
|
self.assertEqual(mock_out.getvalue(),
|
|
'''Available Templates
|
|
[('fedora-31', '1:4.1-20200101', 'qubes-templates-itl')]
|
|
''')
|
|
self.assertEqual(mock_query.mock_calls, [
|
|
mock.call(args, self.app)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def test_152_list_templates_extras_success(self, mock_query):
|
|
mock_query.return_value = [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'2',
|
|
'4.1',
|
|
'2020',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 9, 1, 14, 30,
|
|
tzinfo=datetime.timezone.utc),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-31',
|
|
'Qubes template\n for fedora-31\n'
|
|
)
|
|
]
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n' \
|
|
b'test-vm-2 class=TemplateVM state=Halted\n' \
|
|
b'test-vm-3 class=TemplateVM state=Halted\n' \
|
|
b'non-spec class=TemplateVM state=Halted\n'
|
|
for key, val in [('name', 'test-vm')]:
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-{key}',
|
|
None)] = b'0\0' + val.encode()
|
|
for key, val in [
|
|
('name', 'test-vm-2'),
|
|
('epoch', '1'),
|
|
('version', '4.0'),
|
|
('release', '2019'),
|
|
('reponame', 'qubes-template-itl'),
|
|
('buildtime', '2020-09-02 14:30:00'),
|
|
('installtime', '2020-09-02 15:30:00'),
|
|
('license', 'GPLv2'),
|
|
('url', 'https://qubes-os.org/?'),
|
|
('summary', 'Summary2'),
|
|
('description', 'Desc|desc|2')]:
|
|
self.app.expected_calls[(
|
|
'test-vm-2',
|
|
'admin.vm.feature.Get',
|
|
f'template-{key}',
|
|
None)] = b'0\0' + val.encode()
|
|
for key, val in [('name', 'test-vm-3-non-managed')]:
|
|
self.app.expected_calls[(
|
|
'test-vm-3',
|
|
'admin.vm.feature.Get',
|
|
f'template-{key}',
|
|
None)] = b'0\0' + val.encode()
|
|
for key, val in [
|
|
('name', 'non-spec'),
|
|
('epoch', '1'),
|
|
('version', '4.0'),
|
|
('release', '2019')]:
|
|
self.app.expected_calls[(
|
|
'non-spec',
|
|
'admin.vm.feature.Get',
|
|
f'template-{key}',
|
|
None)] = b'0\0' + val.encode()
|
|
args = argparse.Namespace(
|
|
all=False,
|
|
installed=False,
|
|
available=False,
|
|
extras=True,
|
|
upgrades=False,
|
|
all_versions=True,
|
|
machine_readable=False,
|
|
machine_readable_json=False,
|
|
templates=['test-vm*']
|
|
)
|
|
with mock.patch('sys.stdout', new=io.StringIO()) as mock_out, \
|
|
mock.patch.object(self.app.domains['test-vm-2'],
|
|
'get_disk_utilization') as mock_disk:
|
|
mock_disk.return_value = 1234321
|
|
qubesadmin.tools.qvm_template.list_templates(
|
|
args, self.app, 'list')
|
|
self.assertEqual(mock_out.getvalue(),
|
|
'''Extra Templates
|
|
[('test-vm-2', '1:4.0-2019', 'qubes-template-itl')]
|
|
''')
|
|
self.assertEqual(mock_disk.mock_calls, [mock.call()])
|
|
self.assertEqual(mock_query.mock_calls, [
|
|
mock.call(args, self.app, 'test-vm*')
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def test_153_list_templates_upgrades_success(self, mock_query):
|
|
mock_query.return_value = [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'2',
|
|
'4.1',
|
|
'2020',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 9, 1, 14, 30,
|
|
tzinfo=datetime.timezone.utc),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-31',
|
|
'Qubes template\n for fedora-31\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'0',
|
|
'4.1',
|
|
'2020',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 9, 1, 14, 30,
|
|
tzinfo=datetime.timezone.utc),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-31',
|
|
'Qubes template\n for fedora-31\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm-3',
|
|
'0',
|
|
'4.1',
|
|
'2020',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 9, 1, 14, 30,
|
|
tzinfo=datetime.timezone.utc),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-31',
|
|
'Qubes template\n for fedora-31\n'
|
|
)
|
|
]
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n' \
|
|
b'test-vm-2 class=TemplateVM state=Halted\n' \
|
|
b'test-vm-3 class=TemplateVM state=Halted\n'
|
|
for key, val in [
|
|
('name', 'test-vm'),
|
|
('epoch', '1'),
|
|
('version', '4.0'),
|
|
('release', '2019')]:
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-{key}',
|
|
None)] = b'0\0' + val.encode()
|
|
for key, val in [
|
|
('name', 'test-vm-2'),
|
|
('epoch', '1'),
|
|
('version', '4.0'),
|
|
('release', '2019')]:
|
|
self.app.expected_calls[(
|
|
'test-vm-2',
|
|
'admin.vm.feature.Get',
|
|
f'template-{key}',
|
|
None)] = b'0\0' + val.encode()
|
|
for key, val in [('name', 'test-vm-3-non-managed')]:
|
|
self.app.expected_calls[(
|
|
'test-vm-3',
|
|
'admin.vm.feature.Get',
|
|
f'template-{key}',
|
|
None)] = b'0\0' + val.encode()
|
|
args = argparse.Namespace(
|
|
all=False,
|
|
installed=False,
|
|
available=False,
|
|
extras=False,
|
|
upgrades=True,
|
|
all_versions=True,
|
|
machine_readable=False,
|
|
machine_readable_json=False,
|
|
templates=['test-vm*']
|
|
)
|
|
with mock.patch('sys.stdout', new=io.StringIO()) as mock_out, \
|
|
mock.patch.object(self.app.domains['test-vm-2'],
|
|
'get_disk_utilization') as mock_disk:
|
|
mock_disk.return_value = 1234321
|
|
qubesadmin.tools.qvm_template.list_templates(
|
|
args, self.app, 'list')
|
|
self.assertEqual(mock_out.getvalue(),
|
|
'''Available Upgrades
|
|
[('test-vm', '2:4.1-2020', 'qubes-templates-itl')]
|
|
''')
|
|
self.assertEqual(mock_disk.mock_calls, [])
|
|
self.assertEqual(mock_query.mock_calls, [
|
|
mock.call(args, self.app, 'test-vm*')
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def __test_list_templates_all_success(self, operation,
|
|
args, expected, mock_query):
|
|
mock_query.return_value = [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'2',
|
|
'4.1',
|
|
'2020',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 9, 1, 14, 30,
|
|
tzinfo=datetime.timezone.utc),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-31',
|
|
'Qubes template\n for fedora-31\n'
|
|
)
|
|
]
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm-2 class=TemplateVM state=Halted\n'
|
|
for key, val in [
|
|
('name', 'test-vm-2'),
|
|
('epoch', '1'),
|
|
('version', '4.0'),
|
|
('release', '2019'),
|
|
('reponame', '@commandline'),
|
|
('buildtime', '2020-09-02 14:30:00'),
|
|
('installtime', '2020-09-02 15:30:00'),
|
|
('license', 'GPL'),
|
|
('url', 'https://qubes-os.org'),
|
|
('summary', 'Summary'),
|
|
('description', 'Desc|desc')]:
|
|
self.app.expected_calls[(
|
|
'test-vm-2',
|
|
'admin.vm.feature.Get',
|
|
f'template-{key}',
|
|
None)] = b'0\0' + val.encode()
|
|
with mock.patch('sys.stdout', new=io.StringIO()) as mock_out, \
|
|
mock.patch.object(self.app.domains['test-vm-2'],
|
|
'get_disk_utilization') as mock_disk:
|
|
mock_disk.return_value = 1234321
|
|
qubesadmin.tools.qvm_template.list_templates(
|
|
args, self.app, operation)
|
|
self.assertEqual(mock_out.getvalue(), expected)
|
|
self.assertEqual(mock_disk.mock_calls, [mock.call()])
|
|
self.assertEqual(mock_query.mock_calls, [
|
|
mock.call(args, self.app, 'test-vm*')
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
def test_154_list_templates_all_success(self):
|
|
args = argparse.Namespace(
|
|
all=True,
|
|
installed=False,
|
|
available=False,
|
|
extras=False,
|
|
upgrades=False,
|
|
all_versions=True,
|
|
machine_readable=False,
|
|
machine_readable_json=False,
|
|
templates=['test-vm*']
|
|
)
|
|
expected = \
|
|
'''Installed Templates
|
|
[('test-vm-2', '1:4.0-2019', '@commandline')]
|
|
Available Templates
|
|
[('test-vm', '2:4.1-2020', 'qubes-templates-itl')]
|
|
'''
|
|
self.__test_list_templates_all_success('list', args, expected)
|
|
|
|
def test_155_list_templates_all_implicit_success(self):
|
|
args = argparse.Namespace(
|
|
all=False,
|
|
installed=False,
|
|
available=False,
|
|
extras=False,
|
|
upgrades=False,
|
|
all_versions=True,
|
|
machine_readable=False,
|
|
machine_readable_json=False,
|
|
templates=['test-vm*']
|
|
)
|
|
expected = \
|
|
'''Installed Templates
|
|
[('test-vm-2', '1:4.0-2019', '@commandline')]
|
|
Available Templates
|
|
[('test-vm', '2:4.1-2020', 'qubes-templates-itl')]
|
|
'''
|
|
self.__test_list_templates_all_success('list', args, expected)
|
|
|
|
def test_156_list_templates_info_all_success(self):
|
|
args = argparse.Namespace(
|
|
all=False,
|
|
installed=False,
|
|
available=False,
|
|
extras=False,
|
|
upgrades=False,
|
|
all_versions=True,
|
|
machine_readable=False,
|
|
machine_readable_json=False,
|
|
templates=['test-vm*']
|
|
)
|
|
expected = \
|
|
'''Installed Templates
|
|
[('Name', ':', 'test-vm-2'), ('Epoch', ':', '1'), ('Version', ':', '4.0'), ('Release', ':', '2019'), ('Size', ':', '1.2 MiB'), ('Repository', ':', '@commandline'), ('Buildtime', ':', '2020-09-02 14:30:00'), ('Install time', ':', '2020-09-02 15:30:00'), ('URL', ':', 'https://qubes-os.org'), ('License', ':', 'GPL'), ('Summary', ':', 'Summary'), ('Description', ':', 'Desc'), ('', ':', 'desc'), (' ', ' ', ' ')]
|
|
Available Templates
|
|
[('Name', ':', 'test-vm'), ('Epoch', ':', '2'), ('Version', ':', '4.1'), ('Release', ':', '2020'), ('Size', ':', '1.0 MiB'), ('Repository', ':', 'qubes-templates-itl'), ('Buildtime', ':', '2020-09-01 14:30:00+00:00'), ('URL', ':', 'https://qubes-os.org'), ('License', ':', 'GPL'), ('Summary', ':', 'Qubes template for fedora-31'), ('Description', ':', 'Qubes template'), ('', ':', ' for fedora-31'), (' ', ' ', ' ')]
|
|
'''
|
|
self.__test_list_templates_all_success('info', args, expected)
|
|
|
|
def test_157_list_templates_list_all_machinereadable_success(self):
|
|
args = argparse.Namespace(
|
|
all=False,
|
|
installed=False,
|
|
available=False,
|
|
extras=False,
|
|
upgrades=False,
|
|
all_versions=True,
|
|
machine_readable=True,
|
|
machine_readable_json=False,
|
|
templates=['test-vm*']
|
|
)
|
|
expected = \
|
|
'''installed|test-vm-2|1:4.0-2019|@commandline
|
|
available|test-vm|2:4.1-2020|qubes-templates-itl
|
|
'''
|
|
self.__test_list_templates_all_success('list', args, expected)
|
|
|
|
def test_158_list_templates_info_all_machinereadable_success(self):
|
|
args = argparse.Namespace(
|
|
all=False,
|
|
installed=False,
|
|
available=False,
|
|
extras=False,
|
|
upgrades=False,
|
|
all_versions=True,
|
|
machine_readable=True,
|
|
machine_readable_json=False,
|
|
templates=['test-vm*']
|
|
)
|
|
expected = \
|
|
'''installed|test-vm-2|1|4.0|2019|@commandline|1234321|2020-09-02 14:30:00|2020-09-02 15:30:00|GPL|https://qubes-os.org|Summary|Desc|desc
|
|
available|test-vm|2|4.1|2020|qubes-templates-itl|1048576|2020-09-01 14:30:00||GPL|https://qubes-os.org|Qubes template for fedora-31|Qubes template| for fedora-31|
|
|
'''
|
|
self.__test_list_templates_all_success('info', args, expected)
|
|
|
|
def test_159_list_templates_list_all_machinereadablejson_success(self):
|
|
args = argparse.Namespace(
|
|
all=False,
|
|
installed=False,
|
|
available=False,
|
|
extras=False,
|
|
upgrades=False,
|
|
all_versions=True,
|
|
machine_readable=False,
|
|
machine_readable_json=True,
|
|
templates=['test-vm*']
|
|
)
|
|
expected = \
|
|
'''{"installed": [{"name": "test-vm-2", "evr": "1:4.0-2019", "reponame": "@commandline"}], "available": [{"name": "test-vm", "evr": "2:4.1-2020", "reponame": "qubes-templates-itl"}]}
|
|
'''
|
|
self.__test_list_templates_all_success('list', args, expected)
|
|
|
|
def test_160_list_templates_info_all_machinereadablejson_success(self):
|
|
args = argparse.Namespace(
|
|
all=False,
|
|
installed=False,
|
|
available=False,
|
|
extras=False,
|
|
upgrades=False,
|
|
all_versions=True,
|
|
machine_readable=False,
|
|
machine_readable_json=True,
|
|
templates=['test-vm*']
|
|
)
|
|
expected = \
|
|
r'''{"installed": [{"name": "test-vm-2", "epoch": "1", "version": "4.0", "release": "2019", "reponame": "@commandline", "size": "1234321", "buildtime": "2020-09-02 14:30:00", "installtime": "2020-09-02 15:30:00", "license": "GPL", "url": "https://qubes-os.org", "summary": "Summary", "description": "Desc\ndesc"}], "available": [{"name": "test-vm", "epoch": "2", "version": "4.1", "release": "2020", "reponame": "qubes-templates-itl", "size": "1048576", "buildtime": "2020-09-01 14:30:00", "installtime": "", "license": "GPL", "url": "https://qubes-os.org", "summary": "Qubes template for fedora-31", "description": "Qubes template\n for fedora-31\n"}]}
|
|
'''
|
|
self.__test_list_templates_all_success('info', args, expected)
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def test_161_list_templates_noresults_fail(self, mock_query):
|
|
mock_query.return_value = []
|
|
args = argparse.Namespace(
|
|
all=False,
|
|
installed=False,
|
|
available=True,
|
|
extras=False,
|
|
upgrades=False,
|
|
all_versions=True,
|
|
machine_readable=False,
|
|
machine_readable_json=False,
|
|
templates=[]
|
|
)
|
|
with mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
|
with self.assertRaises(SystemExit):
|
|
qubesadmin.tools.qvm_template.list_templates(
|
|
args, self.app, 'list')
|
|
self.assertTrue('No matching templates' in mock_err.getvalue())
|
|
self.assertEqual(mock_query.mock_calls, [
|
|
mock.call(args, self.app)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def test_170_search_success(self, mock_query):
|
|
mock_query.return_value = [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'2',
|
|
'4.1',
|
|
'2020',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 9, 1, 14, 30,
|
|
tzinfo=datetime.timezone.utc),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-31',
|
|
'Qubes template\n for fedora-31\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'0',
|
|
'4.1',
|
|
'2020',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 9, 1, 14, 30,
|
|
tzinfo=datetime.timezone.utc),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Older Qubes template for fedora-31',
|
|
'Older Qubes template\n for fedora-31\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'should-not-match-3',
|
|
'0',
|
|
'4.1',
|
|
'2020',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 9, 1, 14, 30,
|
|
tzinfo=datetime.timezone.utc),
|
|
'GPL',
|
|
'https://qubes-os.org/test-vm',
|
|
'Qubes template for fedora-31',
|
|
'test-vm Qubes template\n for fedora-31\n'
|
|
)
|
|
]
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm-2 class=TemplateVM state=Halted\n'
|
|
for key, val in [
|
|
('name', 'test-vm-2'),
|
|
('epoch', '1'),
|
|
('version', '4.0'),
|
|
('release', '2019'),
|
|
('reponame', '@commandline'),
|
|
('buildtime', '2020-09-02 14:30:00'),
|
|
('license', 'GPL'),
|
|
('url', 'https://qubes-os.org'),
|
|
('summary', 'Summary'),
|
|
('description', 'Desc|desc')]:
|
|
self.app.expected_calls[(
|
|
'test-vm-2',
|
|
'admin.vm.feature.Get',
|
|
f'template-{key}',
|
|
None)] = b'0\0' + val.encode()
|
|
args = argparse.Namespace(
|
|
all=False,
|
|
templates=['test-vm']
|
|
)
|
|
with mock.patch('sys.stdout', new=io.StringIO()) as mock_out, \
|
|
mock.patch.object(self.app.domains['test-vm-2'],
|
|
'get_disk_utilization') as mock_disk:
|
|
mock_disk.return_value = 1234321
|
|
qubesadmin.tools.qvm_template.search(args, self.app)
|
|
self.assertEqual(mock_out.getvalue(),
|
|
'''=== Name Exactly Matched: test-vm ===
|
|
test-vm : Qubes template for fedora-31
|
|
=== Name Matched: test-vm ===
|
|
test-vm-2 : Summary
|
|
''')
|
|
self.assertEqual(mock_disk.mock_calls, [mock.call()])
|
|
self.assertEqual(mock_query.mock_calls, [
|
|
mock.call(args, self.app)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def test_171_search_summary_success(self, mock_query):
|
|
mock_query.return_value = [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-template',
|
|
'2',
|
|
'4.1',
|
|
'2020',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 9, 1, 14, 30,
|
|
tzinfo=datetime.timezone.utc),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for test-vm :)',
|
|
'Qubes template\n for fedora-31\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-template-exact',
|
|
'2',
|
|
'4.1',
|
|
'2020',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 9, 1, 14, 30,
|
|
tzinfo=datetime.timezone.utc),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'test-vm',
|
|
'Qubes template\n for fedora-31\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'2',
|
|
'4.1',
|
|
'2020',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 9, 1, 14, 30,
|
|
tzinfo=datetime.timezone.utc),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for test-vm',
|
|
'Qubes template\n for fedora-31\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm-2',
|
|
'2',
|
|
'4.1',
|
|
'2020',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 9, 1, 14, 30,
|
|
tzinfo=datetime.timezone.utc),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for test-vm-2',
|
|
'Qubes template\n for fedora-31\n'
|
|
),
|
|
]
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00'
|
|
args = argparse.Namespace(
|
|
all=False,
|
|
templates=['test-vm']
|
|
)
|
|
with mock.patch('sys.stdout', new=io.StringIO()) as mock_out:
|
|
qubesadmin.tools.qvm_template.search(args, self.app)
|
|
self.assertEqual(mock_out.getvalue(),
|
|
'''=== Name & Summary Matched: test-vm ===
|
|
test-vm : Qubes template for test-vm
|
|
test-vm-2 : Qubes template for test-vm-2
|
|
=== Summary Matched: test-vm ===
|
|
test-template : Qubes template for test-vm :)
|
|
=== Summary Exactly Matched: test-vm ===
|
|
test-template-exact : test-vm
|
|
''')
|
|
self.assertEqual(mock_query.mock_calls, [
|
|
mock.call(args, self.app)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def test_172_search_namesummaryexact_success(self, mock_query):
|
|
mock_query.return_value = [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-template-exact',
|
|
'2',
|
|
'4.1',
|
|
'2020',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 9, 1, 14, 30,
|
|
tzinfo=datetime.timezone.utc),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'test-vm',
|
|
'Qubes template\n for fedora-31\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'2',
|
|
'4.1',
|
|
'2020',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 9, 1, 14, 30,
|
|
tzinfo=datetime.timezone.utc),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'test-vm',
|
|
'Qubes template\n for fedora-31\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm-2',
|
|
'2',
|
|
'4.1',
|
|
'2020',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 9, 1, 14, 30,
|
|
tzinfo=datetime.timezone.utc),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'test-vm',
|
|
'Qubes template\n for fedora-31\n'
|
|
)
|
|
]
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00'
|
|
args = argparse.Namespace(
|
|
all=False,
|
|
templates=['test-vm']
|
|
)
|
|
with mock.patch('sys.stdout', new=io.StringIO()) as mock_out:
|
|
qubesadmin.tools.qvm_template.search(args, self.app)
|
|
self.assertEqual(mock_out.getvalue(),
|
|
'''=== Name & Summary Exactly Matched: test-vm ===
|
|
test-vm : test-vm
|
|
=== Name & Summary Matched: test-vm ===
|
|
test-vm-2 : test-vm
|
|
=== Summary Exactly Matched: test-vm ===
|
|
test-template-exact : test-vm
|
|
''')
|
|
self.assertEqual(mock_query.mock_calls, [
|
|
mock.call(args, self.app)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def test_173_search_multiquery_success(self, mock_query):
|
|
mock_query.return_value = [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-template-exact',
|
|
'2',
|
|
'4.1',
|
|
'2020',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 9, 1, 14, 30,
|
|
tzinfo=datetime.timezone.utc),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'test-vm',
|
|
'Qubes template\n for fedora-31\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'2',
|
|
'4.1',
|
|
'2020',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 9, 1, 14, 30,
|
|
tzinfo=datetime.timezone.utc),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'test-vm',
|
|
'Qubes template\n for fedora-31\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'should-not-match',
|
|
'2',
|
|
'4.1',
|
|
'2020',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 9, 1, 14, 30,
|
|
tzinfo=datetime.timezone.utc),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Summary',
|
|
'test-vm Qubes template\n for fedora-31\n'
|
|
)
|
|
]
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00'
|
|
args = argparse.Namespace(
|
|
all=False,
|
|
templates=['test-vm', 'test-template']
|
|
)
|
|
with mock.patch('sys.stdout', new=io.StringIO()) as mock_out:
|
|
qubesadmin.tools.qvm_template.search(args, self.app)
|
|
self.assertEqual(mock_out.getvalue(),
|
|
'''=== Name & Summary Matched: test-template, test-vm ===
|
|
test-template-exact : test-vm
|
|
''')
|
|
self.assertEqual(mock_query.mock_calls, [
|
|
mock.call(args, self.app)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def test_174_search_multiquery_exact_success(self, mock_query):
|
|
mock_query.return_value = [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'2',
|
|
'4.1',
|
|
'2020',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 9, 1, 14, 30,
|
|
tzinfo=datetime.timezone.utc),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'summary',
|
|
'Qubes template\n for fedora-31\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'summary',
|
|
'2',
|
|
'4.1',
|
|
'2020',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 9, 1, 14, 30,
|
|
tzinfo=datetime.timezone.utc),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'test-vm Summary',
|
|
'Qubes template\n for fedora-31\n'
|
|
)
|
|
]
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00'
|
|
args = argparse.Namespace(
|
|
all=False,
|
|
templates=['test-vm', 'summary']
|
|
)
|
|
with mock.patch('sys.stdout', new=io.StringIO()) as mock_out:
|
|
qubesadmin.tools.qvm_template.search(args, self.app)
|
|
self.assertEqual(mock_out.getvalue(),
|
|
'''=== Name & Summary Matched: summary, test-vm ===
|
|
summary : test-vm Summary
|
|
=== Name & Summary Exactly Matched: summary, test-vm ===
|
|
test-vm : summary
|
|
''')
|
|
self.assertEqual(mock_query.mock_calls, [
|
|
mock.call(args, self.app)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def test_175_search_all_success(self, mock_query):
|
|
mock_query.return_value = [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'2',
|
|
'4.1',
|
|
'2020',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 9, 1, 14, 30,
|
|
tzinfo=datetime.timezone.utc),
|
|
'GPL',
|
|
'https://qubes-os.org/keyword-url',
|
|
'summary',
|
|
'Qubes template\n for fedora-31\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm-exact',
|
|
'2',
|
|
'4.1',
|
|
'2020',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 9, 1, 14, 30,
|
|
tzinfo=datetime.timezone.utc),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'test-vm Summary',
|
|
'Qubes template\n for fedora-31\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm-exac2',
|
|
'2',
|
|
'4.1',
|
|
'2020',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 9, 1, 14, 30,
|
|
tzinfo=datetime.timezone.utc),
|
|
'GPL',
|
|
'test-vm-exac2',
|
|
'test-vm Summary',
|
|
'Qubes template\n for fedora-31\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm-2',
|
|
'2',
|
|
'4.1',
|
|
'2020',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 9, 1, 14, 30,
|
|
tzinfo=datetime.timezone.utc),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'test-vm Summary',
|
|
'keyword-desc'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'should-not-match',
|
|
'2',
|
|
'4.1',
|
|
'2020',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 9, 1, 14, 30,
|
|
tzinfo=datetime.timezone.utc),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Summary',
|
|
'Description'
|
|
)
|
|
]
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00'
|
|
args = argparse.Namespace(
|
|
all=True,
|
|
templates=['test-vm-exact', 'test-vm-exac2',
|
|
'keyword-url', 'keyword-desc']
|
|
)
|
|
with mock.patch('sys.stdout', new=io.StringIO()) as mock_out:
|
|
qubesadmin.tools.qvm_template.search(args, self.app)
|
|
self.assertEqual(mock_out.getvalue(),
|
|
'''=== Name & URL Exactly Matched: test-vm-exac2 ===
|
|
test-vm-exac2 : test-vm Summary
|
|
=== Name Exactly Matched: test-vm-exact ===
|
|
test-vm-exact : test-vm Summary
|
|
=== Description Exactly Matched: keyword-desc ===
|
|
test-vm-2 : test-vm Summary
|
|
=== URL Matched: keyword-url ===
|
|
test-vm : summary
|
|
''')
|
|
self.assertEqual(mock_query.mock_calls, [
|
|
mock.call(args, self.app)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_repoquery')
|
|
def test_176_search_wildcard_success(self, mock_query):
|
|
mock_query.return_value = [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'test-vm',
|
|
'2',
|
|
'4.1',
|
|
'2020',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 9, 1, 14, 30,
|
|
tzinfo=datetime.timezone.utc),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-31',
|
|
'Qubes template\n for fedora-31\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'should-not-match-3',
|
|
'0',
|
|
'4.1',
|
|
'2020',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 9, 1, 14, 30,
|
|
tzinfo=datetime.timezone.utc),
|
|
'GPL',
|
|
'https://qubes-os.org/test-vm',
|
|
'Qubes template for fedora-31',
|
|
'test-vm Qubes template\n for fedora-31\n'
|
|
)
|
|
]
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00'
|
|
args = argparse.Namespace(
|
|
all=False,
|
|
templates=['t?st-vm']
|
|
)
|
|
with mock.patch('sys.stdout', new=io.StringIO()) as mock_out:
|
|
qubesadmin.tools.qvm_template.search(args, self.app)
|
|
self.assertEqual(mock_out.getvalue(),
|
|
'''=== Name Matched: t?st-vm ===
|
|
test-vm : Qubes template for fedora-31
|
|
''')
|
|
self.assertEqual(mock_query.mock_calls, [
|
|
mock.call(args, self.app)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
def _mock_qrexec_download(self, args, app, spec, path,
|
|
dlsize=None, refresh=False):
|
|
self.assertFalse(os.path.exists(path),
|
|
'{} should not exist before'.format(path))
|
|
# just create an empty file
|
|
with open(path, 'wb') as f:
|
|
if f is not None:
|
|
f.truncate(dlsize)
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.verify_rpm')
|
|
@mock.patch('qubesadmin.tools.qvm_template.get_dl_list')
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_download')
|
|
def test_180_download_success(self, mock_qrexec, mock_dllist,
|
|
mock_verify_rpm):
|
|
mock_qrexec.side_effect = self._mock_qrexec_download
|
|
with tempfile.TemporaryDirectory() as dir:
|
|
args = argparse.Namespace(
|
|
repo_files=[],
|
|
keyring='/tmp/keyring.gpg',
|
|
releasever='4.1',
|
|
nogpgcheck=False,
|
|
retries=1
|
|
)
|
|
qubesadmin.tools.qvm_template.download(args, self.app, dir, {
|
|
'fedora-31': qubesadmin.tools.qvm_template.DlEntry(
|
|
('1', '2', '3'), 'qubes-templates-itl', 1048576),
|
|
'fedora-32': qubesadmin.tools.qvm_template.DlEntry(
|
|
('0', '1', '2'),
|
|
'qubes-templates-itl-testing',
|
|
2048576)
|
|
})
|
|
self.assertEqual(mock_qrexec.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-fedora-31-1:2-3',
|
|
re_str(dir + '/.*/qubes-template-fedora-31-1:2-3.rpm.UNTRUSTED'),
|
|
1048576),
|
|
mock.call(args, self.app, 'qubes-template-fedora-32-0:1-2',
|
|
re_str(dir + '/.*/qubes-template-fedora-32-0:1-2.rpm.UNTRUSTED'),
|
|
2048576)
|
|
])
|
|
self.assertEqual(mock_dllist.mock_calls, [])
|
|
self.assertEqual(mock_verify_rpm.mock_calls, [
|
|
mock.call(re_str(dir + '/.*/qubes-template-fedora-31-1:2-3.rpm.UNTRUSTED'),
|
|
'/tmp/keyring.gpg', template_name='fedora-31'),
|
|
mock.call(re_str(dir + '/.*/qubes-template-fedora-32-0:1-2.rpm.UNTRUSTED'),
|
|
'/tmp/keyring.gpg', template_name='fedora-32'),
|
|
])
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.verify_rpm')
|
|
@mock.patch('qubesadmin.tools.qvm_template.get_dl_list')
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_download')
|
|
def test_181_download_success_nosuffix(self, mock_qrexec, mock_dllist,
|
|
mock_verify_rpm):
|
|
mock_qrexec.side_effect = self._mock_qrexec_download
|
|
with tempfile.TemporaryDirectory() as dir:
|
|
args = argparse.Namespace(
|
|
retries=1,
|
|
repo_files=[],
|
|
keyring='/tmp/keyring.gpg',
|
|
releasever='4.1',
|
|
nogpgcheck=False,
|
|
downloaddir=dir
|
|
)
|
|
with mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
|
qubesadmin.tools.qvm_template.download(args, self.app, None, {
|
|
'fedora-31': qubesadmin.tools.qvm_template.DlEntry(
|
|
('1', '2', '3'), 'qubes-templates-itl', 1048576)
|
|
})
|
|
self.assertEqual(mock_qrexec.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-fedora-31-1:2-3',
|
|
re_str(dir + '/.*/qubes-template-fedora-31-1:2-3.rpm.UNTRUSTED'),
|
|
1048576)
|
|
])
|
|
self.assertEqual(mock_dllist.mock_calls, [])
|
|
self.assertEqual(mock_verify_rpm.mock_calls, [
|
|
mock.call(re_str(dir + '/.*/qubes-template-fedora-31-1:2-3.rpm.UNTRUSTED'),
|
|
'/tmp/keyring.gpg', template_name='fedora-31'),
|
|
])
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.verify_rpm')
|
|
@mock.patch('qubesadmin.tools.qvm_template.get_dl_list')
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_download')
|
|
def test_182_download_success_getdllist(self, mock_qrexec, mock_dllist,
|
|
mock_verify_rpm):
|
|
mock_qrexec.side_effect = self._mock_qrexec_download
|
|
mock_dllist.return_value = {
|
|
'fedora-31': qubesadmin.tools.qvm_template.DlEntry(
|
|
('1', '2', '3'), 'qubes-templates-itl', 1048576)
|
|
}
|
|
with tempfile.TemporaryDirectory() as dir:
|
|
args = argparse.Namespace(
|
|
retries=1,
|
|
repo_files=[],
|
|
keyring='/tmp/keyring.gpg',
|
|
releasever='4.1',
|
|
nogpgcheck=False,
|
|
)
|
|
qubesadmin.tools.qvm_template.download(args, self.app,
|
|
dir, None,
|
|
qubesadmin.tools.qvm_template.VersionSelector.LATEST_LOWER)
|
|
self.assertEqual(mock_qrexec.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-fedora-31-1:2-3',
|
|
re_str(dir + '/.*/qubes-template-fedora-31-1:2-3.rpm.UNTRUSTED'),
|
|
1048576)
|
|
])
|
|
self.assertEqual(mock_dllist.mock_calls, [
|
|
mock.call(args, self.app,
|
|
version_selector=\
|
|
qubesadmin.tools.qvm_template.\
|
|
VersionSelector.LATEST_LOWER)
|
|
])
|
|
self.assertEqual(mock_verify_rpm.mock_calls, [
|
|
mock.call(re_str(dir + '/.*/qubes-template-fedora-31-1:2-3.rpm.UNTRUSTED'),
|
|
'/tmp/keyring.gpg', template_name='fedora-31'),
|
|
])
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.verify_rpm')
|
|
@mock.patch('qubesadmin.tools.qvm_template.get_dl_list')
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_download')
|
|
def test_183_download_success_downloaddir(self, mock_qrexec, mock_dllist,
|
|
mock_verify_rpm):
|
|
mock_qrexec.side_effect = self._mock_qrexec_download
|
|
with tempfile.TemporaryDirectory() as dir:
|
|
args = argparse.Namespace(
|
|
retries=1,
|
|
repo_files=[],
|
|
keyring='/tmp/keyring.gpg',
|
|
releasever='4.1',
|
|
nogpgcheck=False,
|
|
downloaddir=dir
|
|
)
|
|
qubesadmin.tools.qvm_template.download(args, self.app, None, {
|
|
'fedora-31': qubesadmin.tools.qvm_template.DlEntry(
|
|
('1', '2', '3'), 'qubes-templates-itl', 1048576)
|
|
})
|
|
self.assertEqual(mock_qrexec.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-fedora-31-1:2-3',
|
|
re_str(dir + '/.*/qubes-template-fedora-31-1:2-3.rpm.UNTRUSTED'),
|
|
1048576)
|
|
])
|
|
self.assertEqual(mock_dllist.mock_calls, [])
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.verify_rpm')
|
|
@mock.patch('qubesadmin.tools.qvm_template.get_dl_list')
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_download')
|
|
def test_184_download_success_exists(self, mock_qrexec, mock_dllist,
|
|
mock_verify_rpm):
|
|
mock_qrexec.side_effect = self._mock_qrexec_download
|
|
with tempfile.TemporaryDirectory() as dir:
|
|
with open(os.path.join(
|
|
dir, 'qubes-template-fedora-31-1:2-3.rpm'),
|
|
'w') as _:
|
|
pass
|
|
args = argparse.Namespace(
|
|
retries=1,
|
|
repo_files=[],
|
|
keyring='/tmp/keyring.gpg',
|
|
releasever='4.1',
|
|
nogpgcheck=False,
|
|
downloaddir=dir
|
|
)
|
|
with mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
|
qubesadmin.tools.qvm_template.download(args, self.app, None, {
|
|
'fedora-31': qubesadmin.tools.qvm_template.DlEntry(
|
|
('1', '2', '3'), 'qubes-templates-itl', 1048576),
|
|
'fedora-32': qubesadmin.tools.qvm_template.DlEntry(
|
|
('0', '1', '2'),
|
|
'qubes-templates-itl-testing',
|
|
2048576)
|
|
})
|
|
self.assertTrue('already exists, skipping'
|
|
in mock_err.getvalue())
|
|
self.assertEqual(mock_qrexec.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-fedora-32-0:1-2',
|
|
re_str(dir + '/.*/qubes-template-fedora-32-0:1-2.rpm.UNTRUSTED'),
|
|
2048576)
|
|
])
|
|
self.assertEqual(mock_dllist.mock_calls, [])
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.verify_rpm')
|
|
@mock.patch('qubesadmin.tools.qvm_template.get_dl_list')
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_download')
|
|
def test_185_download_success_existsmove(self, mock_qrexec, mock_dllist,
|
|
mock_verify_rpm):
|
|
mock_qrexec.side_effect = self._mock_qrexec_download
|
|
with tempfile.TemporaryDirectory() as dir:
|
|
with open(os.path.join(
|
|
dir, 'qubes-template-fedora-31-1:2-3.rpm'),
|
|
'w') as _:
|
|
pass
|
|
args = argparse.Namespace(
|
|
retries=1,
|
|
repo_files=[],
|
|
keyring='/tmp/keyring.gpg',
|
|
releasever='4.1',
|
|
nogpgcheck=False,
|
|
downloaddir=dir
|
|
)
|
|
with mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
|
qubesadmin.tools.qvm_template.download(args, self.app, None, {
|
|
'fedora-31': qubesadmin.tools.qvm_template.DlEntry(
|
|
('1', '2', '3'), 'qubes-templates-itl', 1048576)
|
|
})
|
|
self.assertTrue('already exists, skipping'
|
|
in mock_err.getvalue())
|
|
self.assertEqual(mock_qrexec.mock_calls, [])
|
|
self.assertEqual(mock_dllist.mock_calls, [])
|
|
self.assertTrue(os.path.exists(
|
|
dir + '/qubes-template-fedora-31-1:2-3.rpm'))
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.verify_rpm')
|
|
@mock.patch('qubesadmin.tools.qvm_template.get_dl_list')
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_download')
|
|
def test_186_download_success_existsnosuffix(self, mock_qrexec, mock_dllist,
|
|
mock_verify_rpm):
|
|
mock_qrexec.side_effect = self._mock_qrexec_download
|
|
with tempfile.TemporaryDirectory() as dir:
|
|
with open(os.path.join(
|
|
dir, 'qubes-template-fedora-31-1:2-3.rpm'),
|
|
'w') as _:
|
|
pass
|
|
args = argparse.Namespace(
|
|
retries=1,
|
|
repo_files=[],
|
|
keyring='/tmp/keyring.gpg',
|
|
releasever='4.1',
|
|
nogpgcheck=False,
|
|
downloaddir=dir
|
|
)
|
|
with mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
|
qubesadmin.tools.qvm_template.download(args, self.app, None, {
|
|
'fedora-31': qubesadmin.tools.qvm_template.DlEntry(
|
|
('1', '2', '3'), 'qubes-templates-itl', 1048576)
|
|
})
|
|
self.assertTrue('already exists, skipping'
|
|
in mock_err.getvalue())
|
|
self.assertEqual(mock_qrexec.mock_calls, [])
|
|
self.assertEqual(mock_dllist.mock_calls, [])
|
|
self.assertTrue(os.path.exists(
|
|
dir + '/qubes-template-fedora-31-1:2-3.rpm'))
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.verify_rpm')
|
|
@mock.patch('qubesadmin.tools.qvm_template.get_dl_list')
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_download')
|
|
def test_187_download_success_retry(self, mock_qrexec, mock_dllist,
|
|
mock_verify_rpm):
|
|
counter = 0
|
|
def f(*args, **kwargs):
|
|
nonlocal counter
|
|
counter += 1
|
|
if counter == 1:
|
|
raise ConnectionError
|
|
self._mock_qrexec_download(*args, **kwargs)
|
|
mock_qrexec.side_effect = f
|
|
with tempfile.TemporaryDirectory() as dir:
|
|
args = argparse.Namespace(
|
|
retries=2,
|
|
repo_files=[],
|
|
keyring='/tmp/keyring.gpg',
|
|
releasever='4.1',
|
|
nogpgcheck=False,
|
|
downloaddir=dir
|
|
)
|
|
with mock.patch('sys.stderr', new=io.StringIO()) as mock_err, \
|
|
mock.patch('os.remove') as mock_rm:
|
|
qubesadmin.tools.qvm_template.download(args, self.app, None, {
|
|
'fedora-31': qubesadmin.tools.qvm_template.DlEntry(
|
|
('1', '2', '3'), 'qubes-templates-itl', 1048576)
|
|
})
|
|
self.assertTrue('retrying...' in mock_err.getvalue())
|
|
self.assertEqual(mock_rm.mock_calls, [
|
|
mock.call(re_str(dir + '/.*/qubes-template-fedora-31-1:2-3.rpm.UNTRUSTED'))
|
|
])
|
|
self.assertEqual(mock_qrexec.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-fedora-31-1:2-3',
|
|
re_str(dir + '/.*/qubes-template-fedora-31-1:2-3.rpm.UNTRUSTED'),
|
|
1048576),
|
|
mock.call(args, self.app, 'qubes-template-fedora-31-1:2-3',
|
|
re_str(dir + '/.*/qubes-template-fedora-31-1:2-3.rpm.UNTRUSTED'),
|
|
1048576)
|
|
])
|
|
self.assertEqual(mock_dllist.mock_calls, [])
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.verify_rpm')
|
|
@mock.patch('qubesadmin.tools.qvm_template.get_dl_list')
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_download')
|
|
def test_188_download_fail_retry(self, mock_qrexec, mock_dllist,
|
|
mock_verify_rpm):
|
|
mock_qrexec.side_effect = self._mock_qrexec_download
|
|
counter = 0
|
|
def f(*args, **kwargs):
|
|
nonlocal counter
|
|
counter += 1
|
|
if counter <= 3:
|
|
raise ConnectionError
|
|
self._mock_qrexec_download(*args, **kwargs)
|
|
mock_qrexec.side_effect = f
|
|
with tempfile.TemporaryDirectory() as dir:
|
|
args = argparse.Namespace(
|
|
retries=3,
|
|
repo_files=[],
|
|
keyring='/tmp/keyring.gpg',
|
|
releasever='4.1',
|
|
nogpgcheck=False,
|
|
downloaddir=dir
|
|
)
|
|
with mock.patch('sys.stderr', new=io.StringIO()) as mock_err, \
|
|
mock.patch('os.remove') as mock_rm:
|
|
with self.assertRaises(SystemExit):
|
|
qubesadmin.tools.qvm_template.download(
|
|
args, self.app, None, {
|
|
'fedora-31': qubesadmin.tools.qvm_template.DlEntry(
|
|
('1', '2', '3'), 'qubes-templates-itl', 1048576)
|
|
})
|
|
self.assertEqual(mock_err.getvalue().count('retrying...'), 2)
|
|
self.assertTrue('download failed' in mock_err.getvalue())
|
|
self.assertEqual(mock_rm.mock_calls, [
|
|
mock.call(re_str(dir + '/.*/qubes-template-fedora-31-1:2-3.rpm.UNTRUSTED')),
|
|
mock.call(re_str(dir + '/.*/qubes-template-fedora-31-1:2-3.rpm.UNTRUSTED')),
|
|
mock.call(re_str(dir + '/.*/qubes-template-fedora-31-1:2-3.rpm.UNTRUSTED'))
|
|
])
|
|
self.assertEqual(mock_qrexec.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-fedora-31-1:2-3',
|
|
re_str(dir + '/.*/qubes-template-fedora-31-1:2-3.rpm.UNTRUSTED'),
|
|
1048576),
|
|
mock.call(args, self.app, 'qubes-template-fedora-31-1:2-3',
|
|
re_str(dir + '/.*/qubes-template-fedora-31-1:2-3.rpm.UNTRUSTED'),
|
|
1048576),
|
|
mock.call(args, self.app, 'qubes-template-fedora-31-1:2-3',
|
|
re_str(dir + '/.*/qubes-template-fedora-31-1:2-3.rpm.UNTRUSTED'),
|
|
1048576)
|
|
])
|
|
self.assertEqual(mock_dllist.mock_calls, [])
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.verify_rpm')
|
|
@mock.patch('qubesadmin.tools.qvm_template.get_dl_list')
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_download')
|
|
def test_189_download_fail_interrupt(self, mock_qrexec, mock_dllist,
|
|
mock_verify_rpm):
|
|
def f(*args):
|
|
raise RuntimeError
|
|
mock_qrexec.side_effect = f
|
|
with tempfile.TemporaryDirectory() as dir:
|
|
args = argparse.Namespace(
|
|
retries=3,
|
|
repo_files=[],
|
|
keyring='/tmp/keyring.gpg',
|
|
releasever='4.1',
|
|
nogpgcheck=False,
|
|
downloaddir=dir
|
|
)
|
|
with mock.patch('sys.stderr', new=io.StringIO()) as mock_err, \
|
|
mock.patch('os.remove') as mock_rm:
|
|
with self.assertRaises(RuntimeError):
|
|
qubesadmin.tools.qvm_template.download(
|
|
args, self.app, None, {
|
|
'fedora-31': qubesadmin.tools.qvm_template.DlEntry(
|
|
('1', '2', '3'), 'qubes-templates-itl', 1048576)
|
|
})
|
|
self.assertEqual(mock_qrexec.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-fedora-31-1:2-3',
|
|
re_str(dir + '/.*/qubes-template-fedora-31-1:2-3.rpm'),
|
|
1048576)
|
|
])
|
|
self.assertEqual(mock_dllist.mock_calls, [])
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.verify_rpm')
|
|
@mock.patch('qubesadmin.tools.qvm_template.get_dl_list')
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_download')
|
|
def test_190_download_fail_verify(self, mock_qrexec, mock_dllist,
|
|
mock_verify_rpm):
|
|
mock_qrexec.side_effect = self._mock_qrexec_download
|
|
mock_verify_rpm.side_effect = \
|
|
qubesadmin.tools.qvm_template.SignatureVerificationError
|
|
|
|
with tempfile.TemporaryDirectory() as dir:
|
|
args = argparse.Namespace(
|
|
retries=3,
|
|
repo_files=[],
|
|
keyring='/tmp/keyring.gpg',
|
|
releasever='4.1',
|
|
nogpgcheck=True, # make sure it gets ignored
|
|
downloaddir=dir
|
|
)
|
|
with self.assertRaises(qubesadmin.tools.qvm_template.SignatureVerificationError):
|
|
qubesadmin.tools.qvm_template.download(
|
|
args, self.app, None, {
|
|
'fedora-31': qubesadmin.tools.qvm_template.DlEntry(
|
|
('1', '2', '3'), 'qubes-templates-itl', 1048576)
|
|
})
|
|
self.assertEqual(mock_qrexec.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-fedora-31-1:2-3',
|
|
re_str(dir + '/.*/qubes-template-fedora-31-1:2-3.rpm'),
|
|
1048576)
|
|
])
|
|
self.assertEqual(mock_dllist.mock_calls, [])
|
|
self.assertEqual(os.listdir(dir), [])
|
|
self.assertEqual(mock_verify_rpm.mock_calls, [
|
|
mock.call(re_str(dir + '/.*/qubes-template-fedora-31-1:2-3.rpm.UNTRUSTED'),
|
|
'/tmp/keyring.gpg', template_name='fedora-31'),
|
|
])
|
|
|
|
def _mock_qrexec_download_short(self, args, app, spec, path,
|
|
dlsize=None, refresh=False):
|
|
self.assertFalse(os.path.exists(path),
|
|
'{} should not exist before'.format(path))
|
|
# just create an empty file
|
|
with open(path, 'wb') as f:
|
|
if f is not None:
|
|
f.truncate(dlsize // 2)
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_template.verify_rpm')
|
|
@mock.patch('qubesadmin.tools.qvm_template.get_dl_list')
|
|
@mock.patch('qubesadmin.tools.qvm_template.qrexec_download')
|
|
def test_191_download_fail_short(self, mock_qrexec, mock_dllist,
|
|
mock_verify_rpm):
|
|
mock_qrexec.side_effect = self._mock_qrexec_download_short
|
|
with tempfile.TemporaryDirectory() as dir, \
|
|
self.assertRaises(SystemExit):
|
|
args = argparse.Namespace(
|
|
repo_files=[],
|
|
keyring='/tmp/keyring.gpg',
|
|
releasever='4.1',
|
|
nogpgcheck=False,
|
|
retries=1
|
|
)
|
|
qubesadmin.tools.qvm_template.download(args, self.app, dir, {
|
|
'fedora-31': qubesadmin.tools.qvm_template.DlEntry(
|
|
('1', '2', '3'), 'qubes-templates-itl', 1048576),
|
|
})
|
|
self.assertEqual(mock_qrexec.mock_calls, [
|
|
mock.call(args, self.app, 'qubes-template-fedora-31-1:2-3',
|
|
re_str(dir + '/.*/qubes-template-fedora-31-1:2-3.rpm.UNTRUSTED'),
|
|
1048576),
|
|
])
|
|
self.assertEqual(mock_dllist.mock_calls, [])
|
|
self.assertEqual(mock_verify_rpm.mock_calls, [])
|
|
|
|
|
|
@mock.patch('os.remove')
|
|
@mock.patch('os.rename')
|
|
@mock.patch('os.makedirs')
|
|
@mock.patch('subprocess.check_call')
|
|
@mock.patch('qubesadmin.tools.qvm_template.confirm_action')
|
|
@mock.patch('qubesadmin.tools.qvm_template.extract_rpm')
|
|
@mock.patch('qubesadmin.tools.qvm_template.download')
|
|
@mock.patch('qubesadmin.tools.qvm_template.get_dl_list')
|
|
@mock.patch('qubesadmin.tools.qvm_template.verify_rpm')
|
|
def test_200_reinstall_success(
|
|
self,
|
|
mock_verify,
|
|
mock_dl_list,
|
|
mock_dl,
|
|
mock_extract,
|
|
mock_confirm,
|
|
mock_call,
|
|
mock_mkdirs,
|
|
mock_rename,
|
|
mock_remove):
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
build_time = '2020-09-01 14:30:00' # 1598970600
|
|
install_time = '2020-09-01 15:30:00'
|
|
for key, val in [
|
|
('name', 'test-vm'),
|
|
('epoch', '2'),
|
|
('version', '4.1'),
|
|
('release', '2020')]:
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-{key}',
|
|
None)] = b'0\0' + val.encode()
|
|
for key, val in [
|
|
('name', 'test-vm'),
|
|
('epoch', '2'),
|
|
('version', '4.1'),
|
|
('release', '2020'),
|
|
('reponame', 'qubes-templates-itl'),
|
|
('buildtime', build_time),
|
|
('installtime', install_time),
|
|
('license', 'GPL'),
|
|
('url', 'https://qubes-os.org'),
|
|
('summary', 'Summary'),
|
|
('description', 'Desc|desc')]:
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Set',
|
|
f'template-{key}',
|
|
val.encode())] = b'0\0'
|
|
rpm_hdr = {
|
|
rpm.RPMTAG_NAME : 'qubes-template-test-vm',
|
|
rpm.RPMTAG_BUILDTIME : 1598970600,
|
|
rpm.RPMTAG_DESCRIPTION : 'Desc\ndesc',
|
|
rpm.RPMTAG_EPOCHNUM : 2,
|
|
rpm.RPMTAG_LICENSE : 'GPL',
|
|
rpm.RPMTAG_RELEASE : '2020',
|
|
rpm.RPMTAG_SUMMARY : 'Summary',
|
|
rpm.RPMTAG_URL : 'https://qubes-os.org',
|
|
rpm.RPMTAG_VERSION : '4.1'
|
|
}
|
|
mock_dl.return_value = {'test-vm': rpm_hdr}
|
|
dl_list = {
|
|
'test-vm': qubesadmin.tools.qvm_template.DlEntry(
|
|
('2', '4.1', '2020'), 'qubes-templates-itl', 1048576)
|
|
}
|
|
mock_dl_list.return_value = dl_list
|
|
mock_call.side_effect = self.add_new_vm_side_effect
|
|
mock_time = mock.Mock(wraps=datetime.datetime)
|
|
mock_time.now.return_value = \
|
|
datetime.datetime(2020, 9, 1, 15, 30, tzinfo=datetime.timezone.utc)
|
|
selector = qubesadmin.tools.qvm_template.VersionSelector.REINSTALL
|
|
with mock.patch('qubesadmin.tools.qvm_template.LOCK_FILE', '/tmp/test.lock'), \
|
|
mock.patch('datetime.datetime', new=mock_time), \
|
|
mock.patch('tempfile.TemporaryDirectory') as mock_tmpdir:
|
|
args = argparse.Namespace(
|
|
templates=['test-vm'],
|
|
keyring='/tmp/keyring.gpg',
|
|
nogpgcheck=False,
|
|
cachedir='/var/cache/qvm-template',
|
|
repo_files=[],
|
|
releasever='4.1',
|
|
yes=False,
|
|
keep_cache=True,
|
|
allow_pv=False,
|
|
pool=None
|
|
)
|
|
mock_tmpdir.return_value.__enter__.return_value = \
|
|
'/var/tmp/qvm-template-tmpdir'
|
|
qubesadmin.tools.qvm_template.install(args, self.app,
|
|
version_selector=selector,
|
|
override_existing=True)
|
|
# Attempt to get download list
|
|
self.assertEqual(mock_dl_list.mock_calls, [
|
|
mock.call(args, self.app, version_selector=selector)
|
|
])
|
|
mock_dl.assert_called_with(args, self.app,
|
|
dl_list=dl_list, version_selector=selector)
|
|
# already verified by download()
|
|
self.assertEqual(mock_verify.mock_calls, [])
|
|
# Package is extracted
|
|
mock_extract.assert_called_with('test-vm',
|
|
'/var/cache/qvm-template/qubes-template-test-vm-2:4.1-2020.rpm',
|
|
'/var/tmp/qvm-template-tmpdir')
|
|
# Expect override confirmation
|
|
self.assertEqual(mock_confirm.mock_calls,
|
|
[mock.call(re_str(r'.*override changes.*:'), ['test-vm'])])
|
|
# qvm-template-postprocess is called
|
|
self.assertEqual(mock_call.mock_calls, [
|
|
mock.call([
|
|
'qvm-template-postprocess',
|
|
'--really',
|
|
'--no-installed-by-rpm',
|
|
'post-install',
|
|
'test-vm',
|
|
'/var/tmp/qvm-template-tmpdir'
|
|
'/var/lib/qubes/vm-templates/test-vm'
|
|
])
|
|
])
|
|
# Cache directory created
|
|
self.assertEqual(mock_mkdirs.mock_calls, [
|
|
mock.call(args.cachedir, exist_ok=True)
|
|
])
|
|
# Downloaded package should not be removed
|
|
self.assertEqual(mock_remove.mock_calls, [
|
|
mock.call('/tmp/test.lock')
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('os.rename')
|
|
@mock.patch('os.makedirs')
|
|
@mock.patch('subprocess.check_call')
|
|
@mock.patch('qubesadmin.tools.qvm_template.confirm_action')
|
|
@mock.patch('qubesadmin.tools.qvm_template.extract_rpm')
|
|
@mock.patch('qubesadmin.tools.qvm_template.download')
|
|
@mock.patch('qubesadmin.tools.qvm_template.get_dl_list')
|
|
@mock.patch('qubesadmin.tools.qvm_template.verify_rpm')
|
|
def test_201_reinstall_fail_noversion(
|
|
self,
|
|
mock_verify,
|
|
mock_dl_list,
|
|
mock_dl,
|
|
mock_extract,
|
|
mock_confirm,
|
|
mock_call,
|
|
mock_mkdirs,
|
|
mock_rename):
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
for key, val in [
|
|
('name', 'test-vm'),
|
|
('epoch', '2'),
|
|
('version', '4.1'),
|
|
('release', '2021')]:
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-{key}',
|
|
None)] = b'0\0' + val.encode()
|
|
rpm_hdr = {
|
|
rpm.RPMTAG_NAME : 'qubes-template-test-vm',
|
|
rpm.RPMTAG_BUILDTIME : 1598970600,
|
|
rpm.RPMTAG_DESCRIPTION : 'Desc\ndesc',
|
|
rpm.RPMTAG_EPOCHNUM : 2,
|
|
rpm.RPMTAG_LICENSE : 'GPL',
|
|
rpm.RPMTAG_RELEASE : '2020',
|
|
rpm.RPMTAG_SUMMARY : 'Summary',
|
|
rpm.RPMTAG_URL : 'https://qubes-os.org',
|
|
rpm.RPMTAG_VERSION : '4.1'
|
|
}
|
|
mock_dl.return_value = {'test-vm': rpm_hdr}
|
|
dl_list = {
|
|
'test-vm': qubesadmin.tools.qvm_template.DlEntry(
|
|
('1', '4.1', '2020'), 'qubes-templates-itl', 1048576)
|
|
}
|
|
mock_dl_list.return_value = dl_list
|
|
selector = qubesadmin.tools.qvm_template.VersionSelector.REINSTALL
|
|
with mock.patch('qubesadmin.tools.qvm_template.LOCK_FILE', '/tmp/test.lock'), \
|
|
self.assertRaises(SystemExit) as e, \
|
|
mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
|
args = argparse.Namespace(
|
|
templates=['test-vm'],
|
|
keyring='/tmp/keyring.gpg',
|
|
nogpgcheck=False,
|
|
cachedir='/var/cache/qvm-template',
|
|
repo_files=[],
|
|
releasever='4.1',
|
|
yes=False,
|
|
keep_cache=True,
|
|
allow_pv=False,
|
|
pool=None
|
|
)
|
|
qubesadmin.tools.qvm_template.install(args, self.app,
|
|
version_selector=selector,
|
|
override_existing=True)
|
|
self.assertIn(
|
|
'Same version of template \'test-vm\' not found',
|
|
mock_err.getvalue())
|
|
# Attempt to get download list
|
|
self.assertEqual(mock_dl_list.mock_calls, [
|
|
mock.call(args, self.app, version_selector=selector)
|
|
])
|
|
mock_dl.assert_called_with(args, self.app,
|
|
dl_list=dl_list, version_selector=selector)
|
|
# already verified by download()
|
|
self.assertEqual(mock_verify.mock_calls, [])
|
|
# Expect override confirmation
|
|
self.assertEqual(mock_confirm.mock_calls,
|
|
[mock.call(re_str(r'.*override changes.*:'), ['test-vm'])])
|
|
# Nothing extracted / installed
|
|
mock_extract.assert_not_called()
|
|
mock_call.assert_not_called()
|
|
# Cache directory created
|
|
self.assertEqual(mock_mkdirs.mock_calls, [
|
|
mock.call(args.cachedir, exist_ok=True)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('os.rename')
|
|
@mock.patch('os.makedirs')
|
|
@mock.patch('subprocess.check_call')
|
|
@mock.patch('qubesadmin.tools.qvm_template.confirm_action')
|
|
@mock.patch('qubesadmin.tools.qvm_template.extract_rpm')
|
|
@mock.patch('qubesadmin.tools.qvm_template.download')
|
|
@mock.patch('qubesadmin.tools.qvm_template.get_dl_list')
|
|
@mock.patch('qubesadmin.tools.qvm_template.verify_rpm')
|
|
def test_202_reinstall_local_success(
|
|
self,
|
|
mock_verify,
|
|
mock_dl_list,
|
|
mock_dl,
|
|
mock_extract,
|
|
mock_confirm,
|
|
mock_call,
|
|
mock_mkdirs,
|
|
mock_rename):
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
build_time = '2020-09-01 14:30:00' # 1598970600
|
|
install_time = '2020-09-01 15:30:00'
|
|
for key, val in [
|
|
('name', 'test-vm'),
|
|
('epoch', '2'),
|
|
('version', '4.1'),
|
|
('release', '2020')]:
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-{key}',
|
|
None)] = b'0\0' + val.encode()
|
|
for key, val in [
|
|
('name', 'test-vm'),
|
|
('epoch', '2'),
|
|
('version', '4.1'),
|
|
('release', '2020'),
|
|
('reponame', '@commandline'),
|
|
('buildtime', build_time),
|
|
('installtime', install_time),
|
|
('license', 'GPL'),
|
|
('url', 'https://qubes-os.org'),
|
|
('summary', 'Summary'),
|
|
('description', 'Desc|desc')]:
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Set',
|
|
f'template-{key}',
|
|
val.encode())] = b'0\0'
|
|
rpm_hdr = {
|
|
rpm.RPMTAG_NAME : 'qubes-template-test-vm',
|
|
rpm.RPMTAG_BUILDTIME : 1598970600,
|
|
rpm.RPMTAG_DESCRIPTION : 'Desc\ndesc',
|
|
rpm.RPMTAG_EPOCHNUM : 2,
|
|
rpm.RPMTAG_LICENSE : 'GPL',
|
|
rpm.RPMTAG_RELEASE : '2020',
|
|
rpm.RPMTAG_SUMMARY : 'Summary',
|
|
rpm.RPMTAG_URL : 'https://qubes-os.org',
|
|
rpm.RPMTAG_VERSION : '4.1'
|
|
}
|
|
mock_verify.return_value = rpm_hdr
|
|
dl_list = {}
|
|
mock_dl_list.return_value = dl_list
|
|
mock_call.side_effect = self.add_new_vm_side_effect
|
|
mock_time = mock.Mock(wraps=datetime.datetime)
|
|
mock_time.now.return_value = \
|
|
datetime.datetime(2020, 9, 1, 15, 30, tzinfo=datetime.timezone.utc)
|
|
selector = qubesadmin.tools.qvm_template.VersionSelector.REINSTALL
|
|
with mock.patch('qubesadmin.tools.qvm_template.LOCK_FILE', '/tmp/test.lock'), \
|
|
mock.patch('datetime.datetime', new=mock_time), \
|
|
mock.patch('tempfile.TemporaryDirectory') as mock_tmpdir, \
|
|
tempfile.NamedTemporaryFile(suffix='.rpm') as template_file:
|
|
path = template_file.name
|
|
args = argparse.Namespace(
|
|
templates=[path],
|
|
keyring='/tmp/keyring.gpg',
|
|
nogpgcheck=False,
|
|
cachedir='/var/cache/qvm-template',
|
|
repo_files=[],
|
|
releasever='4.1',
|
|
yes=False,
|
|
allow_pv=False,
|
|
pool=None
|
|
)
|
|
mock_tmpdir.return_value.__enter__.return_value = \
|
|
'/var/tmp/qvm-template-tmpdir'
|
|
qubesadmin.tools.qvm_template.install(args, self.app,
|
|
version_selector=selector,
|
|
override_existing=True)
|
|
# Package is extracted
|
|
mock_extract.assert_called_with(
|
|
'test-vm',
|
|
path,
|
|
'/var/tmp/qvm-template-tmpdir')
|
|
# Package verified
|
|
self.assertEqual(mock_verify.mock_calls, [
|
|
mock.call(path, '/tmp/keyring.gpg', nogpgcheck=False)
|
|
])
|
|
# Attempt to get download list
|
|
self.assertEqual(mock_dl_list.mock_calls, [
|
|
mock.call(args, self.app, version_selector=selector)
|
|
])
|
|
mock_dl.assert_called_with(args, self.app,
|
|
dl_list=dl_list, version_selector=selector)
|
|
# Expect override confirmation
|
|
self.assertEqual(mock_confirm.mock_calls,
|
|
[mock.call(re_str(r'.*override changes.*:'), ['test-vm'])])
|
|
# qvm-template-postprocess is called
|
|
self.assertEqual(mock_call.mock_calls, [
|
|
mock.call([
|
|
'qvm-template-postprocess',
|
|
'--really',
|
|
'--no-installed-by-rpm',
|
|
'post-install',
|
|
'test-vm',
|
|
'/var/tmp/qvm-template-tmpdir'
|
|
'/var/lib/qubes/vm-templates/test-vm'
|
|
])
|
|
])
|
|
# Cache directory created
|
|
self.assertEqual(mock_mkdirs.mock_calls, [
|
|
mock.call(args.cachedir, exist_ok=True)
|
|
])
|
|
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()
|
|
|
|
@mock.patch('qubesadmin.tools.qvm_kill.main')
|
|
@mock.patch('qubesadmin.tools.qvm_remove.main')
|
|
@mock.patch('qubesadmin.tools.qvm_template.confirm_action')
|
|
def test_214_remove_disassoc_success_newdummy(
|
|
self,
|
|
mock_confirm,
|
|
mock_remove,
|
|
mock_kill):
|
|
def append_new_vm_side_effect(*args, **kwargs):
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] += \
|
|
b'dummy-1 class=TemplateVM state=Halted\n'
|
|
self.app.domains.clear_cache()
|
|
return self.app.domains['dummy-1']
|
|
self.app.add_new_vm = mock.Mock(side_effect=append_new_vm_side_effect)
|
|
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'dummy 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.Set',
|
|
'template-dummy', b'1')] = \
|
|
b'0\x00'
|
|
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'
|
|
args = argparse.Namespace(
|
|
templates=['vm1'],
|
|
yes=False
|
|
)
|
|
def deps(app, vm):
|
|
if vm == 'vm1':
|
|
return [(self.app.domains['vm2'], 'default_template')]
|
|
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'])
|
|
])
|
|
self.assertEqual(mock_confirm.mock_calls, [
|
|
mock.call(re_str(r'.*completely remove.*'), ['vm1'])
|
|
])
|
|
self.assertEqual(mock_remove.mock_calls, [
|
|
mock.call(['--force', '--', 'vm1'], self.app)
|
|
])
|
|
self.assertEqual(mock_kill.mock_calls, [
|
|
mock.call(['--', 'vm1'], self.app)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
def test_220_get_keys_for_repos_success(self):
|
|
with tempfile.NamedTemporaryFile() as f:
|
|
f.write(
|
|
b'''[qubes-templates-itl]
|
|
name = Qubes Templates repository
|
|
#baseurl = https://yum.qubes-os.org/r$releasever/templates-itl
|
|
#baseurl = http://yum.qubesosfasa4zl44o4tws22di6kepyzfeqv3tg4e3ztknltfxqrymdad.onion/r$releasever/templates-itl
|
|
metalink = https://yum.qubes-os.org/r$releasever/templates-itl/repodata/repomd.xml.metalink
|
|
enabled = 1
|
|
fastestmirror = 1
|
|
metadata_expire = 7d
|
|
gpgcheck = 1
|
|
gpgkey = file:///etc/qubes/repo-templates/keys/RPM-GPG-KEY-qubes-$releasever-primary
|
|
[qubes-templates-itl-testing-nokey]
|
|
name = Qubes Templates repository
|
|
#baseurl = https://yum.qubes-os.org/r$releasever/templates-itl-testing
|
|
#baseurl = http://yum.qubesosfasa4zl44o4tws22di6kepyzfeqv3tg4e3ztknltfxqrymdad.onion/r$releasever/templates-itl-testing
|
|
metalink = https://yum.qubes-os.org/r$releasever/templates-itl-testing/repodata/repomd.xml.metalink
|
|
enabled = 0
|
|
fastestmirror = 1
|
|
gpgcheck = 1
|
|
#gpgkey = file:///etc/qubes/repo-templates/keys/RPM-GPG-KEY-qubes-$releasever-primary
|
|
[qubes-templates-itl-testing]
|
|
name = Qubes Templates repository
|
|
#baseurl = https://yum.qubes-os.org/r$releasever/templates-itl-testing
|
|
#baseurl = http://yum.qubesosfasa4zl44o4tws22di6kepyzfeqv3tg4e3ztknltfxqrymdad.onion/r$releasever/templates-itl-testing
|
|
metalink = https://yum.qubes-os.org/r$releasever/templates-itl-testing/repodata/repomd.xml.metalink
|
|
enabled = 0
|
|
fastestmirror = 1
|
|
gpgcheck = 1
|
|
gpgkey = file:///etc/qubes/repo-templates/keys/RPM-GPG-KEY-qubes-$releasever-primary-testing
|
|
''')
|
|
f.flush()
|
|
ret = qubesadmin.tools.qvm_template.get_keys_for_repos(
|
|
[f.name], 'r4.1')
|
|
self.assertEqual(ret, {
|
|
'qubes-templates-itl':
|
|
'/etc/qubes/repo-templates/keys/RPM-GPG-KEY-qubes-r4.1-primary',
|
|
'qubes-templates-itl-testing':
|
|
'/etc/qubes/repo-templates/keys/RPM-GPG-KEY-qubes-r4.1-primary-testing'
|
|
})
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('os.rename')
|
|
@mock.patch('os.makedirs')
|
|
@mock.patch('subprocess.check_call')
|
|
@mock.patch('qubesadmin.tools.qvm_template.confirm_action')
|
|
@mock.patch('qubesadmin.tools.qvm_template.extract_rpm')
|
|
@mock.patch('qubesadmin.tools.qvm_template.download')
|
|
@mock.patch('qubesadmin.tools.qvm_template.get_dl_list')
|
|
@mock.patch('qubesadmin.tools.qvm_template.verify_rpm')
|
|
def test_220_downgrade_skip_lower(
|
|
self,
|
|
mock_verify,
|
|
mock_dl_list,
|
|
mock_dl,
|
|
mock_extract,
|
|
mock_confirm,
|
|
mock_call,
|
|
mock_mkdirs,
|
|
mock_rename):
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
for key, val in [
|
|
('name', 'test-vm'),
|
|
('epoch', '2'),
|
|
('version', '4.1'),
|
|
('release', '2020')]:
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-{key}',
|
|
None)] = b'0\0' + val.encode()
|
|
rpm_hdr = {
|
|
rpm.RPMTAG_NAME : 'qubes-template-test-vm',
|
|
rpm.RPMTAG_BUILDTIME : 1598970600,
|
|
rpm.RPMTAG_DESCRIPTION : 'Desc\ndesc',
|
|
rpm.RPMTAG_EPOCHNUM : 2,
|
|
rpm.RPMTAG_LICENSE : 'GPL',
|
|
rpm.RPMTAG_RELEASE : '2021',
|
|
rpm.RPMTAG_SUMMARY : 'Summary',
|
|
rpm.RPMTAG_URL : 'https://qubes-os.org',
|
|
rpm.RPMTAG_VERSION : '4.1'
|
|
}
|
|
mock_verify.return_value = rpm_hdr
|
|
mock_dl_list.return_value = {}
|
|
selector = qubesadmin.tools.qvm_template.VersionSelector.LATEST_LOWER
|
|
with mock.patch('qubesadmin.tools.qvm_template.LOCK_FILE', '/tmp/test.lock'), \
|
|
tempfile.NamedTemporaryFile(suffix='.rpm') as template_file, \
|
|
mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
|
args = argparse.Namespace(
|
|
templates=[template_file.name],
|
|
keyring='/tmp/keyring.gpg',
|
|
nogpgcheck=False,
|
|
cachedir='/var/cache/qvm-template',
|
|
repo_files=[],
|
|
releasever='4.1',
|
|
yes=False,
|
|
keep_cache=True,
|
|
allow_pv=False,
|
|
pool=None
|
|
)
|
|
qubesadmin.tools.qvm_template.install(args, self.app,
|
|
version_selector=selector,
|
|
override_existing=True)
|
|
self.assertIn(
|
|
'lower version already installed',
|
|
mock_err.getvalue())
|
|
# Attempt to get download list
|
|
self.assertEqual(mock_dl_list.mock_calls, [
|
|
mock.call(args, self.app, version_selector=selector)
|
|
])
|
|
mock_dl.assert_called_with(args, self.app,
|
|
dl_list={}, version_selector=selector)
|
|
self.assertEqual(mock_verify.mock_calls, [
|
|
mock.call(template_file.name, '/tmp/keyring.gpg', nogpgcheck=False)
|
|
])
|
|
# No confirmation since nothing needs to be done
|
|
mock_confirm.assert_not_called()
|
|
# Nothing extracted / installed
|
|
mock_extract.assert_not_called()
|
|
mock_call.assert_not_called()
|
|
# Cache directory created
|
|
self.assertEqual(mock_mkdirs.mock_calls, [
|
|
mock.call(args.cachedir, exist_ok=True)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('os.rename')
|
|
@mock.patch('os.makedirs')
|
|
@mock.patch('subprocess.check_call')
|
|
@mock.patch('qubesadmin.tools.qvm_template.confirm_action')
|
|
@mock.patch('qubesadmin.tools.qvm_template.extract_rpm')
|
|
@mock.patch('qubesadmin.tools.qvm_template.download')
|
|
@mock.patch('qubesadmin.tools.qvm_template.get_dl_list')
|
|
@mock.patch('qubesadmin.tools.qvm_template.verify_rpm')
|
|
def test_221_upgrade_skip_higher(
|
|
self,
|
|
mock_verify,
|
|
mock_dl_list,
|
|
mock_dl,
|
|
mock_extract,
|
|
mock_confirm,
|
|
mock_call,
|
|
mock_mkdirs,
|
|
mock_rename):
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
for key, val in [
|
|
('name', 'test-vm'),
|
|
('epoch', '2'),
|
|
('version', '4.1'),
|
|
('release', '2021')]:
|
|
self.app.expected_calls[(
|
|
'test-vm',
|
|
'admin.vm.feature.Get',
|
|
f'template-{key}',
|
|
None)] = b'0\0' + val.encode()
|
|
rpm_hdr = {
|
|
rpm.RPMTAG_NAME : 'qubes-template-test-vm',
|
|
rpm.RPMTAG_BUILDTIME : 1598970600,
|
|
rpm.RPMTAG_DESCRIPTION : 'Desc\ndesc',
|
|
rpm.RPMTAG_EPOCHNUM : 2,
|
|
rpm.RPMTAG_LICENSE : 'GPL',
|
|
rpm.RPMTAG_RELEASE : '2020',
|
|
rpm.RPMTAG_SUMMARY : 'Summary',
|
|
rpm.RPMTAG_URL : 'https://qubes-os.org',
|
|
rpm.RPMTAG_VERSION : '4.1'
|
|
}
|
|
mock_verify.return_value = rpm_hdr
|
|
mock_dl_list.return_value = {}
|
|
selector = qubesadmin.tools.qvm_template.VersionSelector.LATEST_HIGHER
|
|
with mock.patch('qubesadmin.tools.qvm_template.LOCK_FILE', '/tmp/test.lock'), \
|
|
tempfile.NamedTemporaryFile(suffix='.rpm') as template_file, \
|
|
mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
|
args = argparse.Namespace(
|
|
templates=[template_file.name],
|
|
keyring='/tmp/keyring.gpg',
|
|
nogpgcheck=False,
|
|
cachedir='/var/cache/qvm-template',
|
|
repo_files=[],
|
|
releasever='4.1',
|
|
yes=False,
|
|
keep_cache=True,
|
|
allow_pv=False,
|
|
pool=None
|
|
)
|
|
qubesadmin.tools.qvm_template.install(args, self.app,
|
|
version_selector=selector,
|
|
override_existing=True)
|
|
self.assertIn(
|
|
'higher version already installed',
|
|
mock_err.getvalue())
|
|
# Attempt to get download list
|
|
self.assertEqual(mock_dl_list.mock_calls, [
|
|
mock.call(args, self.app, version_selector=selector)
|
|
])
|
|
mock_dl.assert_called_with(args, self.app,
|
|
dl_list={}, version_selector=selector)
|
|
self.assertEqual(mock_verify.mock_calls, [
|
|
mock.call(template_file.name, '/tmp/keyring.gpg', nogpgcheck=False)
|
|
])
|
|
# No confirmation since nothing needs to be done
|
|
mock_confirm.assert_not_called()
|
|
# Nothing extracted / installed
|
|
mock_extract.assert_not_called()
|
|
mock_call.assert_not_called()
|
|
# Cache directory created
|
|
self.assertEqual(mock_mkdirs.mock_calls, [
|
|
mock.call(args.cachedir, exist_ok=True)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
def test_230_filter_version_latest(self):
|
|
query_res = [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-31',
|
|
'0',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-31',
|
|
'Qubes template\n for fedora-31\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'0',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-32',
|
|
'Qubes template\n for fedora-32\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'0',
|
|
'4.1',
|
|
'20200102',
|
|
'qubes-templates-itl-testing',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-32',
|
|
'Qubes template\n for fedora-32\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'0',
|
|
'4.1',
|
|
'20200102',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-32',
|
|
'Qubes template\n for fedora-32\n'
|
|
)
|
|
]
|
|
results = qubesadmin.tools.qvm_template.filter_version(
|
|
query_res,
|
|
self.app
|
|
)
|
|
self.assertEqual(sorted(results), [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-31',
|
|
'0',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-31',
|
|
'Qubes template\n for fedora-31\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'0',
|
|
'4.1',
|
|
'20200102',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-32',
|
|
'Qubes template\n for fedora-32\n'
|
|
)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
def test_231_filter_version_reinstall(self):
|
|
query_res = [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-31',
|
|
'0',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-31',
|
|
'Qubes template\n for fedora-31\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'0',
|
|
'4.1',
|
|
'20200102',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-32',
|
|
'Qubes template\n for fedora-32\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'0',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-32',
|
|
'Qubes template\n for fedora-32\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'0',
|
|
'4.1',
|
|
'20200102',
|
|
'qubes-templates-itl-testing',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-32',
|
|
'Qubes template\n for fedora-32\n'
|
|
)
|
|
]
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00fedora-31 class=TemplateVM state=Halted\n' \
|
|
b'fedora-32 class=TemplateVM state=Halted\n'
|
|
for key, val in [
|
|
('name', 'fedora-31'),
|
|
('epoch', '0'),
|
|
('version', '4.1'),
|
|
('release', '20200101')]:
|
|
self.app.expected_calls[(
|
|
'fedora-31',
|
|
'admin.vm.feature.Get',
|
|
f'template-{key}',
|
|
None)] = b'0\0' + val.encode()
|
|
for key, val in [
|
|
('name', 'fedora-32'),
|
|
('epoch', '0'),
|
|
('version', '4.1'),
|
|
('release', '20200101')]:
|
|
self.app.expected_calls[(
|
|
'fedora-32',
|
|
'admin.vm.feature.Get',
|
|
f'template-{key}',
|
|
None)] = b'0\0' + val.encode()
|
|
results = qubesadmin.tools.qvm_template.filter_version(
|
|
query_res,
|
|
self.app,
|
|
qubesadmin.tools.qvm_template.VersionSelector.REINSTALL
|
|
)
|
|
self.assertEqual(sorted(results), [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-31',
|
|
'0',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-31',
|
|
'Qubes template\n for fedora-31\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'0',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-32',
|
|
'Qubes template\n for fedora-32\n'
|
|
)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
def test_232_filter_version_upgrade(self):
|
|
query_res = [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-31',
|
|
'0',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-31',
|
|
'Qubes template\n for fedora-31\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'0',
|
|
'4.1',
|
|
'20200102',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-32',
|
|
'Qubes template\n for fedora-32\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'0',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-32',
|
|
'Qubes template\n for fedora-32\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'0',
|
|
'4.1',
|
|
'20200102',
|
|
'qubes-templates-itl-testing',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-32',
|
|
'Qubes template\n for fedora-32\n'
|
|
)
|
|
]
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00fedora-31 class=TemplateVM state=Halted\n' \
|
|
b'fedora-32 class=TemplateVM state=Halted\n'
|
|
for key, val in [
|
|
('name', 'fedora-31'),
|
|
('epoch', '0'),
|
|
('version', '4.1'),
|
|
('release', '20200101')]:
|
|
self.app.expected_calls[(
|
|
'fedora-31',
|
|
'admin.vm.feature.Get',
|
|
f'template-{key}',
|
|
None)] = b'0\0' + val.encode()
|
|
for key, val in [
|
|
('name', 'fedora-32'),
|
|
('epoch', '0'),
|
|
('version', '4.1'),
|
|
('release', '20200101')]:
|
|
self.app.expected_calls[(
|
|
'fedora-32',
|
|
'admin.vm.feature.Get',
|
|
f'template-{key}',
|
|
None)] = b'0\0' + val.encode()
|
|
results = qubesadmin.tools.qvm_template.filter_version(
|
|
query_res,
|
|
self.app,
|
|
qubesadmin.tools.qvm_template.VersionSelector.LATEST_HIGHER
|
|
)
|
|
self.assertEqual(sorted(results), [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'0',
|
|
'4.1',
|
|
'20200102',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-32',
|
|
'Qubes template\n for fedora-32\n'
|
|
)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
def test_233_filter_version_downgrade(self):
|
|
query_res = [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-31',
|
|
'0',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-31',
|
|
'Qubes template\n for fedora-31\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'0',
|
|
'4.1',
|
|
'20200102',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-32',
|
|
'Qubes template\n for fedora-32\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'0',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-32',
|
|
'Qubes template\n for fedora-32\n'
|
|
),
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'0',
|
|
'4.1',
|
|
'20200102',
|
|
'qubes-templates-itl-testing',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-32',
|
|
'Qubes template\n for fedora-32\n'
|
|
)
|
|
]
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00fedora-31 class=TemplateVM state=Halted\n' \
|
|
b'fedora-32 class=TemplateVM state=Halted\n'
|
|
for key, val in [
|
|
('name', 'fedora-31'),
|
|
('epoch', '0'),
|
|
('version', '4.1'),
|
|
('release', '20200101')]:
|
|
self.app.expected_calls[(
|
|
'fedora-31',
|
|
'admin.vm.feature.Get',
|
|
f'template-{key}',
|
|
None)] = b'0\0' + val.encode()
|
|
for key, val in [
|
|
('name', 'fedora-32'),
|
|
('epoch', '0'),
|
|
('version', '4.1'),
|
|
('release', '20200102')]:
|
|
self.app.expected_calls[(
|
|
'fedora-32',
|
|
'admin.vm.feature.Get',
|
|
f'template-{key}',
|
|
None)] = b'0\0' + val.encode()
|
|
results = qubesadmin.tools.qvm_template.filter_version(
|
|
query_res,
|
|
self.app,
|
|
qubesadmin.tools.qvm_template.VersionSelector.LATEST_LOWER
|
|
)
|
|
self.assertEqual(sorted(results), [
|
|
qubesadmin.tools.qvm_template.Template(
|
|
'fedora-32',
|
|
'0',
|
|
'4.1',
|
|
'20200101',
|
|
'qubes-templates-itl',
|
|
1048576,
|
|
datetime.datetime(2020, 1, 23, 4, 56),
|
|
'GPL',
|
|
'https://qubes-os.org',
|
|
'Qubes template for fedora-32',
|
|
'Qubes template\n for fedora-32\n'
|
|
)
|
|
])
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('os.path.exists')
|
|
def test_240_qubes_release(self, mock_exists):
|
|
# /usr/share/qubes/marker-vm does not exist
|
|
mock_exists.return_value = False
|
|
marker_vm = '''
|
|
NAME=Qubes
|
|
VERSION="4.2 (R4.2)"
|
|
ID=qubes
|
|
# Some comments here
|
|
VERSION_ID=4.2
|
|
PRETTY_NAME="Qubes 4.2 (R4.2)"
|
|
ANSI_COLOR="0;31"
|
|
CPE_NAME="cpe:/o:ITL:qubes:4.2"
|
|
'''
|
|
with mock.patch('builtins.open', mock.mock_open(read_data=marker_vm)) \
|
|
as mock_open:
|
|
ret = qubesadmin.tools.qvm_template.qubes_release()
|
|
self.assertEqual(ret, '4.2')
|
|
self.assertEqual(mock_exists.mock_calls, [
|
|
mock.call('/usr/share/qubes/marker-vm')
|
|
])
|
|
mock_open.assert_called_with('/etc/os-release', 'r')
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('os.path.exists')
|
|
def test_241_qubes_release_quotes(self, mock_exists):
|
|
# /usr/share/qubes/marker-vm does not exist
|
|
mock_exists.return_value = False
|
|
os_rel = '''
|
|
NAME=Qubes
|
|
VERSION="4.2 (R4.2)"
|
|
ID=qubes
|
|
# Some comments here
|
|
VERSION_ID="4.2"
|
|
PRETTY_NAME="Qubes 4.2 (R4.2)"
|
|
ANSI_COLOR="0;31"
|
|
CPE_NAME="cpe:/o:ITL:qubes:4.2"
|
|
'''
|
|
with mock.patch('builtins.open', mock.mock_open(read_data=os_rel)) \
|
|
as mock_open:
|
|
ret = qubesadmin.tools.qvm_template.qubes_release()
|
|
self.assertEqual(ret, '4.2')
|
|
self.assertEqual(mock_exists.mock_calls, [
|
|
mock.call('/usr/share/qubes/marker-vm')
|
|
])
|
|
mock_open.assert_called_with('/etc/os-release', 'r')
|
|
self.assertAllCalled()
|
|
|
|
@mock.patch('os.path.exists')
|
|
def test_242_qubes_release_quotes(self, mock_exists):
|
|
# /usr/share/qubes/marker-vm does exist
|
|
mock_exists.return_value = True
|
|
marker_vm = '''
|
|
# This is just a marker file for Qubes OS VM.
|
|
# This VM have tools for Qubes version:
|
|
4.2
|
|
'''
|
|
with mock.patch('builtins.open', mock.mock_open(read_data=marker_vm)) \
|
|
as mock_open:
|
|
ret = qubesadmin.tools.qvm_template.qubes_release()
|
|
self.assertEqual(ret, '4.2')
|
|
self.assertEqual(mock_exists.mock_calls, [
|
|
mock.call('/usr/share/qubes/marker-vm')
|
|
])
|
|
mock_open.assert_called_with('/usr/share/qubes/marker-vm', 'r')
|
|
self.assertAllCalled()
|
|
|
|
def test_250_qrexec_download_success(self):
|
|
rand_bytes = os.urandom(128)
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
self.app.expected_service_calls[
|
|
('test-vm', 'qubes.TemplateDownload')] = rand_bytes
|
|
args = argparse.Namespace(
|
|
repo_files=[],
|
|
releasever='4.1',
|
|
updatevm='test-vm',
|
|
enablerepo=[],
|
|
disablerepo=[],
|
|
repoid=[],
|
|
quiet=True
|
|
)
|
|
with tempfile.NamedTemporaryFile() as fd:
|
|
qubesadmin.tools.qvm_template.qrexec_download(
|
|
args, self.app, 'fedora-31:4.0', path=fd.name)
|
|
with open(fd.name, 'rb') as fd2:
|
|
result = fd2.read()
|
|
self.assertEqual(rand_bytes, result)
|
|
self.assertAllCalled()
|
|
|
|
def test_251_qrexec_download_fail(self):
|
|
rand_bytes = os.urandom(128)
|
|
self.app.expected_calls[('dom0', 'admin.vm.List', None, None)] = \
|
|
b'0\x00test-vm class=TemplateVM state=Halted\n'
|
|
self.app.expected_service_calls[
|
|
('test-vm', 'qubes.TemplateDownload')] = rand_bytes
|
|
args = argparse.Namespace(
|
|
repo_files=[],
|
|
releasever='4.1',
|
|
updatevm='test-vm',
|
|
enablerepo=[],
|
|
disablerepo=[],
|
|
repoid=[],
|
|
quiet=True
|
|
)
|
|
with tempfile.NamedTemporaryFile() as fd, \
|
|
mock.patch('qubesadmin.tests.TestProcess.wait') as mock_wait:
|
|
mock_wait.return_value = 1
|
|
with self.assertRaises(ConnectionError):
|
|
qubesadmin.tools.qvm_template.qrexec_download(
|
|
args, self.app, 'fedora-31:4.0', path=fd.name)
|
|
self.assertAllCalled()
|