Detect file system errors during downloads.

content/browser/download/base_file_unittest.cc

 // Copyright (c) 2011 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/file_util.h"
 #include "base/message_loop.h"
 #include "base/scoped_temp_dir.h"
 #include "base/string_number_conversions.h"
+#include "chrome/browser/download/download_util.h"
 #include "content/browser/browser_thread.h"
 #include "content/browser/download/base_file.h"
 #include "net/base/file_stream.h"
+#include "net/base/net_errors.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
 
+class MockFileStream : public net::FileStream {
+ public:
+  MockFileStream() : forced_error_(0) {}
+  MockFileStream(base::PlatformFile file, int flags)
+      : net::FileStream(file, flags), forced_error_(0) {}
+
+  void set_forced_error(int error) { forced_error_ = error; }
+  void clear_forced_error() { forced_error_ = 0; }
+
+  int Open(const FilePath& path, int open_flags) {
+    path_ = path;
+    return get_error(net::FileStream::Open(path, open_flags));
+  }
+
+  int64 Seek(net::Whence whence, int64 offset) {
+    return get_error64(net::FileStream::Seek(whence, offset));
+  }
+
+  int64 Available() {
+    return get_error64(net::FileStream::Available());
+  }
+
+  int Read(char* buf, int buf_len, net::CompletionCallback* callback) {
+    return get_error(net::FileStream::Read(buf, buf_len, callback));
+  }
+
+  int ReadUntilComplete(char *buf, int buf_len) {
+    return get_error(net::FileStream::ReadUntilComplete(buf, buf_len));
+  }
+
+  int Write(const char* buf, int buf_len, net::CompletionCallback* callback) {
+    return get_error(net::FileStream::Write(buf, buf_len, callback));
+  }
+  int64 Truncate(int64 bytes) {
+    return get_error64(net::FileStream::Truncate(bytes));
+  }
+
+  int Flush() {
+    return get_error(net::FileStream::Flush());
+  }
+
+  const FilePath& get_path() const { return path_; }
+
+ private:
+  int get_error(int function_error) {
+    if (forced_error_ != 0) {
+      int ret = forced_error_;
+      clear_forced_error();
+      return ret;
+    }
+
+    return function_error;
+  }
+
+  int64 get_error64(int64 function_error) {
+    if (forced_error_ != 0) {
+      int64 ret = forced_error_;
+      clear_forced_error();
+      return ret;
+    }
+
+    return function_error;
+  }
+
+   int forced_error_;
+   FilePath path_;
+};
+
 const char kTestData1[] = "Let's write some data to the file!\n";
 const char kTestData2[] = "Writing more data.\n";
 const char kTestData3[] = "Final line.";
 
+}  // namespace
+
 class BaseFileTest : public testing::Test {
  public:
   BaseFileTest()
       : expect_file_survives_(false),
         file_thread_(BrowserThread::FILE, &message_loop_) {
   }
 
   virtual void SetUp() {
     ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
     base_file_.reset(
@@ skipping 14 matching lines @@
       EXPECT_EQ(expected_data_, disk_data);
     }
 
     // Make sure the mock BrowserThread outlives the BaseFile to satisfy
     // thread checks inside it.
     base_file_.reset();
 
     EXPECT_EQ(expect_file_survives_, file_util::PathExists(full_path));
   }
 
-  void AppendDataToFile(const std::string& data) {
-    ASSERT_TRUE(base_file_->in_progress());
-    base_file_->AppendDataToFile(data.data(), data.size());
+  bool OpenMockFileStream() {
+    DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE));
+
+    FilePath path;
+    if (!download_util::CreateTemporaryFileForDownload(&path))
+      return false;
+
+    // Create a new file stream if it is not provided.
+    mock_file_stream_.reset(new MockFileStream);
+    if (mock_file_stream_->Open(
+        path,
+        base::PLATFORM_FILE_OPEN_ALWAYS | base::PLATFORM_FILE_WRITE) != 0) {
+      mock_file_stream_.reset();
+      return false;
+    }
+
+    // We may be re-opening the file after rename. Always make sure we're
+    // writing at the end of the file.
+    if (mock_file_stream_->Seek(net::FROM_END, 0) < 0) {
+      mock_file_stream_.reset();
+      return false;
+    }
+
+    return true;
+  }
+
+  void ForceError(net::Error error) {
+    mock_file_stream_->set_forced_error(error);
+  }
+
+  bool AppendDataToFile(const std::string& data) {
+    EXPECT_TRUE(base_file_->in_progress());
+    bool appended = base_file_->AppendDataToFile(data.data(), data.size());
     expected_data_ += data;
     EXPECT_EQ(static_cast<int64>(expected_data_.size()),
               base_file_->bytes_so_far());
+    return appended;
   }
 
  protected:
   linked_ptr<net::FileStream> file_stream_;
+  linked_ptr<MockFileStream> mock_file_stream_;
 
   // BaseClass instance we are testing.
   scoped_ptr<BaseFile> base_file_;
 
   // Temporary directory for renamed downloads.
   ScopedTempDir temp_dir_;
 
   // Expect the file to survive deletion of the BaseFile instance.
   bool expect_file_survives_;
 
@@ skipping 146 matching lines @@
   EXPECT_TRUE(base_file_->in_progress());
   EXPECT_TRUE(base_file_->Rename(new_path));
   EXPECT_FALSE(file_util::PathExists(initial_path));
   EXPECT_TRUE(file_util::PathExists(new_path));
 
   AppendDataToFile(kTestData2);
 
   base_file_->Finish();
 }
 
-}  // namespace
+// Write data to the file multiple times.
+TEST_F(BaseFileTest, MultipleWritesWithError) {
+  ASSERT_TRUE(OpenMockFileStream());
+  base_file_.reset(new BaseFile(mock_file_stream_->get_path(),
+                                GURL(), GURL(), 0, mock_file_stream_));
+  ASSERT_TRUE(base_file_->Initialize(false));
+  ASSERT_TRUE(AppendDataToFile(kTestData1));
+  ASSERT_TRUE(AppendDataToFile(kTestData2));
+  ForceError(net::ERR_FAILED);
+  ASSERT_FALSE(AppendDataToFile(kTestData3));
+  std::string hash;
+  EXPECT_FALSE(base_file_->GetSha256Hash(&hash));
+  base_file_->Finish();
+}