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/utf_string_conversions.h" | 7 #include "base/utf_string_conversions.h" |
| 8 #include "base/test/test_reg_util_win.h" |
8 #include "base/version.h" | 9 #include "base/version.h" |
9 #include "base/win/registry.h" | 10 #include "base/win/registry.h" |
10 #include "chrome/installer/util/browser_distribution.h" | 11 #include "chrome/installer/util/browser_distribution.h" |
11 #include "chrome/installer/util/google_update_constants.h" | 12 #include "chrome/installer/util/google_update_constants.h" |
12 #include "chrome/installer/util/installation_state.h" | 13 #include "chrome/installer/util/installation_state.h" |
13 #include "chrome/installer/util/product_unittest.h" | 14 #include "chrome/installer/util/product_unittest.h" |
14 #include "chrome/installer/util/util_constants.h" | 15 #include "chrome/installer/util/util_constants.h" |
15 #include "testing/gtest/include/gtest/gtest.h" | 16 #include "testing/gtest/include/gtest/gtest.h" |
16 | 17 |
17 using base::win::RegKey; | 18 using base::win::RegKey; |
18 using installer::ProductState; | 19 using installer::ProductState; |
| 20 using registry_util::RegistryOverrideManager; |
19 | 21 |
20 class ProductStateTest : public testing::Test { | 22 class ProductStateTest : public testing::Test { |
21 protected: | 23 protected: |
22 static void SetUpTestCase(); | 24 static void SetUpTestCase(); |
23 static void TearDownTestCase(); | 25 static void TearDownTestCase(); |
24 | 26 |
25 virtual void SetUp(); | 27 virtual void SetUp(); |
26 virtual void TearDown(); | 28 virtual void TearDown(); |
27 | 29 |
28 void ApplyUninstallCommand(const wchar_t* exe_path, const wchar_t* args); | 30 void ApplyUninstallCommand(const wchar_t* exe_path, const wchar_t* args); |
(...skipping 12 matching lines...) Expand all Loading... |
41 | 43 |
42 // static | 44 // static |
43 void ProductStateTest::SetUpTestCase() { | 45 void ProductStateTest::SetUpTestCase() { |
44 testing::Test::SetUpTestCase(); | 46 testing::Test::SetUpTestCase(); |
45 | 47 |
46 // We'll use Chrome as our test subject. | 48 // We'll use Chrome as our test subject. |
47 dist_ = BrowserDistribution::GetSpecificDistribution( | 49 dist_ = BrowserDistribution::GetSpecificDistribution( |
48 BrowserDistribution::CHROME_BROWSER); | 50 BrowserDistribution::CHROME_BROWSER); |
49 | 51 |
50 // And we'll play in HKCU here: | 52 // And we'll play in HKCU here: |
51 temp_key_path_.assign(TempRegKeyOverride::kTempTestKeyPath) | 53 temp_key_path_.assign(RegistryOverrideManager::kTempTestKeyPath) |
52 .append(1, L'\\') | 54 .append(1, L'\\') |
53 .append(L"ProductStateTest"); | 55 .append(L"ProductStateTest"); |
54 } | 56 } |
55 | 57 |
56 // static | 58 // static |
57 void ProductStateTest::TearDownTestCase() { | 59 void ProductStateTest::TearDownTestCase() { |
58 temp_key_path_.clear(); | 60 temp_key_path_.clear(); |
59 dist_ = NULL; | 61 dist_ = NULL; |
60 | 62 |
61 testing::Test::TearDownTestCase(); | 63 testing::Test::TearDownTestCase(); |
62 } | 64 } |
63 | 65 |
64 void ProductStateTest::SetUp() { | 66 void ProductStateTest::SetUp() { |
65 testing::Test::SetUp(); | 67 testing::Test::SetUp(); |
66 | 68 |
67 // Create/open the keys for the product we'll test. | 69 // Create/open the keys for the product we'll test. |
68 system_install_ = true; | 70 system_install_ = true; |
69 overridden_ = (system_install_ ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER); | 71 overridden_ = (system_install_ ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER); |
70 | 72 |
71 // Override for test purposes. We don't use TempRegKeyOverride | 73 // Override for test purposes. We don't use ScopedRegistryKeyOverride |
72 // directly because it doesn't suit itself to our use here. | 74 // directly because it doesn't suit itself to our use here. |
73 RegKey temp_key; | 75 RegKey temp_key; |
74 EXPECT_EQ(ERROR_SUCCESS, | 76 EXPECT_EQ(ERROR_SUCCESS, |
75 temp_key.Create(HKEY_CURRENT_USER, temp_key_path_.c_str(), | 77 temp_key.Create(HKEY_CURRENT_USER, temp_key_path_.c_str(), |
76 KEY_ALL_ACCESS)); | 78 KEY_ALL_ACCESS)); |
77 EXPECT_EQ(ERROR_SUCCESS, | 79 EXPECT_EQ(ERROR_SUCCESS, |
78 ::RegOverridePredefKey(overridden_, temp_key.Handle())); | 80 ::RegOverridePredefKey(overridden_, temp_key.Handle())); |
79 | 81 |
80 EXPECT_EQ(ERROR_SUCCESS, | 82 EXPECT_EQ(ERROR_SUCCESS, |
81 clients_.Create(overridden_, dist_->GetVersionKey().c_str(), | 83 clients_.Create(overridden_, dist_->GetVersionKey().c_str(), |
82 KEY_ALL_ACCESS)); | 84 KEY_ALL_ACCESS)); |
83 EXPECT_EQ(ERROR_SUCCESS, | 85 EXPECT_EQ(ERROR_SUCCESS, |
84 client_state_.Create(overridden_, dist_->GetStateKey().c_str(), | 86 client_state_.Create(overridden_, dist_->GetStateKey().c_str(), |
85 KEY_ALL_ACCESS)); | 87 KEY_ALL_ACCESS)); |
86 } | 88 } |
87 | 89 |
88 void ProductStateTest::TearDown() { | 90 void ProductStateTest::TearDown() { |
89 // Done with the keys. | 91 // Done with the keys. |
90 client_state_.Close(); | 92 client_state_.Close(); |
91 clients_.Close(); | 93 clients_.Close(); |
92 EXPECT_EQ(ERROR_SUCCESS, ::RegOverridePredefKey(overridden_, NULL)); | 94 EXPECT_EQ(ERROR_SUCCESS, ::RegOverridePredefKey(overridden_, NULL)); |
93 overridden_ = NULL; | 95 overridden_ = NULL; |
94 system_install_ = false; | 96 system_install_ = false; |
95 | 97 |
96 // Shotgun approach to clearing out data we may have written. | 98 // Shotgun approach to clearing out data we may have written. |
97 TempRegKeyOverride::DeleteAllTempKeys(); | 99 RegistryOverrideManager::DeleteAllTempKeys(); |
98 | 100 |
99 testing::Test::TearDown(); | 101 testing::Test::TearDown(); |
100 } | 102 } |
101 | 103 |
102 void ProductStateTest::MinimallyInstallProduct(const wchar_t* version) { | 104 void ProductStateTest::MinimallyInstallProduct(const wchar_t* version) { |
103 EXPECT_EQ(ERROR_SUCCESS, | 105 EXPECT_EQ(ERROR_SUCCESS, |
104 clients_.WriteValue(google_update::kRegVersionField, version)); | 106 clients_.WriteValue(google_update::kRegVersionField, version)); |
105 } | 107 } |
106 | 108 |
107 void ProductStateTest::ApplyUninstallCommand(const wchar_t* exe_path, | 109 void ProductStateTest::ApplyUninstallCommand(const wchar_t* exe_path, |
(...skipping 309 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
417 | 419 |
418 // Uninstall command with --multi-install is multi install. | 420 // Uninstall command with --multi-install is multi install. |
419 { | 421 { |
420 ProductState state; | 422 ProductState state; |
421 ApplyUninstallCommand(L"setup.exe", | 423 ApplyUninstallCommand(L"setup.exe", |
422 L"--uninstall --chrome --multi-install"); | 424 L"--uninstall --chrome --multi-install"); |
423 EXPECT_TRUE(state.Initialize(system_install_, dist_)); | 425 EXPECT_TRUE(state.Initialize(system_install_, dist_)); |
424 EXPECT_TRUE(state.is_multi_install()); | 426 EXPECT_TRUE(state.is_multi_install()); |
425 } | 427 } |
426 } | 428 } |
OLD | NEW |