| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/chromeos/login/signed_settings_helper.h" | 5 #include "chrome/browser/chromeos/login/signed_settings_helper.h" |
| 6 | 6 |
| 7 #include "chrome/browser/chromeos/cros/cros_library.h" | 7 #include "chrome/browser/chromeos/cros/cros_library.h" |
| 8 #include "chrome/browser/chromeos/cros_settings_names.h" | 8 #include "chrome/browser/chromeos/cros_settings_names.h" |
| 9 #include "chrome/browser/chromeos/dbus/dbus_thread_manager.h" | 9 #include "chrome/browser/chromeos/dbus/dbus_thread_manager.h" |
| 10 #include "chrome/browser/chromeos/login/mock_ownership_service.h" | 10 #include "chrome/browser/chromeos/login/mock_ownership_service.h" |
| (...skipping 22 matching lines...) Expand all Loading... |
| 33 public: | 33 public: |
| 34 MOCK_METHOD2(OnCheckWhitelistCompleted, void( | 34 MOCK_METHOD2(OnCheckWhitelistCompleted, void( |
| 35 SignedSettings::ReturnCode code, const std::string& email)); | 35 SignedSettings::ReturnCode code, const std::string& email)); |
| 36 MOCK_METHOD2(OnWhitelistCompleted, void( | 36 MOCK_METHOD2(OnWhitelistCompleted, void( |
| 37 SignedSettings::ReturnCode code, const std::string& email)); | 37 SignedSettings::ReturnCode code, const std::string& email)); |
| 38 MOCK_METHOD2(OnUnwhitelistCompleted, void( | 38 MOCK_METHOD2(OnUnwhitelistCompleted, void( |
| 39 SignedSettings::ReturnCode code, const std::string& email)); | 39 SignedSettings::ReturnCode code, const std::string& email)); |
| 40 MOCK_METHOD3(OnStorePropertyCompleted, void( | 40 MOCK_METHOD3(OnStorePropertyCompleted, void( |
| 41 SignedSettings::ReturnCode code, | 41 SignedSettings::ReturnCode code, |
| 42 const std::string& name, | 42 const std::string& name, |
| 43 const std::string& value)); | 43 const base::Value& value)); |
| 44 MOCK_METHOD3(OnRetrievePropertyCompleted, void( | 44 MOCK_METHOD3(OnRetrievePropertyCompleted, void( |
| 45 SignedSettings::ReturnCode code, | 45 SignedSettings::ReturnCode code, |
| 46 const std::string& name, | 46 const std::string& name, |
| 47 const std::string& value)); | 47 const base::Value* value)); |
| 48 }; | 48 }; |
| 49 | 49 |
| 50 class SignedSettingsHelperTest : public testing::Test, | 50 class SignedSettingsHelperTest : public testing::Test, |
| 51 public SignedSettingsHelper::TestDelegate { | 51 public SignedSettingsHelper::TestDelegate { |
| 52 public: | 52 public: |
| 53 SignedSettingsHelperTest() | 53 SignedSettingsHelperTest() |
| 54 : fake_email_("fakey@example.com"), | 54 : fake_email_("fakey@example.com"), |
| 55 fake_prop_(kAccountsPrefAllowGuest), | 55 fake_prop_(kReleaseChannel), |
| 56 fake_value_("false"), | 56 fake_value_("false"), |
| 57 message_loop_(MessageLoop::TYPE_UI), | 57 message_loop_(MessageLoop::TYPE_UI), |
| 58 ui_thread_(BrowserThread::UI, &message_loop_), | 58 ui_thread_(BrowserThread::UI, &message_loop_), |
| 59 file_thread_(BrowserThread::FILE), | 59 file_thread_(BrowserThread::FILE), |
| 60 pending_ops_(0) { | 60 pending_ops_(0) { |
| 61 } | 61 } |
| 62 | 62 |
| 63 virtual void SetUp() { | 63 virtual void SetUp() { |
| 64 file_thread_.Start(); | 64 file_thread_.Start(); |
| 65 SignedSettingsHelper::Get()->set_test_delegate(this); | 65 SignedSettingsHelper::Get()->set_test_delegate(this); |
| (...skipping 26 matching lines...) Expand all Loading... |
| 92 em::PolicyData BuildPolicyData() { | 92 em::PolicyData BuildPolicyData() { |
| 93 em::PolicyData to_return; | 93 em::PolicyData to_return; |
| 94 em::ChromeDeviceSettingsProto pol; | 94 em::ChromeDeviceSettingsProto pol; |
| 95 to_return.set_policy_type(SignedSettings::kDevicePolicyType); | 95 to_return.set_policy_type(SignedSettings::kDevicePolicyType); |
| 96 to_return.set_policy_value(pol.SerializeAsString()); | 96 to_return.set_policy_value(pol.SerializeAsString()); |
| 97 return to_return; | 97 return to_return; |
| 98 } | 98 } |
| 99 | 99 |
| 100 const std::string fake_email_; | 100 const std::string fake_email_; |
| 101 const std::string fake_prop_; | 101 const std::string fake_prop_; |
| 102 const std::string fake_value_; | 102 const base::StringValue fake_value_; |
| 103 MockOwnershipService m_; | 103 MockOwnershipService m_; |
| 104 | 104 |
| 105 MessageLoop message_loop_; | 105 MessageLoop message_loop_; |
| 106 BrowserThread ui_thread_; | 106 BrowserThread ui_thread_; |
| 107 BrowserThread file_thread_; | 107 BrowserThread file_thread_; |
| 108 | 108 |
| 109 int pending_ops_; | 109 int pending_ops_; |
| 110 | 110 |
| 111 ScopedStubCrosEnabler stub_cros_enabler_; | 111 ScopedStubCrosEnabler stub_cros_enabler_; |
| 112 }; | 112 }; |
| 113 | 113 |
| 114 ACTION(ConsumeValue) { delete arg0; } |
| 115 |
| 114 TEST_F(SignedSettingsHelperTest, SerializedOps) { | 116 TEST_F(SignedSettingsHelperTest, SerializedOps) { |
| 115 MockSignedSettingsHelperCallback cb; | 117 MockSignedSettingsHelperCallback cb; |
| 116 | 118 |
| 117 EXPECT_CALL(m_, GetStatus(_)) | 119 EXPECT_CALL(m_, GetStatus(_)) |
| 118 .Times(2) | 120 .Times(2) |
| 119 .WillRepeatedly(Return(OwnershipService::OWNERSHIP_TAKEN)); | 121 .WillRepeatedly(Return(OwnershipService::OWNERSHIP_TAKEN)); |
| 120 EXPECT_CALL(m_, has_cached_policy()) | 122 EXPECT_CALL(m_, has_cached_policy()) |
| 121 .Times(5) | 123 .Times(5) |
| 122 .WillRepeatedly(Return(true)); | 124 .WillRepeatedly(Return(true)); |
| 123 em::PolicyData fake_pol = BuildPolicyData(); | 125 em::PolicyData fake_pol = BuildPolicyData(); |
| (...skipping 17 matching lines...) Expand all Loading... |
| 141 .WillOnce(WithArg<1>(Invoke(&SignedSettingsHelperTest::OnKeyOpComplete))); | 143 .WillOnce(WithArg<1>(Invoke(&SignedSettingsHelperTest::OnKeyOpComplete))); |
| 142 EXPECT_CALL(cb, OnUnwhitelistCompleted(SignedSettings::SUCCESS, _)) | 144 EXPECT_CALL(cb, OnUnwhitelistCompleted(SignedSettings::SUCCESS, _)) |
| 143 .Times(1); | 145 .Times(1); |
| 144 | 146 |
| 145 EXPECT_CALL(m_, StartSigningAttempt(_, A<OwnerManager::Delegate*>())) | 147 EXPECT_CALL(m_, StartSigningAttempt(_, A<OwnerManager::Delegate*>())) |
| 146 .WillOnce(WithArg<1>(Invoke(&SignedSettingsHelperTest::OnKeyOpComplete))); | 148 .WillOnce(WithArg<1>(Invoke(&SignedSettingsHelperTest::OnKeyOpComplete))); |
| 147 EXPECT_CALL(cb, OnStorePropertyCompleted(SignedSettings::SUCCESS, _, _)) | 149 EXPECT_CALL(cb, OnStorePropertyCompleted(SignedSettings::SUCCESS, _, _)) |
| 148 .Times(1); | 150 .Times(1); |
| 149 | 151 |
| 150 EXPECT_CALL(cb, OnRetrievePropertyCompleted(SignedSettings::SUCCESS, _, _)) | 152 EXPECT_CALL(cb, OnRetrievePropertyCompleted(SignedSettings::SUCCESS, _, _)) |
| 151 .Times(1); | 153 .WillOnce(WithArg<2>(ConsumeValue())); |
| 152 | 154 |
| 153 pending_ops_ = 5; | 155 pending_ops_ = 5; |
| 154 SignedSettingsHelper::Get()->StartWhitelistOp(fake_email_, true, &cb); | 156 SignedSettingsHelper::Get()->StartWhitelistOp(fake_email_, true, &cb); |
| 155 SignedSettingsHelper::Get()->StartCheckWhitelistOp(fake_email_, &cb); | 157 SignedSettingsHelper::Get()->StartCheckWhitelistOp(fake_email_, &cb); |
| 156 SignedSettingsHelper::Get()->StartWhitelistOp(fake_email_, false, &cb); | 158 SignedSettingsHelper::Get()->StartWhitelistOp(fake_email_, false, &cb); |
| 157 SignedSettingsHelper::Get()->StartStorePropertyOp(fake_prop_, fake_value_, | 159 SignedSettingsHelper::Get()->StartStorePropertyOp(fake_prop_, fake_value_, |
| 158 &cb); | 160 &cb); |
| 159 SignedSettingsHelper::Get()->StartRetrieveProperty(fake_prop_, &cb); | 161 SignedSettingsHelper::Get()->StartRetrieveProperty(fake_prop_, &cb); |
| 160 | 162 |
| 161 message_loop_.Run(); | 163 message_loop_.Run(); |
| (...skipping 33 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 195 | 197 |
| 196 // CheckWhitelistOp for cb_to_be_canceled still gets executed but callback | 198 // CheckWhitelistOp for cb_to_be_canceled still gets executed but callback |
| 197 // does not happen. | 199 // does not happen. |
| 198 | 200 |
| 199 EXPECT_CALL(m_, StartSigningAttempt(_, A<OwnerManager::Delegate*>())) | 201 EXPECT_CALL(m_, StartSigningAttempt(_, A<OwnerManager::Delegate*>())) |
| 200 .WillOnce(WithArg<1>(Invoke(&SignedSettingsHelperTest::OnKeyOpComplete))); | 202 .WillOnce(WithArg<1>(Invoke(&SignedSettingsHelperTest::OnKeyOpComplete))); |
| 201 EXPECT_CALL(cb, OnStorePropertyCompleted(SignedSettings::SUCCESS, _, _)) | 203 EXPECT_CALL(cb, OnStorePropertyCompleted(SignedSettings::SUCCESS, _, _)) |
| 202 .Times(1); | 204 .Times(1); |
| 203 | 205 |
| 204 EXPECT_CALL(cb, OnRetrievePropertyCompleted(SignedSettings::SUCCESS, _, _)) | 206 EXPECT_CALL(cb, OnRetrievePropertyCompleted(SignedSettings::SUCCESS, _, _)) |
| 205 .Times(1); | 207 .WillOnce(WithArg<2>(ConsumeValue())); |
| 206 | 208 |
| 207 pending_ops_ = 6; | 209 pending_ops_ = 6; |
| 208 SignedSettingsHelper::Get()->StartWhitelistOp(fake_email_, true, &cb); | 210 SignedSettingsHelper::Get()->StartWhitelistOp(fake_email_, true, &cb); |
| 209 SignedSettingsHelper::Get()->StartCheckWhitelistOp(fake_email_, &cb); | 211 SignedSettingsHelper::Get()->StartCheckWhitelistOp(fake_email_, &cb); |
| 210 SignedSettingsHelper::Get()->StartWhitelistOp(fake_email_, false, &cb); | 212 SignedSettingsHelper::Get()->StartWhitelistOp(fake_email_, false, &cb); |
| 211 | 213 |
| 212 MockSignedSettingsHelperCallback cb_to_be_canceled; | 214 MockSignedSettingsHelperCallback cb_to_be_canceled; |
| 213 SignedSettingsHelper::Get()->StartCheckWhitelistOp(fake_email_, | 215 SignedSettingsHelper::Get()->StartCheckWhitelistOp(fake_email_, |
| 214 &cb_to_be_canceled); | 216 &cb_to_be_canceled); |
| 215 SignedSettingsHelper::Get()->CancelCallback(&cb_to_be_canceled); | 217 SignedSettingsHelper::Get()->CancelCallback(&cb_to_be_canceled); |
| 216 | 218 |
| 217 SignedSettingsHelper::Get()->StartStorePropertyOp(fake_prop_, fake_value_, | 219 SignedSettingsHelper::Get()->StartStorePropertyOp(fake_prop_, fake_value_, |
| 218 &cb); | 220 &cb); |
| 219 SignedSettingsHelper::Get()->StartRetrieveProperty(fake_prop_, &cb); | 221 SignedSettingsHelper::Get()->StartRetrieveProperty(fake_prop_, &cb); |
| 220 | 222 |
| 221 message_loop_.Run(); | 223 message_loop_.Run(); |
| 222 } | 224 } |
| 223 | 225 |
| 224 } // namespace chromeos | 226 } // namespace chromeos |
| OLD | NEW |