| 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/browser/fileapi/local_file_stream_writer.h" | 5 #include "webkit/browser/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 19 matching lines...) Expand all Loading... |
| 30 | 30 |
| 31 protected: | 31 protected: |
| 32 base::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.get(), 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( |
| 45 callback.callback()); | 45 drainable.get(), drainable->BytesRemaining(), 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 base::FilePath& path) { | 55 std::string GetFileContent(const base::FilePath& path) { |
| (...skipping 83 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 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 base::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 = |
| 150 writer->Write(buffer.get(), buffer->size(), base::Bind(&NeverCalled)); |
| 150 ASSERT_EQ(net::ERR_IO_PENDING, result); | 151 ASSERT_EQ(net::ERR_IO_PENDING, result); |
| 151 | 152 |
| 152 net::TestCompletionCallback callback; | 153 net::TestCompletionCallback callback; |
| 153 writer->Cancel(callback.callback()); | 154 writer->Cancel(callback.callback()); |
| 154 int cancel_result = callback.WaitForResult(); | 155 int cancel_result = callback.WaitForResult(); |
| 155 EXPECT_EQ(net::OK, cancel_result); | 156 EXPECT_EQ(net::OK, cancel_result); |
| 156 } | 157 } |
| OLD | NEW |