Index: chrome/browser/chromeos/login/signed_settings_helper_unittest.cc |
diff --git a/chrome/browser/chromeos/login/signed_settings_helper_unittest.cc b/chrome/browser/chromeos/login/signed_settings_helper_unittest.cc |
index 81a553e56209fb1bf7273a72d905266f01d50cb2..4de86d645d5dd98327686e64e027e126de31f44e 100644 |
--- a/chrome/browser/chromeos/login/signed_settings_helper_unittest.cc |
+++ b/chrome/browser/chromeos/login/signed_settings_helper_unittest.cc |
@@ -30,12 +30,8 @@ namespace chromeos { |
class MockSignedSettingsHelperCallback : public SignedSettingsHelper::Callback { |
public: |
- MOCK_METHOD2(OnCheckWhitelistCompleted, void( |
- SignedSettings::ReturnCode code, const std::string& email)); |
- MOCK_METHOD2(OnWhitelistCompleted, void( |
- SignedSettings::ReturnCode code, const std::string& email)); |
- MOCK_METHOD2(OnUnwhitelistCompleted, void( |
- SignedSettings::ReturnCode code, const std::string& email)); |
+ virtual ~MockSignedSettingsHelperCallback() {} |
+ |
MOCK_METHOD3(OnStorePropertyCompleted, void( |
SignedSettings::ReturnCode code, |
const std::string& name, |
@@ -115,42 +111,27 @@ TEST_F(SignedSettingsHelperTest, SerializedOps) { |
.Times(2) |
.WillRepeatedly(Return(OwnershipService::OWNERSHIP_TAKEN)); |
EXPECT_CALL(m_, has_cached_policy()) |
- .Times(5) |
+ .Times(2) |
.WillRepeatedly(Return(true)); |
em::PolicyData fake_pol = BuildPolicyData(); |
EXPECT_CALL(m_, cached_policy()) |
- .Times(5) |
+ .Times(2) |
.WillRepeatedly(ReturnRef(fake_pol)); |
EXPECT_CALL(m_, set_cached_policy(A<const em::PolicyData&>())) |
- .Times(3) |
+ .Times(1) |
.WillRepeatedly(SaveArg<0>(&fake_pol)); |
InSequence s; |
EXPECT_CALL(m_, StartSigningAttempt(_, A<OwnerManager::Delegate*>())) |
.WillOnce(WithArg<1>(Invoke(&SignedSettingsHelperTest::OnKeyOpComplete))); |
- EXPECT_CALL(cb, OnWhitelistCompleted(SignedSettings::SUCCESS, _)) |
- .Times(1); |
- |
- EXPECT_CALL(cb, OnCheckWhitelistCompleted(SignedSettings::SUCCESS, _)) |
- .Times(1); |
- |
- EXPECT_CALL(m_, StartSigningAttempt(_, A<OwnerManager::Delegate*>())) |
- .WillOnce(WithArg<1>(Invoke(&SignedSettingsHelperTest::OnKeyOpComplete))); |
- EXPECT_CALL(cb, OnUnwhitelistCompleted(SignedSettings::SUCCESS, _)) |
- .Times(1); |
- |
- EXPECT_CALL(m_, StartSigningAttempt(_, A<OwnerManager::Delegate*>())) |
- .WillOnce(WithArg<1>(Invoke(&SignedSettingsHelperTest::OnKeyOpComplete))); |
EXPECT_CALL(cb, OnStorePropertyCompleted(SignedSettings::SUCCESS, _, _)) |
.Times(1); |
EXPECT_CALL(cb, OnRetrievePropertyCompleted(SignedSettings::SUCCESS, _, _)) |
.Times(1); |
- pending_ops_ = 5; |
- SignedSettingsHelper::Get()->StartWhitelistOp(fake_email_, true, &cb); |
- SignedSettingsHelper::Get()->StartCheckWhitelistOp(fake_email_, &cb); |
- SignedSettingsHelper::Get()->StartWhitelistOp(fake_email_, false, &cb); |
+ |
+ pending_ops_ = 2; |
SignedSettingsHelper::Get()->StartStorePropertyOp(fake_prop_, fake_value_, |
&cb); |
SignedSettingsHelper::Get()->StartRetrieveProperty(fake_prop_, &cb); |
@@ -162,53 +143,34 @@ TEST_F(SignedSettingsHelperTest, CanceledOps) { |
MockSignedSettingsHelperCallback cb; |
EXPECT_CALL(m_, GetStatus(_)) |
- .Times(2) |
+ .Times(3) |
.WillRepeatedly(Return(OwnershipService::OWNERSHIP_TAKEN)); |
EXPECT_CALL(m_, has_cached_policy()) |
- .Times(6) |
+ .Times(3) |
.WillRepeatedly(Return(true)); |
em::PolicyData fake_pol = BuildPolicyData(); |
EXPECT_CALL(m_, cached_policy()) |
- .Times(7) |
+ .Times(3) |
.WillRepeatedly(ReturnRef(fake_pol)); |
EXPECT_CALL(m_, set_cached_policy(A<const em::PolicyData&>())) |
- .Times(3) |
+ .Times(1) |
.WillRepeatedly(SaveArg<0>(&fake_pol)); |
InSequence s; |
- EXPECT_CALL(m_, StartSigningAttempt(_, A<OwnerManager::Delegate*>())) |
- .WillOnce(WithArg<1>(Invoke(&SignedSettingsHelperTest::OnKeyOpComplete))); |
- EXPECT_CALL(cb, OnWhitelistCompleted(SignedSettings::SUCCESS, _)) |
- .Times(1); |
- |
- EXPECT_CALL(cb, OnCheckWhitelistCompleted(SignedSettings::SUCCESS, _)) |
- .Times(1); |
- |
- EXPECT_CALL(m_, StartSigningAttempt(_, A<OwnerManager::Delegate*>())) |
- .WillOnce(WithArg<1>(Invoke(&SignedSettingsHelperTest::OnKeyOpComplete))); |
- EXPECT_CALL(cb, OnUnwhitelistCompleted(SignedSettings::SUCCESS, _)) |
- .Times(1); |
- |
- // CheckWhitelistOp for cb_to_be_canceled still gets executed but callback |
+ // RetrievePropertyOp for cb_to_be_canceled still gets executed but callback |
// does not happen. |
- |
EXPECT_CALL(m_, StartSigningAttempt(_, A<OwnerManager::Delegate*>())) |
.WillOnce(WithArg<1>(Invoke(&SignedSettingsHelperTest::OnKeyOpComplete))); |
EXPECT_CALL(cb, OnStorePropertyCompleted(SignedSettings::SUCCESS, _, _)) |
.Times(1); |
- |
EXPECT_CALL(cb, OnRetrievePropertyCompleted(SignedSettings::SUCCESS, _, _)) |
.Times(1); |
- pending_ops_ = 6; |
- SignedSettingsHelper::Get()->StartWhitelistOp(fake_email_, true, &cb); |
- SignedSettingsHelper::Get()->StartCheckWhitelistOp(fake_email_, &cb); |
- SignedSettingsHelper::Get()->StartWhitelistOp(fake_email_, false, &cb); |
- |
+ pending_ops_ = 3; |
MockSignedSettingsHelperCallback cb_to_be_canceled; |
- SignedSettingsHelper::Get()->StartCheckWhitelistOp(fake_email_, |
- &cb_to_be_canceled); |
+ SignedSettingsHelper::Get()->StartRetrieveProperty(fake_prop_, |
+ &cb_to_be_canceled); |
SignedSettingsHelper::Get()->CancelCallback(&cb_to_be_canceled); |
SignedSettingsHelper::Get()->StartStorePropertyOp(fake_prop_, fake_value_, |