mash: Exit the root process if the window manager service crashes

services/service_manager/background/tests/background_service_manager_unittest.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 "services/service_manager/background/background_service_manager.h"
 
 #include <memory>
 
 #include "base/memory/ptr_util.h"
 #include "base/run_loop.h"
 #include "services/service_manager/background/tests/test.mojom.h"
 #include "services/service_manager/public/cpp/connector.h"
 #include "services/service_manager/public/cpp/service.h"
 #include "services/service_manager/public/cpp/service_context.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace service_manager {
 namespace {
 
 const char kTestName[] = "background_service_manager_unittest";
 const char kAppName[] = "background_service_manager_test_service";
 
+// The parent unit test suite service, not the underlying test service.
 class ServiceImpl : public Service {
  public:
   ServiceImpl() {}
   ~ServiceImpl() override {}
 
   // Service:
   bool OnConnect(const ServiceInfo& remote_info,
                  InterfaceRegistry* registry) override {
     return false;
   }
 
  private:
   DISALLOW_COPY_AND_ASSIGN(ServiceImpl);
 };
 
 void SetFlagAndRunClosure(bool* flag, const base::Closure& closure) {
   *flag = true;
   closure.Run();
 }
 
-}  // namespace
-
 // Uses BackgroundServiceManager to start the service manager in the background
 // and connects to background_service_manager_test_service, verifying we can
 // send a message to the service.
 #if defined(OS_ANDROID)
 // TODO(crbug.com/589784): This test is disabled, as it fails
 // on the Android GN bot.
 #define MAYBE_Basic DISABLED_Basic
 #else
 #define MAYBE_Basic Basic
 #endif
 TEST(BackgroundServiceManagerTest, MAYBE_Basic) {
   BackgroundServiceManager background_service_manager(nullptr, nullptr);
   base::MessageLoop message_loop;
   mojom::ServicePtr service;
   ServiceContext service_context(base::MakeUnique<ServiceImpl>(),
                                  mojom::ServiceRequest(&service));
   background_service_manager.RegisterService(
       Identity(kTestName, mojom::kRootUserID), std::move(service), nullptr);
 
   mojom::TestServicePtr test_service;
   service_context.connector()->BindInterface(kAppName, &test_service);
   base::RunLoop run_loop;
   bool got_result = false;
   test_service->Test(
       base::Bind(&SetFlagAndRunClosure, &got_result, run_loop.QuitClosure()));
   run_loop.Run();
   EXPECT_TRUE(got_result);
 }
 
+// The out param cannot be last due to base::Bind() currying.
+void QuitCallback(bool* callback_called,
+                  const base::Closure& quit_closure,
+                  const Identity& identity) {
+  EXPECT_EQ(kAppName, identity.name());
+  *callback_called = true;
+  quit_closure.Run();
+}
+
+// Verifies that quitting a service invokes the quit callback.
+TEST(BackgroundServiceManagerTest, SetInstanceQuitCallback) {
+  BackgroundServiceManager background_service_manager(nullptr, nullptr);
+  base::MessageLoop message_loop;
+  mojom::ServicePtr service;
+  ServiceContext service_context(base::MakeUnique<ServiceImpl>(),
+                                 mojom::ServiceRequest(&service));
+  background_service_manager.RegisterService(
+      Identity(kTestName, mojom::kRootUserID), std::move(service), nullptr);
+
+  mojom::TestServicePtr test_service;
+  service_context.connector()->BindInterface(kAppName, &test_service);
+
+  // Set a callback for when the service quits that will quit |run_loop|.
+  base::RunLoop run_loop;
+  bool callback_called = false;
+  background_service_manager.SetInstanceQuitCallback(
+      base::Bind(&QuitCallback, &callback_called, run_loop.QuitClosure()));
+
+  // Ask the service to quit itself.
+  test_service->Quit();
+  run_loop.Run();
+
+  // The run loop was quit by the callback and not by something else.
+  EXPECT_TRUE(callback_called);
+}
+
+}  // namespace
 }  // namespace service_manager