Index: delta_performer_unittest.cc |
diff --git a/delta_performer_unittest.cc b/delta_performer_unittest.cc |
index 0c2d9b9a1dd25c56c65c34c10508ecd21725ebd8..a90367db18e4ee35972cf2f047edca6d23ba681d 100755 |
--- a/delta_performer_unittest.cc |
+++ b/delta_performer_unittest.cc |
@@ -450,18 +450,23 @@ void DoSmallImageTest(bool full_kernel, bool full_rootfs, bool noop, |
strlen(new_data_string))); |
EXPECT_TRUE(utils::FileExists(kUnittestPublicKeyPath)); |
- bool expect_verify_success = |
- signature_test != kSignatureNone && |
- signature_test != kSignatureGeneratedShellBadKey; |
+ const bool expect_verify_success = |
+ signature_test != kSignatureNone; |
+ const bool expect_public_verify_failure = |
+ signature_test == kSignatureGeneratedShellBadKey; |
+ bool public_verify_failure = false; |
EXPECT_EQ(expect_verify_success, |
performer.VerifyPayload( |
kUnittestPublicKeyPath, |
OmahaHashCalculator::OmahaHashOfData(delta), |
- delta.size())); |
+ delta.size(), |
+ &public_verify_failure)); |
+ EXPECT_EQ(expect_public_verify_failure, public_verify_failure); |
EXPECT_TRUE(performer.VerifyPayload( |
"/public/key/does/not/exists", |
OmahaHashCalculator::OmahaHashOfData(delta), |
- delta.size())); |
+ delta.size(), |
+ NULL)); |
uint64_t new_kernel_size; |
vector<char> new_kernel_hash; |