Chromium Code Reviews
|
| OLD | NEW |
|---|---|
| (Empty) | |
| 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. | |
| 4 | |
| 5 #include "base/logging.h" | |
| 6 #include "base/test/test_reg_util_win.h" | |
| 7 #include "testing/gtest/include/gtest/gtest.h" | |
| 8 | |
| 9 namespace reg_util { | |
| 10 | |
| 11 const wchar_t TempRegKeyOverride::kTempTestKeyPath[] = | |
| 12 L"Software\\Chromium\\TempTestKeys"; | |
| 13 | |
| 14 TempRegKeyOverride::TempRegKeyOverride(HKEY override, const wchar_t* temp_name) | |
| 15 : override_(override), temp_name_(temp_name) { | |
|
Paweł Hajdan Jr.
2011/08/18 22:42:47
nit: Each one on its own line please.
Roger Tawa OOO till Jul 10th
2011/08/22 19:24:42
Done.
| |
| 16 DCHECK(temp_name && lstrlenW(temp_name)); | |
|
Paweł Hajdan Jr.
2011/08/18 22:42:47
Split that into two, so it's easier to see what fa
Roger Tawa OOO till Jul 10th
2011/08/22 19:24:42
Done.
| |
| 17 std::wstring key_path(kTempTestKeyPath); | |
| 18 key_path += L"\\" + temp_name_; | |
| 19 EXPECT_EQ(ERROR_SUCCESS, | |
| 20 temp_key_.Create(HKEY_CURRENT_USER, key_path.c_str(), KEY_ALL_ACCESS)); | |
| 21 EXPECT_EQ(ERROR_SUCCESS, | |
| 22 ::RegOverridePredefKey(override_, temp_key_.Handle())); | |
| 23 } | |
| 24 | |
| 25 TempRegKeyOverride::~TempRegKeyOverride() { | |
| 26 ::RegOverridePredefKey(override_, NULL); | |
| 27 // The temp key will be deleted via a call to DeleteAllTempKeys(). | |
|
Paweł Hajdan Jr.
2011/08/18 22:42:47
This relies on the caller calling DeleteAllTempKey
Roger Tawa OOO till Jul 10th
2011/08/22 19:24:42
added a scoper.
| |
| 28 } | |
| 29 | |
| 30 // static | |
| 31 void TempRegKeyOverride::DeleteAllTempKeys() { | |
| 32 base::win::RegKey key; | |
| 33 if (key.Open(HKEY_CURRENT_USER, L"", KEY_ALL_ACCESS) == ERROR_SUCCESS) { | |
| 34 key.DeleteKey(kTempTestKeyPath); | |
| 35 } | |
| 36 } | |
| 37 | |
| 38 } // namespace reg_util | |
| OLD | NEW |