|
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 |