Change primordial pipes to ShellClient

mojo/shell/background/tests/test_service.cc

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "mojo/public/c/system/main.h"
 #include "mojo/public/cpp/bindings/binding_set.h"
 #include "mojo/shell/background/tests/test.mojom.h"
 #include "mojo/shell/public/cpp/application_runner.h"
 #include "mojo/shell/public/cpp/connection.h"
 #include "mojo/shell/public/cpp/shell_client.h"
 
 namespace mojo {
 namespace shell {
 
 class TestClient : public ShellClient,
                    public InterfaceFactory<mojom::TestService>,
                    public mojom::TestService {
  public:
-  TestClient() {}
+  explicit TestClient(mojo::ApplicationRunner* runner) : runner_(runner) {}
   ~TestClient() override {}
 
  private:
   // ShellClient:
   bool AcceptConnection(Connection* connection) override {
     connection->AddInterface(this);
     return true;
   }
+  void ShellConnectionLost() override {
+    runner_->Quit();
+  }
 
   // InterfaceFactory<mojom::TestService>:
   void Create(Connection* connection,
               InterfaceRequest<mojom::TestService> request) override {
     bindings_.AddBinding(this, std::move(request));
   }
 
   // mojom::TestService
   void Test(const TestCallback& callback) override { callback.Run(); }
 
+  mojo::ApplicationRunner* const runner_;
   BindingSet<mojom::TestService> bindings_;
 
   DISALLOW_COPY_AND_ASSIGN(TestClient);
 };
 
 }  // namespace shell
 }  // namespace mojo
 
 MojoResult MojoMain(MojoHandle shell_handle) {
-  mojo::ApplicationRunner runner(new mojo::shell::TestClient);
+  mojo::ApplicationRunner runner(new mojo::shell::TestClient(&runner));
   return runner.Run(shell_handle);
 }