commit f1416f3d830920b2c472e45d036ec2f0d10b70aa
author Ryan Cui <rcui@chromium.org> Tue Jan 22 18:43:41 2013 -0800
committer ChromeBot <chrome-bot@google.com> Wed Jan 23 13:58:16 2013 -0800
tree 8a6ff87a74cab319f093c6a873e9e0ef4c615982
parent 4424e1a9ee4710fa0bfc37a2b34b66ce2dad34ad

MockTestCase - StartPatcher and PatchObject returns mock.

Have StartPatcher() return the PartialMock/mock.Mock instance, for easy
assignment. i.e.,

self.gs_mock = self.StartPatcher(gs_unittest.GSContextMock())

BUG=None
TEST=unittests

Change-Id: I755acb82e5818c9feed32abf1fd628fca41646b5
Reviewed-on: https://gerrit.chromium.org/gerrit/41801
Reviewed-by: David James <davidjames@chromium.org>
Tested-by: Ryan Cui <rcui@chromium.org>
Commit-Queue: Ryan Cui <rcui@chromium.org>

scripts/cros_generate_sysroot_unittest.py

diff --git a/scripts/cros_generate_sysroot_unittest.py b/scripts/cros_generate_sysroot_unittest.py
index 90801c5..8f326ea 100755
--- a/scripts/cros_generate_sysroot_unittest.py
+++ b/scripts/cros_generate_sysroot_unittest.py
@@ -55,8 +55,7 @@
 class OverallTest(cros_test_lib.MockTempDirTestCase):
 
   def setUp(self):
-    self.cg_mock = CrosGenMock()
-    self.StartPatcher(self.cg_mock)
+    self.cg_mock = self.StartPatcher(CrosGenMock())
 
   def testTarballGeneration(self):
     """End-to-end test of tarball generation."""