#! /usr/bin/python3

# SPDX-License-Identifier: GPL-2.0-or-later
#
# avocado/build.py - Tests for various make targets
#
# Copyright Red Hat
# Author: David Gibson <david@gibson.dropbear.id.au>

import os
import os.path
import shutil
import sys

import avocado

import common


class BuildTest(common.BaseTest):
    # Some of these can take a little while
    timeout = 30.0

    def setUp(self):
        super().setUp()

        # Make a temporary copy of the sources
        srcfiles = self.hostx('git ls-files').splitlines()
        for src in srcfiles:
            dst = os.path.join(bytes(self.workdir, 'utf-8'), src)
            os.makedirs(os.path.dirname(dst), exist_ok=True)
            shutil.copy(src, dst)
        os.chdir(self.workdir)

    def build_target(self, target, outputs=None):
        if outputs is None:
            outputs = [target]
        for o in outputs:
            self.assertFalse(os.path.exists(o))
        self.hostx('make {} CFLAGS="-Werror"'.format(target))
        for o in outputs:
            self.assertTrue(os.path.isfile(o))
        self.hostx('make clean')
        for o in outputs:
            self.assertFalse(os.path.exists(o))

    def test_make_passt(self):
        self.build_target('passt')

    def test_make_pasta(self):
        self.build_target('pasta')

    def test_make_qrap(self):
        self.build_target('qrap')

    def test_make_all(self):
        self.build_target('all', ['passt', 'pasta', 'qrap'])

    def test_make_install_uninstall(self):
        prefix = os.path.join(self.workdir, "prefix")
        os.makedirs(prefix)

        # Install
        self.hostx('make install CFLAGS="-Werror" prefix={}'.format(prefix))
        self.assertTrue(os.path.isfile(os.path.join(prefix, 'bin/passt')))
        self.assertTrue(os.path.isfile(os.path.join(prefix, 'bin/pasta')))
        self.assertTrue(os.path.isfile(os.path.join(prefix, 'bin/qrap')))
        mandir = os.path.join(prefix, 'share/man')
        self.hostx('man -M {} -W passt'.format(mandir))
        self.hostx('man -M {} -W pasta'.format(mandir))
        self.hostx('man -M {} -W qrap'.format(mandir))

        # Uninstall
        self.hostx('make uninstall prefix={}'.format(prefix))
        self.assertFalse(os.path.exists(os.path.join(prefix, 'bin/passt')))
        self.assertFalse(os.path.exists(os.path.join(prefix, 'bin/pasta')))
        self.assertFalse(os.path.exists(os.path.join(prefix, 'bin/qrap')))
        self.assertRaises(avocado.utils.process.CmdError,
                          self.hostx, 'man -M {} -W passt'.format(mandir))
        self.assertRaises(avocado.utils.process.CmdError,
                          self.hostx, 'man -M {} -W pasta'.format(mandir))
        self.assertRaises(avocado.utils.process.CmdError,
                          self.hostx, 'man -M {} -W qrap'.format(mandir))