1 files changed, 8 insertions, 0 deletions
diff --git a/webkit/quota/mock_special_storage_policy.h b/webkit/quota/mock_special_storage_policy.h
index b04d16d..7e54a6e 100644
--- a/webkit/quota/mock_special_storage_policy.h
+++ b/webkit/quota/mock_special_storage_policy.h
@@ -19,6 +19,7 @@ class MockSpecialStoragePolicy : public quota::SpecialStoragePolicy {
 
   virtual bool IsStorageProtected(const GURL& origin);
   virtual bool IsStorageUnlimited(const GURL& origin);
+  virtual bool IsStorageSessionOnly(const GURL& origin);
   virtual bool IsFileHandler(const std::string& extension_id);
 
   void AddProtected(const GURL& origin) {
@@ -29,6 +30,10 @@ class MockSpecialStoragePolicy : public quota::SpecialStoragePolicy {
     unlimited_.insert(origin);
   }
 
+  void AddSessionOnly(const GURL& origin) {
+    session_only_.insert(origin);
+  }
+
   void AddFileHandler(const std::string& id) {
     file_handlers_.insert(id);
   }
@@ -40,7 +45,9 @@ class MockSpecialStoragePolicy : public quota::SpecialStoragePolicy {
   void Reset() {
     protected_.clear();
     unlimited_.clear();
+    session_only_.clear();
     file_handlers_.clear();
+    all_unlimited_ = false;
   }
 
   void NotifyChanged() {
@@ -50,6 +57,7 @@ class MockSpecialStoragePolicy : public quota::SpecialStoragePolicy {
  private:
   std::set<GURL> protected_;
   std::set<GURL> unlimited_;
+  std::set<GURL> session_only_;
   std::set<std::string> file_handlers_;
 
   bool all_unlimited_;