| 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 "base/bind.h" | 5 #include "base/bind.h" |
| 6 #include "base/run_loop.h" | 6 #include "base/run_loop.h" |
| 7 #include "base/strings/utf_string_conversions.h" | 7 #include "base/strings/utf_string_conversions.h" |
| 8 #include "content/public/browser/browser_thread.h" | 8 #include "content/public/browser/browser_thread.h" |
| 9 #include "content/public/browser/utility_process_host.h" | 9 #include "content/public/browser/utility_process_host.h" |
| 10 #include "content/public/browser/utility_process_host_client.h" | 10 #include "content/public/browser/utility_process_host_client.h" |
| 11 #include "content/public/common/service_registry.h" | 11 #include "content/public/common/service_registry.h" |
| 12 #include "content/public/test/content_browser_test.h" | 12 #include "content/public/test/content_browser_test.h" |
| 13 #include "content/public/test/content_browser_test_utils.h" | 13 #include "content/public/test/content_browser_test_utils.h" |
| 14 #include "content/public/test/test_mojo_service.mojom.h" | 14 #include "content/public/test/test_mojo_service.mojom.h" |
| 15 | 15 |
| 16 namespace content { | 16 namespace content { |
| 17 | 17 |
| 18 class UtilityProcessHostImplBrowserTest : public ContentBrowserTest { | 18 class UtilityProcessHostImplBrowserTest : public ContentBrowserTest { |
| 19 public: | 19 public: |
| 20 void RunUtilityProcess(bool elevated) { | 20 void RunUtilityProcess() { |
| 21 base::RunLoop run_loop; | 21 base::RunLoop run_loop; |
| 22 done_closure_ = run_loop.QuitClosure(); | 22 done_closure_ = run_loop.QuitClosure(); |
| 23 BrowserThread::PostTask( | 23 BrowserThread::PostTask( |
| 24 BrowserThread::IO, FROM_HERE, | 24 BrowserThread::IO, FROM_HERE, |
| 25 base::Bind( | 25 base::Bind( |
| 26 &UtilityProcessHostImplBrowserTest::RunUtilityProcessOnIOThread, | 26 &UtilityProcessHostImplBrowserTest::RunUtilityProcessOnIOThread, |
| 27 base::Unretained(this), elevated)); | 27 base::Unretained(this))); |
| 28 run_loop.Run(); | 28 run_loop.Run(); |
| 29 } | 29 } |
| 30 | 30 |
| 31 protected: | 31 protected: |
| 32 void RunUtilityProcessOnIOThread(bool elevated) { | 32 void RunUtilityProcessOnIOThread() { |
| 33 UtilityProcessHost* host = UtilityProcessHost::Create(nullptr, nullptr); | 33 UtilityProcessHost* host = UtilityProcessHost::Create(nullptr, nullptr); |
| 34 host->SetName(base::ASCIIToUTF16("TestProcess")); | 34 host->SetName(base::ASCIIToUTF16("TestProcess")); |
| 35 #if defined(OS_WIN) | |
| 36 if (elevated) | |
| 37 host->ElevatePrivileges(); | |
| 38 #endif | |
| 39 EXPECT_TRUE(host->Start()); | 35 EXPECT_TRUE(host->Start()); |
| 40 | 36 |
| 41 ServiceRegistry* service_registry = host->GetServiceRegistry(); | 37 ServiceRegistry* service_registry = host->GetServiceRegistry(); |
| 42 service_registry->ConnectToRemoteService(mojo::GetProxy(&service_)); | 38 service_registry->ConnectToRemoteService(mojo::GetProxy(&service_)); |
| 43 service_->DoSomething(base::Bind( | 39 service_->DoSomething(base::Bind( |
| 44 &UtilityProcessHostImplBrowserTest::OnSomething, | 40 &UtilityProcessHostImplBrowserTest::OnSomething, |
| 45 base::Unretained(this))); | 41 base::Unretained(this))); |
| 46 } | 42 } |
| 47 | 43 |
| 48 void OnSomething() { | 44 void OnSomething() { |
| 49 service_.reset(); | 45 service_.reset(); |
| 50 BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, done_closure_); | 46 BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, done_closure_); |
| 51 } | 47 } |
| 52 | 48 |
| 53 mojom::TestMojoServicePtr service_; | 49 mojom::TestMojoServicePtr service_; |
| 54 base::Closure done_closure_; | 50 base::Closure done_closure_; |
| 55 }; | 51 }; |
| 56 | 52 |
| 57 IN_PROC_BROWSER_TEST_F(UtilityProcessHostImplBrowserTest, LaunchProcess) { | 53 IN_PROC_BROWSER_TEST_F(UtilityProcessHostImplBrowserTest, LaunchProcess) { |
| 58 RunUtilityProcess(false); | 54 RunUtilityProcess(); |
| 59 } | 55 } |
| 60 | 56 |
| 61 #if defined(OS_WIN) | |
| 62 IN_PROC_BROWSER_TEST_F(UtilityProcessHostImplBrowserTest, | |
| 63 LaunchElevatedProcess) { | |
| 64 RunUtilityProcess(true); | |
| 65 } | |
| 66 #endif | |
| 67 | |
| 68 } // namespace content | 57 } // namespace content |
| OLD | NEW |