Restart codereview issue 6792042

buildbot/cbuildbot_commands_unittest.py

 #!/usr/bin/python
 
 # Copyright (c) 2011 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 commands.  Needs to be run inside of chroot for mox."""
 
 import __builtin__
 import mox
 import os
 import shutil
 import sys
 import unittest
 
 import constants
 sys.path.append(constants.SOURCE_ROOT)
 import chromite.buildbot.cbuildbot_commands as commands
 import chromite.lib.cros_build_lib as cros_lib
 
 class CBuildBotTest(mox.MoxTestBase):
 
   def setUp(self):
     mox.MoxTestBase.setUp(self)
     # Always stub RunCommmand out as we use it in every method.
     self.mox.StubOutWithMock(cros_lib, 'OldRunCommand')
+    self.mox.StubOutWithMock(cros_lib, 'RunCommand')
     self.tracking_branch = 'cros/master'
     self._test_repos = [['kernel', 'third_party/kernel/files'],
                         ['login_manager', 'platform/login_manager']
                        ]
     self._test_cros_workon_packages = (
         'chromeos-base/kernel\nchromeos-base/chromeos-login\n')
     self._test_board = 'test-board'
     self._buildroot = '.'
     self._test_dict = {'kernel': ['chromos-base/kernel', 'dev-util/perf'],
                        'cros': ['chromos-base/libcros']
@@ skipping 113 matching lines @@
     the second time."""
     cros_lib.OldRunCommand(mox.In('sync'),
                            cwd=self._buildroot).AndRaise(Exception("failed"))
     cros_lib.OldRunCommand(mox.In('sync'), cwd=self._buildroot)
     cros_lib.OldRunCommand(mox.In('forall'), cwd=self._buildroot)
     cros_lib.OldRunCommand(mox.In('manifest'), cwd=self._buildroot)
     self.mox.ReplayAll()
     commands._RepoSync(self._buildroot, retries=2)
     self.mox.VerifyAll()
 
+  def testBuildIncremental(self):
+    """Base case where Build is called."""
+    buildroot = '/bob/'
+    cros_lib.RunCommand(mox.IgnoreArg(), cwd=mox.StrContains(buildroot),
+        enter_chroot=True, extra_env=mox.IgnoreArg())
+    self.mox.ReplayAll()
+    commands.Build(buildroot, False)
+    self.mox.VerifyAll()
+
+  def testBuildClobber(self):
+    """Base case where Build with emptytree is called."""
+    buildroot = '/bob/'
+    cros_lib.RunCommand(mox.IgnoreArg(), cwd=mox.StrContains(buildroot),
+        enter_chroot=True,
+        extra_env=mox.ContainsKeyValue('EXTRA_BOARD_FLAGS','--emptytree'))
+    self.mox.ReplayAll()
+    commands.Build(buildroot, True)
+    self.mox.VerifyAll()
+
+  def testBuildWithEnv(self):
+    """Case where Build is called with a custom environment."""
+    buildroot = '/bob/'
+    extra = {'A' :'Av', 'B' : 'Bv'}
+    cros_lib.RunCommand( mox.IgnoreArg(), cwd=mox.StrContains(buildroot),
+        enter_chroot=True,extra_env=mox.And(
+        mox.ContainsKeyValue('A','Av'), mox.ContainsKeyValue('B','Bv')))
+    self.mox.ReplayAll()
+    commands.Build(buildroot, True, extra_env=extra)
+    self.mox.VerifyAll()
+
+
 
 if __name__ == '__main__':
   unittest.main()