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>
diff --git a/scripts/deploy_chrome_unittest.py b/scripts/deploy_chrome_unittest.py
index cf1892b..43409d2 100755
--- a/scripts/deploy_chrome_unittest.py
+++ b/scripts/deploy_chrome_unittest.py
@@ -104,8 +104,7 @@
options, self.tempdir, os.path.join(self.tempdir, 'staging'))
def setUp(self):
- self.deploy_mock = DeployChromeMock()
- self.StartPatcher(self.deploy_mock)
+ self.deploy_mock = self.StartPatcher(DeployChromeMock())
self.deploy = self._GetDeployChrome()