scripts/cros_unittest.py

# -*- coding: utf-8 -*-
# Copyright (c) 2013 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Unittests for cros."""

from __future__ import print_function

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):
    self.PatchObject(cros, '_RunSubCommand', autospec=True)

  def testDefaultLogLevel(self):
    """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',
                                  add_help=False)

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