Index: ios/chrome/test/ios_chrome_unit_test_suite.cc |
diff --git a/ios/chrome/test/ios_chrome_unit_test_suite.cc b/ios/chrome/test/ios_chrome_unit_test_suite.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..206553614fe2431909d32699cbc3de451d228419 |
--- /dev/null |
+++ b/ios/chrome/test/ios_chrome_unit_test_suite.cc |
@@ -0,0 +1,67 @@ |
+// Copyright 2015 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. |
+ |
+#include "ios/chrome/test/ios_chrome_unit_test_suite.h" |
+ |
+#include "ios/public/test/test_chrome_browser_provider.h" |
+#include "ios/public/test/test_chrome_provider_initializer.h" |
+#include "ios/web/public/web_client.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+#include "ui/base/ui_base_paths.h" |
+#include "url/url_util.h" |
+ |
+namespace { |
+ |
+class IOSChromeUnitTestSuiteInitializer |
+ : public testing::EmptyTestEventListener { |
+ public: |
+ IOSChromeUnitTestSuiteInitializer() {} |
+ virtual ~IOSChromeUnitTestSuiteInitializer() {} |
sdefresne
2015/01/30 15:44:08
~IOSChromeUnitTestSuiteInitializer() {}
droger
2015/01/30 15:57:22
Done, with override also.
|
+ |
+ void OnTestStart(const testing::TestInfo& test_info) override { |
+ web_client_.reset(new web::WebClient()); |
+ web::SetWebClient(web_client_.get()); |
+ test_ios_chrome_provider_initializer_.reset( |
+ new ios::TestChromeProviderInitializer()); |
+ } |
+ |
+ void OnTestEnd(const testing::TestInfo& test_info) override { |
+ web_client_.reset(); |
+ web::SetWebClient(NULL); |
+ test_ios_chrome_provider_initializer_.reset(); |
+ } |
+ |
+ private: |
+ scoped_ptr<web::WebClient> web_client_; |
+ scoped_ptr<ios::TestChromeProviderInitializer> |
+ test_ios_chrome_provider_initializer_; |
+ DISALLOW_COPY_AND_ASSIGN(IOSChromeUnitTestSuiteInitializer); |
+}; |
+ |
+} // namespace |
+ |
+IOSChromeUnitTestSuite::IOSChromeUnitTestSuite(int argc, char** argv) |
+ : base::TestSuite(argc, argv) { |
+} |
+ |
+IOSChromeUnitTestSuite::~IOSChromeUnitTestSuite() { |
+} |
+ |
+void IOSChromeUnitTestSuite::Initialize() { |
+ // Add an additional listener to do the extra initialization for unit tests. |
+ // It will be started before the base class listeners and ended after the |
+ // base class listeners. |
+ testing::TestEventListeners& listeners = |
+ testing::UnitTest::GetInstance()->listeners(); |
+ listeners.Append(new IOSChromeUnitTestSuiteInitializer); |
+ |
+ ui::RegisterPathProvider(); |
+ |
+ { |
+ ios::TestChromeBrowserProvider provider; |
+ url::AddStandardScheme(provider.GetChromeUIScheme()); |
+ } |
+ |
+ base::TestSuite::Initialize(); |
+} |