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
3 // found in the LICENSE file.
5 #include "base/test/test_reg_util_win.h"
7 #include "base/logging.h"
8 #include "testing/gtest/include/gtest/gtest.h"
10 namespace registry_util {
12 const wchar_t RegistryOverrideManager::kTempTestKeyPath[] =
13 L"Software\\Chromium\\TempTestKeys";
15 RegistryOverrideManager::ScopedRegistryKeyOverride::ScopedRegistryKeyOverride(
17 const std::wstring& temp_name)
18 : override_(override),
19 temp_name_(temp_name) {
20 DCHECK(!temp_name_.empty());
21 std::wstring key_path(RegistryOverrideManager::kTempTestKeyPath);
22 key_path += L"\\" + temp_name_;
23 EXPECT_EQ(ERROR_SUCCESS,
24 temp_key_.Create(HKEY_CURRENT_USER, key_path.c_str(), KEY_ALL_ACCESS));
25 EXPECT_EQ(ERROR_SUCCESS,
26 ::RegOverridePredefKey(override_, temp_key_.Handle()));
29 RegistryOverrideManager::
30 ScopedRegistryKeyOverride::~ScopedRegistryKeyOverride() {
31 ::RegOverridePredefKey(override_, NULL);
32 // The temp key will be deleted via a call to DeleteAllTempKeys().
35 RegistryOverrideManager::RegistryOverrideManager() {
39 RegistryOverrideManager::~RegistryOverrideManager() {
43 void RegistryOverrideManager::OverrideRegistry(HKEY override,
44 const std::wstring& temp_name) {
45 overrides_.push_back(new ScopedRegistryKeyOverride(override, temp_name));
48 void RegistryOverrideManager::RemoveAllOverrides() {
49 while (!overrides_.empty()) {
50 delete overrides_.back();
51 overrides_.pop_back();
58 void RegistryOverrideManager::DeleteAllTempKeys() {
59 base::win::RegKey key;
60 if (key.Open(HKEY_CURRENT_USER, L"", KEY_ALL_ACCESS) == ERROR_SUCCESS) {
61 key.DeleteKey(kTempTestKeyPath);
65 } // namespace registry_util