blob: b71831f2bf3c2f78bea939ce6d1c1ad8a47bbfeb (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
// Copyright (c) 2011 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef BASE_TEST_TEST_REG_UTIL_H_
#define BASE_TEST_TEST_REG_UTIL_H_
// Registry utility functions used only by tests.
#include <string>
#include <vector>
#include "base/basictypes.h"
#include "base/win/registry.h"
namespace registry_util {
// Allows a test to easily override registry hives so that it can start from a
// known good state, or make sure to not leave any side effects once the test
// completes.
class RegistryOverrideManager {
public:
// All overridden hives will be descendents of this registry path under the
// main HKCU hive.
static const wchar_t kTempTestKeyPath[];
RegistryOverrideManager();
~RegistryOverrideManager();
// Override the given registry hive using a temporary key named by temp_name
// under the temporary test key path.
void OverrideRegistry(HKEY override, const std::wstring& temp_name);
// Deletes all temporary test keys used by the overrides.
static void DeleteAllTempKeys();
// Removes all overrides and deletes all temporary test keys used by the
// overrides.
void RemoveAllOverrides();
private:
// Keeps track of one override.
class ScopedRegistryKeyOverride {
public:
ScopedRegistryKeyOverride(HKEY override, const std::wstring& temp_name);
~ScopedRegistryKeyOverride();
private:
HKEY override_;
base::win::RegKey temp_key_;
std::wstring temp_name_;
DISALLOW_COPY_AND_ASSIGN(ScopedRegistryKeyOverride);
};
std::vector<ScopedRegistryKeyOverride*> overrides_;
DISALLOW_COPY_AND_ASSIGN(RegistryOverrideManager);
};
} // namespace registry_util
#endif // BASE_TEST_TEST_REG_UTIL_H_
|
