Index: content/browser/mojo_shell_browsertest.cc |
diff --git a/content/browser/mojo_shell_browsertest.cc b/content/browser/mojo_shell_browsertest.cc |
index 5e9eac1e313538e44f8c59473ae6b69439c3c481..f481cf4339da1722ebbe738223acfc82d2a17896 100644 |
--- a/content/browser/mojo_shell_browsertest.cc |
+++ b/content/browser/mojo_shell_browsertest.cc |
@@ -8,9 +8,11 @@ |
#include "base/run_loop.h" |
#include "content/browser/mojo/mojo_shell_context.h" |
#include "content/public/browser/mojo_app_connection.h" |
+#include "content/public/browser/web_contents.h" |
#include "content/public/test/content_browser_test.h" |
#include "content/public/test/test_mojo_app.h" |
#include "content/public/test/test_mojo_service.mojom.h" |
+#include "content/shell/browser/shell.h" |
#include "url/gurl.h" |
namespace content { |
@@ -24,6 +26,11 @@ class MojoShellTest : public ContentBrowserTest { |
MojoShellContext::SetApplicationsForTest(&test_apps_); |
} |
+ protected: |
+ BrowserContext* GetBrowserContext() { |
+ return shell()->web_contents()->GetBrowserContext(); |
+ } |
+ |
private: |
static scoped_ptr<mojo::ShellClient> CreateTestApp() { |
return scoped_ptr<mojo::ShellClient>(new TestMojoApp); |
@@ -35,7 +42,8 @@ class MojoShellTest : public ContentBrowserTest { |
}; |
IN_PROC_BROWSER_TEST_F(MojoShellTest, TestBrowserConnection) { |
- auto test_app = MojoAppConnection::Create(kInProcessTestMojoAppName, |
+ auto test_app = MojoAppConnection::Create(GetBrowserContext(), |
+ kInProcessTestMojoAppName, |
kBrowserMojoAppUrl); |
TestMojoServicePtr test_service; |
test_app->GetInterface(&test_service); |
@@ -49,7 +57,8 @@ IN_PROC_BROWSER_TEST_F(MojoShellTest, TestUtilityConnection) { |
// With no loader registered at this URL, the shell should spawn a utility |
// process and connect us to it. content_shell's utility process always hosts |
// a TestMojoApp at |kTestMojoAppUrl|. |
- auto test_app = MojoAppConnection::Create(kTestMojoAppUrl, |
+ auto test_app = MojoAppConnection::Create(GetBrowserContext(), |
+ kTestMojoAppUrl, |
kBrowserMojoAppUrl); |
TestMojoServicePtr test_service; |
test_app->GetInterface(&test_service); |