| Index: chrome_frame/test/chrome_frame_test_utils.cc
|
| ===================================================================
|
| --- chrome_frame/test/chrome_frame_test_utils.cc (revision 98031)
|
| +++ chrome_frame/test/chrome_frame_test_utils.cc (working copy)
|
| @@ -51,9 +51,6 @@
|
| const int kChromeFrameLongNavigationTimeoutInSeconds = 10;
|
| const int kChromeFrameVeryLongNavigationTimeoutInSeconds = 30;
|
|
|
| -const wchar_t TempRegKeyOverride::kTempTestKeyPath[] =
|
| - L"Software\\Chromium\\TempTestKeys";
|
| -
|
| // Callback function for EnumThreadWindows.
|
| BOOL CALLBACK CloseWindowsThreadCallback(HWND hwnd, LPARAM param) {
|
| int& count = *reinterpret_cast<int*>(param);
|
| @@ -616,42 +613,18 @@
|
| }
|
| }
|
|
|
| -TempRegKeyOverride::TempRegKeyOverride(HKEY override, const wchar_t* temp_name)
|
| - : override_(override), temp_name_(temp_name) {
|
| - DCHECK(temp_name && lstrlenW(temp_name));
|
| - std::wstring key_path(kTempTestKeyPath);
|
| - key_path += L"\\" + temp_name_;
|
| - EXPECT_EQ(ERROR_SUCCESS, temp_key_.Create(HKEY_CURRENT_USER, key_path.c_str(),
|
| - KEY_ALL_ACCESS));
|
| - EXPECT_EQ(ERROR_SUCCESS,
|
| - ::RegOverridePredefKey(override_, temp_key_.Handle()));
|
| -}
|
| -
|
| -TempRegKeyOverride::~TempRegKeyOverride() {
|
| - ::RegOverridePredefKey(override_, NULL);
|
| - // The temp key will be deleted via a call to DeleteAllTempKeys().
|
| -}
|
| -
|
| -// static
|
| -void TempRegKeyOverride::DeleteAllTempKeys() {
|
| - base::win::RegKey key;
|
| - if (key.Open(HKEY_CURRENT_USER, L"", KEY_ALL_ACCESS) == ERROR_SUCCESS) {
|
| - key.DeleteKey(kTempTestKeyPath);
|
| - }
|
| -}
|
| -
|
| ScopedVirtualizeHklmAndHkcu::ScopedVirtualizeHklmAndHkcu() {
|
| - TempRegKeyOverride::DeleteAllTempKeys();
|
| - hklm_.reset(new TempRegKeyOverride(HKEY_LOCAL_MACHINE, L"hklm_fake"));
|
| - hkcu_.reset(new TempRegKeyOverride(HKEY_CURRENT_USER, L"hkcu_fake"));
|
| + override_manager_.OverrideRegistry(HKEY_LOCAL_MACHINE, L"hklm_fake");
|
| + override_manager_.OverrideRegistry(HKEY_CURRENT_USER, L"hkcu_fake");
|
| }
|
|
|
| ScopedVirtualizeHklmAndHkcu::~ScopedVirtualizeHklmAndHkcu() {
|
| - hkcu_.reset(NULL);
|
| - hklm_.reset(NULL);
|
| - TempRegKeyOverride::DeleteAllTempKeys();
|
| }
|
|
|
| +void ScopedVirtualizeHklmAndHkcu::RemoveAllOverrides() {
|
| + override_manager_.RemoveAllOverrides();
|
| +}
|
| +
|
| bool KillProcesses(const std::wstring& executable_name, int exit_code,
|
| bool wait) {
|
| bool result = true;
|
|
|