Implement correct logic for URLPatternSet set operators.

extensions/common/url_pattern_set_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 "extensions/common/url_pattern_set.h"
 
 #include "base/values.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "url/gurl.h"
 
 namespace extensions {
 
 namespace {
 
 void AddPattern(URLPatternSet* set, const std::string& pattern) {
   int schemes = URLPattern::SCHEME_ALL;
   set->AddPattern(URLPattern(schemes, pattern));
 }
 
+URLPatternSet GetAllURLsPattern() {
+  URLPatternSet set;
+  URLPattern pattern;
+  pattern.SetMatchAllURLs(true);
+  set.AddPattern(pattern);
+  return set;
+}
+
 URLPatternSet Patterns(const std::string& pattern) {
   URLPatternSet set;
   AddPattern(&set, pattern);
   return set;
 }
 
 URLPatternSet Patterns(const std::string& pattern1,
                        const std::string& pattern2) {
   URLPatternSet set;
   AddPattern(&set, pattern1);
@@ skipping 356 matching lines @@
     AddPattern(&expected, yahoo_a);
     AddPattern(&expected, yahoo_b);
     AddPattern(&expected, yahoo_c);
     AddPattern(&expected, reddit_a);
     AddPattern(&expected, reddit_b);
     AddPattern(&expected, reddit_c);
     EXPECT_EQ(expected, result);
   }
 }
 
+TEST(URLPatternSetTest, AllURLsSpecialCases) {
+  URLPatternSet empty;
+  URLPatternSet all_urls = GetAllURLsPattern();
+  URLPatternSet specific;
+  AddPattern(&specific, "http://www.google.com/*");
+  AddPattern(&specific, "http://www.yahoo.com/*");
+
+  {
+    URLPatternSet result;
+    result.AddPatterns(specific);
+    EXPECT_EQ(specific, result);
+    result.AddPatterns(all_urls);
+    EXPECT_EQ(all_urls, result);
+  }
+  {
+    URLPatternSet result;
+    result.AddPatterns(all_urls);
+    EXPECT_EQ(all_urls, result);
+    result.AddPatterns(specific);
+    EXPECT_EQ(all_urls, result);
+  }
+  {
+    URLPatternSet result;
+    URLPatternSet::CreateUnion(all_urls, specific, &result);
+    EXPECT_EQ(all_urls, result);
+  }
+  {
+    URLPatternSet result;
+    URLPatternSet::CreateUnion(specific, all_urls, &result);
+    EXPECT_EQ(all_urls, result);
+  }
+  {
+    URLPatternSet result;
+    URLPatternSet::CreateIntersection(all_urls, specific, &result);
+    EXPECT_EQ(specific, result);
+  }
+  {
+    URLPatternSet result;
+    URLPatternSet::CreateIntersection(specific, all_urls, &result);
+    EXPECT_EQ(specific, result);
+  }
+  {
+    URLPatternSet result;
+    URLPatternSet::CreateDifference(all_urls, specific, &result);
+    EXPECT_EQ(all_urls, result);
+  }
+  {
+    URLPatternSet result;
+    URLPatternSet::CreateDifference(specific, all_urls, &result);
+    EXPECT_EQ(empty, result);
+  }
+}
+
+TEST(URLPatternSetTest, OverlappingURLsSpecialCases) {
+  URLPatternSet empty;
+  URLPatternSet specific;
+  AddPattern(&specific, "http://www.google.com/*");
+  AddPattern(&specific, "http://www.yahoo.com/*");
+
+  URLPatternSet overlapping;
+  AddPattern(&overlapping, "*://www.google.com/*");
+  AddPattern(&overlapping, "http://*.google.com/*");
+
+  {
+    URLPatternSet result;
+    result.AddPatterns(overlapping);
+    EXPECT_EQ(overlapping, result);
+
+    result.AddPatterns(specific);
+    URLPatternSet expected;
+    expected.AddPatterns(overlapping);
+    AddPattern(&expected, "http://www.yahoo.com/*");
+    EXPECT_EQ(expected, result);
+  }
+  {
+    URLPatternSet result;
+    URLPatternSet::CreateUnion(specific, overlapping, &result);
+
+    URLPatternSet expected;
+    expected.AddPatterns(overlapping);
+    AddPattern(&expected, "http://www.yahoo.com/*");
+    EXPECT_EQ(expected, result);
+  }
+  {
+    URLPatternSet result;
+    URLPatternSet::CreateIntersection(specific, overlapping, &result);
+    URLPatternSet expected;
+    AddPattern(&expected, "http://www.google.com/*");
+    EXPECT_EQ(expected, result);
+  }
+  {
+    URLPatternSet result;
+    URLPatternSet::CreateDifference(specific, overlapping, &result);
+    URLPatternSet expected;
+    AddPattern(&expected, "http://www.yahoo.com/*");
+    EXPECT_EQ(expected, result);
+  }
+  {
+    URLPatternSet result;
+    URLPatternSet::CreateDifference(overlapping, specific, &result);
+    EXPECT_EQ(overlapping, result);
+  }
+}
+
 }  // namespace extensions