Index: chrome_frame/test/test_scrubber.cc |
diff --git a/chrome_frame/test/test_scrubber.cc b/chrome_frame/test/test_scrubber.cc |
deleted file mode 100644 |
index 558a6fa2ca3b9460415b20fd3eb3fc89486d1db7..0000000000000000000000000000000000000000 |
--- a/chrome_frame/test/test_scrubber.cc |
+++ /dev/null |
@@ -1,147 +0,0 @@ |
-// Copyright (c) 2012 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. |
- |
-// The test scrubber uses a Google Test event listener to trigger scrubs. |
-// Scrubs are performed at the start of the test program and at the conclusion |
-// of each test. |
- |
-#include "chrome_frame/test/test_scrubber.h" |
- |
-#include <windows.h> |
- |
-#include "base/compiler_specific.h" |
-#include "base/file_util.h" |
-#include "base/files/file_path.h" |
-#include "base/lazy_instance.h" |
-#include "base/logging.h" |
-#include "base/process/kill.h" |
-#include "base/strings/string16.h" |
-#include "base/strings/utf_string_conversions.h" |
-#include "base/test/test_process_killer_win.h" |
-#include "base/win/registry.h" |
-#include "base/win/scoped_handle.h" |
-#include "chrome/common/chrome_constants.h" |
-#include "chrome/common/chrome_switches.h" |
-#include "chrome_frame/test/chrome_frame_test_utils.h" |
-#include "testing/gtest/include/gtest/gtest.h" |
- |
-namespace chrome_frame_test { |
- |
-namespace { |
- |
-class TestScrubber { |
- public: |
- TestScrubber(); |
- ~TestScrubber(); |
- |
- // Initializes the instance, causing it to provide its services for all |
- // subsequent tests. |
- void Initialize(testing::UnitTest* unit_test); |
- |
- // Sets the user data directory for the current test. |
- void set_data_directory_override(const base::StringPiece16& user_data_dir) { |
- user_data_dir.as_string().swap(data_directory_override_); |
- } |
- |
- // Kills all instances of IE and all instances of chrome.exe that have |
- // --chrome-frame on their command-lines. Also deletes the user data |
- // directory used by the test. Ordinarily, this is the default Chrome Frame |
- // user data directory for IE. Individual tests that use a specific directory |
- // can override this via |set_data_directory_override|. |
- void CleanUpFromTestRun(); |
- |
- private: |
- // A listener that calls back to the scrubber at the start of the test program |
- // and at the end of each test. |
- class EventListener : public testing::EmptyTestEventListener { |
- public: |
- explicit EventListener(TestScrubber* scrubber) : scrubber_(scrubber) { |
- } |
- |
- virtual void OnTestProgramStart(const testing::UnitTest&) OVERRIDE { |
- scrubber_->CleanUpFromTestRun(); |
- } |
- |
- virtual void OnTestEnd(const testing::TestInfo&) OVERRIDE { |
- scrubber_->CleanUpFromTestRun(); |
- } |
- |
- private: |
- TestScrubber* scrubber_; |
- DISALLOW_COPY_AND_ASSIGN(EventListener); |
- }; |
- |
- bool is_initialized() const { return !default_data_directory_.empty(); } |
- |
- base::string16 default_data_directory_; |
- base::string16 data_directory_override_; |
- |
- DISALLOW_COPY_AND_ASSIGN(TestScrubber); |
-}; |
- |
-TestScrubber::TestScrubber() { |
-} |
- |
-TestScrubber::~TestScrubber() { |
-} |
- |
-void TestScrubber::Initialize(testing::UnitTest* unit_test) { |
- DCHECK(!is_initialized()); |
- |
- default_data_directory_ = GetProfilePathForIE().value(); |
- data_directory_override_.clear(); |
- unit_test->listeners().Append(new EventListener(this)); |
-} |
- |
-void TestScrubber::CleanUpFromTestRun() { |
- // Kill all iexplore.exe and ieuser.exe processes. |
- base::KillProcesses(chrome_frame_test::kIEImageName, 0, NULL); |
- base::KillProcesses(chrome_frame_test::kIEBrokerImageName, 0, NULL); |
- |
- // Kill all chrome_launcher processes trying to launch Chrome. |
- base::KillProcesses(chrome_frame_test::kChromeLauncher, 0, NULL); |
- |
- // Kill all chrome.exe processes with --chrome-frame. |
- base::KillAllNamedProcessesWithArgument( |
- chrome::kBrowserProcessExecutableName, |
- base::ASCIIToWide(switches::kChromeFrame)); |
- |
- // Delete the user data directory. |
- base::FilePath data_directory(data_directory_override_.empty() ? |
- default_data_directory_ : |
- data_directory_override_); |
- |
- VLOG_IF(1, base::PathExists(data_directory)) |
- << __FUNCTION__ << " deleting user data directory " |
- << data_directory.value(); |
- bool deleted = base::DeleteFile(data_directory, true); |
- LOG_IF(ERROR, !deleted) |
- << "Failed to delete user data directory directory " |
- << data_directory.value(); |
- |
- // Clear the overridden data directory for the next test. |
- data_directory_override_.clear(); |
-} |
- |
-base::LazyInstance<TestScrubber> g_scrubber = LAZY_INSTANCE_INITIALIZER; |
- |
-} // namespace |
- |
-void InstallTestScrubber(testing::UnitTest* unit_test) { |
- // Must be called before running any tests. |
- DCHECK(unit_test); |
- DCHECK(!unit_test->current_test_case()); |
- |
- g_scrubber.Get().Initialize(unit_test); |
-} |
- |
-void OverrideDataDirectoryForThisTest( |
- const base::StringPiece16& user_data_dir) { |
- // Must be called within the context of a test. |
- DCHECK(testing::UnitTest::GetInstance()->current_test_info()); |
- |
- g_scrubber.Get().set_data_directory_override(user_data_dir); |
-} |
- |
-} // namespace chrome_frame_test |