Teaching BrowsingDataRemover how to delete application data.

webkit/dom_storage/dom_storage_context_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 "base/bind.h"
 #include "base/file_util.h"
 #include "base/message_loop.h"
 #include "base/message_loop_proxy.h"
 #include "base/scoped_temp_dir.h"
 #include "base/threading/sequenced_worker_pool.h"
 #include "base/time.h"
 #include "base/utf_string_conversions.h"
+#include "chrome/browser/extensions/mock_extension_special_storage_policy.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "webkit/dom_storage/dom_storage_area.h"
 #include "webkit/dom_storage/dom_storage_context.h"
 #include "webkit/dom_storage/dom_storage_namespace.h"
 #include "webkit/dom_storage/dom_storage_task_runner.h"
 #include "webkit/dom_storage/dom_storage_types.h"
 #include "webkit/quota/mock_special_storage_policy.h"
 
 namespace dom_storage {
 
@@ skipping 21 matching lines @@
     context_ = new DomStorageContext(temp_dir_.path(),
                                      FilePath(),
                                      storage_policy_,
                                      task_runner_);
   }
 
   virtual void TearDown() {
     MessageLoop::current()->RunAllPending();
   }
 
+  void RemoveDataSince(const base::Time& cutoff,
+                       bool include_protected_origins) {
+    // Use a new instance to remove the contexts of temp_dir_.
+    scoped_refptr<DomStorageContext> context = new DomStorageContext(
+        temp_dir_.path(), FilePath(), storage_policy_, NULL);
+    context->DeleteDataModifiedSince(cutoff, include_protected_origins);
+  }
+
   void VerifySingleOriginRemains(const GURL& origin) {
     // Use a new instance to examine the contexts of temp_dir_.
     scoped_refptr<DomStorageContext> context =
         new DomStorageContext(temp_dir_.path(), FilePath(), NULL, NULL);
     std::vector<DomStorageContext::UsageInfo> infos;
     context->GetUsageInfo(&infos, kDontIncludeFileInfo);
     EXPECT_EQ(1u, infos.size());
-    EXPECT_EQ(origin, infos[0].origin);
+    if (infos.size() > 0)
+      EXPECT_EQ(origin, infos[0].origin);
   }
 
  protected:
   ScopedTempDir temp_dir_;
   scoped_refptr<quota::MockSpecialStoragePolicy> storage_policy_;
   scoped_refptr<MockDomStorageTaskRunner> task_runner_;
   scoped_refptr<DomStorageContext> context_;
   DISALLOW_COPY_AND_ASSIGN(DomStorageContextTest);
 };
 
 TEST_F(DomStorageContextTest, Basics) {
   // This test doesn't do much, checks that the constructor
   // initializes members properly and that invoking methods
   // on a newly created object w/o any data on disk do no harm.
   EXPECT_EQ(temp_dir_.path(), context_->localstorage_directory());
   EXPECT_EQ(FilePath(), context_->sessionstorage_directory());
   EXPECT_EQ(storage_policy_.get(), context_->special_storage_policy_.get());
   context_->PurgeMemory();
   context_->DeleteOrigin(GURL("http://chromium.org/"));
-  context_->DeleteDataModifiedSince(base::Time::Now());
+  context_->DeleteDataModifiedSince(base::Time::Now(), false);
   const int kFirstSessionStorageNamespaceId = 1;
   EXPECT_TRUE(context_->GetStorageNamespace(kLocalStorageNamespaceId));
   EXPECT_FALSE(context_->GetStorageNamespace(kFirstSessionStorageNamespaceId));
   EXPECT_EQ(kFirstSessionStorageNamespaceId, context_->AllocateSessionId());
   std::vector<DomStorageContext::UsageInfo> infos;
   context_->GetUsageInfo(&infos, kDontIncludeFileInfo);
   EXPECT_TRUE(infos.empty());
   context_->Shutdown();
 }
 
+TEST_F(DomStorageContextTest, RemoveUnprotectedOriginData) {
+  const GURL kProtectedOrigin("http://www.protected.com/");
+  storage_policy_->AddProtected(kProtectedOrigin);
+
+  // Store data for a normal and a protected origin, then remove data since
+  // the epoch.
+  NullableString16 old_value;
+  EXPECT_TRUE(context_->GetStorageNamespace(kLocalStorageNamespaceId)->
+      OpenStorageArea(kOrigin)->SetItem(kKey, kValue, &old_value));
+  EXPECT_TRUE(context_->GetStorageNamespace(kLocalStorageNamespaceId)->
+      OpenStorageArea(kProtectedOrigin)->SetItem(kKey, kValue, &old_value));
+  context_ = NULL;
+  MessageLoop::current()->RunAllPending();
+
+  RemoveDataSince(base::Time(), false);
+  VerifySingleOriginRemains(kProtectedOrigin);
+}
+
+TEST_F(DomStorageContextTest, RemoveProtectedOriginData) {
+  const GURL kProtectedOrigin("http://www.protected.com/");
+  storage_policy_->AddProtected(kProtectedOrigin);
+
+  // Store data for a normal and a protected origin, then remove data since
+  // the epoch.
+  NullableString16 old_value;
+  EXPECT_TRUE(context_->GetStorageNamespace(kLocalStorageNamespaceId)->
+      OpenStorageArea(kOrigin)->SetItem(kKey, kValue, &old_value));
+  EXPECT_TRUE(context_->GetStorageNamespace(kLocalStorageNamespaceId)->
+      OpenStorageArea(kProtectedOrigin)->SetItem(kKey, kValue, &old_value));
+  context_ = NULL;
+  MessageLoop::current()->RunAllPending();
+
+  RemoveDataSince(base::Time(), true);
+
+  // Create a new context that points to the same directory, see that
+  // both origins have been removed.
+  std::vector<DomStorageContext::UsageInfo> infos;
+  context_ = new DomStorageContext(temp_dir_.path(), FilePath(), NULL, NULL);
+  context_->GetUsageInfo(&infos, kDontIncludeFileInfo);
+  EXPECT_TRUE(infos.empty());
+}
+
 TEST_F(DomStorageContextTest, UsageInfo) {
   // Should be empty initially
   std::vector<DomStorageContext::UsageInfo> infos;
   context_->GetUsageInfo(&infos, kDontIncludeFileInfo);
   EXPECT_TRUE(infos.empty());
   context_->GetUsageInfo(&infos, kDoIncludeFileInfo);
   EXPECT_TRUE(infos.empty());
 
   // Put some data into local storage and shutdown the context
   // to ensure data is written to disk.
@@ skipping 71 matching lines @@
   context_->SetClearLocalState(true);
   context_->SaveSessionState();  // Should override clear behavior.
   context_->Shutdown();
   context_ = NULL;
   MessageLoop::current()->RunAllPending();
 
   VerifySingleOriginRemains(kSessionOnlyOrigin);
 }
 
 }  // namespace dom_storage