Create a ProfileDependencyManager to order ProfileKeyedService destruction.

chrome/browser/profiles/profile_dependency_manager_unittest.cc

+// Copyright (c) 2011 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 "testing/gtest/include/gtest/gtest.h"
+
+#include "chrome/browser/profiles/profile_dependency_manager.h"
+#include "chrome/browser/profiles/profile_keyed_service_factory.h"
+#include "chrome/test/testing_profile.h"
+
+class ProfileDependencyManagerUnittests : public ::testing::Test {
+ protected:
+  virtual ~ProfileDependencyManagerUnittests() {
+    EXPECT_TRUE(dependency_manager_.all_components_.empty());
+    EXPECT_TRUE(dependency_manager_.edges_.empty());
+    EXPECT_TRUE(dependency_manager_.destruction_order_.empty());
+  }
+
+  // To get around class access:
+  void DependOn(ProfileKeyedServiceFactory* child,
+                ProfileKeyedServiceFactory* parent) {
+    child->DependsOn(parent);
+  }
+
+  ProfileDependencyManager* manager() { return &dependency_manager_; }
+
+  std::vector<std::string>* shutdown_order() { return &shutdown_order_; }
+
+ private:
+  ProfileDependencyManager dependency_manager_;
+
+  std::vector<std::string> shutdown_order_;
+};
+
+class TestService : public ProfileKeyedServiceFactory {
+ public:
+  TestService(const std::string& name,
+              std::vector<std::string>* fill_on_shutdown,
+              ProfileDependencyManager* manager)
+      : ProfileKeyedServiceFactory(manager),
+        name_(name),
+        fill_on_shutdown_(fill_on_shutdown) {
+  }
+
+  virtual ProfileKeyedService* BuildServiceInstanceFor(
+      Profile* profile) const {
+    NOTREACHED() << "This isn't part of the tests!";

[Paweł Hajdan Jr., 2011/03/29 19:10:28]

nit: NOTREACHED() will fire only for the Debug builds, and will crash the entire
test binary. Of course if the NULL returned below is dereferenced it will also
crash, but could you just switch to ADD_FAILURE above to consistently apply the
rule "avoid NOTREACHED() in unit tests"?

+    return NULL;
+  }
+
+  virtual void ProfileShutdown(Profile* profile) {
+    fill_on_shutdown_->push_back(name_);
+  }
+
+ private:
+  const std::string name_;
+  std::vector<std::string>* fill_on_shutdown_;
+};
+
+// Tests that we can deal with a single component.
+TEST_F(ProfileDependencyManagerUnittests, SingleCase) {
+  TestService service("service", shutdown_order(), manager());
+
+  {
+    TestingProfile profile;
+  }
+
+  ASSERT_EQ(1U, shutdown_order()->size());
+  EXPECT_STREQ("service", (*shutdown_order())[0].c_str());
+}
+
+// Tests that we get a simple one component depends on the other case.
+TEST_F(ProfileDependencyManagerUnittests, SimpleDependency) {
+  TestService parent("parent", shutdown_order(), manager());
+  TestService child("child", shutdown_order(), manager());
+  DependOn(&child, &parent);
+
+  {
+    TestingProfile profile;
+  }
+
+  ASSERT_EQ(2U, shutdown_order()->size());
+  EXPECT_STREQ("child", (*shutdown_order())[0].c_str());
+  EXPECT_STREQ("parent", (*shutdown_order())[1].c_str());
+}
+
+// Tests two children, one parent
+TEST_F(ProfileDependencyManagerUnittests, TwoChildrenOneParent) {
+  TestService parent("parent", shutdown_order(), manager());
+  TestService child1("child1", shutdown_order(), manager());
+  TestService child2("child2", shutdown_order(), manager());
+  DependOn(&child1, &parent);
+  DependOn(&child2, &parent);
+
+  {
+    TestingProfile profile;
+  }
+
+  ASSERT_EQ(3U, shutdown_order()->size());
+  EXPECT_STREQ("child2", (*shutdown_order())[0].c_str());
+  EXPECT_STREQ("child1", (*shutdown_order())[1].c_str());
+  EXPECT_STREQ("parent", (*shutdown_order())[2].c_str());
+}
+
+// Tests an M configuration
+TEST_F(ProfileDependencyManagerUnittests, MConfiguration) {
+  TestService parent1("parent1", shutdown_order(), manager());
+  TestService parent2("parent2", shutdown_order(), manager());
+
+  TestService child_of_1("child_of_1", shutdown_order(), manager());
+  DependOn(&child_of_1, &parent1);
+
+  TestService child_of_12("child_of_12", shutdown_order(), manager());
+  DependOn(&child_of_12, &parent1);
+  DependOn(&child_of_12, &parent2);
+
+  TestService child_of_2("child_of_2", shutdown_order(), manager());
+  DependOn(&child_of_2, &parent2);
+
+  {
+    TestingProfile profile;
+  }
+
+  ASSERT_EQ(5U, shutdown_order()->size());
+  EXPECT_STREQ("child_of_2", (*shutdown_order())[0].c_str());
+  EXPECT_STREQ("child_of_12", (*shutdown_order())[1].c_str());
+  EXPECT_STREQ("child_of_1", (*shutdown_order())[2].c_str());
+  EXPECT_STREQ("parent2", (*shutdown_order())[3].c_str());
+  EXPECT_STREQ("parent1", (*shutdown_order())[4].c_str());
+}
+
+// Tests that it can deal with a simple diamond.
+TEST_F(ProfileDependencyManagerUnittests, DiamondConfiguration) {
+  TestService parent("parent", shutdown_order(), manager());
+
+  TestService middle_row_1("middle_row_1", shutdown_order(), manager());
+  DependOn(&middle_row_1, &parent);
+
+  TestService middle_row_2("middle_row_2", shutdown_order(), manager());
+  DependOn(&middle_row_2, &parent);
+
+  TestService bottom("bottom", shutdown_order(), manager());
+  DependOn(&bottom, &middle_row_1);
+  DependOn(&bottom, &middle_row_2);
+
+  {
+    TestingProfile profile;
+  }
+
+  ASSERT_EQ(4U, shutdown_order()->size());
+  EXPECT_STREQ("bottom", (*shutdown_order())[0].c_str());
+  EXPECT_STREQ("middle_row_2", (*shutdown_order())[1].c_str());
+  EXPECT_STREQ("middle_row_1", (*shutdown_order())[2].c_str());
+  EXPECT_STREQ("parent", (*shutdown_order())[3].c_str());
+}
+
+// A final test that works with a more complex graph.
+TEST_F(ProfileDependencyManagerUnittests, ComplexGraph) {
+  TestService everything_depends_on_me("everything_depends_on_me",
+                                       shutdown_order(), manager());
+
+  TestService intermediary_service("intermediary_service",
+                                   shutdown_order(), manager());
+  DependOn(&intermediary_service, &everything_depends_on_me);
+
+  TestService specialized_service("specialized_service",
+                                  shutdown_order(), manager());
+  DependOn(&specialized_service, &everything_depends_on_me);
+  DependOn(&specialized_service, &intermediary_service);
+
+  TestService other_root("other_root", shutdown_order(), manager());
+
+  TestService other_intermediary("other_intermediary",
+                                 shutdown_order(), manager());
+  DependOn(&other_intermediary, &other_root);
+
+  TestService bottom("bottom", shutdown_order(), manager());
+  DependOn(&bottom, &specialized_service);
+  DependOn(&bottom, &other_intermediary);
+
+  {
+    TestingProfile profile;
+  }
+
+  ASSERT_EQ(6U, shutdown_order()->size());
+  EXPECT_STREQ("bottom", (*shutdown_order())[0].c_str());
+  EXPECT_STREQ("specialized_service", (*shutdown_order())[1].c_str());
+  EXPECT_STREQ("other_intermediary", (*shutdown_order())[2].c_str());
+  EXPECT_STREQ("intermediary_service", (*shutdown_order())[3].c_str());
+  EXPECT_STREQ("other_root", (*shutdown_order())[4].c_str());
+  EXPECT_STREQ("everything_depends_on_me", (*shutdown_order())[5].c_str());
+}