d5512fd6ff
The module backports features to Python 2.6 and 2.7. If it is available, the tests will use it. If it is not available, it will fall back to regular unittest. On Python 2.7, the tests pass anyway. On Python 2.6, there are failures with Python 2.6. Signed-off-by: Lubomír Sedlář <lsedlar@redhat.com>
64 lines
2.7 KiB
Python
Executable File
64 lines
2.7 KiB
Python
Executable File
#!/usr/bin/env python2
|
|
# -*- coding: utf-8 -*-
|
|
|
|
try:
|
|
import unittest2 as unittest
|
|
except ImportError:
|
|
import unittest
|
|
|
|
import os
|
|
import sys
|
|
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
|
|
|
|
from pungi.wrappers.lorax import LoraxWrapper
|
|
|
|
|
|
class LoraxWrapperTest(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.lorax = LoraxWrapper()
|
|
|
|
def test_get_command_with_minimal_arguments(self):
|
|
cmd = self.lorax.get_lorax_cmd("product", "version", "release",
|
|
"/mnt/repo_baseurl", "/mnt/output_dir")
|
|
|
|
self.assertEqual(cmd[0], 'lorax')
|
|
self.assertItemsEqual(cmd[1:],
|
|
['--product=product',
|
|
'--version=version',
|
|
'--release=release',
|
|
'--source=file:///mnt/repo_baseurl',
|
|
'/mnt/output_dir'])
|
|
|
|
def test_get_command_with_all_arguments(self):
|
|
cmd = self.lorax.get_lorax_cmd("product", "version", "release",
|
|
"/mnt/repo_baseurl", "/mnt/output_dir",
|
|
variant="Server", bugurl="http://example.com/",
|
|
nomacboot=True, noupgrade=True, is_final=True,
|
|
buildarch='x86_64', volid='VOLUME_ID',
|
|
buildinstallpackages=['bash', 'vim'],
|
|
add_template=['t1', 't2'],
|
|
add_arch_template=['ta1', 'ta2'],
|
|
add_template_var=['v1', 'v2'],
|
|
add_arch_template_var=['va1', 'va2'])
|
|
|
|
self.assertEqual(cmd[0], 'lorax')
|
|
self.assertItemsEqual(cmd[1:],
|
|
['--product=product', '--version=version',
|
|
'--release=release', '--variant=Server',
|
|
'--source=file:///mnt/repo_baseurl',
|
|
'--bugurl=http://example.com/',
|
|
'--buildarch=x86_64', '--volid=VOLUME_ID',
|
|
'--nomacboot', '--noupgrade', '--isfinal',
|
|
'--installpkgs=bash', '--installpkgs=vim',
|
|
'--add-template=t1', '--add-template=t2',
|
|
'--add-arch-template=ta1', '--add-arch-template=ta2',
|
|
'--add-template-var=v1', '--add-template-var=v2',
|
|
'--add-arch-template-var=va1', '--add-arch-template-var=va2',
|
|
'/mnt/output_dir'])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|