# Copyright 2013 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 cros."""

from chromite.lib import commandline
from chromite.lib import cros_test_lib
from chromite.scripts import cros


class RunScriptTest(cros_test_lib.MockTempDirTestCase):
    """Test the main functionality."""

    def setUp(self) -> None:
        self.PatchObject(cros, "_RunSubCommand", autospec=True)

    def testDefaultLogLevel(self) -> None:
        """Test that the default log level is set to notice."""
        arg_parser = self.PatchObject(
            commandline,
            "ArgumentParser",
            return_value=commandline.ArgumentParser(),
        )
        cros.GetOptions()
        arg_parser.assert_called_with(caching=True, default_log_level="notice")

    def testSubcommand(self) -> None:
        """Test parser when given a subcommand."""
        parser = cros.GetOptions("lint")
        opts = parser.parse_args(["lint"])
        self.assertEqual(opts.subcommand, "lint")