chrome/browser/service_process/service_process_control_browsertest.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/service_process/service_process_control.h"
   6
   7 #include "base/bind.h"
   8 #include "base/bind_helpers.h"
   9 #include "base/command_line.h"
  10 #include "base/location.h"
  11 #include "base/path_service.h"
  12 #include "base/process/kill.h"
  13 #include "base/process/process.h"
  14 #include "base/process/process_iterator.h"
  15 #include "base/single_thread_task_runner.h"
  16 #include "base/test/test_timeouts.h"
  17 #include "base/thread_task_runner_handle.h"
  18 #include "chrome/browser/ui/browser.h"
  19 #include "chrome/common/chrome_constants.h"
  20 #include "chrome/common/service_process_util.h"
  21 #include "chrome/test/base/in_process_browser_test.h"
  22 #include "components/version_info/version_info.h"
  23 #include "content/public/common/content_paths.h"
  24 #include "content/public/common/content_switches.h"
  25 #include "content/public/test/test_utils.h"
  26 #include "testing/gmock/include/gmock/gmock.h"
  27
  28 class ServiceProcessControlBrowserTest
  29     : public InProcessBrowserTest {
  30  public:
  31   ServiceProcessControlBrowserTest() {
  32   }
  33   virtual ~ServiceProcessControlBrowserTest() {
  34   }
  35
  36   void HistogramsCallback() {
  37     MockHistogramsCallback();
  38     QuitMessageLoop();
  39   }
  40
  41   MOCK_METHOD0(MockHistogramsCallback, void());
  42
  43  protected:
  44   void LaunchServiceProcessControl() {
  45     // Launch the process asynchronously.
  46     ServiceProcessControl::GetInstance()->Launch(
  47         base::Bind(&ServiceProcessControlBrowserTest::ProcessControlLaunched,
  48                    this),
  49         base::Bind(
  50             &ServiceProcessControlBrowserTest::ProcessControlLaunchFailed,
  51             this));
  52
  53     // Then run the message loop to keep things running.
  54     content::RunMessageLoop();
  55   }
  56
  57   static void QuitMessageLoop() {
  58     base::MessageLoop::current()->Quit();
  59   }
  60
  61   static void CloudPrintInfoCallback(
  62       const cloud_print::CloudPrintProxyInfo& proxy_info) {
  63     QuitMessageLoop();
  64   }
  65
  66   void Disconnect() {
  67     // This will close the IPC connection.
  68     ServiceProcessControl::GetInstance()->Disconnect();
  69   }
  70
  71   void SetUp() override {
  72     InProcessBrowserTest::SetUp();
  73
  74     // This should not be needed because TearDown() ends with a closed
  75     // service_process_, but HistogramsTimeout and Histograms fail without this
  76     // on Mac.
  77     service_process_.Close();
  78   }
  79
  80   void TearDown() override {
  81     if (ServiceProcessControl::GetInstance()->IsConnected())
  82       EXPECT_TRUE(ServiceProcessControl::GetInstance()->Shutdown());
  83 #if defined(OS_MACOSX)
  84     // ForceServiceProcessShutdown removes the process from launched on Mac.
  85     ForceServiceProcessShutdown("", 0);
  86 #endif  // OS_MACOSX
  87     if (service_process_.IsValid()) {
  88       int exit_code;
  89       EXPECT_TRUE(service_process_.WaitForExitWithTimeout(
  90           TestTimeouts::action_max_timeout(), &exit_code));
  91       EXPECT_EQ(0, exit_code);
  92       service_process_.Close();
  93     }
  94
  95     InProcessBrowserTest::TearDown();
  96   }
  97
  98   void ProcessControlLaunched() {
  99     base::ProcessId service_pid;
 100     EXPECT_TRUE(GetServiceProcessData(NULL, &service_pid));
 101     EXPECT_NE(static_cast<base::ProcessId>(0), service_pid);
 102 #if defined(OS_WIN)
 103     service_process_ =
 104         base::Process::OpenWithAccess(service_pid,
 105                                       SYNCHRONIZE | PROCESS_QUERY_INFORMATION);
 106 #else
 107     service_process_ = base::Process::Open(service_pid);
 108 #endif
 109     EXPECT_TRUE(service_process_.IsValid());
 110     // Quit the current message. Post a QuitTask instead of just calling Quit()
 111     // because this can get invoked in the context of a Launch() call and we
 112     // may not be in Run() yet.
 113     base::ThreadTaskRunnerHandle::Get()->PostTask(
 114         FROM_HERE, base::MessageLoop::QuitClosure());
 115   }
 116
 117   void ProcessControlLaunchFailed() {
 118     ADD_FAILURE();
 119     // Quit the current message.
 120     base::ThreadTaskRunnerHandle::Get()->PostTask(
 121         FROM_HERE, base::MessageLoop::QuitClosure());
 122   }
 123
 124  private:
 125   base::Process service_process_;
 126 };
 127
 128 class RealServiceProcessControlBrowserTest
 129       : public ServiceProcessControlBrowserTest {
 130  public:
 131   void SetUpCommandLine(base::CommandLine* command_line) override {
 132     ServiceProcessControlBrowserTest::SetUpCommandLine(command_line);
 133     base::FilePath exe;
 134     PathService::Get(base::DIR_EXE, &exe);
 135 #if defined(OS_MACOSX)
 136     exe = exe.DirName().DirName().DirName();
 137 #endif
 138     exe = exe.Append(chrome::kHelperProcessExecutablePath);
 139     // Run chrome instead of browser_tests.exe.
 140     EXPECT_TRUE(base::PathExists(exe));
 141     command_line->AppendSwitchPath(switches::kBrowserSubprocessPath, exe);
 142   }
 143 };
 144
 145 // TODO(vitalybuka): Fix crbug.com/340563
 146 IN_PROC_BROWSER_TEST_F(RealServiceProcessControlBrowserTest,
 147                        DISABLED_LaunchAndIPC) {
 148   LaunchServiceProcessControl();
 149
 150   // Make sure we are connected to the service process.
 151   ASSERT_TRUE(ServiceProcessControl::GetInstance()->IsConnected());
 152   ServiceProcessControl::GetInstance()->GetCloudPrintProxyInfo(
 153         base::Bind(&ServiceProcessControlBrowserTest::CloudPrintInfoCallback));
 154   content::RunMessageLoop();
 155
 156   // And then shutdown the service process.
 157   EXPECT_TRUE(ServiceProcessControl::GetInstance()->Shutdown());
 158 }
 159
 160 IN_PROC_BROWSER_TEST_F(ServiceProcessControlBrowserTest, LaunchAndIPC) {
 161   LaunchServiceProcessControl();
 162
 163   // Make sure we are connected to the service process.
 164   ASSERT_TRUE(ServiceProcessControl::GetInstance()->IsConnected());
 165   ServiceProcessControl::GetInstance()->GetCloudPrintProxyInfo(
 166         base::Bind(&ServiceProcessControlBrowserTest::CloudPrintInfoCallback));
 167   content::RunMessageLoop();
 168
 169   // And then shutdown the service process.
 170   EXPECT_TRUE(ServiceProcessControl::GetInstance()->Shutdown());
 171 }
 172
 173 // This tests the case when a service process is launched when the browser
 174 // starts but we try to launch it again while setting up Cloud Print.
 175 // Flaky on Mac ASan. http://crbug.com/517420
 176 #if defined(OS_MACOSX)
 177 #define MAYBE_LaunchTwice DISABLED_LaunchTwice
 178 #else
 179 #define MAYBE_LaunchTwice LaunchTwice
 180 #endif
 181 IN_PROC_BROWSER_TEST_F(ServiceProcessControlBrowserTest, MAYBE_LaunchTwice) {
 182   // Launch the service process the first time.
 183   LaunchServiceProcessControl();
 184
 185   // Make sure we are connected to the service process.
 186   ASSERT_TRUE(ServiceProcessControl::GetInstance()->IsConnected());
 187   EXPECT_TRUE(ServiceProcessControl::GetInstance()->GetCloudPrintProxyInfo(
 188         base::Bind(&ServiceProcessControlBrowserTest::CloudPrintInfoCallback)));
 189   content::RunMessageLoop();
 190
 191   // Launch the service process again.
 192   LaunchServiceProcessControl();
 193   ASSERT_TRUE(ServiceProcessControl::GetInstance()->IsConnected());
 194   EXPECT_TRUE(ServiceProcessControl::GetInstance()->GetCloudPrintProxyInfo(
 195         base::Bind(&ServiceProcessControlBrowserTest::CloudPrintInfoCallback)));
 196   content::RunMessageLoop();
 197 }
 198
 199 static void DecrementUntilZero(int* count) {
 200   (*count)--;
 201   if (!(*count))
 202     base::ThreadTaskRunnerHandle::Get()->PostTask(
 203         FROM_HERE, base::MessageLoop::QuitClosure());
 204 }
 205
 206 // Flaky on Mac ASan. http://crbug.com/517420
 207 #if defined(OS_MACOSX)
 208 #define MAYBE_MultipleLaunchTasks DISABLED_MultipleLaunchTasks
 209 #else
 210 #define MAYBE_MultipleLaunchTasks MultipleLaunchTasks
 211 #endif
 212 // Invoke multiple Launch calls in succession and ensure that all the tasks
 213 // get invoked.
 214 IN_PROC_BROWSER_TEST_F(ServiceProcessControlBrowserTest,
 215                        MAYBE_MultipleLaunchTasks) {
 216   ServiceProcessControl* process = ServiceProcessControl::GetInstance();
 217   int launch_count = 5;
 218   for (int i = 0; i < launch_count; i++) {
 219     // Launch the process asynchronously.
 220     process->Launch(base::Bind(&DecrementUntilZero, &launch_count),
 221                     base::MessageLoop::QuitClosure());
 222   }
 223   // Then run the message loop to keep things running.
 224   content::RunMessageLoop();
 225   EXPECT_EQ(0, launch_count);
 226 }
 227
 228 // Make sure using the same task for success and failure tasks works.
 229 IN_PROC_BROWSER_TEST_F(ServiceProcessControlBrowserTest, SameLaunchTask) {
 230   ServiceProcessControl* process = ServiceProcessControl::GetInstance();
 231   int launch_count = 5;
 232   for (int i = 0; i < launch_count; i++) {
 233     // Launch the process asynchronously.
 234     base::Closure task = base::Bind(&DecrementUntilZero, &launch_count);
 235     process->Launch(task, task);
 236   }
 237   // Then run the message loop to keep things running.
 238   content::RunMessageLoop();
 239   EXPECT_EQ(0, launch_count);
 240 }
 241
 242 // Tests whether disconnecting from the service IPC causes the service process
 243 // to die.
 244 // Flaky on Mac ASan. http://crbug.com/517420
 245 #if defined(OS_MACOSX)
 246 #define MAYBE_DieOnDisconnect DISABLED_DieOnDisconnect
 247 #else
 248 #define MAYBE_DieOnDisconnect DieOnDisconnect
 249 #endif
 250 IN_PROC_BROWSER_TEST_F(ServiceProcessControlBrowserTest,
 251                        MAYBE_DieOnDisconnect) {
 252   // Launch the service process.
 253   LaunchServiceProcessControl();
 254   // Make sure we are connected to the service process.
 255   ASSERT_TRUE(ServiceProcessControl::GetInstance()->IsConnected());
 256   Disconnect();
 257 }
 258
 259 IN_PROC_BROWSER_TEST_F(ServiceProcessControlBrowserTest, ForceShutdown) {
 260   // Launch the service process.
 261   LaunchServiceProcessControl();
 262   // Make sure we are connected to the service process.
 263   ASSERT_TRUE(ServiceProcessControl::GetInstance()->IsConnected());
 264   base::ProcessId service_pid;
 265   EXPECT_TRUE(GetServiceProcessData(NULL, &service_pid));
 266   EXPECT_NE(static_cast<base::ProcessId>(0), service_pid);
 267   ForceServiceProcessShutdown(version_info::GetVersionNumber(), service_pid);
 268 }
 269
 270 // Flaky on Mac ASan. http://crbug.com/517420
 271 #if defined(OS_MACOSX)
 272 #define MAYBE_CheckPid DISABLED_CheckPid
 273 #else
 274 #define MAYBE_CheckPid CheckPid
 275 #endif
 276 IN_PROC_BROWSER_TEST_F(ServiceProcessControlBrowserTest, MAYBE_CheckPid) {
 277   base::ProcessId service_pid;
 278   EXPECT_FALSE(GetServiceProcessData(NULL, &service_pid));
 279   // Launch the service process.
 280   LaunchServiceProcessControl();
 281   EXPECT_TRUE(GetServiceProcessData(NULL, &service_pid));
 282   EXPECT_NE(static_cast<base::ProcessId>(0), service_pid);
 283   // Disconnect from service process.
 284   Disconnect();
 285 }
 286
 287 IN_PROC_BROWSER_TEST_F(ServiceProcessControlBrowserTest, HistogramsNoService) {
 288   ASSERT_FALSE(ServiceProcessControl::GetInstance()->IsConnected());
 289   EXPECT_CALL(*this, MockHistogramsCallback()).Times(0);
 290   EXPECT_FALSE(ServiceProcessControl::GetInstance()->GetHistograms(
 291       base::Bind(&ServiceProcessControlBrowserTest::HistogramsCallback,
 292                  base::Unretained(this)),
 293       base::TimeDelta()));
 294 }
 295
 296 // Histograms disabled on OSX http://crbug.com/406227
 297 #if defined(OS_MACOSX)
 298 #define MAYBE_HistogramsTimeout DISABLED_HistogramsTimeout
 299 #define MAYBE_Histograms DISABLED_Histograms
 300 #else
 301 #define MAYBE_HistogramsTimeout HistogramsTimeout
 302 #define MAYBE_Histograms Histograms
 303 #endif
 304 IN_PROC_BROWSER_TEST_F(ServiceProcessControlBrowserTest,
 305                        MAYBE_HistogramsTimeout) {
 306   LaunchServiceProcessControl();
 307   ASSERT_TRUE(ServiceProcessControl::GetInstance()->IsConnected());
 308   // Callback should not be called during GetHistograms call.
 309   EXPECT_CALL(*this, MockHistogramsCallback()).Times(0);
 310   EXPECT_TRUE(ServiceProcessControl::GetInstance()->GetHistograms(
 311       base::Bind(&ServiceProcessControlBrowserTest::HistogramsCallback,
 312                  base::Unretained(this)),
 313       base::TimeDelta::FromMilliseconds(100)));
 314   EXPECT_CALL(*this, MockHistogramsCallback()).Times(1);
 315   EXPECT_TRUE(ServiceProcessControl::GetInstance()->Shutdown());
 316   content::RunMessageLoop();
 317 }
 318
 319 IN_PROC_BROWSER_TEST_F(ServiceProcessControlBrowserTest, MAYBE_Histograms) {
 320   LaunchServiceProcessControl();
 321   ASSERT_TRUE(ServiceProcessControl::GetInstance()->IsConnected());
 322   // Callback should not be called during GetHistograms call.
 323   EXPECT_CALL(*this, MockHistogramsCallback()).Times(0);
 324   // Wait for real callback by providing large timeout value.
 325   EXPECT_TRUE(ServiceProcessControl::GetInstance()->GetHistograms(
 326       base::Bind(&ServiceProcessControlBrowserTest::HistogramsCallback,
 327                 base::Unretained(this)),
 328       base::TimeDelta::FromHours(1)));
 329   EXPECT_CALL(*this, MockHistogramsCallback()).Times(1);
 330   content::RunMessageLoop();
 331 }