| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef CHROME_FRAME_TEST_CHROME_FRAME_TEST_UTILS_H_ | 5 #ifndef CHROME_FRAME_TEST_CHROME_FRAME_TEST_UTILS_H_ |
| 6 #define CHROME_FRAME_TEST_CHROME_FRAME_TEST_UTILS_H_ | 6 #define CHROME_FRAME_TEST_CHROME_FRAME_TEST_UTILS_H_ |
| 7 | 7 |
| 8 #include <windows.h> | 8 #include <windows.h> |
| 9 | 9 |
| 10 #include <atlbase.h> | 10 #include <atlbase.h> |
| 11 #include <atlwin.h> | 11 #include <atlwin.h> |
| 12 | 12 |
| 13 #include <string> | 13 #include <string> |
| 14 | 14 |
| 15 #include "base/basictypes.h" | 15 #include "base/basictypes.h" |
| 16 #include "base/compiler_specific.h" | 16 #include "base/compiler_specific.h" |
| 17 #include "base/memory/scoped_ptr.h" | 17 #include "base/memory/scoped_ptr.h" |
| 18 #include "base/message_loop.h" | 18 #include "base/message_loop.h" |
| 19 #include "base/process_util.h" | 19 #include "base/process_util.h" |
| 20 #include "base/test/test_reg_util_win.h" |
| 20 #include "base/win/registry.h" | 21 #include "base/win/registry.h" |
| 21 #include "base/win/scoped_comptr.h" | 22 #include "base/win/scoped_comptr.h" |
| 22 | 23 |
| 23 #include "chrome_frame/test/simulate_input.h" | 24 #include "chrome_frame/test/simulate_input.h" |
| 24 #include "chrome_frame/test_utils.h" | 25 #include "chrome_frame/test_utils.h" |
| 25 #include "chrome_frame/utils.h" | 26 #include "chrome_frame/utils.h" |
| 26 | 27 |
| 27 // Include without path to make GYP build see it. | 28 // Include without path to make GYP build see it. |
| 28 #include "chrome_tab.h" // NOLINT | 29 #include "chrome_tab.h" // NOLINT |
| 29 | 30 |
| (...skipping 248 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 278 class CloseIeAtEndOfScope { | 279 class CloseIeAtEndOfScope { |
| 279 public: | 280 public: |
| 280 CloseIeAtEndOfScope() {} | 281 CloseIeAtEndOfScope() {} |
| 281 ~CloseIeAtEndOfScope(); | 282 ~CloseIeAtEndOfScope(); |
| 282 }; | 283 }; |
| 283 | 284 |
| 284 // Starts the Chrome crash service which enables us to gather crash dumps | 285 // Starts the Chrome crash service which enables us to gather crash dumps |
| 285 // during test runs. | 286 // during test runs. |
| 286 base::ProcessHandle StartCrashService(); | 287 base::ProcessHandle StartCrashService(); |
| 287 | 288 |
| 288 class TempRegKeyOverride { | |
| 289 public: | |
| 290 static const wchar_t kTempTestKeyPath[]; | |
| 291 | |
| 292 TempRegKeyOverride(HKEY override, const wchar_t* temp_name); | |
| 293 ~TempRegKeyOverride(); | |
| 294 | |
| 295 static void DeleteAllTempKeys(); | |
| 296 | |
| 297 protected: | |
| 298 HKEY override_; | |
| 299 base::win::RegKey temp_key_; | |
| 300 std::wstring temp_name_; | |
| 301 }; | |
| 302 | |
| 303 // Used in tests where we reference the registry and don't want to run into | 289 // Used in tests where we reference the registry and don't want to run into |
| 304 // problems where existing registry settings might conflict with the | 290 // problems where existing registry settings might conflict with the |
| 305 // expectations of the test. | 291 // expectations of the test. |
| 306 class ScopedVirtualizeHklmAndHkcu { | 292 class ScopedVirtualizeHklmAndHkcu { |
| 307 public: | 293 public: |
| 308 ScopedVirtualizeHklmAndHkcu(); | 294 ScopedVirtualizeHklmAndHkcu(); |
| 309 ~ScopedVirtualizeHklmAndHkcu(); | 295 ~ScopedVirtualizeHklmAndHkcu(); |
| 310 | 296 |
| 297 // Removes all overrides and deletes all temporary test keys used by the |
| 298 // overrides. |
| 299 void RemoveAllOverrides(); |
| 300 |
| 311 protected: | 301 protected: |
| 312 scoped_ptr<TempRegKeyOverride> hklm_; | 302 registry_util::RegistryOverrideManager override_manager_; |
| 313 scoped_ptr<TempRegKeyOverride> hkcu_; | |
| 314 }; | 303 }; |
| 315 | 304 |
| 316 // Attempts to kill all the processes on the current machine that were launched | 305 // Attempts to kill all the processes on the current machine that were launched |
| 317 // from the given executable name, ending them with the given exit code. If | 306 // from the given executable name, ending them with the given exit code. If |
| 318 // filter is non-null, then only processes selected by the filter are killed. | 307 // filter is non-null, then only processes selected by the filter are killed. |
| 319 // Returns true if all processes were able to be killed off, false if at least | 308 // Returns true if all processes were able to be killed off, false if at least |
| 320 // one couldn't be killed. | 309 // one couldn't be killed. |
| 321 // NOTE: This function is based on the base\process_util.h helper function | 310 // NOTE: This function is based on the base\process_util.h helper function |
| 322 // KillProcesses. Takes in the wait flag as a parameter. | 311 // KillProcesses. Takes in the wait flag as a parameter. |
| 323 bool KillProcesses(const std::wstring& executable_name, int exit_code, | 312 bool KillProcesses(const std::wstring& executable_name, int exit_code, |
| 324 bool wait); | 313 bool wait); |
| 325 | 314 |
| 326 // Returns the type of test bed, PER_USER or SYSTEM_LEVEL. | 315 // Returns the type of test bed, PER_USER or SYSTEM_LEVEL. |
| 327 ScopedChromeFrameRegistrar::RegistrationType GetTestBedType(); | 316 ScopedChromeFrameRegistrar::RegistrationType GetTestBedType(); |
| 328 | 317 |
| 329 // Clears IE8 session restore history. | 318 // Clears IE8 session restore history. |
| 330 void ClearIESessionHistory(); | 319 void ClearIESessionHistory(); |
| 331 | 320 |
| 332 } // namespace chrome_frame_test | 321 } // namespace chrome_frame_test |
| 333 | 322 |
| 334 // TODO(tommi): This is a temporary workaround while we're getting our | 323 // TODO(tommi): This is a temporary workaround while we're getting our |
| 335 // Singleton story straight. Ideally each test should clear up any singletons | 324 // Singleton story straight. Ideally each test should clear up any singletons |
| 336 // it might have created, but test cases do not implicitly have their own | 325 // it might have created, but test cases do not implicitly have their own |
| 337 // AtExitManager, so we have this workaround method for tests that depend on | 326 // AtExitManager, so we have this workaround method for tests that depend on |
| 338 // "fresh" singletons. The implementation is in chrome_frame_unittest_main.cc. | 327 // "fresh" singletons. The implementation is in chrome_frame_unittest_main.cc. |
| 339 void DeleteAllSingletons(); | 328 void DeleteAllSingletons(); |
| 340 | 329 |
| 341 #endif // CHROME_FRAME_TEST_CHROME_FRAME_TEST_UTILS_H_ | 330 #endif // CHROME_FRAME_TEST_CHROME_FRAME_TEST_UTILS_H_ |
| OLD | NEW |