Implement rules removal for WebRequestRulesRegistry

chrome/browser/extensions/api/declarative_webrequest/webrequest_rules_registry_unittest.cc

 // Copyright (c) 2012 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 "chrome/browser/extensions/api/declarative_webrequest/webrequest_rules_registry.h"
 
 #include <vector>
 
 #include "base/memory/linked_ptr.h"
 #include "base/message_loop.h"
 #include "base/values.h"
 #include "content/test/test_browser_thread.h"
 #include "net/url_request/url_request_test_util.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
 const char kExtensionId[] = "ext1";
+const char kExtensionId2[] = "ext2";
 const char kRuleId1[] = "rule1";
 const char kRuleId2[] = "rule2";
 
 const char kCancelRequestType[] = "experimental.webRequest.CancelRequest";
 const char kRequestMatcher[] = "experimental.webRequest.RequestMatcher";
 const char kInstanceType[] = "instanceType";
 }
 
 namespace extensions {
 
@@ skipping 105 matching lines @@
       matches.end());
 
   GURL foobar_url("http://www.foobar.com");
   TestURLRequest foobar_request(foobar_url, NULL);
   matches = registry->GetMatches(&foobar_request);
   EXPECT_EQ(1u, matches.size());
   EXPECT_TRUE(matches.find(std::make_pair(kExtensionId, kRuleId2)) !=
       matches.end());
 }
 
+TEST_F(WebRequestRulesRegistryTest, RemoveRulesImpl) {
+  scoped_refptr<WebRequestRulesRegistry> registry(new WebRequestRulesRegistry);
+  std::string error;
+
+  // Setup RulesRegistry to contain two rules.
+  std::vector<linked_ptr<RulesRegistry::Rule> > rules_to_add;
+  rules_to_add.push_back(CreateRule1());
+  rules_to_add.push_back(CreateRule2());
+  error = registry->AddRules(kExtensionId, rules_to_add);
+  EXPECT_TRUE(error.empty());
+
+  // Verify initial state.
+  std::vector<linked_ptr<RulesRegistry::Rule> > registered_rules;
+  registry->GetAllRules(kExtensionId, &registered_rules);
+  EXPECT_EQ(2u, registered_rules.size());
+
+  // Remove first rule.
+  std::vector<std::string> rules_to_remove;
+  rules_to_remove.push_back(kRuleId1);
+  error = registry->RemoveRules(kExtensionId, rules_to_remove);
+  EXPECT_TRUE(error.empty());
+
+  // Verify that only one rule is left.
+  registered_rules.clear();
+  registry->GetAllRules(kExtensionId, &registered_rules);
+  EXPECT_EQ(1u, registered_rules.size());
+
+  // Now rules_to_remove contains both rules, i.e. one that does not exist in
+  // the rules registry anymore. Effectively we only remove the second rule.
+  rules_to_remove.push_back(kRuleId2);
+  error = registry->RemoveRules(kExtensionId, rules_to_remove);
+  EXPECT_TRUE(error.empty());
+
+  // Verify that everything is gone.
+  registered_rules.clear();
+  registry->GetAllRules(kExtensionId, &registered_rules);
+  EXPECT_EQ(0u, registered_rules.size());
+
+  EXPECT_TRUE(registry->IsEmpty());
+}
+
+TEST_F(WebRequestRulesRegistryTest, RemoveAllRulesImpl) {
+  scoped_refptr<WebRequestRulesRegistry> registry(new WebRequestRulesRegistry);
+  std::string error;
+
+  // Setup RulesRegistry to contain two rules, one for each extension.
+  std::vector<linked_ptr<RulesRegistry::Rule> > rules_to_add(1);
+  rules_to_add[0] = CreateRule1();
+  error = registry->AddRules(kExtensionId, rules_to_add);
+  EXPECT_TRUE(error.empty());
+
+  rules_to_add[0] = CreateRule2();
+  error = registry->AddRules(kExtensionId2, rules_to_add);
+  EXPECT_TRUE(error.empty());
+
+  // Verify initial state.
+  std::vector<linked_ptr<RulesRegistry::Rule> > registered_rules;
+  registry->GetAllRules(kExtensionId, &registered_rules);
+  EXPECT_EQ(1u, registered_rules.size());
+  registered_rules.clear();
+  registry->GetAllRules(kExtensionId2, &registered_rules);
+  EXPECT_EQ(1u, registered_rules.size());
+
+  // Remove rule of first extension.
+  error = registry->RemoveAllRules(kExtensionId);
+  EXPECT_TRUE(error.empty());
+
+  // Verify that only the first rule is deleted.
+  registered_rules.clear();
+  registry->GetAllRules(kExtensionId, &registered_rules);
+  EXPECT_EQ(0u, registered_rules.size());
+  registered_rules.clear();
+  registry->GetAllRules(kExtensionId2, &registered_rules);
+  EXPECT_EQ(1u, registered_rules.size());
+
+  // Test removing rules if none exist.
+  error = registry->RemoveAllRules(kExtensionId);
+  EXPECT_TRUE(error.empty());
+
+  // Remove rule from second extension.
+  error = registry->RemoveAllRules(kExtensionId2);
+  EXPECT_TRUE(error.empty());
+
+  EXPECT_TRUE(registry->IsEmpty());
+}
+
 }  // namespace extensions