1 files changed, 180 insertions, 0 deletions
diff --git a/chromium/net/url_request/url_fetcher_response_writer_unittest.cc b/chromium/net/url_request/url_fetcher_response_writer_unittest.cc
new file mode 100644
index 00000000000..629e4bc3ce6
--- /dev/null
+++ b/chromium/net/url_request/url_fetcher_response_writer_unittest.cc
@@ -0,0 +1,180 @@
+// Copyright 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 "net/url_request/url_fetcher_response_writer.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 "net/base/test_completion_callback.h"
+#include "testing/platform_test.h"
+
+namespace net {
+
+namespace {
+
+const char kData[] = "Hello!";
+
+}  // namespace
+
+class URLFetcherStringWriterTest : public PlatformTest {
+ protected:
+  virtual void SetUp() OVERRIDE {
+    writer_.reset(new URLFetcherStringWriter);
+    buf_ = new StringIOBuffer(kData);
+  }
+
+  scoped_ptr<URLFetcherStringWriter> writer_;
+  scoped_refptr<StringIOBuffer> buf_;
+};
+
+TEST_F(URLFetcherStringWriterTest, Basic) {
+  int rv = 0;
+  // Initialize(), Write() and Finish().
+  TestCompletionCallback callback;
+  rv = writer_->Initialize(callback.callback());
+  EXPECT_EQ(OK, callback.GetResult(rv));
+  rv = writer_->Write(buf_.get(), buf_->size(), callback.callback());
+  EXPECT_EQ(buf_->size(), callback.GetResult(rv));
+  rv = writer_->Finish(callback.callback());
+  EXPECT_EQ(OK, callback.GetResult(rv));
+
+  // Verify the result.
+  EXPECT_EQ(kData, writer_->data());
+
+  // Initialize() again to reset.
+  rv = writer_->Initialize(callback.callback());
+  EXPECT_EQ(OK, callback.GetResult(rv));
+  EXPECT_TRUE(writer_->data().empty());
+}
+
+class URLFetcherFileWriterTest : public PlatformTest {
+ protected:
+  virtual void SetUp() OVERRIDE {
+    ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
+    file_path_ = temp_dir_.path().AppendASCII("test.txt");
+    writer_.reset(new URLFetcherFileWriter(
+        base::MessageLoopProxy::current(), file_path_));
+    buf_ = new StringIOBuffer(kData);
+  }
+
+  base::ScopedTempDir temp_dir_;
+  base::FilePath file_path_;
+  scoped_ptr<URLFetcherFileWriter> writer_;
+  scoped_refptr<StringIOBuffer> buf_;
+};
+
+TEST_F(URLFetcherFileWriterTest, WriteToFile) {
+  int rv = 0;
+  // Initialize(), Write() and Finish().
+  TestCompletionCallback callback;
+  rv = writer_->Initialize(callback.callback());
+  EXPECT_EQ(OK, callback.GetResult(rv));
+  rv = writer_->Write(buf_.get(), buf_->size(), callback.callback());
+  EXPECT_EQ(buf_->size(), callback.GetResult(rv));
+  rv = writer_->Finish(callback.callback());
+  EXPECT_EQ(OK, callback.GetResult(rv));
+
+  // Verify the result.
+  EXPECT_EQ(file_path_.value(), writer_->file_path().value());
+  std::string file_contents;
+  EXPECT_TRUE(base::ReadFileToString(writer_->file_path(), &file_contents));
+  EXPECT_EQ(kData, file_contents);
+
+  // Destroy the writer. File should be deleted.
+  writer_.reset();
+  base::RunLoop().RunUntilIdle();
+  EXPECT_FALSE(base::PathExists(file_path_));
+}
+
+TEST_F(URLFetcherFileWriterTest, InitializeAgain) {
+  int rv = 0;
+  // Initialize(), Write() and Finish().
+  TestCompletionCallback callback;
+  rv = writer_->Initialize(callback.callback());
+  EXPECT_EQ(OK, callback.GetResult(rv));
+  rv = writer_->Write(buf_.get(), buf_->size(), callback.callback());
+  EXPECT_EQ(buf_->size(), callback.GetResult(rv));
+  rv = writer_->Finish(callback.callback());
+  EXPECT_EQ(OK, callback.GetResult(rv));
+
+  // Verify the result.
+  std::string file_contents;
+  EXPECT_TRUE(base::ReadFileToString(writer_->file_path(), &file_contents));
+  EXPECT_EQ(kData, file_contents);
+
+  // Initialize() again to reset. Write different data.
+  const std::string data2 = "Bye!";
+  scoped_refptr<StringIOBuffer> buf2(new StringIOBuffer(data2));
+
+  rv = writer_->Initialize(callback.callback());
+  EXPECT_EQ(OK, callback.GetResult(rv));
+  rv = writer_->Write(buf2.get(), buf2->size(), callback.callback());
+  EXPECT_EQ(buf2->size(), callback.GetResult(rv));
+  rv = writer_->Finish(callback.callback());
+  EXPECT_EQ(OK, callback.GetResult(rv));
+
+  // Verify the result.
+  file_contents.clear();
+  EXPECT_TRUE(base::ReadFileToString(writer_->file_path(), &file_contents));
+  EXPECT_EQ(data2, file_contents);
+}
+
+TEST_F(URLFetcherFileWriterTest, DisownFile) {
+  int rv = 0;
+  // Initialize() and Finish() to create a file.
+  TestCompletionCallback callback;
+  rv = writer_->Initialize(callback.callback());
+  EXPECT_EQ(OK, callback.GetResult(rv));
+  rv = writer_->Finish(callback.callback());
+  EXPECT_EQ(OK, callback.GetResult(rv));
+
+  // Disown file.
+  writer_->DisownFile();
+
+  // File is not deleted even after the writer gets destroyed.
+  writer_.reset();
+  base::RunLoop().RunUntilIdle();
+  EXPECT_TRUE(base::PathExists(file_path_));
+}
+
+class URLFetcherFileWriterTemporaryFileTest : public PlatformTest {
+ protected:
+  virtual void SetUp() OVERRIDE {
+    writer_.reset(new URLFetcherFileWriter(
+        base::MessageLoopProxy::current(), base::FilePath()));
+    buf_ = new StringIOBuffer(kData);
+  }
+
+  scoped_ptr<URLFetcherFileWriter> writer_;
+  scoped_refptr<StringIOBuffer> buf_;
+};
+
+TEST_F(URLFetcherFileWriterTemporaryFileTest, WriteToTemporaryFile) {
+  int rv = 0;
+  // Initialize(), Write() and Finish().
+  TestCompletionCallback callback;
+  rv = writer_->Initialize(callback.callback());
+  EXPECT_EQ(OK, callback.GetResult(rv));
+  rv = writer_->Write(buf_.get(), buf_->size(), callback.callback());
+  EXPECT_EQ(buf_->size(), callback.GetResult(rv));
+  rv = writer_->Finish(callback.callback());
+  EXPECT_EQ(OK, callback.GetResult(rv));
+
+  // Verify the result.
+  std::string file_contents;
+  EXPECT_TRUE(base::ReadFileToString(writer_->file_path(), &file_contents));
+  EXPECT_EQ(kData, file_contents);
+
+  // Destroy the writer. File should be deleted.
+  const base::FilePath file_path = writer_->file_path();
+  writer_.reset();
+  base::RunLoop().RunUntilIdle();
+  EXPECT_FALSE(base::PathExists(file_path));
+}
+
+}  // namespace net