qvm-template: Add tests for qrexec_payload
This commit is contained in:
WillyPillow
parent
89895038b5
commit
3f75e6e49e
@ -769,3 +769,293 @@ class TC_00_qvm_template(qubesadmin.tests.QubesTestCase):
|
|||||||
mock.call('/var/tmp/qvm-template.lck')
|
mock.call('/var/tmp/qvm-template.lck')
|
||||||
])
|
])
|
||||||
self.assertAllCalled()
|
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:///usr/share/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:///usr/share/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:///usr/share/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:///usr/share/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 self.assertRaises(SystemExit), \
|
||||||
|
mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
||||||
|
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:///usr/share/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 self.assertRaises(SystemExit), \
|
||||||
|
mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
||||||
|
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:///usr/share/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 self.assertRaises(SystemExit), \
|
||||||
|
mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
||||||
|
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:///usr/share/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 self.assertRaises(SystemExit), \
|
||||||
|
mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
||||||
|
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:///usr/share/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 self.assertRaises(SystemExit), \
|
||||||
|
mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
||||||
|
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:///usr/share/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 self.assertRaises(SystemExit), \
|
||||||
|
mock.patch('sys.stderr', new=io.StringIO()) as mock_err:
|
||||||
|
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()
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user