OLD | NEW |
1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 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 // Unit tests for the CEEE module-wide utilities. | 5 // Unit tests for the CEEE module-wide utilities. |
6 | 6 |
7 #include "ceee/ie/common/ceee_module_util.h" | 7 #include "ceee/ie/common/ceee_module_util.h" |
8 | 8 |
9 #include <wtypes.h> | 9 #include <wtypes.h> |
10 #include <string> | 10 #include <string> |
(...skipping 189 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
200 EXPECT_EQ(L"Software\\Google\\Update\\ClientState\\" | 200 EXPECT_EQ(L"Software\\Google\\Update\\ClientState\\" |
201 L"{8BA986DA-5100-405E-AA35-86F34A02ACBF}", | 201 L"{8BA986DA-5100-405E-AA35-86F34A02ACBF}", |
202 ceee_module_util::GetCromeFrameClientStateKey()); | 202 ceee_module_util::GetCromeFrameClientStateKey()); |
203 } | 203 } |
204 | 204 |
205 TEST_F(CeeeModuleUtilTest, GetCollectStatsConsentFromHkcu) { | 205 TEST_F(CeeeModuleUtilTest, GetCollectStatsConsentFromHkcu) { |
206 EXPECT_FALSE(ceee_module_util::GetCollectStatsConsent()); | 206 EXPECT_FALSE(ceee_module_util::GetCollectStatsConsent()); |
207 base::win::RegKey hkcu(HKEY_CURRENT_USER, | 207 base::win::RegKey hkcu(HKEY_CURRENT_USER, |
208 ceee_module_util::GetCromeFrameClientStateKey().c_str(), KEY_WRITE); | 208 ceee_module_util::GetCromeFrameClientStateKey().c_str(), KEY_WRITE); |
209 | 209 |
210 ASSERT_TRUE(hkcu.WriteValue(google_update::kRegUsageStatsField, kTrue)); | 210 ASSERT_EQ(ERROR_SUCCESS, |
| 211 hkcu.WriteValue(google_update::kRegUsageStatsField, kTrue)); |
211 EXPECT_TRUE(ceee_module_util::GetCollectStatsConsent()); | 212 EXPECT_TRUE(ceee_module_util::GetCollectStatsConsent()); |
212 | 213 |
213 ASSERT_TRUE(hkcu.WriteValue(google_update::kRegUsageStatsField, kFalse)); | 214 ASSERT_EQ(ERROR_SUCCESS, |
| 215 hkcu.WriteValue(google_update::kRegUsageStatsField, kFalse)); |
214 EXPECT_FALSE(ceee_module_util::GetCollectStatsConsent()); | 216 EXPECT_FALSE(ceee_module_util::GetCollectStatsConsent()); |
215 | 217 |
216 ASSERT_TRUE(hkcu.WriteValue(google_update::kRegUsageStatsField, kInvalid)); | 218 ASSERT_EQ(ERROR_SUCCESS, |
| 219 hkcu.WriteValue(google_update::kRegUsageStatsField, kInvalid)); |
217 EXPECT_FALSE(ceee_module_util::GetCollectStatsConsent()); | 220 EXPECT_FALSE(ceee_module_util::GetCollectStatsConsent()); |
218 | 221 |
219 ASSERT_TRUE(hkcu.DeleteValue(google_update::kRegUsageStatsField)); | 222 ASSERT_EQ(ERROR_SUCCESS, |
| 223 hkcu.DeleteValue(google_update::kRegUsageStatsField)); |
220 EXPECT_FALSE(ceee_module_util::GetCollectStatsConsent()); | 224 EXPECT_FALSE(ceee_module_util::GetCollectStatsConsent()); |
221 } | 225 } |
222 | 226 |
223 TEST_F(CeeeModuleUtilTest, GetCollectStatsConsentFromHklm) { | 227 TEST_F(CeeeModuleUtilTest, GetCollectStatsConsentFromHklm) { |
224 EXPECT_FALSE(ceee_module_util::GetCollectStatsConsent()); | 228 EXPECT_FALSE(ceee_module_util::GetCollectStatsConsent()); |
225 base::win::RegKey hklm(HKEY_LOCAL_MACHINE, | 229 base::win::RegKey hklm(HKEY_LOCAL_MACHINE, |
226 ceee_module_util::GetCromeFrameClientStateKey().c_str(), KEY_WRITE); | 230 ceee_module_util::GetCromeFrameClientStateKey().c_str(), KEY_WRITE); |
227 | 231 |
228 ASSERT_TRUE(hklm.WriteValue(google_update::kRegUsageStatsField, kTrue)); | 232 ASSERT_EQ(ERROR_SUCCESS, |
| 233 hklm.WriteValue(google_update::kRegUsageStatsField, kTrue)); |
229 EXPECT_TRUE(ceee_module_util::GetCollectStatsConsent()); | 234 EXPECT_TRUE(ceee_module_util::GetCollectStatsConsent()); |
230 | 235 |
231 ASSERT_TRUE(hklm.WriteValue(google_update::kRegUsageStatsField, kFalse)); | 236 ASSERT_EQ(ERROR_SUCCESS, |
| 237 hklm.WriteValue(google_update::kRegUsageStatsField, kFalse)); |
232 EXPECT_FALSE(ceee_module_util::GetCollectStatsConsent()); | 238 EXPECT_FALSE(ceee_module_util::GetCollectStatsConsent()); |
233 | 239 |
234 ASSERT_TRUE(hklm.WriteValue(google_update::kRegUsageStatsField, kInvalid)); | 240 ASSERT_EQ(ERROR_SUCCESS, |
| 241 hklm.WriteValue(google_update::kRegUsageStatsField, kInvalid)); |
235 EXPECT_FALSE(ceee_module_util::GetCollectStatsConsent()); | 242 EXPECT_FALSE(ceee_module_util::GetCollectStatsConsent()); |
236 | 243 |
237 ASSERT_TRUE(hklm.DeleteValue(google_update::kRegUsageStatsField)); | 244 ASSERT_EQ(ERROR_SUCCESS, |
| 245 hklm.DeleteValue(google_update::kRegUsageStatsField)); |
238 EXPECT_FALSE(ceee_module_util::GetCollectStatsConsent()); | 246 EXPECT_FALSE(ceee_module_util::GetCollectStatsConsent()); |
239 } | 247 } |
240 | 248 |
241 TEST_F(CeeeModuleUtilTest, GetCollectStatsConsentHkcuBeforeHklm) { | 249 TEST_F(CeeeModuleUtilTest, GetCollectStatsConsentHkcuBeforeHklm) { |
242 EXPECT_FALSE(ceee_module_util::GetCollectStatsConsent()); | 250 EXPECT_FALSE(ceee_module_util::GetCollectStatsConsent()); |
243 base::win::RegKey hkcu(HKEY_CURRENT_USER, | 251 base::win::RegKey hkcu(HKEY_CURRENT_USER, |
244 ceee_module_util::GetCromeFrameClientStateKey().c_str(), KEY_WRITE); | 252 ceee_module_util::GetCromeFrameClientStateKey().c_str(), KEY_WRITE); |
245 | 253 |
246 base::win::RegKey hklm(HKEY_LOCAL_MACHINE, | 254 base::win::RegKey hklm(HKEY_LOCAL_MACHINE, |
247 ceee_module_util::GetCromeFrameClientStateKey().c_str(), KEY_WRITE); | 255 ceee_module_util::GetCromeFrameClientStateKey().c_str(), KEY_WRITE); |
248 | 256 |
249 ASSERT_TRUE(hklm.WriteValue(google_update::kRegUsageStatsField, kTrue)); | 257 ASSERT_EQ(ERROR_SUCCESS, |
| 258 hklm.WriteValue(google_update::kRegUsageStatsField, kTrue)); |
250 ASSERT_TRUE(ceee_module_util::GetCollectStatsConsent()); | 259 ASSERT_TRUE(ceee_module_util::GetCollectStatsConsent()); |
251 | 260 |
252 ASSERT_TRUE(hkcu.WriteValue(google_update::kRegUsageStatsField, kFalse)); | 261 ASSERT_EQ(ERROR_SUCCESS, |
| 262 hkcu.WriteValue(google_update::kRegUsageStatsField, kFalse)); |
253 EXPECT_FALSE(ceee_module_util::GetCollectStatsConsent()); | 263 EXPECT_FALSE(ceee_module_util::GetCollectStatsConsent()); |
254 | 264 |
255 ASSERT_TRUE(hkcu.DeleteValue(google_update::kRegUsageStatsField)); | 265 ASSERT_EQ(ERROR_SUCCESS, |
256 ASSERT_TRUE(hklm.DeleteValue(google_update::kRegUsageStatsField)); | 266 hkcu.DeleteValue(google_update::kRegUsageStatsField)); |
| 267 ASSERT_EQ(ERROR_SUCCESS, |
| 268 hklm.DeleteValue(google_update::kRegUsageStatsField)); |
257 } | 269 } |
258 | 270 |
259 } // namespace | 271 } // namespace |
OLD | NEW |