scripts/cros_unittest.py

#!/usr/bin/python

# 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.

import os
import sys

sys.path.insert(0, os.path.abspath('%s/../..' % os.path.dirname(__file__)))
from chromite.lib import cros_test_lib
from chromite.lib import stats
from chromite.lib import stats_unittest
from chromite.scripts import cros


# pylint: disable=W0212


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

  def setUp(self):
    self.StartPatcher(stats_unittest.StatsModuleMock())
    self.PatchObject(cros, '_RunSubCommand', autospec=True)

  def testStatsUpload(self):
    """Test stats uploading."""
    return_value = cros.main(['chrome-sdk', '--board', 'lumpy'])
    self.assertEquals(stats.StatsUploader._Upload.call_count, 1)
    self.assertEquals(return_value, 0)


if __name__ == '__main__':
  cros_test_lib.main()