# -*- coding: utf-8 -*-

try:
    import unittest2 as unittest
except ImportError:
    import unittest

import os
import sys

import six

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",
                                       dracut_args=["--foo", "bar"])

        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",
             "--dracut-arg=--foo",
             "--dracut-arg=bar",
             "/mnt/output_dir"],
        )