Index: mojo/shell/tests/connect/connect_test_package.cc |
diff --git a/mojo/shell/tests/connect/connect_test_package.cc b/mojo/shell/tests/connect/connect_test_package.cc |
index 1458e65f2e554210248a353739c909b995c2eb57..c5a1a0765dcd158e3828ba738c96cdef9cedc074 100644 |
--- a/mojo/shell/tests/connect/connect_test_package.cc |
+++ b/mojo/shell/tests/connect/connect_test_package.cc |
@@ -184,6 +184,12 @@ |
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, |