Rename PartialMock.Start to start().
So that partial mocks can be managed by calling
MockTestCase.StartPatcher().
BUG=None
TEST=unit tests.
Change-Id: I72cfd3033c5da400712465d6b33d9b321b694ca0
Reviewed-on: https://gerrit.chromium.org/gerrit/39685
Reviewed-by: David James <davidjames@chromium.org>
Tested-by: Ryan Cui <rcui@chromium.org>
Commit-Ready: Ryan Cui <rcui@chromium.org>
diff --git a/scripts/deploy_chrome_unittest.py b/scripts/deploy_chrome_unittest.py
index daabb8f..867446c 100755
--- a/scripts/deploy_chrome_unittest.py
+++ b/scripts/deploy_chrome_unittest.py
@@ -76,13 +76,11 @@
returnvalue,
side_effect=None if returnvalue else hook)
- def Start(self):
- partial_mock.PartialMock.Start(self)
- self.rsh_mock.Start()
+ def PreStart(self):
+ self.rsh_mock.start()
- def Stop(self):
- partial_mock.PartialMock.Stop(self)
- self.rsh_mock.Stop()
+ def PreStop(self):
+ self.rsh_mock.stop()
def _CheckRootfsWriteable(self, _inst):
return self.rootfs_writeable
@@ -95,7 +93,7 @@
pass
-class DeployChromeTest(cros_test_lib.TempDirTestCase):
+class DeployChromeTest(cros_test_lib.MockTempDirTestCase):
def _GetDeployChrome(self):
options, _ = _ParseCommandLine(list(_REGULAR_TO) + ['--gs-path', _GS_PATH])
@@ -104,12 +102,9 @@
def setUp(self):
self.deploy_mock = DeployChromeMock()
- self.deploy_mock.Start()
+ self.StartPatcher(self.deploy_mock)
self.deploy = self._GetDeployChrome()
- def tearDown(self):
- self.deploy_mock.Stop()
-
class TestPrepareTarget(DeployChromeTest):
"""Testing disabling of rootfs verification and RO mode."""