OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "components/keyed_service/content/browser_context_keyed_service_factory
.h" | 5 #include "components/keyed_service/content/browser_context_keyed_service_factory
.h" |
6 | 6 |
7 #include <map> | 7 #include <map> |
8 | 8 |
9 #include "base/logging.h" | 9 #include "base/logging.h" |
10 #include "base/stl_util.h" | 10 #include "base/stl_util.h" |
11 #include "components/keyed_service/content/browser_context_dependency_manager.h" | 11 #include "components/keyed_service/content/browser_context_dependency_manager.h" |
12 #include "components/keyed_service/core/keyed_service.h" | 12 #include "components/keyed_service/core/keyed_service.h" |
13 #include "content/public/browser/browser_context.h" | 13 #include "content/public/browser/browser_context.h" |
14 | 14 |
15 void BrowserContextKeyedServiceFactory::SetTestingFactory( | 15 void BrowserContextKeyedServiceFactory::SetTestingFactory( |
16 content::BrowserContext* context, | 16 content::BrowserContext* context, |
17 TestingFactoryFunction testing_factory) { | 17 TestingFactoryFunction testing_factory) { |
| 18 if (!context) { |
| 19 testing_factories_[NULL] = testing_factory; |
| 20 return; |
| 21 } |
| 22 |
18 // Destroying the context may cause us to lose data about whether |context| | 23 // Destroying the context may cause us to lose data about whether |context| |
19 // has our preferences registered on it (since the context object itself | 24 // has our preferences registered on it (since the context object itself |
20 // isn't dead). See if we need to readd it once we've gone through normal | 25 // isn't dead). See if we need to readd it once we've gone through normal |
21 // destruction. | 26 // destruction. |
22 bool add_context = ArePreferencesSetOn(context); | 27 bool add_context = ArePreferencesSetOn(context); |
23 | 28 |
24 #ifndef NDEBUG | 29 #ifndef NDEBUG |
25 // Ensure that |context| is not marked as stale (e.g., due to it aliasing an | 30 // Ensure that |context| is not marked as stale (e.g., due to it aliasing an |
26 // instance that was destroyed in an earlier test) in order to avoid accesses | 31 // instance that was destroyed in an earlier test) in order to avoid accesses |
27 // to |context| in |BrowserContextShutdown| from causing | 32 // to |context| in |BrowserContextShutdown| from causing |
(...skipping 42 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
70 BrowserContextKeyedServices::const_iterator it = mapping_.find(context); | 75 BrowserContextKeyedServices::const_iterator it = mapping_.find(context); |
71 if (it != mapping_.end()) | 76 if (it != mapping_.end()) |
72 return it->second; | 77 return it->second; |
73 | 78 |
74 // Object not found. | 79 // Object not found. |
75 if (!create) | 80 if (!create) |
76 return NULL; // And we're forbidden from creating one. | 81 return NULL; // And we're forbidden from creating one. |
77 | 82 |
78 // Create new object. | 83 // Create new object. |
79 // Check to see if we have a per-BrowserContext testing factory that we should | 84 // Check to see if we have a per-BrowserContext testing factory that we should |
80 // use instead of default behavior. | 85 // use instead of default behavior. Check for a global factory as well. |
81 KeyedService* service = NULL; | 86 KeyedService* service = NULL; |
82 BrowserContextOverriddenTestingFunctions::const_iterator jt = | 87 BrowserContextOverriddenTestingFunctions::const_iterator jt = |
83 testing_factories_.find(context); | 88 testing_factories_.find(context); |
| 89 if (jt == testing_factories_.end()) |
| 90 jt = testing_factories_.find(NULL); |
84 if (jt != testing_factories_.end()) { | 91 if (jt != testing_factories_.end()) { |
85 if (jt->second) { | 92 if (jt->second) { |
86 if (!context->IsOffTheRecord()) | 93 if (!context->IsOffTheRecord()) |
87 RegisterUserPrefsOnBrowserContextForTest(context); | 94 RegisterUserPrefsOnBrowserContextForTest(context); |
88 service = jt->second(context); | 95 service = jt->second(context); |
89 } | 96 } |
90 } else { | 97 } else { |
91 service = BuildServiceInstanceFor(context); | 98 service = BuildServiceInstanceFor(context); |
92 } | 99 } |
93 | 100 |
(...skipping 39 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
133 | 140 |
134 void BrowserContextKeyedServiceFactory::SetEmptyTestingFactory( | 141 void BrowserContextKeyedServiceFactory::SetEmptyTestingFactory( |
135 content::BrowserContext* context) { | 142 content::BrowserContext* context) { |
136 SetTestingFactory(context, NULL); | 143 SetTestingFactory(context, NULL); |
137 } | 144 } |
138 | 145 |
139 void BrowserContextKeyedServiceFactory::CreateServiceNow( | 146 void BrowserContextKeyedServiceFactory::CreateServiceNow( |
140 content::BrowserContext* context) { | 147 content::BrowserContext* context) { |
141 GetServiceForBrowserContext(context, true); | 148 GetServiceForBrowserContext(context, true); |
142 } | 149 } |
OLD | NEW |