import os
import infra.basetest
import subprocess

CHECK_FS_CMD = "mount | grep 'rootfs on / type rootfs'"


def boot_img(emulator, builddir):
    img = os.path.join(builddir, "images", "rootfs.cpio")
    emulator.boot(arch="armv7",
                  kernel="builtin",
                  options=["-initrd", "{}".format(img)])
    emulator.login()
    _, exit_code = emulator.run(CHECK_FS_CMD)
    return exit_code


class TestCpioFull(infra.basetest.BRTest):
    config = infra.basetest.BASIC_TOOLCHAIN_CONFIG + \
        """
        BR2_INIT_BUSYBOX=y
        BR2_TARGET_ROOTFS_CPIO=y
        # BR2_TARGET_ROOTFS_TAR is not set
        """

    def test_run(self):

        exit_code = boot_img(self.emulator,
                             self.builddir)
        self.assertEqual(exit_code, 0)


class TestCpioDracutBase(infra.basetest.BRTest):
    config = \
        """
        BR2_arm=y
        BR2_TOOLCHAIN_EXTERNAL=y
        BR2_TOOLCHAIN_EXTERNAL_BOOTLIN=y
        BR2_INIT_BUSYBOX=y
        BR2_PACKAGE_PV=y
        BR2_TARGET_ROOTFS_CPIO=y
        BR2_TARGET_ROOTFS_CPIO_DRACUT=y
        # BR2_TARGET_ROOTFS_TAR is not set
        """

    def check_dracut(self):
        out = subprocess.check_output(["cpio", "--list"],
                                      stdin=open(os.path.join(self.builddir, "images/rootfs.cpio")),
                                      stderr=open(os.devnull, "w"),
                                      cwd=self.builddir,
                                      env={"LANG": "C"},
                                      universal_newlines=True)
        # pv should *not* be included in cpio image
        self.assertEqual(out.find("bin/pv"), -1)

        exit_code = boot_img(self.emulator,
                             self.builddir)
        self.assertEqual(exit_code, 0)


class TestCpioDracutUclibc(TestCpioDracutBase):
    config = TestCpioDracutBase.config + \
        """
        BR2_TOOLCHAIN_EXTERNAL_BOOTLIN_ARMV5_EABI_UCLIBC_STABLE=y
        """

    def test_run(self):
        self.check_dracut()


class TestCpioDracutGlibc(TestCpioDracutBase):
    config = TestCpioDracutBase.config + \
        """
        BR2_TOOLCHAIN_EXTERNAL_BOOTLIN_ARMV5_EABI_GLIBC_STABLE=y
        """

    def test_run(self):
        self.check_dracut()


class TestCpioDracutMusl(TestCpioDracutBase):
    config = TestCpioDracutBase.config + \
        """
        BR2_TOOLCHAIN_EXTERNAL_BOOTLIN_ARMV5_EABI_MUSL_STABLE=y
        """

    def test_run(self):
        self.check_dracut()


class TestCpioDracutUclibcMergedUsr(TestCpioDracutBase):
    config = TestCpioDracutBase.config + \
        """
        BR2_ROOTFS_MERGED_USR=y
        BR2_TOOLCHAIN_EXTERNAL_BOOTLIN_ARMV5_EABI_UCLIBC_STABLE=y
        """

    def test_run(self):
        self.check_dracut()


class TestCpioDracutGlibcMergedUsr(TestCpioDracutBase):
    config = TestCpioDracutBase.config + \
        """
        BR2_ROOTFS_MERGED_USR=y
        BR2_TOOLCHAIN_EXTERNAL_BOOTLIN_ARMV5_EABI_GLIBC_STABLE=y
        """

    def test_run(self):
        self.check_dracut()


class TestCpioDracutMuslMergedUsr(TestCpioDracutBase):
    config = TestCpioDracutBase.config + \
        """
        BR2_ROOTFS_MERGED_USR=y
        BR2_TOOLCHAIN_EXTERNAL_BOOTLIN_ARMV5_EABI_MUSL_STABLE=y
        """

    def test_run(self):
        self.check_dracut()