OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 "chrome/browser/extensions/api/declarative_webrequest/webrequest_rules_
registry.h" | 5 #include "chrome/browser/extensions/api/declarative_webrequest/webrequest_rules_
registry.h" |
6 | 6 |
7 #include <vector> | 7 #include <vector> |
8 | 8 |
9 #include "base/memory/linked_ptr.h" | 9 #include "base/memory/linked_ptr.h" |
10 #include "base/message_loop.h" | 10 #include "base/message_loop.h" |
(...skipping 24 matching lines...) Expand all Loading... |
35 class TestWebRequestRulesRegistry : public WebRequestRulesRegistry { | 35 class TestWebRequestRulesRegistry : public WebRequestRulesRegistry { |
36 public: | 36 public: |
37 TestWebRequestRulesRegistry() | 37 TestWebRequestRulesRegistry() |
38 : WebRequestRulesRegistry(NULL, NULL), | 38 : WebRequestRulesRegistry(NULL, NULL), |
39 num_clear_cache_calls_(0) {} | 39 num_clear_cache_calls_(0) {} |
40 | 40 |
41 // Returns how often the in-memory caches of the renderers were instructed | 41 // Returns how often the in-memory caches of the renderers were instructed |
42 // to be cleared. | 42 // to be cleared. |
43 int num_clear_cache_calls() const { return num_clear_cache_calls_; } | 43 int num_clear_cache_calls() const { return num_clear_cache_calls_; } |
44 | 44 |
| 45 // How many rules are there which have some conditions not triggered by URL |
| 46 // matches. |
| 47 size_t RulesWithoutTriggers() const { |
| 48 return rules_with_untriggered_conditions_for_test().size(); |
| 49 } |
| 50 |
45 protected: | 51 protected: |
46 virtual ~TestWebRequestRulesRegistry() {} | 52 virtual ~TestWebRequestRulesRegistry() {} |
47 | 53 |
48 virtual base::Time GetExtensionInstallationTime( | 54 virtual base::Time GetExtensionInstallationTime( |
49 const std::string& extension_id) const { | 55 const std::string& extension_id) const { |
50 if (extension_id == kExtensionId) | 56 if (extension_id == kExtensionId) |
51 return base::Time() + base::TimeDelta::FromDays(1); | 57 return base::Time() + base::TimeDelta::FromDays(1); |
52 else if (extension_id == kExtensionId2) | 58 else if (extension_id == kExtensionId2) |
53 return base::Time() + base::TimeDelta::FromDays(2); | 59 return base::Time() + base::TimeDelta::FromDays(2); |
54 else | 60 else |
(...skipping 201 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
256 rules_to_add.push_back(CreateRule1()); | 262 rules_to_add.push_back(CreateRule1()); |
257 rules_to_add.push_back(CreateRule2()); | 263 rules_to_add.push_back(CreateRule2()); |
258 error = registry->AddRules(kExtensionId, rules_to_add); | 264 error = registry->AddRules(kExtensionId, rules_to_add); |
259 EXPECT_EQ("", error); | 265 EXPECT_EQ("", error); |
260 EXPECT_EQ(1, registry->num_clear_cache_calls()); | 266 EXPECT_EQ(1, registry->num_clear_cache_calls()); |
261 | 267 |
262 // Verify initial state. | 268 // Verify initial state. |
263 std::vector<linked_ptr<RulesRegistry::Rule> > registered_rules; | 269 std::vector<linked_ptr<RulesRegistry::Rule> > registered_rules; |
264 registry->GetAllRules(kExtensionId, ®istered_rules); | 270 registry->GetAllRules(kExtensionId, ®istered_rules); |
265 EXPECT_EQ(2u, registered_rules.size()); | 271 EXPECT_EQ(2u, registered_rules.size()); |
| 272 EXPECT_EQ(1u, registry->RulesWithoutTriggers()); |
266 | 273 |
267 // Remove first rule. | 274 // Remove first rule. |
268 std::vector<std::string> rules_to_remove; | 275 std::vector<std::string> rules_to_remove; |
269 rules_to_remove.push_back(kRuleId1); | 276 rules_to_remove.push_back(kRuleId1); |
270 error = registry->RemoveRules(kExtensionId, rules_to_remove); | 277 error = registry->RemoveRules(kExtensionId, rules_to_remove); |
271 EXPECT_EQ("", error); | 278 EXPECT_EQ("", error); |
272 EXPECT_EQ(2, registry->num_clear_cache_calls()); | 279 EXPECT_EQ(2, registry->num_clear_cache_calls()); |
273 | 280 |
274 // Verify that only one rule is left. | 281 // Verify that only one rule is left. |
275 registered_rules.clear(); | 282 registered_rules.clear(); |
276 registry->GetAllRules(kExtensionId, ®istered_rules); | 283 registry->GetAllRules(kExtensionId, ®istered_rules); |
277 EXPECT_EQ(1u, registered_rules.size()); | 284 EXPECT_EQ(1u, registered_rules.size()); |
| 285 EXPECT_EQ(1u, registry->RulesWithoutTriggers()); |
278 | 286 |
279 // Now rules_to_remove contains both rules, i.e. one that does not exist in | 287 // Now rules_to_remove contains both rules, i.e. one that does not exist in |
280 // the rules registry anymore. Effectively we only remove the second rule. | 288 // the rules registry anymore. Effectively we only remove the second rule. |
281 rules_to_remove.push_back(kRuleId2); | 289 rules_to_remove.push_back(kRuleId2); |
282 error = registry->RemoveRules(kExtensionId, rules_to_remove); | 290 error = registry->RemoveRules(kExtensionId, rules_to_remove); |
283 EXPECT_EQ("", error); | 291 EXPECT_EQ("", error); |
284 EXPECT_EQ(3, registry->num_clear_cache_calls()); | 292 EXPECT_EQ(3, registry->num_clear_cache_calls()); |
285 | 293 |
286 // Verify that everything is gone. | 294 // Verify that everything is gone. |
287 registered_rules.clear(); | 295 registered_rules.clear(); |
288 registry->GetAllRules(kExtensionId, ®istered_rules); | 296 registry->GetAllRules(kExtensionId, ®istered_rules); |
289 EXPECT_EQ(0u, registered_rules.size()); | 297 EXPECT_EQ(0u, registered_rules.size()); |
| 298 EXPECT_EQ(0u, registry->RulesWithoutTriggers()); |
290 | 299 |
291 EXPECT_TRUE(registry->IsEmpty()); | 300 EXPECT_TRUE(registry->IsEmpty()); |
292 } | 301 } |
293 | 302 |
294 TEST_F(WebRequestRulesRegistryTest, RemoveAllRulesImpl) { | 303 TEST_F(WebRequestRulesRegistryTest, RemoveAllRulesImpl) { |
295 scoped_refptr<TestWebRequestRulesRegistry> registry( | 304 scoped_refptr<TestWebRequestRulesRegistry> registry( |
296 new TestWebRequestRulesRegistry()); | 305 new TestWebRequestRulesRegistry()); |
297 std::string error; | 306 std::string error; |
298 | 307 |
299 // Setup RulesRegistry to contain two rules, one for each extension. | 308 // Setup RulesRegistry to contain two rules, one for each extension. |
(...skipping 120 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
420 // The redirect by the first extension is ignored due to the ignore rule. | 429 // The redirect by the first extension is ignored due to the ignore rule. |
421 ASSERT_EQ(1u, deltas.size()); | 430 ASSERT_EQ(1u, deltas.size()); |
422 LinkedPtrEventResponseDelta effective_rule = *(deltas.begin()); | 431 LinkedPtrEventResponseDelta effective_rule = *(deltas.begin()); |
423 | 432 |
424 EXPECT_EQ(kExtensionId2, effective_rule->extension_id); | 433 EXPECT_EQ(kExtensionId2, effective_rule->extension_id); |
425 EXPECT_EQ(base::Time() + base::TimeDelta::FromDays(2), | 434 EXPECT_EQ(base::Time() + base::TimeDelta::FromDays(2), |
426 effective_rule->extension_install_time); | 435 effective_rule->extension_install_time); |
427 EXPECT_EQ(GURL("http://www.bar.com"), effective_rule->new_url); | 436 EXPECT_EQ(GURL("http://www.bar.com"), effective_rule->new_url); |
428 } | 437 } |
429 } // namespace extensions | 438 } // namespace extensions |
OLD | NEW |