| OLD | NEW |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 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_TEST_UNIT_CHROME_TEST_SUITE_H_ | 5 #ifndef CHROME_TEST_UNIT_CHROME_TEST_SUITE_H_ |
| 6 #define CHROME_TEST_UNIT_CHROME_TEST_SUITE_H_ | 6 #define CHROME_TEST_UNIT_CHROME_TEST_SUITE_H_ |
| 7 #pragma once | 7 #pragma once |
| 8 | 8 |
| 9 #include <string> | 9 #include <string> |
| 10 | 10 |
| 11 #include "build/build_config.h" | 11 #include "build/build_config.h" |
| 12 | 12 |
| 13 #include "app/app_paths.h" | 13 #include "app/app_paths.h" |
| 14 #include "app/resource_bundle.h" | 14 #include "app/resource_bundle.h" |
| 15 #include "base/command_line.h" | |
| 16 #include "base/stats_table.h" | 15 #include "base/stats_table.h" |
| 17 #include "base/file_util.h" | 16 #include "base/file_util.h" |
| 18 #include "base/path_service.h" | 17 #include "base/path_service.h" |
| 19 #include "base/process_util.h" | |
| 20 #include "base/ref_counted.h" | 18 #include "base/ref_counted.h" |
| 21 #include "base/scoped_nsautorelease_pool.h" | 19 #include "base/scoped_nsautorelease_pool.h" |
| 22 #include "base/test/test_suite.h" | 20 #include "base/test/test_suite.h" |
| 23 #include "base/utf_string_conversions.h" | 21 #include "base/utf_string_conversions.h" |
| 24 #include "chrome/app/scoped_ole_initializer.h" | 22 #include "chrome/app/scoped_ole_initializer.h" |
| 25 #include "chrome/browser/browser_process.h" | 23 #include "chrome/browser/browser_process.h" |
| 26 #include "chrome/common/chrome_constants.h" | 24 #include "chrome/common/chrome_constants.h" |
| 27 #include "chrome/common/chrome_paths.h" | 25 #include "chrome/common/chrome_paths.h" |
| 28 #include "chrome/common/chrome_switches.h" | 26 #include "chrome/common/chrome_switches.h" |
| 29 #include "chrome/common/url_constants.h" | 27 #include "chrome/common/url_constants.h" |
| 30 #include "chrome/test/testing_browser_process.h" | 28 #include "chrome/test/testing_browser_process.h" |
| 31 #include "net/base/mock_host_resolver.h" | 29 #include "net/base/mock_host_resolver.h" |
| 32 #include "net/base/net_util.h" | 30 #include "net/base/net_util.h" |
| 33 #include "testing/gtest/include/gtest/gtest.h" | |
| 34 | 31 |
| 35 #if defined(OS_MACOSX) | 32 #if defined(OS_MACOSX) |
| 36 #include "base/mac_util.h" | 33 #include "base/mac_util.h" |
| 37 #endif | 34 #endif |
| 38 | 35 |
| 39 #if defined(OS_POSIX) | 36 #if defined(OS_POSIX) |
| 40 #include "base/shared_memory.h" | 37 #include "base/shared_memory.h" |
| 41 #endif | 38 #endif |
| 42 | 39 |
| 43 static void RemoveSharedMemoryFile(const std::string& filename) { | 40 static void RemoveSharedMemoryFile(const std::string& filename) { |
| (...skipping 33 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 77 // Make the test fail so it's harder to ignore. | 74 // Make the test fail so it's harder to ignore. |
| 78 // If you really need to make real DNS query, use | 75 // If you really need to make real DNS query, use |
| 79 // net::RuleBasedHostResolverProc and its AllowDirectLookup method. | 76 // net::RuleBasedHostResolverProc and its AllowDirectLookup method. |
| 80 EXPECT_TRUE(local) << "Making external DNS lookup of " << host; | 77 EXPECT_TRUE(local) << "Making external DNS lookup of " << host; |
| 81 | 78 |
| 82 return ResolveUsingPrevious(host, address_family, host_resolver_flags, | 79 return ResolveUsingPrevious(host, address_family, host_resolver_flags, |
| 83 addrlist, os_error); | 80 addrlist, os_error); |
| 84 } | 81 } |
| 85 }; | 82 }; |
| 86 | 83 |
| 87 class ChromeTestSuite : public base::TestSuite { | 84 class ChromeTestSuite : public TestSuite { |
| 88 public: | 85 public: |
| 89 ChromeTestSuite(int argc, char** argv) | 86 ChromeTestSuite(int argc, char** argv) |
| 90 : base::TestSuite(argc, argv), | 87 : TestSuite(argc, argv), |
| 91 stats_table_(NULL), | 88 stats_table_(NULL), |
| 92 created_user_data_dir_(false) { | 89 created_user_data_dir_(false) { |
| 93 } | 90 } |
| 94 | 91 |
| 95 protected: | 92 protected: |
| 96 | 93 |
| 97 virtual void Initialize() { | 94 virtual void Initialize() { |
| 98 base::ScopedNSAutoreleasePool autorelease_pool; | 95 base::ScopedNSAutoreleasePool autorelease_pool; |
| 99 | 96 |
| 100 base::TestSuite::Initialize(); | 97 TestSuite::Initialize(); |
| 101 | 98 |
| 102 chrome::RegisterChromeSchemes(); | 99 chrome::RegisterChromeSchemes(); |
| 103 host_resolver_proc_ = new WarningHostResolverProc(); | 100 host_resolver_proc_ = new WarningHostResolverProc(); |
| 104 scoped_host_resolver_proc_.Init(host_resolver_proc_.get()); | 101 scoped_host_resolver_proc_.Init(host_resolver_proc_.get()); |
| 105 | 102 |
| 106 chrome::RegisterPathProvider(); | 103 chrome::RegisterPathProvider(); |
| 107 app::RegisterPathProvider(); | 104 app::RegisterPathProvider(); |
| 108 g_browser_process = new TestingBrowserProcess; | 105 g_browser_process = new TestingBrowserProcess; |
| 109 | 106 |
| 110 // Notice a user data override, and otherwise default to using a custom | 107 // Notice a user data override, and otherwise default to using a custom |
| (...skipping 57 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 168 // Delete the test_user_data dir recursively | 165 // Delete the test_user_data dir recursively |
| 169 // NOTE: user_data_dir will be deleted only if it was automatically | 166 // NOTE: user_data_dir will be deleted only if it was automatically |
| 170 // created. | 167 // created. |
| 171 FilePath user_data_dir; | 168 FilePath user_data_dir; |
| 172 if (created_user_data_dir_ && | 169 if (created_user_data_dir_ && |
| 173 PathService::Get(chrome::DIR_USER_DATA, &user_data_dir) && | 170 PathService::Get(chrome::DIR_USER_DATA, &user_data_dir) && |
| 174 !user_data_dir.empty()) { | 171 !user_data_dir.empty()) { |
| 175 file_util::Delete(user_data_dir, true); | 172 file_util::Delete(user_data_dir, true); |
| 176 file_util::Delete(user_data_dir.DirName(), false); | 173 file_util::Delete(user_data_dir.DirName(), false); |
| 177 } | 174 } |
| 178 base::TestSuite::Shutdown(); | 175 TestSuite::Shutdown(); |
| 179 } | 176 } |
| 180 | 177 |
| 181 void SetBrowserDirectory(const FilePath& browser_dir) { | 178 void SetBrowserDirectory(const FilePath& browser_dir) { |
| 182 browser_dir_ = browser_dir; | 179 browser_dir_ = browser_dir; |
| 183 } | 180 } |
| 184 | 181 |
| 185 StatsTable* stats_table_; | 182 StatsTable* stats_table_; |
| 186 // The name used for the stats file so it can be cleaned up on posix during | 183 // The name used for the stats file so it can be cleaned up on posix during |
| 187 // test shutdown. | 184 // test shutdown. |
| 188 std::string stats_filename_; | 185 std::string stats_filename_; |
| 189 | 186 |
| 190 // Alternative path to browser binaries. | 187 // Alternative path to browser binaries. |
| 191 FilePath browser_dir_; | 188 FilePath browser_dir_; |
| 192 | 189 |
| 193 ScopedOleInitializer ole_initializer_; | 190 ScopedOleInitializer ole_initializer_; |
| 194 scoped_refptr<WarningHostResolverProc> host_resolver_proc_; | 191 scoped_refptr<WarningHostResolverProc> host_resolver_proc_; |
| 195 net::ScopedDefaultHostResolverProc scoped_host_resolver_proc_; | 192 net::ScopedDefaultHostResolverProc scoped_host_resolver_proc_; |
| 196 | 193 |
| 197 // Flag indicating whether user_data_dir was automatically created or not. | 194 // Flag indicating whether user_data_dir was automatically created or not. |
| 198 bool created_user_data_dir_; | 195 bool created_user_data_dir_; |
| 199 }; | 196 }; |
| 200 | 197 |
| 201 #endif // CHROME_TEST_UNIT_CHROME_TEST_SUITE_H_ | 198 #endif // CHROME_TEST_UNIT_CHROME_TEST_SUITE_H_ |
| OLD | NEW |