1// Copyright 2014 The Chromium Authors. All rights reserved. 2// Use of this source code is governed by a BSD-style license that can be 3// found in the LICENSE file. 4 5#include "components/keyed_service/content/browser_context_dependency_manager.h" 6#include "components/keyed_service/content/browser_context_keyed_service_factory.h" 7#include "testing/gtest/include/gtest/gtest.h" 8 9class BrowserContextDependencyManagerUnittests : public ::testing::Test { 10 protected: 11 // To get around class access: 12 void DependOn(BrowserContextKeyedServiceFactory* child, 13 BrowserContextKeyedServiceFactory* parent) { 14 child->DependsOn(parent); 15 } 16 17 BrowserContextDependencyManager* manager() { return &dependency_manager_; } 18 19 std::vector<std::string>* shutdown_order() { return &shutdown_order_; } 20 21 private: 22 BrowserContextDependencyManager dependency_manager_; 23 24 std::vector<std::string> shutdown_order_; 25}; 26 27class TestService : public BrowserContextKeyedServiceFactory { 28 public: 29 TestService(const std::string& name, 30 std::vector<std::string>* fill_on_shutdown, 31 BrowserContextDependencyManager* manager) 32 : BrowserContextKeyedServiceFactory("TestService", manager), 33 name_(name), 34 fill_on_shutdown_(fill_on_shutdown) {} 35 36 virtual KeyedService* BuildServiceInstanceFor( 37 content::BrowserContext* context) const OVERRIDE { 38 ADD_FAILURE() << "This isn't part of the tests!"; 39 return NULL; 40 } 41 42 virtual void BrowserContextShutdown(content::BrowserContext* context) 43 OVERRIDE { 44 fill_on_shutdown_->push_back(name_); 45 } 46 47 private: 48 const std::string name_; 49 std::vector<std::string>* fill_on_shutdown_; 50}; 51 52// Tests that we can deal with a single component. 53TEST_F(BrowserContextDependencyManagerUnittests, SingleCase) { 54 TestService service("service", shutdown_order(), manager()); 55 56 manager()->DestroyBrowserContextServices(NULL); 57 58 ASSERT_EQ(1U, shutdown_order()->size()); 59 EXPECT_STREQ("service", (*shutdown_order())[0].c_str()); 60} 61 62// Tests that we get a simple one component depends on the other case. 63TEST_F(BrowserContextDependencyManagerUnittests, SimpleDependency) { 64 TestService parent("parent", shutdown_order(), manager()); 65 TestService child("child", shutdown_order(), manager()); 66 DependOn(&child, &parent); 67 68 manager()->DestroyBrowserContextServices(NULL); 69 70 ASSERT_EQ(2U, shutdown_order()->size()); 71 EXPECT_STREQ("child", (*shutdown_order())[0].c_str()); 72 EXPECT_STREQ("parent", (*shutdown_order())[1].c_str()); 73} 74 75// Tests two children, one parent 76TEST_F(BrowserContextDependencyManagerUnittests, TwoChildrenOneParent) { 77 TestService parent("parent", shutdown_order(), manager()); 78 TestService child1("child1", shutdown_order(), manager()); 79 TestService child2("child2", shutdown_order(), manager()); 80 DependOn(&child1, &parent); 81 DependOn(&child2, &parent); 82 83 manager()->DestroyBrowserContextServices(NULL); 84 85 ASSERT_EQ(3U, shutdown_order()->size()); 86 EXPECT_STREQ("child2", (*shutdown_order())[0].c_str()); 87 EXPECT_STREQ("child1", (*shutdown_order())[1].c_str()); 88 EXPECT_STREQ("parent", (*shutdown_order())[2].c_str()); 89} 90 91// Tests an M configuration 92TEST_F(BrowserContextDependencyManagerUnittests, MConfiguration) { 93 TestService parent1("parent1", shutdown_order(), manager()); 94 TestService parent2("parent2", shutdown_order(), manager()); 95 96 TestService child_of_1("child_of_1", shutdown_order(), manager()); 97 DependOn(&child_of_1, &parent1); 98 99 TestService child_of_12("child_of_12", shutdown_order(), manager()); 100 DependOn(&child_of_12, &parent1); 101 DependOn(&child_of_12, &parent2); 102 103 TestService child_of_2("child_of_2", shutdown_order(), manager()); 104 DependOn(&child_of_2, &parent2); 105 106 manager()->DestroyBrowserContextServices(NULL); 107 108 ASSERT_EQ(5U, shutdown_order()->size()); 109 EXPECT_STREQ("child_of_2", (*shutdown_order())[0].c_str()); 110 EXPECT_STREQ("child_of_12", (*shutdown_order())[1].c_str()); 111 EXPECT_STREQ("child_of_1", (*shutdown_order())[2].c_str()); 112 EXPECT_STREQ("parent2", (*shutdown_order())[3].c_str()); 113 EXPECT_STREQ("parent1", (*shutdown_order())[4].c_str()); 114} 115 116// Tests that it can deal with a simple diamond. 117TEST_F(BrowserContextDependencyManagerUnittests, DiamondConfiguration) { 118 TestService parent("parent", shutdown_order(), manager()); 119 120 TestService middle_row_1("middle_row_1", shutdown_order(), manager()); 121 DependOn(&middle_row_1, &parent); 122 123 TestService middle_row_2("middle_row_2", shutdown_order(), manager()); 124 DependOn(&middle_row_2, &parent); 125 126 TestService bottom("bottom", shutdown_order(), manager()); 127 DependOn(&bottom, &middle_row_1); 128 DependOn(&bottom, &middle_row_2); 129 130 manager()->DestroyBrowserContextServices(NULL); 131 132 ASSERT_EQ(4U, shutdown_order()->size()); 133 EXPECT_STREQ("bottom", (*shutdown_order())[0].c_str()); 134 EXPECT_STREQ("middle_row_2", (*shutdown_order())[1].c_str()); 135 EXPECT_STREQ("middle_row_1", (*shutdown_order())[2].c_str()); 136 EXPECT_STREQ("parent", (*shutdown_order())[3].c_str()); 137} 138 139// A final test that works with a more complex graph. 140TEST_F(BrowserContextDependencyManagerUnittests, ComplexGraph) { 141 TestService everything_depends_on_me( 142 "everything_depends_on_me", shutdown_order(), manager()); 143 144 TestService intermediary_service( 145 "intermediary_service", shutdown_order(), manager()); 146 DependOn(&intermediary_service, &everything_depends_on_me); 147 148 TestService specialized_service( 149 "specialized_service", shutdown_order(), manager()); 150 DependOn(&specialized_service, &everything_depends_on_me); 151 DependOn(&specialized_service, &intermediary_service); 152 153 TestService other_root("other_root", shutdown_order(), manager()); 154 155 TestService other_intermediary( 156 "other_intermediary", shutdown_order(), manager()); 157 DependOn(&other_intermediary, &other_root); 158 159 TestService bottom("bottom", shutdown_order(), manager()); 160 DependOn(&bottom, &specialized_service); 161 DependOn(&bottom, &other_intermediary); 162 163 manager()->DestroyBrowserContextServices(NULL); 164 165 ASSERT_EQ(6U, shutdown_order()->size()); 166 EXPECT_STREQ("bottom", (*shutdown_order())[0].c_str()); 167 EXPECT_STREQ("specialized_service", (*shutdown_order())[1].c_str()); 168 EXPECT_STREQ("other_intermediary", (*shutdown_order())[2].c_str()); 169 EXPECT_STREQ("intermediary_service", (*shutdown_order())[3].c_str()); 170 EXPECT_STREQ("other_root", (*shutdown_order())[4].c_str()); 171 EXPECT_STREQ("everything_depends_on_me", (*shutdown_order())[5].c_str()); 172} 173