2f54745715
It was needed to provide assertItemsEqual method. Starting with Python 3.2, there's assertCountEqual that does the same thing. Six provides a helper that will dispatch to the existing method. With this change, unittest 2 is only needed on Python 2.6 to backport the method. Signed-off-by: Lubomír Sedlář <lsedlar@redhat.com>
69 lines
2.5 KiB
Python
69 lines
2.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
try:
|
|
import unittest2 as unittest
|
|
except ImportError:
|
|
import unittest
|
|
|
|
import os
|
|
import sys
|
|
|
|
import six
|
|
|
|
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')
|
|
six.assertCountEqual(
|
|
self,
|
|
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'],
|
|
log_dir='/tmp')
|
|
|
|
self.assertEqual(cmd[0], 'lorax')
|
|
six.assertCountEqual(
|
|
self,
|
|
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",
|
|
"--logfile=/tmp/lorax.log",
|
|
"/mnt/output_dir"],
|
|
)
|