| |
@@ -363,5 +363,70 @@
|
| |
'--rootfs-size=None',
|
| |
self.output])])
|
| |
|
| |
+
|
| |
+ class OstreeLiveImageScriptTest(helpers.PungiTestCase):
|
| |
+ def setUp(self):
|
| |
+ super(OstreeLiveImageScriptTest, self).setUp()
|
| |
+ self.ks = os.path.join(self.topdir, 'fedora-atomic-pxe-live.ks')
|
| |
+ self.iso = os.path.join(self.topdir, 'install.iso')
|
| |
+ self.output_dir = os.path.join(self.topdir, 'output')
|
| |
+ self.logdir = os.path.join(self.topdir, 'logs')
|
| |
+
|
| |
+ @mock.patch('kobo.shortcuts.run')
|
| |
+ def test_run_virt_install(self, run):
|
| |
+ args = ['liveimage',
|
| |
+ '--ks=%s' % self.ks,
|
| |
+ '--iso=%s' % self.iso,
|
| |
+ '--output=%s' % self.output_dir,
|
| |
+ '--logdir=%s' % self.logdir]
|
| |
+
|
| |
+ ostree.main(args)
|
| |
+
|
| |
+ self.maxDiff = None
|
| |
+ self.assertItemsEqual(run.mock_calls,
|
| |
+ [mock.call(['livemedia-creator',
|
| |
+ '--make-ostree-live',
|
| |
+ '--ks', self.ks,
|
| |
+ '--iso', self.iso,
|
| |
+ '--resultdir', self.output_dir,
|
| |
+ '--logfile', os.path.join(self.logdir, 'lmc-make-ostree-live.log')],
|
| |
+ logfile='%s/pungi-make-ostree-live-image.log' % self.logdir,
|
| |
+ show_cmd=True,
|
| |
+ stdout=True)])
|
| |
+
|
| |
+ @mock.patch('kobo.shortcuts.run')
|
| |
+ def test_run_virt_install_without_iso(self, run):
|
| |
+ args = ['liveimage',
|
| |
+ '--ks=%s' % self.ks,
|
| |
+ '--output=%s' % self.output_dir,
|
| |
+ '--logdir=%s' % self.logdir]
|
| |
+
|
| |
+ with self.assertRaises(RuntimeError) as ctx:
|
| |
+ ostree.main(args)
|
| |
+
|
| |
+ self.assertIn('iso is required for virt install', str(ctx.exception))
|
| |
+
|
| |
+ @mock.patch('kobo.shortcuts.run')
|
| |
+ def test_run_no_virt_install(self, run):
|
| |
+ args = ['liveimage',
|
| |
+ '--ks=%s' % self.ks,
|
| |
+ '--no-virt',
|
| |
+ '--output=%s' % self.output_dir,
|
| |
+ '--logdir=%s' % self.logdir]
|
| |
+
|
| |
+ ostree.main(args)
|
| |
+
|
| |
+ self.maxDiff = None
|
| |
+ self.assertItemsEqual(run.mock_calls,
|
| |
+ [mock.call(['livemedia-creator',
|
| |
+ '--make-ostree-live',
|
| |
+ '--ks', self.ks,
|
| |
+ '--no-virt',
|
| |
+ '--resultdir', self.output_dir,
|
| |
+ '--logfile', os.path.join(self.logdir, 'lmc-make-ostree-live.log')],
|
| |
+ logfile='%s/pungi-make-ostree-live-image.log' % self.logdir,
|
| |
+ show_cmd=True,
|
| |
+ stdout=True)])
|
| |
+
|
| |
if __name__ == '__main__':
|
| |
unittest.main()
|
| |
liveimage is added to support creating OSTree live images
Signed-off-by: Qixiang Wan qwan@redhat.com