OLD | NEW |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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 #include <algorithm> | 5 #include <algorithm> |
6 | 6 |
7 #include "base/at_exit.h" | 7 #include "base/at_exit.h" |
8 #include "base/command_line.h" | 8 #include "base/command_line.h" |
9 #include "base/debug/debugger.h" | 9 #include "base/debug/debugger.h" |
10 #include "base/memory/ptr_util.h" | 10 #include "base/memory/ptr_util.h" |
11 #include "base/process/launch.h" | 11 #include "base/process/launch.h" |
12 #include "base/sys_info.h" | 12 #include "base/sys_info.h" |
13 #include "chrome/test/base/chrome_test_launcher.h" | 13 #include "chrome/test/base/chrome_test_launcher.h" |
14 #include "chrome/test/base/chrome_test_suite.h" | 14 #include "chrome/test/base/chrome_test_suite.h" |
15 #include "chrome/test/base/mojo_test_connector.h" | 15 #include "chrome/test/base/mojo_test_connector.h" |
16 #include "content/public/common/mojo_shell_connection.h" | 16 #include "content/public/common/mojo_shell_connection.h" |
17 #include "content/public/test/test_launcher.h" | 17 #include "content/public/test/test_launcher.h" |
18 #include "services/shell/public/cpp/connector.h" | 18 #include "services/shell/public/cpp/connector.h" |
19 #include "services/shell/public/cpp/shell_client.h" | 19 #include "services/shell/public/cpp/shell_client.h" |
20 #include "services/shell/public/cpp/shell_connection.h" | 20 #include "services/shell/public/cpp/shell_connection.h" |
21 #include "services/shell/runner/common/switches.h" | 21 #include "services/shell/runner/common/switches.h" |
22 #include "services/shell/runner/host/child_process.h" | 22 #include "services/shell/runner/host/child_process.h" |
23 #include "services/shell/runner/init.h" | 23 #include "services/shell/runner/init.h" |
24 | 24 |
25 namespace { | 25 namespace { |
26 | 26 |
27 void ConnectToDefaultApps(mojo::Connector* connector) { | 27 void ConnectToDefaultApps(shell::Connector* connector) { |
28 connector->Connect("mojo:mash_session"); | 28 connector->Connect("mojo:mash_session"); |
29 } | 29 } |
30 | 30 |
31 class MashTestSuite : public ChromeTestSuite { | 31 class MashTestSuite : public ChromeTestSuite { |
32 public: | 32 public: |
33 MashTestSuite(int argc, char** argv) : ChromeTestSuite(argc, argv) {} | 33 MashTestSuite(int argc, char** argv) : ChromeTestSuite(argc, argv) {} |
34 | 34 |
35 void SetMojoTestConnector(std::unique_ptr<MojoTestConnector> connector) { | 35 void SetMojoTestConnector(std::unique_ptr<MojoTestConnector> connector) { |
36 mojo_test_connector_ = std::move(connector); | 36 mojo_test_connector_ = std::move(connector); |
37 } | 37 } |
(...skipping 35 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
73 test_suite_.reset(new MashTestSuite(argc, argv)); | 73 test_suite_.reset(new MashTestSuite(argc, argv)); |
74 const int result = test_suite_->Run(); | 74 const int result = test_suite_->Run(); |
75 test_suite_.reset(); | 75 test_suite_.reset(); |
76 return result; | 76 return result; |
77 } | 77 } |
78 std::unique_ptr<content::TestState> PreRunTest( | 78 std::unique_ptr<content::TestState> PreRunTest( |
79 base::CommandLine* command_line, | 79 base::CommandLine* command_line, |
80 base::TestLauncher::LaunchOptions* test_launch_options) override { | 80 base::TestLauncher::LaunchOptions* test_launch_options) override { |
81 if (!mojo_test_connector_) { | 81 if (!mojo_test_connector_) { |
82 mojo_test_connector_.reset(new MojoTestConnector); | 82 mojo_test_connector_.reset(new MojoTestConnector); |
83 shell_client_.reset(new mojo::ShellClient); | 83 shell_client_.reset(new shell::ShellClient); |
84 shell_connection_.reset(new mojo::ShellConnection( | 84 shell_connection_.reset(new shell::ShellConnection( |
85 shell_client_.get(), mojo_test_connector_->Init())); | 85 shell_client_.get(), mojo_test_connector_->Init())); |
86 ConnectToDefaultApps(shell_connection_->connector()); | 86 ConnectToDefaultApps(shell_connection_->connector()); |
87 } | 87 } |
88 return mojo_test_connector_->PrepareForTest(command_line, | 88 return mojo_test_connector_->PrepareForTest(command_line, |
89 test_launch_options); | 89 test_launch_options); |
90 } | 90 } |
91 void OnDoneRunningTests() override { | 91 void OnDoneRunningTests() override { |
92 // We have to shutdown this state here, while an AtExitManager is still | 92 // We have to shutdown this state here, while an AtExitManager is still |
93 // valid. | 93 // valid. |
94 shell_connection_.reset(); | 94 shell_connection_.reset(); |
95 shell_client_.reset(); | 95 shell_client_.reset(); |
96 mojo_test_connector_.reset(); | 96 mojo_test_connector_.reset(); |
97 } | 97 } |
98 | 98 |
99 std::unique_ptr<MashTestSuite> test_suite_; | 99 std::unique_ptr<MashTestSuite> test_suite_; |
100 std::unique_ptr<MojoTestConnector> mojo_test_connector_; | 100 std::unique_ptr<MojoTestConnector> mojo_test_connector_; |
101 std::unique_ptr<mojo::ShellClient> shell_client_; | 101 std::unique_ptr<shell::ShellClient> shell_client_; |
102 std::unique_ptr<mojo::ShellConnection> shell_connection_; | 102 std::unique_ptr<shell::ShellConnection> shell_connection_; |
103 | 103 |
104 DISALLOW_COPY_AND_ASSIGN(MashTestLauncherDelegate); | 104 DISALLOW_COPY_AND_ASSIGN(MashTestLauncherDelegate); |
105 }; | 105 }; |
106 | 106 |
107 void CreateMojoShellConnection(MashTestLauncherDelegate* delegate) { | 107 void CreateMojoShellConnection(MashTestLauncherDelegate* delegate) { |
108 const bool is_external_shell = true; | 108 const bool is_external_shell = true; |
109 content::MojoShellConnection::Create( | 109 content::MojoShellConnection::Create( |
110 delegate->GetMojoTestConnectorForSingleProcess()->Init(), | 110 delegate->GetMojoTestConnectorForSingleProcess()->Init(), |
111 is_external_shell); | 111 is_external_shell); |
112 ConnectToDefaultApps(content::MojoShellConnection::Get()->GetConnector()); | 112 ConnectToDefaultApps(content::MojoShellConnection::Get()->GetConnector()); |
113 } | 113 } |
114 | 114 |
115 } // namespace | 115 } // namespace |
116 | 116 |
117 bool RunMashBrowserTests(int argc, char** argv, int* exit_code) { | 117 bool RunMashBrowserTests(int argc, char** argv, int* exit_code) { |
118 base::CommandLine::Init(argc, argv); | 118 base::CommandLine::Init(argc, argv); |
119 const base::CommandLine& command_line = | 119 const base::CommandLine& command_line = |
120 *base::CommandLine::ForCurrentProcess(); | 120 *base::CommandLine::ForCurrentProcess(); |
121 if (!command_line.HasSwitch("run-in-mash")) | 121 if (!command_line.HasSwitch("run-in-mash")) |
122 return false; | 122 return false; |
123 | 123 |
124 if (command_line.HasSwitch(switches::kChildProcess) && | 124 if (command_line.HasSwitch(switches::kChildProcess) && |
125 !command_line.HasSwitch(MojoTestConnector::kTestSwitch)) { | 125 !command_line.HasSwitch(MojoTestConnector::kTestSwitch)) { |
126 base::AtExitManager at_exit; | 126 base::AtExitManager at_exit; |
127 mojo::shell::InitializeLogging(); | 127 shell::InitializeLogging(); |
128 mojo::shell::WaitForDebuggerIfNecessary(); | 128 shell::WaitForDebuggerIfNecessary(); |
129 #if !defined(OFFICIAL_BUILD) && defined(OS_WIN) | 129 #if !defined(OFFICIAL_BUILD) && defined(OS_WIN) |
130 base::RouteStdioToConsole(false); | 130 base::RouteStdioToConsole(false); |
131 #endif | 131 #endif |
132 *exit_code = mojo::shell::ChildProcessMain(); | 132 *exit_code = shell::ChildProcessMain(); |
133 return true; | 133 return true; |
134 } | 134 } |
135 | 135 |
136 int default_jobs = std::max(1, base::SysInfo::NumberOfProcessors() / 2); | 136 int default_jobs = std::max(1, base::SysInfo::NumberOfProcessors() / 2); |
137 MashTestLauncherDelegate delegate; | 137 MashTestLauncherDelegate delegate; |
138 // --single_process and no primoridal pipe token indicate we were run directly | 138 // --single_process and no primoridal pipe token indicate we were run directly |
139 // from the command line. In this case we have to start up MojoShellConnection | 139 // from the command line. In this case we have to start up MojoShellConnection |
140 // as though we were embedded. | 140 // as though we were embedded. |
141 content::MojoShellConnection::Factory shell_connection_factory; | 141 content::MojoShellConnection::Factory shell_connection_factory; |
142 if (command_line.HasSwitch(content::kSingleProcessTestsFlag) && | 142 if (command_line.HasSwitch(content::kSingleProcessTestsFlag) && |
143 !command_line.HasSwitch(switches::kPrimordialPipeToken)) { | 143 !command_line.HasSwitch(switches::kPrimordialPipeToken)) { |
144 shell_connection_factory = | 144 shell_connection_factory = |
145 base::Bind(&CreateMojoShellConnection, &delegate); | 145 base::Bind(&CreateMojoShellConnection, &delegate); |
146 content::MojoShellConnection::SetFactoryForTest(&shell_connection_factory); | 146 content::MojoShellConnection::SetFactoryForTest(&shell_connection_factory); |
147 } | 147 } |
148 *exit_code = LaunchChromeTests(default_jobs, &delegate, argc, argv); | 148 *exit_code = LaunchChromeTests(default_jobs, &delegate, argc, argv); |
149 return true; | 149 return true; |
150 } | 150 } |
OLD | NEW |