Teaching BrowsingDataRemover how to delete application data.

chrome/browser/browsing_data_helper_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/browsing_data_helper.h"
 
 #include "base/stringprintf.h"
+#include "chrome/browser/extensions/mock_extension_special_storage_policy.h"
 #include "chrome/common/url_constants.h"
 #include "content/public/common/url_constants.h"
+#include "googleurl/src/gurl.h"
 #include "testing/gtest/include/gtest/gtest.h"
-#include "googleurl/src/gurl.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebString.h"
 
 namespace {
 
+const char kTestOrigin1[] = "http://host1:1/";
+const char kTestOrigin2[] = "http://host2:1/";
+const char kTestOrigin3[] = "http://host3:1/";
+const char kTestOriginExt[] = "chrome-extension://abcdefghijklmnopqrstuvwxyz/";
+const char kTestOriginDevTools[] = "chrome-devtools://abcdefghijklmnopqrstuvw/";
+
+const GURL kOrigin1(kTestOrigin1);
+const GURL kOrigin2(kTestOrigin2);
+const GURL kOrigin3(kTestOrigin3);
+const GURL kOriginExt(kTestOriginExt);
+const GURL kOriginDevTools(kTestOriginDevTools);
+
+const int kExtension = BrowsingDataHelper::EXTENSION;
+const int kProtected = BrowsingDataHelper::PROTECTED_WEB;
+const int kUnprotected = BrowsingDataHelper::UNPROTECTED_WEB;
+
 class BrowsingDataHelperTest : public testing::Test {
  public:
   BrowsingDataHelperTest() {}
   virtual ~BrowsingDataHelperTest() {}
 
   bool IsValidScheme(const std::string& scheme) {
     GURL test(scheme + "://example.com");
     return (BrowsingDataHelper::HasValidScheme(test) &&
             BrowsingDataHelper::IsValidScheme(scheme) &&
             BrowsingDataHelper::IsValidScheme(
                 WebKit::WebString::fromUTF8(scheme)));
   }
 
+  bool Match(const GURL& origin,
+             int mask,
+             ExtensionSpecialStoragePolicy* policy) {
+    return BrowsingDataHelper::DoesOriginMatchMask(origin, mask, policy);
+  }
+
  private:
   DISALLOW_COPY_AND_ASSIGN(BrowsingDataHelperTest);
 };
 
 TEST_F(BrowsingDataHelperTest, WebSafeSchemesAreValid) {
   EXPECT_TRUE(IsValidScheme(chrome::kHttpScheme));
   EXPECT_TRUE(IsValidScheme(chrome::kHttpsScheme));
   EXPECT_TRUE(IsValidScheme(chrome::kFtpScheme));
   EXPECT_TRUE(IsValidScheme(chrome::kDataScheme));
   EXPECT_TRUE(IsValidScheme("feed"));
   EXPECT_TRUE(IsValidScheme(chrome::kBlobScheme));
   EXPECT_TRUE(IsValidScheme(chrome::kFileSystemScheme));
 
   EXPECT_FALSE(IsValidScheme("invalid-scheme-i-just-made-up"));
 }
 
 TEST_F(BrowsingDataHelperTest, ChromeSchemesAreInvalid) {
   EXPECT_FALSE(IsValidScheme(chrome::kExtensionScheme));
   EXPECT_FALSE(IsValidScheme(chrome::kAboutScheme));
   EXPECT_FALSE(IsValidScheme(chrome::kChromeDevToolsScheme));
   EXPECT_FALSE(IsValidScheme(chrome::kChromeInternalScheme));
   EXPECT_FALSE(IsValidScheme(chrome::kChromeUIScheme));
   EXPECT_FALSE(IsValidScheme(chrome::kJavaScriptScheme));
   EXPECT_FALSE(IsValidScheme(chrome::kMailToScheme));
   EXPECT_FALSE(IsValidScheme(chrome::kMetadataScheme));
   EXPECT_FALSE(IsValidScheme(chrome::kSwappedOutScheme));
   EXPECT_FALSE(IsValidScheme(chrome::kViewSourceScheme));
 }
 
+TEST_F(BrowsingDataHelperTest, TestMatches) {
+  scoped_refptr<MockExtensionSpecialStoragePolicy> mock_policy =
+          new MockExtensionSpecialStoragePolicy;
+  // Protect kOrigin1.
+  mock_policy->AddProtected(kOrigin1.GetOrigin());
+
+  EXPECT_FALSE(Match(kOrigin1, kUnprotected, mock_policy));

[Bernhard Bauer, 2012/05/31 15:57:31]

You might be able to make this a bit easier to test if you add a method that
classifies a given origin as one of (protected, unprotected, extension); then
you can test that method instead of testing all bitmask combinations.

[Mike West, 2012/06/01 13:35:17]

Extensions are both protected and extensions. Bookmark applications are both
unprotected and extensions.

I'm not sure how we'd get around that with a single method.

[Bernhard Bauer, 2012/06/01 13:59:35]

We're defining extension as "everything with a chrome-extension:// scheme" here.
In that sense, an extension is not protected web content (as it's not web
content), and a bookmark application is not an extension (as it doesn't contain
anything with a chrome-extension:// scheme). That means that every origin
matches at most one of (protected web, unprotected web, extension). Or am I
missing something here?

[Mike West, 2012/06/01 14:11:53]

No, you're right. Again, I successfully confused myself while talking to you
yesterday.

But I still think I need to check all bitmask combinations. :)

[Bernhard Bauer, 2012/06/04 09:06:43]

In that case I feel obligated to point out that you're missing kUnprotected |
kExtension here ;-)

(If you would for example enumerate all the bitmask values in a loop, you could
avoid that. Just sayin' ;-)

+  EXPECT_TRUE(Match(kOrigin2, kUnprotected, mock_policy));
+  EXPECT_FALSE(Match(kOriginExt, kUnprotected, mock_policy));
+  EXPECT_FALSE(Match(kOriginDevTools, kUnprotected, mock_policy));
+
+  EXPECT_TRUE(Match(kOrigin1, kProtected, mock_policy));
+  EXPECT_FALSE(Match(kOrigin2, kProtected, mock_policy));
+  EXPECT_FALSE(Match(kOriginExt, kProtected, mock_policy));
+  EXPECT_FALSE(Match(kOriginDevTools, kProtected, mock_policy));
+
+  EXPECT_FALSE(Match(kOrigin1, kExtension, mock_policy));
+  EXPECT_FALSE(Match(kOrigin2, kExtension, mock_policy));
+  EXPECT_TRUE(Match(kOriginExt, kExtension, mock_policy));
+  EXPECT_TRUE(Match(kOriginDevTools, kExtension, mock_policy));
+
+  EXPECT_TRUE(Match(kOrigin1, kUnprotected | kProtected, mock_policy));
+  EXPECT_TRUE(Match(kOrigin2, kUnprotected | kProtected, mock_policy));
+  EXPECT_FALSE(Match(kOriginExt, kUnprotected | kProtected, mock_policy));
+  EXPECT_FALSE(Match(kOriginDevTools, kUnprotected | kProtected, mock_policy));
+
+  EXPECT_TRUE(Match(kOrigin1, kProtected | kExtension, mock_policy));
+  EXPECT_FALSE(Match(kOrigin2, kProtected | kExtension, mock_policy));
+  EXPECT_TRUE(Match(kOriginExt, kProtected | kExtension, mock_policy));
+  EXPECT_TRUE(Match(kOriginDevTools, kProtected | kExtension, mock_policy));
+
+  EXPECT_TRUE(Match(kOrigin1, kUnprotected | kProtected | kExtension,
+        mock_policy));
+  EXPECT_TRUE(Match(kOrigin2, kUnprotected | kProtected | kExtension,
+        mock_policy));
+  EXPECT_TRUE(Match(kOriginExt, kUnprotected | kProtected | kExtension,
+        mock_policy));
+  EXPECT_TRUE(Match(kOriginDevTools, kUnprotected | kProtected | kExtension,
+        mock_policy));
+}
+
 }  // namespace