Appcache::OnCorruptionDetected handling

webkit/browser/appcache/appcache_disk_cache_unittest.cc

+// Copyright (c) 2014 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/files/scoped_temp_dir.h"
+#include "base/message_loop/message_loop_proxy.h"
+#include "base/run_loop.h"
+#include "net/base/io_buffer.h"
+#include "net/base/net_errors.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "webkit/browser/appcache/appcache_disk_cache.h"
+
+namespace appcache {
+
+class AppCacheDiskCacheTest : public testing::Test {
+ public:
+  AppCacheDiskCacheTest() {
+  }

[jsbell, 2014/01/24 21:13:44]

Nit: put on previous line

[michaeln, 2014/01/28 22:12:16]

Done.

+
+  virtual void SetUp() OVERRIDE {
+    // Use the current thread for the DiskCache's cache_thread.
+    message_loop_.reset(new base::MessageLoopForIO());
+    cache_thread_ = base::MessageLoopProxy::current();
+    ASSERT_TRUE(directory_.CreateUniqueTempDir());
+    completion_callback_ = base::Bind(
+        &AppCacheDiskCacheTest::OnComplete,
+        base::Unretained(this));
+  }
+
+  virtual void TearDown() OVERRIDE {
+    base::RunLoop().RunUntilIdle();
+    message_loop_.reset(NULL);
+  }
+
+  void FlushCacheTasks() {
+    base::RunLoop().RunUntilIdle();
+  }
+
+  void OnComplete(int err) {
+    completion_results_.push_back(err);
+  }
+
+  base::ScopedTempDir directory_;
+  scoped_ptr<base::MessageLoop> message_loop_;
+  scoped_refptr<base::MessageLoopProxy> cache_thread_;
+  net::CompletionCallback completion_callback_;
+  std::vector<int> completion_results_;
+
+  static const int k10MBytes = 10 * 1024 * 1024;
+};
+
+TEST_F(AppCacheDiskCacheTest, DisablePriorToInitCompletion) {
+  AppCacheDiskCache::Entry* entry = NULL;
+
+  // Create an instance and start it initializing, queue up
+  // one of each kind of "entry" function.
+  scoped_ptr<AppCacheDiskCache> disk_cache (new AppCacheDiskCache);
+  EXPECT_FALSE(disk_cache->is_disabled());
+  disk_cache->InitWithDiskBackend(
+      directory_.path(), k10MBytes, false, cache_thread_,
+      completion_callback_);
+  disk_cache->CreateEntry(1, &entry, completion_callback_);
+  disk_cache->OpenEntry(2, &entry, completion_callback_);
+  disk_cache->DoomEntry(3, completion_callback_);
+
+  // Pull the plug on all that.
+  EXPECT_FALSE(disk_cache->is_disabled());
+  disk_cache->Disable();
+  EXPECT_TRUE(disk_cache->is_disabled());
+
+  FlushCacheTasks();
+
+  EXPECT_EQ(NULL, entry);
+  EXPECT_EQ(4u, completion_results_.size());
+  for (std::vector<int>::const_iterator iter = completion_results_.begin();
+       iter < completion_results_.end(); ++iter) {
+    EXPECT_EQ(net::ERR_ABORTED, *iter);
+  }
+
+  // Ensure the directory can be deleted at this point.
+  EXPECT_TRUE(base::DirectoryExists(directory_.path()));
+  EXPECT_FALSE(base::IsDirectoryEmpty(directory_.path()));
+  EXPECT_TRUE(base::DeleteFile(directory_.path(), true));
+  EXPECT_FALSE(base::DirectoryExists(directory_.path()));
+}
+
+TEST_F(AppCacheDiskCacheTest, DisableAfterInitted) {
+  // Create an instance and let it fully init.
+  scoped_ptr<AppCacheDiskCache> disk_cache (new AppCacheDiskCache);
+  EXPECT_FALSE(disk_cache->is_disabled());
+  disk_cache->InitWithDiskBackend(
+      directory_.path(), k10MBytes, false, cache_thread_,
+      completion_callback_);
+  FlushCacheTasks();
+  EXPECT_EQ(1u, completion_results_.size());
+  EXPECT_EQ(net::OK, completion_results_[0]);
+
+  // Pull the plug
+  disk_cache->Disable();
+  FlushCacheTasks();
+
+  // Ensure the directory can be deleted at this point.
+  EXPECT_TRUE(base::DirectoryExists(directory_.path()));
+  EXPECT_FALSE(base::IsDirectoryEmpty(directory_.path()));
+  EXPECT_TRUE(base::DeleteFile(directory_.path(), true));
+  EXPECT_FALSE(base::DirectoryExists(directory_.path()));
+
+  // Methods should return immediately when disabled and not invoke
+  // the callback at all.
+  AppCacheDiskCache::Entry* entry = NULL;
+  completion_results_.clear();
+  EXPECT_EQ(net::ERR_ABORTED,
+            disk_cache->CreateEntry(1, &entry, completion_callback_));
+  EXPECT_EQ(net::ERR_ABORTED,
+            disk_cache->OpenEntry(2, &entry, completion_callback_));
+  EXPECT_EQ(net::ERR_ABORTED,
+            disk_cache->DoomEntry(3, completion_callback_));
+  FlushCacheTasks();
+  EXPECT_TRUE(completion_results_.empty());
+}
+
+TEST_F(AppCacheDiskCacheTest, DisableWithEntriesOpen) {
+  // Create an instance and let it fully init.
+  scoped_ptr<AppCacheDiskCache> disk_cache (new AppCacheDiskCache);
+  EXPECT_FALSE(disk_cache->is_disabled());
+  disk_cache->InitWithDiskBackend(
+      directory_.path(), k10MBytes, false, cache_thread_,
+      completion_callback_);
+  FlushCacheTasks();
+  EXPECT_EQ(1u, completion_results_.size());
+  EXPECT_EQ(net::OK, completion_results_[0]);
+
+  // Create/open some entries.
+  AppCacheDiskCache::Entry* entry1 = NULL;
+  AppCacheDiskCache::Entry* entry2 = NULL;
+  completion_results_.clear();
+  EXPECT_EQ(net::ERR_IO_PENDING,
+            disk_cache->CreateEntry(1, &entry1, completion_callback_));
+  EXPECT_EQ(net::ERR_IO_PENDING,
+            disk_cache->CreateEntry(2, &entry2, completion_callback_));
+  FlushCacheTasks();
+  EXPECT_EQ(2u, completion_results_.size());
+  EXPECT_EQ(net::OK, completion_results_[0]);
+  EXPECT_EQ(net::OK, completion_results_[1]);
+  EXPECT_TRUE(entry1);
+  EXPECT_TRUE(entry2);
+  completion_results_.clear();
+
+  // Write something to one of the entries and flush it.
+  const char* kData = "Hello";
+  const int kDataLen = strlen(kData) + 1;
+  scoped_refptr<net::IOBuffer> write_buf(new net::WrappedIOBuffer(kData));
+  EXPECT_EQ(net::ERR_IO_PENDING,
+            entry1->Write(0, 0, write_buf, kDataLen, completion_callback_));
+  FlushCacheTasks();
+  EXPECT_EQ(1u, completion_results_.size());
+  EXPECT_EQ(kDataLen, completion_results_[0]);
+  completion_results_.clear();
+
+  // Queue up a read and a write.
+  scoped_refptr<net::IOBuffer> read_buf = new net::IOBuffer(kDataLen);
+  EXPECT_EQ(
+      net::ERR_IO_PENDING,
+      entry1->Read(0, 0, read_buf.get(), kDataLen, completion_callback_));
+  EXPECT_EQ(
+      net::ERR_IO_PENDING,
+      entry2->Write(0, 0, write_buf.get(), kDataLen, completion_callback_));
+
+  // Pull the plug
+  disk_cache->Disable();
+  FlushCacheTasks();
+
+  // The entries are abandoned and the read/write callbacks
+  // should not be called.
+  EXPECT_TRUE(completion_results_.empty());
+
+  // Ensure the directory can be deleted at this point.
+  EXPECT_TRUE(base::DirectoryExists(directory_.path()));
+  EXPECT_FALSE(base::IsDirectoryEmpty(directory_.path()));
+  EXPECT_TRUE(base::DeleteFile(directory_.path(), true));
+  EXPECT_FALSE(base::DirectoryExists(directory_.path()));
+
+  disk_cache.reset(NULL);
+
+  // New IO operations should fail immediately.
+  EXPECT_EQ(
+      net::ERR_ABORTED,
+      entry1->Read(0, 0, read_buf.get(), kDataLen, completion_callback_));
+  entry1->Close();
+  entry2->Close();
+
+  FlushCacheTasks();
+  EXPECT_TRUE(completion_results_.empty());
+}
+
+}  // namespace appcache