Add better support for getting the path to crosutils.

lib/cros_build_lib_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.
 
 """Unit tests for cros_build_lib."""
 
+import mox
 import os
 import tempfile
 import unittest
 
 import cros_build_lib
 
-class CrosBuildLibTest(unittest.TestCase):
+class CrosBuildLibTest(mox.MoxTestBase):
   """Test class for cros_build_lib."""
 
   def testRunCommandSimple(self):
     """Test that RunCommand can run a simple successful command."""
     result = cros_build_lib.RunCommand(['ls'],
                                        # Keep the test quiet options
                                        print_cmd=False,
                                        redirect_stdout=True,
                                        redirect_stderr=True,
                                        # Test specific options
@@ skipping 70 matching lines @@
                                # Keep the test quiet options
                                print_cmd=False,
                                # Test specific options
                                log_to_file=log_file)
     log_fh = open(log_file)
     log_data = log_fh.read()
     self.assertEquals('Hi', log_data)
     log_fh.close()
     os.remove(log_file)
 
+  def testGetCrosUtilsPathInChroot(self):
+    """Tests whether we can get crosutils from chroot."""
+    self.mox.StubOutWithMock(cros_build_lib, 'IsInsideChroot')
+    crosutils_path_src = '/home/' + os.getenv('USER') + 'trunk/src/scripts'
+    crosutils_path_installed = '/usr/lib/crosutils'
+
+    cros_build_lib.IsInsideChroot().MultipleTimes().AndReturn(True)
+
+    self.mox.ReplayAll()
+    self.assertTrue(cros_build_lib.GetCrosUtilsPath(source_dir_path=True),
+                    crosutils_path_src)
+    self.assertTrue(cros_build_lib.GetCrosUtilsPath(source_dir_path=False),
+                    crosutils_path_installed)
+    self.mox.VerifyAll()
+
+  def testGetCrosUtilsPathOutsideChroot(self):
+    """Tests whether we can get crosutils from outside chroot."""
+    self.mox.StubOutWithMock(cros_build_lib, 'IsInsideChroot')
+    path = os.path.join(os.path.dirname(os.path.realpath(__file__)), '..')
+    cros_build_lib.IsInsideChroot().MultipleTimes().AndReturn(False)
+
+    self.mox.ReplayAll()
+    self.assertTrue(cros_build_lib.GetCrosUtilsPath(), path)
+    self.mox.VerifyAll()
+
+  def testGetCrosUtilsBinPath(self):
+    """Tests whether we can get crosutilsbin correctly."""
+    self.mox.StubOutWithMock(cros_build_lib, 'IsInsideChroot')
+    self.mox.StubOutWithMock(cros_build_lib, 'GetCrosUtilsPath')
+    src_path = '/fake/src'
+    chroot_src_path = '/chroot/fake/src'
+    chroot_path = '/usr/bin'
+
+    cros_build_lib.IsInsideChroot().AndReturn(False)
+    cros_build_lib.GetCrosUtilsPath(True).AndReturn(src_path)
+    cros_build_lib.IsInsideChroot().AndReturn(True)
+    cros_build_lib.GetCrosUtilsPath(True).AndReturn(chroot_src_path)
+    cros_build_lib.IsInsideChroot().AndReturn(True)
+
+    self.mox.ReplayAll()
+    # Outside chroot.
+    self.assertTrue(cros_build_lib.GetCrosUtilsBinPath(source_dir_path=True),
+                    src_path + '/bin')
+    # Rest inside chroot.
+    self.assertTrue(cros_build_lib.GetCrosUtilsBinPath(source_dir_path=True),
+                    chroot_src_path + '/bin')
+    self.assertTrue(cros_build_lib.GetCrosUtilsBinPath(source_dir_path=False),
+                    chroot_path)
+    self.mox.VerifyAll()
+
+
 
 if __name__ == '__main__':
   unittest.main()