# Copyright 2018 The ChromiumOS Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Test install_toolchain."""

import os

from chromite.lib import cros_test_lib
from chromite.lib import osutils
from chromite.lib import sysroot_lib
from chromite.lib import unittest_lib
from chromite.scripts import install_toolchain


class ParseArgsTest(cros_test_lib.TempDirTestCase):
    """Tests for argument parsing and validation rules."""

    # pylint: disable=protected-access
    def setUp(self) -> None:
        D = cros_test_lib.Directory
        filesystem = (
            D("build", (D("board", (D("etc", ("make.conf.board_setup",)),)),)),
        )
        cros_test_lib.CreateOnDiskHierarchy(self.tempdir, filesystem)

        self.chost_value = "chost"
        osutils.WriteFile(
            os.path.join(self.tempdir, "build/board/etc/make.conf.board_setup"),
            'CHOST="%s"' % self.chost_value,
        )

        self.sysroot_dir = os.path.join(self.tempdir, "build/board")
        # make.conf needs to exist to correctly read back config.
        unittest_lib.create_stub_make_conf(self.sysroot_dir)

    def testInvalidArgs(self) -> None:
        """Test invalid argument parsing."""
        with self.assertRaises(SystemExit):
            install_toolchain._ParseArgs([])

    def testSysrootCreate(self) -> None:
        """Test the sysroot is correctly built."""
        opts = install_toolchain._ParseArgs(["--sysroot", self.sysroot_dir])
        self.assertIsInstance(opts.sysroot, sysroot_lib.Sysroot)
        self.assertEqual(self.sysroot_dir, opts.sysroot.path)

    def testToolchain(self) -> None:
        """Test toolchain arg precedence/fallback."""
        toolchain = "toolchain"
        opts = install_toolchain._ParseArgs(
            ["--sysroot", self.sysroot_dir, "--toolchain", toolchain]
        )
        self.assertEqual(toolchain, opts.toolchain)

        opts = install_toolchain._ParseArgs(["--sysroot", self.sysroot_dir])
        self.assertEqual(self.chost_value, opts.toolchain)