Index: delta_performer_unittest.cc |
diff --git a/delta_performer_unittest.cc b/delta_performer_unittest.cc |
index 12de8a9a24e1784d2e2e6e5ad20dad154165116c..cf3ff462ef78e79ec68452bc44e4ff961a1be631 100755 |
--- a/delta_performer_unittest.cc |
+++ b/delta_performer_unittest.cc |
@@ -247,8 +247,10 @@ TEST(DeltaPerformerTest, RunAsRootSmallImageTest) { |
.WillRepeatedly(Return(true)); |
EXPECT_CALL(prefs, SetInt64(kPrefsUpdateStateNextDataOffset, _)) |
.WillRepeatedly(Return(true)); |
- EXPECT_CALL(prefs, SetString(kPrefsUpdateStateSignedSHA256Context, _)) |
+ EXPECT_CALL(prefs, SetString(kPrefsUpdateStateSHA256Context, _)) |
.WillRepeatedly(Return(true)); |
+ EXPECT_CALL(prefs, SetString(kPrefsUpdateStateSignedSHA256Context, _)) |
+ .WillOnce(Return(true)); |
// Update the A image in place. |
DeltaPerformer performer(&prefs); |
@@ -274,7 +276,10 @@ TEST(DeltaPerformerTest, RunAsRootSmallImageTest) { |
strlen(new_data_string))); |
EXPECT_TRUE(utils::FileExists(kUnittestPublicKeyPath)); |
- EXPECT_TRUE(performer.VerifyPayload(kUnittestPublicKeyPath)); |
+ EXPECT_TRUE(performer.VerifyPayload( |
+ kUnittestPublicKeyPath, |
+ OmahaHashCalculator::OmahaHashOfData(delta), |
+ delta.size())); |
} |
} // namespace chromeos_update_engine |