summaryrefslogtreecommitdiffstats
path: root/mojo/shell/tests/connect/connect_test_package.cc
diff options
context:
space:
mode:
Diffstat (limited to 'mojo/shell/tests/connect/connect_test_package.cc')
-rw-r--r--mojo/shell/tests/connect/connect_test_package.cc6
1 files changed, 6 insertions, 0 deletions
diff --git a/mojo/shell/tests/connect/connect_test_package.cc b/mojo/shell/tests/connect/connect_test_package.cc
index 1458e65..c5a1a07 100644
--- a/mojo/shell/tests/connect/connect_test_package.cc
+++ b/mojo/shell/tests/connect/connect_test_package.cc
@@ -184,6 +184,12 @@ class ConnectTestShellClient
connection->AddInterface<test::mojom::ConnectTestService>(this);
return true;
}
+ void ShellConnectionLost() override {
+ if (base::MessageLoop::current() &&
+ base::MessageLoop::current()->is_running()) {
+ base::MessageLoop::current()->QuitWhenIdle();
+ }
+ }
// InterfaceFactory<mojom::ShellClientFactory>:
void Create(Connection* connection,