Cascade shutdown of instances

mojo/shell/tests/lifecycle/lifecycle_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 "base/bind.h"
 #include "base/command_line.h"
 #include "base/macros.h"
 #include "base/process/process.h"
 #include "base/run_loop.h"
 #include "mojo/shell/public/cpp/identity.h"
 #include "mojo/shell/public/cpp/shell_test.h"
 #include "mojo/shell/public/interfaces/shell.mojom.h"
 #include "mojo/shell/tests/lifecycle/lifecycle_unittest.mojom.h"
 #include "mojo/shell/tests/util.h"
 
 namespace mojo {
 namespace shell {
 namespace {
 
 const char kTestAppName[] = "mojo:lifecycle_unittest_app";
+const char kTestParentName[] = "mojo:lifecycle_unittest_parent";
 const char kTestExeName[] = "exe:lifecycle_unittest_exe";
 const char kTestPackageName[] = "mojo:lifecycle_unittest_package";
 const char kTestPackageAppNameA[] = "mojo:lifecycle_unittest_package_app_a";
 const char kTestPackageAppNameB[] = "mojo:lifecycle_unittest_package_app_b";
 const char kTestName[] = "mojo:lifecycle_unittest";
 
 void QuitLoop(base::RunLoop* loop) {
   loop->Quit();
 }
 
@@ skipping 400 matching lines @@
   base::RunLoop loop;
   lifecycle.set_connection_error_handler(base::Bind(&QuitLoop, &loop));
   process.Terminate(9, true);
   loop.Run();
 
   WaitForInstanceDestruction();
   EXPECT_FALSE(instances()->HasInstanceForName(kTestExeName));
   EXPECT_EQ(0u, instances()->GetNewInstanceCount());
 }
 
+TEST_F(LifecycleTest, ShutdownTree) {
+  // Verifies that Instances are destroyed when their creator is.
+  scoped_ptr<Connection> parent_connection =
+      connector()->Connect(kTestParentName);
+  test::mojom::ParentPtr parent;
+  parent_connection->GetInterface(&parent);
+
+  // This asks kTestParentName to open a connection to kTestAppName and blocks
+  // on a response from a Ping().
+  {
+    base::RunLoop loop;
+    parent->ConnectToChild(base::Bind(&QuitLoop, &loop));
+    loop.Run();
+  }
+
+  // Should now have two new instances (parent and child).
+  EXPECT_EQ(2u, instances()->GetNewInstanceCount());
+  EXPECT_TRUE(instances()->HasInstanceForName(kTestParentName));
+  EXPECT_TRUE(instances()->HasInstanceForName(kTestAppName));
+
+  parent->Quit();
+
+  // Quitting the parent should cascade-quit the child.
+  WaitForInstanceDestruction();
+  EXPECT_EQ(0u, instances()->GetNewInstanceCount());
+  EXPECT_FALSE(instances()->HasInstanceForName(kTestParentName));
+  EXPECT_FALSE(instances()->HasInstanceForName(kTestAppName));
+}
 
 }  // namespace shell
 }  // namespace mojo