Index: webkit/quota/mock_special_storage_policy.h |
diff --git a/webkit/quota/mock_special_storage_policy.h b/webkit/quota/mock_special_storage_policy.h |
index b04d16d3dd6a0218067523b9a34e345cde503e47..7e54a6e311c657e129226f3a206b61b5925d4350 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_; |