Chromium Code Reviews| 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 <windows.h> | 5 #include <windows.h> |
| 6 | 6 |
| 7 #include "base/strings/utf_string_conversions.h" | 7 #include "base/strings/utf_string_conversions.h" |
| 8 #include "base/test/test_reg_util_win.h" | 8 #include "base/test/test_reg_util_win.h" |
| 9 #include "base/version.h" | 9 #include "base/version.h" |
| 10 #include "base/win/registry.h" | 10 #include "base/win/registry.h" |
| (...skipping 52 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 63 testing::Test::TearDownTestCase(); | 63 testing::Test::TearDownTestCase(); |
| 64 } | 64 } |
| 65 | 65 |
| 66 void ProductStateTest::SetUp() { | 66 void ProductStateTest::SetUp() { |
| 67 testing::Test::SetUp(); | 67 testing::Test::SetUp(); |
| 68 | 68 |
| 69 // Create/open the keys for the product we'll test. | 69 // Create/open the keys for the product we'll test. |
| 70 system_install_ = true; | 70 system_install_ = true; |
| 71 overridden_ = (system_install_ ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER); | 71 overridden_ = (system_install_ ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER); |
| 72 | 72 |
| 73 // Override for test purposes. We don't use ScopedRegistryKeyOverride | 73 // Override for test purposes. We don't use ScopedRegistryKeyOverride |
|
grt (UTC plus 2)
2013/11/07 02:39:33
Can RegistryOverrideManager replace this? As it is
tommycli
2013/11/07 18:02:07
Done.
| |
| 74 // directly because it doesn't suit itself to our use here. | 74 // directly because it doesn't suit itself to our use here. |
| 75 RegKey temp_key; | 75 RegKey temp_key; |
| 76 EXPECT_EQ(ERROR_SUCCESS, | 76 EXPECT_EQ(ERROR_SUCCESS, |
| 77 temp_key.Create(HKEY_CURRENT_USER, temp_key_path_.c_str(), | 77 temp_key.Create(HKEY_CURRENT_USER, temp_key_path_.c_str(), |
| 78 KEY_ALL_ACCESS)); | 78 KEY_ALL_ACCESS)); |
| 79 EXPECT_EQ(ERROR_SUCCESS, | 79 EXPECT_EQ(ERROR_SUCCESS, |
| 80 ::RegOverridePredefKey(overridden_, temp_key.Handle())); | 80 ::RegOverridePredefKey(overridden_, temp_key.Handle())); |
| 81 | 81 |
| 82 EXPECT_EQ(ERROR_SUCCESS, | 82 EXPECT_EQ(ERROR_SUCCESS, |
| 83 clients_.Create(overridden_, dist_->GetVersionKey().c_str(), | 83 clients_.Create(overridden_, dist_->GetVersionKey().c_str(), |
| 84 KEY_ALL_ACCESS)); | 84 KEY_ALL_ACCESS)); |
| 85 EXPECT_EQ(ERROR_SUCCESS, | 85 EXPECT_EQ(ERROR_SUCCESS, |
| 86 client_state_.Create(overridden_, dist_->GetStateKey().c_str(), | 86 client_state_.Create(overridden_, dist_->GetStateKey().c_str(), |
| 87 KEY_ALL_ACCESS)); | 87 KEY_ALL_ACCESS)); |
| 88 } | 88 } |
| 89 | 89 |
| 90 void ProductStateTest::TearDown() { | 90 void ProductStateTest::TearDown() { |
| 91 // Done with the keys. | 91 // Done with the keys. |
| 92 client_state_.Close(); | 92 client_state_.Close(); |
| 93 clients_.Close(); | 93 clients_.Close(); |
| 94 EXPECT_EQ(ERROR_SUCCESS, ::RegOverridePredefKey(overridden_, NULL)); | 94 EXPECT_EQ(ERROR_SUCCESS, ::RegOverridePredefKey(overridden_, NULL)); |
| 95 overridden_ = NULL; | 95 overridden_ = NULL; |
| 96 system_install_ = false; | 96 system_install_ = false; |
| 97 | 97 |
| 98 // Shotgun approach to clearing out data we may have written. | |
| 99 RegistryOverrideManager::DeleteAllTempKeys(); | |
| 100 | |
| 101 testing::Test::TearDown(); | 98 testing::Test::TearDown(); |
| 102 } | 99 } |
| 103 | 100 |
| 104 void ProductStateTest::MinimallyInstallProduct(const wchar_t* version) { | 101 void ProductStateTest::MinimallyInstallProduct(const wchar_t* version) { |
| 105 EXPECT_EQ(ERROR_SUCCESS, | 102 EXPECT_EQ(ERROR_SUCCESS, |
| 106 clients_.WriteValue(google_update::kRegVersionField, version)); | 103 clients_.WriteValue(google_update::kRegVersionField, version)); |
| 107 } | 104 } |
| 108 | 105 |
| 109 void ProductStateTest::ApplyUninstallCommand(const wchar_t* exe_path, | 106 void ProductStateTest::ApplyUninstallCommand(const wchar_t* exe_path, |
| 110 const wchar_t* args) { | 107 const wchar_t* args) { |
| (...skipping 308 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 419 | 416 |
| 420 // Uninstall command with --multi-install is multi install. | 417 // Uninstall command with --multi-install is multi install. |
| 421 { | 418 { |
| 422 ProductState state; | 419 ProductState state; |
| 423 ApplyUninstallCommand(L"setup.exe", | 420 ApplyUninstallCommand(L"setup.exe", |
| 424 L"--uninstall --chrome --multi-install"); | 421 L"--uninstall --chrome --multi-install"); |
| 425 EXPECT_TRUE(state.Initialize(system_install_, dist_)); | 422 EXPECT_TRUE(state.Initialize(system_install_, dist_)); |
| 426 EXPECT_TRUE(state.is_multi_install()); | 423 EXPECT_TRUE(state.is_multi_install()); |
| 427 } | 424 } |
| 428 } | 425 } |
| OLD | NEW |