Index: autoupdate_unittest.py |
diff --git a/autoupdate_unittest.py b/autoupdate_unittest.py |
index 0aaa5d373ceed95afa00606976d304c7c354ed12..0cf0062ecd5082576cedf2cc7f0e618e9ed04a92 100755 |
--- a/autoupdate_unittest.py |
+++ b/autoupdate_unittest.py |
@@ -60,14 +60,14 @@ class AutoupdateTest(mox.MoxTestBase): |
return dummy |
def testGenerateLatestUpdateImageWithForced(self): |
- self.mox.StubOutWithMock(autoupdate.Autoupdate, |
- 'GenerateUpdateImageWithCache') |
+ self.mox.StubOutWithMock(autoupdate.Autoupdate, 'GenerateUpdateImage') |
autoupdate.Autoupdate._GetLatestImageDir(self.test_board).AndReturn( |
'%s/%s/%s' % (self.build_root, self.test_board, self.latest_dir)) |
- autoupdate.Autoupdate.GenerateUpdateImageWithCache( |
+ autoupdate.Autoupdate.GenerateUpdateImage( |
'%s/%s/%s/chromiumos_image.bin' % (self.build_root, self.test_board, |
self.latest_dir), |
- static_image_dir=self.static_image_dir).AndReturn('update.gz') |
+ move_to_static_dir=True, |
+ static_image_dir=self.static_image_dir).AndReturn(True) |
self.mox.ReplayAll() |
au_mock = self._DummyAutoupdateConstructor() |
@@ -77,14 +77,14 @@ class AutoupdateTest(mox.MoxTestBase): |
self.mox.VerifyAll() |
def testHandleUpdatePingForForcedImage(self): |
- self.mox.StubOutWithMock(autoupdate.Autoupdate, |
- 'GenerateUpdateImageWithCache') |
+ self.mox.StubOutWithMock(autoupdate.Autoupdate, 'GenerateUpdateImage') |
test_data = _TEST_REQUEST % self.test_dict |
- autoupdate.Autoupdate.GenerateUpdateImageWithCache( |
+ autoupdate.Autoupdate.GenerateUpdateImage( |
self.forced_image_path, |
- static_image_dir=self.static_image_dir).AndReturn('update.gz') |
+ move_to_static_dir=True, |
+ static_image_dir=self.static_image_dir).AndReturn(True) |
autoupdate.Autoupdate._GetHash(os.path.join( |
self.static_image_dir, 'update.gz')).AndReturn(self.hash) |
autoupdate.Autoupdate._GetSHA256(os.path.join( |
@@ -107,8 +107,7 @@ class AutoupdateTest(mox.MoxTestBase): |
test_data = _TEST_REQUEST % self.test_dict |
autoupdate.Autoupdate.GenerateLatestUpdateImage( |
- self.test_board, 'ForcedUpdate', self.static_image_dir).AndReturn( |
- 'update.gz') |
+ self.test_board, 'ForcedUpdate', self.static_image_dir).AndReturn(True) |
autoupdate.Autoupdate._GetHash(os.path.join( |
self.static_image_dir, 'update.gz')).AndReturn(self.hash) |
autoupdate.Autoupdate._GetSHA256(os.path.join( |
@@ -130,7 +129,7 @@ class AutoupdateTest(mox.MoxTestBase): |
test_data = _TEST_REQUEST % self.test_dict |
autoupdate.Autoupdate.GenerateImageFromZip( |
- self.static_image_dir).AndReturn('update.gz') |
+ self.static_image_dir).AndReturn(True) |
autoupdate.Autoupdate._GetHash(os.path.join( |
self.static_image_dir, 'update.gz')).AndReturn(self.hash) |
autoupdate.Autoupdate._GetSHA256(os.path.join( |