Index: webkit/fileapi/file_system_context_unittest.cc |
diff --git a/webkit/fileapi/file_system_context_unittest.cc b/webkit/fileapi/file_system_context_unittest.cc |
index 2992dfb9006b05a7d2b7feb983329cd29870432f..19da15777426e790b81a322de97d263289afd8ef 100644 |
--- a/webkit/fileapi/file_system_context_unittest.cc |
+++ b/webkit/fileapi/file_system_context_unittest.cc |
@@ -12,6 +12,7 @@ |
#include "base/string_number_conversions.h" |
#include "googleurl/src/gurl.h" |
#include "testing/gtest/include/gtest/gtest.h" |
+#include "webkit/quota/mock_special_storage_policy.h" |
#include "webkit/quota/quota_manager.h" |
namespace fileapi { |
@@ -24,21 +25,6 @@ static const char* const kTestOrigins[] = { |
"file:///", |
}; |
-class TestSpecialStoragePolicy : public quota::SpecialStoragePolicy { |
- public: |
- virtual bool IsStorageProtected(const GURL& origin) { |
- return false; |
- } |
- |
- virtual bool IsStorageUnlimited(const GURL& origin) { |
- return origin == GURL(kTestOrigins[1]); |
- } |
- |
- virtual bool IsFileHandler(const std::string& extension_id) { |
- return false; |
- } |
-}; |
- |
scoped_refptr<FileSystemContext> NewFileSystemContext( |
bool allow_file_access, |
bool unlimited_quota, |
@@ -81,7 +67,10 @@ TEST(FileSystemContextTest, IsStorageUnlimited) { |
} |
// With SpecialStoragePolicy. |
- scoped_refptr<TestSpecialStoragePolicy> policy(new TestSpecialStoragePolicy); |
+ scoped_refptr<quota::MockSpecialStoragePolicy> policy( |
+ new quota::MockSpecialStoragePolicy); |
+ policy->AddUnlimited(GURL(kTestOrigins[1])); |
+ |
context = NewFileSystemContext(false, false, policy); |
for (size_t i = 0; i < ARRAYSIZE_UNSAFE(kTestOrigins); ++i) { |
SCOPED_TRACE(testing::Message() << "IsStorageUnlimited /w policy #" |