scripts/strip_package_unittest.py

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

"""Unittests for strip_package.py"""

import os

from chromite.lib import build_target_lib
from chromite.lib import cros_test_lib
from chromite.lib import install_mask
from chromite.scripts import strip_package


class StripPackageTest(cros_test_lib.MockTestCase):
    """Tests for strip_package."""

    def setUp(self) -> None:
        self.sysroot_path = "/build/testboard"
        self.builder_mock = self.PatchObject(
            strip_package.builder, "UpdateGmergeBinhost"
        )
        self.PatchObject(
            build_target_lib,
            "get_default_sysroot_path",
            return_value=self.sysroot_path,
        )

    def testDefaultSysroot(self) -> None:
        """Test the base case."""
        strip_package.main(["--board=testboard", "foo"])
        self.builder_mock.assert_called_with(self.sysroot_path, ["foo"], False)

    def testMultiplePkg(self) -> None:
        """Test multiple package input."""
        strip_package.main(["--board=testboard", "foo", "foo1"])
        self.builder_mock.assert_called_with(
            self.sysroot_path, ["foo", "foo1"], False
        )

    def testCustomSysroot(self) -> None:
        """Test user given custom sysroot path."""
        strip_package.main(["--sysroot=/build/sysroot", "foo"])
        self.builder_mock.assert_called_with("/build/sysroot", ["foo"], False)

    def testInstallMask(self) -> None:
        """Test install mask environment variable."""
        strip_package.main(["--board=testboard", "foo"])
        self.assertEqual(
            os.environ.get("DEFAULT_INSTALL_MASK"),
            "\n".join(install_mask.DEFAULT),
        )

    def testDeepOption(self) -> None:
        """Test Deep option."""
        strip_package.main(["--board=testboard", "--deep", "foo"])
        self.builder_mock.assert_called_with(self.sysroot_path, ["foo"], True)