| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "webkit/fileapi/local_file_stream_writer.h" | 5 #include "webkit/fileapi/local_file_stream_writer.h" |
| 6 | 6 |
| 7 #include <string> | 7 #include <string> |
| 8 | 8 |
| 9 #include "base/callback.h" | 9 #include "base/callback.h" |
| 10 #include "base/file_util.h" | 10 #include "base/file_util.h" |
| (...skipping 11 matching lines...) Expand all Loading... |
| 22 | 22 |
| 23 class LocalFileStreamWriterTest : public testing::Test { | 23 class LocalFileStreamWriterTest : public testing::Test { |
| 24 public: | 24 public: |
| 25 LocalFileStreamWriterTest() : message_loop_(MessageLoop::TYPE_IO) {} | 25 LocalFileStreamWriterTest() : message_loop_(MessageLoop::TYPE_IO) {} |
| 26 | 26 |
| 27 virtual void SetUp() OVERRIDE { | 27 virtual void SetUp() OVERRIDE { |
| 28 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); | 28 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); |
| 29 } | 29 } |
| 30 | 30 |
| 31 protected: | 31 protected: |
| 32 FilePath Path(const std::string& name) { | 32 base::FilePath Path(const std::string& name) { |
| 33 return temp_dir_.path().AppendASCII(name); | 33 return temp_dir_.path().AppendASCII(name); |
| 34 } | 34 } |
| 35 | 35 |
| 36 int WriteStringToWriter(LocalFileStreamWriter* writer, | 36 int WriteStringToWriter(LocalFileStreamWriter* writer, |
| 37 const std::string& data) { | 37 const std::string& data) { |
| 38 scoped_refptr<net::StringIOBuffer> buffer(new net::StringIOBuffer(data)); | 38 scoped_refptr<net::StringIOBuffer> buffer(new net::StringIOBuffer(data)); |
| 39 scoped_refptr<net::DrainableIOBuffer> drainable( | 39 scoped_refptr<net::DrainableIOBuffer> drainable( |
| 40 new net::DrainableIOBuffer(buffer, buffer->size())); | 40 new net::DrainableIOBuffer(buffer, buffer->size())); |
| 41 | 41 |
| 42 while (drainable->BytesRemaining() > 0) { | 42 while (drainable->BytesRemaining() > 0) { |
| 43 net::TestCompletionCallback callback; | 43 net::TestCompletionCallback callback; |
| 44 int result = writer->Write(drainable, drainable->BytesRemaining(), | 44 int result = writer->Write(drainable, drainable->BytesRemaining(), |
| 45 callback.callback()); | 45 callback.callback()); |
| 46 if (result == net::ERR_IO_PENDING) | 46 if (result == net::ERR_IO_PENDING) |
| 47 result = callback.WaitForResult(); | 47 result = callback.WaitForResult(); |
| 48 if (result <= 0) | 48 if (result <= 0) |
| 49 return result; | 49 return result; |
| 50 drainable->DidConsume(result); | 50 drainable->DidConsume(result); |
| 51 } | 51 } |
| 52 return net::OK; | 52 return net::OK; |
| 53 } | 53 } |
| 54 | 54 |
| 55 std::string GetFileContent(const FilePath& path) { | 55 std::string GetFileContent(const base::FilePath& path) { |
| 56 std::string content; | 56 std::string content; |
| 57 file_util::ReadFileToString(path, &content); | 57 file_util::ReadFileToString(path, &content); |
| 58 return content; | 58 return content; |
| 59 } | 59 } |
| 60 | 60 |
| 61 FilePath CreateFileWithContent(const std::string& name, | 61 base::FilePath CreateFileWithContent(const std::string& name, |
| 62 const std::string& data) { | 62 const std::string& data) { |
| 63 FilePath path = Path(name); | 63 base::FilePath path = Path(name); |
| 64 file_util::WriteFile(path, data.c_str(), data.size()); | 64 file_util::WriteFile(path, data.c_str(), data.size()); |
| 65 return path; | 65 return path; |
| 66 } | 66 } |
| 67 | 67 |
| 68 private: | 68 private: |
| 69 MessageLoop message_loop_; | 69 MessageLoop message_loop_; |
| 70 base::ScopedTempDir temp_dir_; | 70 base::ScopedTempDir temp_dir_; |
| 71 }; | 71 }; |
| 72 | 72 |
| 73 void NeverCalled(int unused) { | 73 void NeverCalled(int unused) { |
| 74 ADD_FAILURE(); | 74 ADD_FAILURE(); |
| 75 } | 75 } |
| 76 | 76 |
| 77 } // namespace | 77 } // namespace |
| 78 | 78 |
| 79 TEST_F(LocalFileStreamWriterTest, Write) { | 79 TEST_F(LocalFileStreamWriterTest, Write) { |
| 80 FilePath path = CreateFileWithContent("file_a", ""); | 80 base::FilePath path = CreateFileWithContent("file_a", ""); |
| 81 scoped_ptr<LocalFileStreamWriter> writer(new LocalFileStreamWriter(path, 0)); | 81 scoped_ptr<LocalFileStreamWriter> writer(new LocalFileStreamWriter(path, 0)); |
| 82 EXPECT_EQ(net::OK, WriteStringToWriter(writer.get(), "foo")); | 82 EXPECT_EQ(net::OK, WriteStringToWriter(writer.get(), "foo")); |
| 83 EXPECT_EQ(net::OK, WriteStringToWriter(writer.get(), "bar")); | 83 EXPECT_EQ(net::OK, WriteStringToWriter(writer.get(), "bar")); |
| 84 writer.reset(); | 84 writer.reset(); |
| 85 MessageLoop::current()->RunUntilIdle(); | 85 MessageLoop::current()->RunUntilIdle(); |
| 86 EXPECT_TRUE(file_util::PathExists(path)); | 86 EXPECT_TRUE(file_util::PathExists(path)); |
| 87 EXPECT_EQ("foobar", GetFileContent(path)); | 87 EXPECT_EQ("foobar", GetFileContent(path)); |
| 88 } | 88 } |
| 89 | 89 |
| 90 TEST_F(LocalFileStreamWriterTest, WriteMiddle) { | 90 TEST_F(LocalFileStreamWriterTest, WriteMiddle) { |
| 91 FilePath path = CreateFileWithContent("file_a", "foobar"); | 91 base::FilePath path = CreateFileWithContent("file_a", "foobar"); |
| 92 scoped_ptr<LocalFileStreamWriter> writer(new LocalFileStreamWriter(path, 2)); | 92 scoped_ptr<LocalFileStreamWriter> writer(new LocalFileStreamWriter(path, 2)); |
| 93 EXPECT_EQ(net::OK, WriteStringToWriter(writer.get(), "xxx")); | 93 EXPECT_EQ(net::OK, WriteStringToWriter(writer.get(), "xxx")); |
| 94 writer.reset(); | 94 writer.reset(); |
| 95 MessageLoop::current()->RunUntilIdle(); | 95 MessageLoop::current()->RunUntilIdle(); |
| 96 EXPECT_TRUE(file_util::PathExists(path)); | 96 EXPECT_TRUE(file_util::PathExists(path)); |
| 97 EXPECT_EQ("foxxxr", GetFileContent(path)); | 97 EXPECT_EQ("foxxxr", GetFileContent(path)); |
| 98 } | 98 } |
| 99 | 99 |
| 100 TEST_F(LocalFileStreamWriterTest, WriteEnd) { | 100 TEST_F(LocalFileStreamWriterTest, WriteEnd) { |
| 101 FilePath path = CreateFileWithContent("file_a", "foobar"); | 101 base::FilePath path = CreateFileWithContent("file_a", "foobar"); |
| 102 scoped_ptr<LocalFileStreamWriter> writer(new LocalFileStreamWriter(path, 6)); | 102 scoped_ptr<LocalFileStreamWriter> writer(new LocalFileStreamWriter(path, 6)); |
| 103 EXPECT_EQ(net::OK, WriteStringToWriter(writer.get(), "xxx")); | 103 EXPECT_EQ(net::OK, WriteStringToWriter(writer.get(), "xxx")); |
| 104 writer.reset(); | 104 writer.reset(); |
| 105 MessageLoop::current()->RunUntilIdle(); | 105 MessageLoop::current()->RunUntilIdle(); |
| 106 EXPECT_TRUE(file_util::PathExists(path)); | 106 EXPECT_TRUE(file_util::PathExists(path)); |
| 107 EXPECT_EQ("foobarxxx", GetFileContent(path)); | 107 EXPECT_EQ("foobarxxx", GetFileContent(path)); |
| 108 } | 108 } |
| 109 | 109 |
| 110 TEST_F(LocalFileStreamWriterTest, WriteFailForNonexistingFile) { | 110 TEST_F(LocalFileStreamWriterTest, WriteFailForNonexistingFile) { |
| 111 FilePath path = Path("file_a"); | 111 base::FilePath path = Path("file_a"); |
| 112 ASSERT_FALSE(file_util::PathExists(path)); | 112 ASSERT_FALSE(file_util::PathExists(path)); |
| 113 scoped_ptr<LocalFileStreamWriter> writer(new LocalFileStreamWriter(path, 0)); | 113 scoped_ptr<LocalFileStreamWriter> writer(new LocalFileStreamWriter(path, 0)); |
| 114 EXPECT_EQ(net::ERR_FILE_NOT_FOUND, WriteStringToWriter(writer.get(), "foo")); | 114 EXPECT_EQ(net::ERR_FILE_NOT_FOUND, WriteStringToWriter(writer.get(), "foo")); |
| 115 writer.reset(); | 115 writer.reset(); |
| 116 MessageLoop::current()->RunUntilIdle(); | 116 MessageLoop::current()->RunUntilIdle(); |
| 117 EXPECT_FALSE(file_util::PathExists(path)); | 117 EXPECT_FALSE(file_util::PathExists(path)); |
| 118 } | 118 } |
| 119 | 119 |
| 120 TEST_F(LocalFileStreamWriterTest, CancelBeforeOperation) { | 120 TEST_F(LocalFileStreamWriterTest, CancelBeforeOperation) { |
| 121 FilePath path = Path("file_a"); | 121 base::FilePath path = Path("file_a"); |
| 122 scoped_ptr<LocalFileStreamWriter> writer(new LocalFileStreamWriter(path, 0)); | 122 scoped_ptr<LocalFileStreamWriter> writer(new LocalFileStreamWriter(path, 0)); |
| 123 // Cancel immediately fails when there's no in-flight operation. | 123 // Cancel immediately fails when there's no in-flight operation. |
| 124 int cancel_result = writer->Cancel(base::Bind(&NeverCalled)); | 124 int cancel_result = writer->Cancel(base::Bind(&NeverCalled)); |
| 125 EXPECT_EQ(net::ERR_UNEXPECTED, cancel_result); | 125 EXPECT_EQ(net::ERR_UNEXPECTED, cancel_result); |
| 126 } | 126 } |
| 127 | 127 |
| 128 TEST_F(LocalFileStreamWriterTest, CancelAfterFinishedOperation) { | 128 TEST_F(LocalFileStreamWriterTest, CancelAfterFinishedOperation) { |
| 129 FilePath path = CreateFileWithContent("file_a", ""); | 129 base::FilePath path = CreateFileWithContent("file_a", ""); |
| 130 scoped_ptr<LocalFileStreamWriter> writer(new LocalFileStreamWriter(path, 0)); | 130 scoped_ptr<LocalFileStreamWriter> writer(new LocalFileStreamWriter(path, 0)); |
| 131 EXPECT_EQ(net::OK, WriteStringToWriter(writer.get(), "foo")); | 131 EXPECT_EQ(net::OK, WriteStringToWriter(writer.get(), "foo")); |
| 132 | 132 |
| 133 // Cancel immediately fails when there's no in-flight operation. | 133 // Cancel immediately fails when there's no in-flight operation. |
| 134 int cancel_result = writer->Cancel(base::Bind(&NeverCalled)); | 134 int cancel_result = writer->Cancel(base::Bind(&NeverCalled)); |
| 135 EXPECT_EQ(net::ERR_UNEXPECTED, cancel_result); | 135 EXPECT_EQ(net::ERR_UNEXPECTED, cancel_result); |
| 136 | 136 |
| 137 writer.reset(); | 137 writer.reset(); |
| 138 MessageLoop::current()->RunUntilIdle(); | 138 MessageLoop::current()->RunUntilIdle(); |
| 139 // Write operation is already completed. | 139 // Write operation is already completed. |
| 140 EXPECT_TRUE(file_util::PathExists(path)); | 140 EXPECT_TRUE(file_util::PathExists(path)); |
| 141 EXPECT_EQ("foo", GetFileContent(path)); | 141 EXPECT_EQ("foo", GetFileContent(path)); |
| 142 } | 142 } |
| 143 | 143 |
| 144 TEST_F(LocalFileStreamWriterTest, CancelWrite) { | 144 TEST_F(LocalFileStreamWriterTest, CancelWrite) { |
| 145 FilePath path = CreateFileWithContent("file_a", "foobar"); | 145 base::FilePath path = CreateFileWithContent("file_a", "foobar"); |
| 146 scoped_ptr<LocalFileStreamWriter> writer(new LocalFileStreamWriter(path, 0)); | 146 scoped_ptr<LocalFileStreamWriter> writer(new LocalFileStreamWriter(path, 0)); |
| 147 | 147 |
| 148 scoped_refptr<net::StringIOBuffer> buffer(new net::StringIOBuffer("xxx")); | 148 scoped_refptr<net::StringIOBuffer> buffer(new net::StringIOBuffer("xxx")); |
| 149 int result = writer->Write(buffer, buffer->size(), base::Bind(&NeverCalled)); | 149 int result = writer->Write(buffer, buffer->size(), base::Bind(&NeverCalled)); |
| 150 ASSERT_EQ(net::ERR_IO_PENDING, result); | 150 ASSERT_EQ(net::ERR_IO_PENDING, result); |
| 151 | 151 |
| 152 net::TestCompletionCallback callback; | 152 net::TestCompletionCallback callback; |
| 153 writer->Cancel(callback.callback()); | 153 writer->Cancel(callback.callback()); |
| 154 int cancel_result = callback.WaitForResult(); | 154 int cancel_result = callback.WaitForResult(); |
| 155 EXPECT_EQ(net::OK, cancel_result); | 155 EXPECT_EQ(net::OK, cancel_result); |
| 156 } | 156 } |
| OLD | NEW |