blob: 16a6542419173eb6368ff988c9bff43f15afb8ab [file] [log] [blame]
amistrydc1c91062016-05-12 00:59:521// Copyright 2016 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 "base/bind.h"
6#include "base/run_loop.h"
7#include "base/strings/utf_string_conversions.h"
8#include "content/public/browser/browser_thread.h"
9#include "content/public/browser/utility_process_host.h"
10#include "content/public/browser/utility_process_host_client.h"
amistrydc1c91062016-05-12 00:59:5211#include "content/public/test/content_browser_test.h"
12#include "content/public/test/content_browser_test_utils.h"
13#include "content/public/test/test_mojo_service.mojom.h"
bend1dd50f52016-06-26 22:10:4814#include "services/shell/public/cpp/interface_provider.h"
15#include "services/shell/public/cpp/interface_registry.h"
amistrydc1c91062016-05-12 00:59:5216
17namespace content {
18
19class UtilityProcessHostImplBrowserTest : public ContentBrowserTest {
20 public:
amistryb6138912016-05-31 01:54:1221 void RunUtilityProcess(bool elevated) {
amistrydc1c91062016-05-12 00:59:5222 base::RunLoop run_loop;
23 done_closure_ = run_loop.QuitClosure();
24 BrowserThread::PostTask(
25 BrowserThread::IO, FROM_HERE,
26 base::Bind(
27 &UtilityProcessHostImplBrowserTest::RunUtilityProcessOnIOThread,
amistryb6138912016-05-31 01:54:1228 base::Unretained(this), elevated));
amistrydc1c91062016-05-12 00:59:5229 run_loop.Run();
30 }
31
32 protected:
amistryb6138912016-05-31 01:54:1233 void RunUtilityProcessOnIOThread(bool elevated) {
amistrydc1c91062016-05-12 00:59:5234 UtilityProcessHost* host = UtilityProcessHost::Create(nullptr, nullptr);
35 host->SetName(base::ASCIIToUTF16("TestProcess"));
amistryb6138912016-05-31 01:54:1236#if defined(OS_WIN)
37 if (elevated)
38 host->ElevatePrivileges();
39#endif
amistrydc1c91062016-05-12 00:59:5240 EXPECT_TRUE(host->Start());
41
bend1dd50f52016-06-26 22:10:4842 host->GetRemoteInterfaces()->GetInterface(&service_);
amistrydc1c91062016-05-12 00:59:5243 service_->DoSomething(base::Bind(
44 &UtilityProcessHostImplBrowserTest::OnSomething,
45 base::Unretained(this)));
46 }
47
48 void OnSomething() {
49 service_.reset();
50 BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, done_closure_);
51 }
52
53 mojom::TestMojoServicePtr service_;
54 base::Closure done_closure_;
55};
56
57IN_PROC_BROWSER_TEST_F(UtilityProcessHostImplBrowserTest, LaunchProcess) {
amistryb6138912016-05-31 01:54:1258 RunUtilityProcess(false);
amistrydc1c91062016-05-12 00:59:5259}
60
amistryb6138912016-05-31 01:54:1261#if defined(OS_WIN)
62IN_PROC_BROWSER_TEST_F(UtilityProcessHostImplBrowserTest,
63 LaunchElevatedProcess) {
64 RunUtilityProcess(true);
65}
66#endif
67
amistrydc1c91062016-05-12 00:59:5268} // namespace content