src/third_party/chromite/scripts/cros_unittest.py

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