blob: bdcbfdea38838f066f557b52c774cd559f8f6992 [file] [log] [blame]
Ryan Cui47f80e42013-04-01 19:01:54 -07001#!/usr/bin/python
2
3# Copyright (c) 2013 The Chromium OS Authors. All rights reserved.
4# Use of this source code is governed by a BSD-style license that can be
5# found in the LICENSE file.
6
7import os
8import sys
9
10sys.path.insert(0, os.path.abspath('%s/../..' % os.path.dirname(__file__)))
11from chromite.lib import cros_test_lib
12from chromite.lib import stats
13from chromite.lib import stats_unittest
14from chromite.scripts import cros
15
16
17# pylint: disable=W0212
18
19
20class RunScriptTest(cros_test_lib.MockTempDirTestCase):
21 """Test the main functionality."""
22
23 def setUp(self):
Ryan Cuicbd9bb62013-04-30 11:17:02 -070024 self.stats_module_mock = stats_unittest.StatsModuleMock()
25 self.StartPatcher(self.stats_module_mock)
Ryan Cui47f80e42013-04-01 19:01:54 -070026 self.PatchObject(cros, '_RunSubCommand', autospec=True)
27
Ryan Cuicbd9bb62013-04-30 11:17:02 -070028 def testStatsUpload(self, upload_count=1, return_value=0):
Ryan Cui47f80e42013-04-01 19:01:54 -070029 """Test stats uploading."""
30 return_value = cros.main(['chrome-sdk', '--board', 'lumpy'])
Ryan Cuicbd9bb62013-04-30 11:17:02 -070031 self.assertEquals(stats.StatsUploader._Upload.call_count, upload_count)
32 self.assertEquals(return_value, return_value)
33
34 def testStatsUploadError(self):
35 """We don't upload stats if the stats creation failed."""
36 self.stats_module_mock.stats_mock.init_exception = True
37 with cros_test_lib.LoggingCapturer():
38 self.testStatsUpload(upload_count=0)
Ryan Cui47f80e42013-04-01 19:01:54 -070039
40
41if __name__ == '__main__':
42 cros_test_lib.main()