blob: 7510fcad0c52257b1b4f2c2f9408f0bbaa61b4fd (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
# 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.lib import stats
from chromite.lib import stats_unittest
from chromite.scripts import cros
class RunScriptTest(cros_test_lib.WorkspaceTestCase):
"""Test the main functionality."""
def setUp(self):
self.stats_module_mock = stats_unittest.StatsModuleMock()
self.StartPatcher(self.stats_module_mock)
self.PatchObject(cros, '_RunSubCommand', autospec=True)
def testStatsUpload(self, upload_count=1, return_value=0):
"""Test stats uploading."""
return_value = cros.main(['chrome-sdk', '--board', 'lumpy'])
# pylint: disable=protected-access
self.assertEquals(stats.StatsUploader._Upload.call_count, upload_count)
# pylint: enable=protected-access
self.assertEquals(return_value, return_value)
def testStatsUploadError(self):
"""We don't upload stats if the stats creation failed."""
self.stats_module_mock.stats_mock.init_exception = True
with cros_test_lib.LoggingCapturer():
self.testStatsUpload(upload_count=0)
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')
|
