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 "net/base/upload_data_stream.h" | 5 #include "net/base/upload_data_stream.h" |
6 | 6 |
| 7 #include <algorithm> |
7 #include <vector> | 8 #include <vector> |
8 | 9 |
9 #include "base/basictypes.h" | 10 #include "base/basictypes.h" |
10 #include "base/bind.h" | 11 #include "base/bind.h" |
11 #include "base/file_path.h" | 12 #include "base/file_path.h" |
12 #include "base/file_util.h" | 13 #include "base/file_util.h" |
13 #include "base/memory/scoped_ptr.h" | 14 #include "base/memory/scoped_ptr.h" |
14 #include "base/message_loop.h" | 15 #include "base/message_loop.h" |
15 #include "base/time.h" | 16 #include "base/time.h" |
16 #include "net/base/io_buffer.h" | 17 #include "net/base/io_buffer.h" |
(...skipping 15 matching lines...) Expand all Loading... |
32 | 33 |
33 const char kTestData[] = "0123456789"; | 34 const char kTestData[] = "0123456789"; |
34 const size_t kTestDataSize = arraysize(kTestData) - 1; | 35 const size_t kTestDataSize = arraysize(kTestData) - 1; |
35 const size_t kTestBufferSize = 1 << 14; // 16KB. | 36 const size_t kTestBufferSize = 1 << 14; // 16KB. |
36 | 37 |
37 // Reads data from the upload data stream, and returns the data as string. | 38 // Reads data from the upload data stream, and returns the data as string. |
38 std::string ReadFromUploadDataStream(UploadDataStream* stream) { | 39 std::string ReadFromUploadDataStream(UploadDataStream* stream) { |
39 std::string data_read; | 40 std::string data_read; |
40 scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize); | 41 scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize); |
41 while (!stream->IsEOF()) { | 42 while (!stream->IsEOF()) { |
42 const int bytes_read = stream->Read(buf, kTestBufferSize); | 43 const int bytes_read = stream->ReadSync(buf, kTestBufferSize); |
43 data_read.append(buf->data(), bytes_read); | 44 data_read.append(buf->data(), bytes_read); |
44 } | 45 } |
45 return data_read; | 46 return data_read; |
46 } | 47 } |
47 | 48 |
48 // A mock class of UploadElementReader. | 49 // A mock class of UploadElementReader. |
49 class MockUploadElementReader : public UploadElementReader { | 50 class MockUploadElementReader : public UploadElementReader { |
50 public: | 51 public: |
51 MockUploadElementReader() : init_result_(OK) {} | 52 MockUploadElementReader(int content_length, bool is_in_memory) |
| 53 : content_length_(content_length), |
| 54 bytes_remaining_(content_length), |
| 55 is_in_memory_(is_in_memory), |
| 56 init_result_(OK), |
| 57 read_result_(OK) {} |
| 58 |
52 virtual ~MockUploadElementReader() {} | 59 virtual ~MockUploadElementReader() {} |
53 | 60 |
| 61 // UploadElementReader overrides. |
54 MOCK_METHOD1(Init, int(const CompletionCallback& callback)); | 62 MOCK_METHOD1(Init, int(const CompletionCallback& callback)); |
55 MOCK_CONST_METHOD0(GetContentLength, uint64()); | 63 virtual uint64 GetContentLength() const OVERRIDE { return content_length_; } |
56 MOCK_CONST_METHOD0(BytesRemaining, uint64()); | 64 virtual uint64 BytesRemaining() const OVERRIDE { return bytes_remaining_; } |
57 MOCK_CONST_METHOD0(IsInMemory, bool()); | 65 virtual bool IsInMemory() const OVERRIDE { return is_in_memory_; } |
58 MOCK_METHOD2(ReadSync, int(char* buf, int buf_length)); | 66 MOCK_METHOD3(Read, int(IOBuffer* buf, |
| 67 int buf_length, |
| 68 const CompletionCallback& callback)); |
59 | 69 |
60 // Sets expectation to return the specified result from Init() asynchronously. | 70 // Sets expectation to return the specified result from Init() asynchronously. |
61 void SetAsyncInitExpectation(int result) { | 71 void SetAsyncInitExpectation(int result) { |
62 init_result_ = result; | 72 init_result_ = result; |
63 EXPECT_CALL(*this, Init(_)) | 73 EXPECT_CALL(*this, Init(_)) |
64 .WillOnce(DoAll(Invoke(this, &MockUploadElementReader::OnInit), | 74 .WillOnce(DoAll(Invoke(this, &MockUploadElementReader::OnInit), |
65 Return(ERR_IO_PENDING))); | 75 Return(ERR_IO_PENDING))); |
66 } | 76 } |
67 | 77 |
| 78 // Sets expectation to return the specified result from Read(). |
| 79 void SetReadExpectation(int result) { |
| 80 read_result_ = result; |
| 81 EXPECT_CALL(*this, Read(_, _, _)) |
| 82 .WillOnce(Invoke(this, &MockUploadElementReader::OnRead)); |
| 83 } |
| 84 |
68 private: | 85 private: |
69 void OnInit(const CompletionCallback& callback) { | 86 void OnInit(const CompletionCallback& callback) { |
70 MessageLoop::current()->PostTask(FROM_HERE, | 87 MessageLoop::current()->PostTask(FROM_HERE, |
71 base::Bind(callback, init_result_)); | 88 base::Bind(callback, init_result_)); |
72 } | 89 } |
73 | 90 |
| 91 int OnRead(IOBuffer* buf, |
| 92 int buf_length, |
| 93 const CompletionCallback& callback) { |
| 94 bytes_remaining_ = std::max(0, bytes_remaining_ - read_result_); |
| 95 if (IsInMemory()) { |
| 96 return read_result_; |
| 97 } else { |
| 98 MessageLoop::current()->PostTask(FROM_HERE, |
| 99 base::Bind(callback, read_result_)); |
| 100 return ERR_IO_PENDING; |
| 101 } |
| 102 } |
| 103 |
| 104 int content_length_; |
| 105 int bytes_remaining_; |
| 106 bool is_in_memory_; |
| 107 |
74 // Result value returned from Init(). | 108 // Result value returned from Init(). |
75 int init_result_; | 109 int init_result_; |
| 110 |
| 111 // Result value returned from Read(). |
| 112 int read_result_; |
76 }; | 113 }; |
77 | 114 |
78 // A mock CompletionCallback. | 115 // A mock CompletionCallback. |
79 class MockCompletionCallback { | 116 class MockCompletionCallback { |
80 public: | 117 public: |
81 MOCK_METHOD1(Run, void(int result)); | 118 MOCK_METHOD1(Run, void(int result)); |
82 | 119 |
83 CompletionCallback CreateCallback() { | 120 CompletionCallback CreateCallback() { |
84 return base::Bind(&MockCompletionCallback::Run, base::Unretained(this)); | 121 return base::Bind(&MockCompletionCallback::Run, base::Unretained(this)); |
85 } | 122 } |
86 }; | 123 }; |
87 | 124 |
88 } // namespace | 125 } // namespace |
89 | 126 |
90 class UploadDataStreamTest : public PlatformTest { | 127 class UploadDataStreamTest : public PlatformTest { |
91 public: | 128 public: |
92 UploadDataStreamTest() : upload_data_(new UploadData) { } | 129 UploadDataStreamTest() : upload_data_(new UploadData) { } |
93 | 130 |
94 void FileChangedHelper(const FilePath& file_path, | 131 void FileChangedHelper(const FilePath& file_path, |
95 const base::Time& time, | 132 const base::Time& time, |
96 bool error_expected); | 133 bool error_expected); |
97 | 134 |
98 scoped_refptr<UploadData> upload_data_; | 135 scoped_refptr<UploadData> upload_data_; |
99 }; | 136 }; |
100 | 137 |
101 TEST_F(UploadDataStreamTest, EmptyUploadData) { | 138 TEST_F(UploadDataStreamTest, EmptyUploadData) { |
102 upload_data_->AppendBytes("", 0); | 139 upload_data_->AppendBytes("", 0); |
103 scoped_ptr<UploadDataStream> stream(new UploadDataStream(upload_data_)); | 140 UploadDataStream stream(upload_data_); |
104 ASSERT_EQ(OK, stream->InitSync()); | 141 ASSERT_EQ(OK, stream.InitSync()); |
105 EXPECT_TRUE(stream->IsInMemory()); | 142 EXPECT_TRUE(stream.IsInMemory()); |
106 ASSERT_TRUE(stream.get()); | 143 EXPECT_EQ(0U, stream.size()); |
107 EXPECT_EQ(0U, stream->size()); | 144 EXPECT_EQ(0U, stream.position()); |
108 EXPECT_EQ(0U, stream->position()); | 145 EXPECT_TRUE(stream.IsEOF()); |
109 EXPECT_TRUE(stream->IsEOF()); | |
110 } | 146 } |
111 | 147 |
112 TEST_F(UploadDataStreamTest, ConsumeAllBytes) { | 148 TEST_F(UploadDataStreamTest, ConsumeAllBytes) { |
113 upload_data_->AppendBytes(kTestData, kTestDataSize); | 149 upload_data_->AppendBytes(kTestData, kTestDataSize); |
114 scoped_ptr<UploadDataStream> stream(new UploadDataStream(upload_data_)); | 150 UploadDataStream stream(upload_data_); |
115 ASSERT_EQ(OK, stream->InitSync()); | 151 ASSERT_EQ(OK, stream.InitSync()); |
116 EXPECT_TRUE(stream->IsInMemory()); | 152 EXPECT_TRUE(stream.IsInMemory()); |
117 ASSERT_TRUE(stream.get()); | 153 EXPECT_EQ(kTestDataSize, stream.size()); |
118 EXPECT_EQ(kTestDataSize, stream->size()); | 154 EXPECT_EQ(0U, stream.position()); |
119 EXPECT_EQ(0U, stream->position()); | 155 EXPECT_FALSE(stream.IsEOF()); |
120 EXPECT_FALSE(stream->IsEOF()); | |
121 scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize); | 156 scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize); |
122 while (!stream->IsEOF()) { | 157 while (!stream.IsEOF()) { |
123 int bytes_read = stream->Read(buf, kTestBufferSize); | 158 int bytes_read = stream.ReadSync(buf, kTestBufferSize); |
124 ASSERT_LE(0, bytes_read); // Not an error. | 159 ASSERT_LE(0, bytes_read); // Not an error. |
125 } | 160 } |
126 EXPECT_EQ(kTestDataSize, stream->position()); | 161 EXPECT_EQ(kTestDataSize, stream.position()); |
127 ASSERT_TRUE(stream->IsEOF()); | 162 ASSERT_TRUE(stream.IsEOF()); |
128 } | 163 } |
129 | 164 |
130 TEST_F(UploadDataStreamTest, File) { | 165 TEST_F(UploadDataStreamTest, File) { |
131 FilePath temp_file_path; | 166 FilePath temp_file_path; |
132 ASSERT_TRUE(file_util::CreateTemporaryFile(&temp_file_path)); | 167 ASSERT_TRUE(file_util::CreateTemporaryFile(&temp_file_path)); |
133 ASSERT_EQ(static_cast<int>(kTestDataSize), | 168 ASSERT_EQ(static_cast<int>(kTestDataSize), |
134 file_util::WriteFile(temp_file_path, kTestData, kTestDataSize)); | 169 file_util::WriteFile(temp_file_path, kTestData, kTestDataSize)); |
135 | 170 |
136 std::vector<UploadElement> elements; | 171 std::vector<UploadElement> elements; |
137 UploadElement element; | 172 UploadElement element; |
138 element.SetToFilePath(temp_file_path); | 173 element.SetToFilePath(temp_file_path); |
139 elements.push_back(element); | 174 elements.push_back(element); |
140 upload_data_->SetElements(elements); | 175 upload_data_->SetElements(elements); |
141 | 176 |
142 scoped_ptr<UploadDataStream> stream(new UploadDataStream(upload_data_)); | 177 UploadDataStream stream(upload_data_); |
143 ASSERT_EQ(OK, stream->InitSync()); | 178 ASSERT_EQ(OK, stream.InitSync()); |
144 EXPECT_FALSE(stream->IsInMemory()); | 179 EXPECT_FALSE(stream.IsInMemory()); |
145 ASSERT_TRUE(stream.get()); | 180 EXPECT_EQ(kTestDataSize, stream.size()); |
146 EXPECT_EQ(kTestDataSize, stream->size()); | 181 EXPECT_EQ(0U, stream.position()); |
147 EXPECT_EQ(0U, stream->position()); | 182 EXPECT_FALSE(stream.IsEOF()); |
148 EXPECT_FALSE(stream->IsEOF()); | |
149 scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize); | 183 scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize); |
150 while (!stream->IsEOF()) { | 184 while (!stream.IsEOF()) { |
151 int bytes_read = stream->Read(buf, kTestBufferSize); | 185 int bytes_read = stream.ReadSync(buf, kTestBufferSize); |
152 ASSERT_LE(0, bytes_read); // Not an error. | 186 ASSERT_LE(0, bytes_read); // Not an error. |
153 } | 187 } |
154 EXPECT_EQ(kTestDataSize, stream->position()); | 188 EXPECT_EQ(kTestDataSize, stream.position()); |
155 ASSERT_TRUE(stream->IsEOF()); | 189 ASSERT_TRUE(stream.IsEOF()); |
156 file_util::Delete(temp_file_path, false); | 190 file_util::Delete(temp_file_path, false); |
157 } | 191 } |
158 | 192 |
159 TEST_F(UploadDataStreamTest, FileSmallerThanLength) { | 193 TEST_F(UploadDataStreamTest, FileSmallerThanLength) { |
160 FilePath temp_file_path; | 194 FilePath temp_file_path; |
161 ASSERT_TRUE(file_util::CreateTemporaryFile(&temp_file_path)); | 195 ASSERT_TRUE(file_util::CreateTemporaryFile(&temp_file_path)); |
162 ASSERT_EQ(static_cast<int>(kTestDataSize), | 196 ASSERT_EQ(static_cast<int>(kTestDataSize), |
163 file_util::WriteFile(temp_file_path, kTestData, kTestDataSize)); | 197 file_util::WriteFile(temp_file_path, kTestData, kTestDataSize)); |
164 const uint64 kFakeSize = kTestDataSize*2; | 198 const uint64 kFakeSize = kTestDataSize*2; |
165 | 199 |
166 UploadFileElementReader::ScopedOverridingContentLengthForTests | 200 UploadFileElementReader::ScopedOverridingContentLengthForTests |
167 overriding_content_length(kFakeSize); | 201 overriding_content_length(kFakeSize); |
168 | 202 |
169 std::vector<UploadElement> elements; | 203 std::vector<UploadElement> elements; |
170 UploadElement element; | 204 UploadElement element; |
171 element.SetToFilePath(temp_file_path); | 205 element.SetToFilePath(temp_file_path); |
172 elements.push_back(element); | 206 elements.push_back(element); |
173 upload_data_->SetElements(elements); | 207 upload_data_->SetElements(elements); |
174 | 208 |
175 scoped_ptr<UploadDataStream> stream(new UploadDataStream(upload_data_)); | 209 UploadDataStream stream(upload_data_); |
176 ASSERT_EQ(OK, stream->InitSync()); | 210 ASSERT_EQ(OK, stream.InitSync()); |
177 EXPECT_FALSE(stream->IsInMemory()); | 211 EXPECT_FALSE(stream.IsInMemory()); |
178 ASSERT_TRUE(stream.get()); | 212 EXPECT_EQ(kFakeSize, stream.size()); |
179 EXPECT_EQ(kFakeSize, stream->size()); | 213 EXPECT_EQ(0U, stream.position()); |
180 EXPECT_EQ(0U, stream->position()); | 214 EXPECT_FALSE(stream.IsEOF()); |
181 EXPECT_FALSE(stream->IsEOF()); | |
182 uint64 read_counter = 0; | 215 uint64 read_counter = 0; |
183 scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize); | 216 scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize); |
184 while (!stream->IsEOF()) { | 217 while (!stream.IsEOF()) { |
185 int bytes_read = stream->Read(buf, kTestBufferSize); | 218 int bytes_read = stream.ReadSync(buf, kTestBufferSize); |
186 ASSERT_LE(0, bytes_read); // Not an error. | 219 ASSERT_LE(0, bytes_read); // Not an error. |
187 read_counter += bytes_read; | 220 read_counter += bytes_read; |
188 EXPECT_EQ(read_counter, stream->position()); | 221 EXPECT_EQ(read_counter, stream.position()); |
189 } | 222 } |
190 // UpdateDataStream will pad out the file with 0 bytes so that the HTTP | 223 // UpdateDataStream will pad out the file with 0 bytes so that the HTTP |
191 // transaction doesn't hang. Therefore we expected the full size. | 224 // transaction doesn't hang. Therefore we expected the full size. |
192 EXPECT_EQ(kFakeSize, read_counter); | 225 EXPECT_EQ(kFakeSize, read_counter); |
193 EXPECT_EQ(read_counter, stream->position()); | 226 EXPECT_EQ(read_counter, stream.position()); |
194 | 227 |
195 file_util::Delete(temp_file_path, false); | 228 file_util::Delete(temp_file_path, false); |
196 } | 229 } |
197 | 230 |
198 TEST_F(UploadDataStreamTest, FileAndBytes) { | 231 TEST_F(UploadDataStreamTest, FileAndBytes) { |
199 FilePath temp_file_path; | 232 FilePath temp_file_path; |
200 ASSERT_TRUE(file_util::CreateTemporaryFile(&temp_file_path)); | 233 ASSERT_TRUE(file_util::CreateTemporaryFile(&temp_file_path)); |
201 ASSERT_EQ(static_cast<int>(kTestDataSize), | 234 ASSERT_EQ(static_cast<int>(kTestDataSize), |
202 file_util::WriteFile(temp_file_path, kTestData, kTestDataSize)); | 235 file_util::WriteFile(temp_file_path, kTestData, kTestDataSize)); |
203 | 236 |
204 const uint64 kFileRangeOffset = 1; | 237 const uint64 kFileRangeOffset = 1; |
205 const uint64 kFileRangeLength = 4; | 238 const uint64 kFileRangeLength = 4; |
206 upload_data_->AppendFileRange( | 239 upload_data_->AppendFileRange( |
207 temp_file_path, kFileRangeOffset, kFileRangeLength, base::Time()); | 240 temp_file_path, kFileRangeOffset, kFileRangeLength, base::Time()); |
208 | 241 |
209 upload_data_->AppendBytes(kTestData, kTestDataSize); | 242 upload_data_->AppendBytes(kTestData, kTestDataSize); |
210 | 243 |
211 const uint64 kStreamSize = kTestDataSize + kFileRangeLength; | 244 const uint64 kStreamSize = kTestDataSize + kFileRangeLength; |
212 scoped_ptr<UploadDataStream> stream(new UploadDataStream(upload_data_)); | 245 UploadDataStream stream(upload_data_); |
213 ASSERT_EQ(OK, stream->InitSync()); | 246 ASSERT_EQ(OK, stream.InitSync()); |
214 EXPECT_FALSE(stream->IsInMemory()); | 247 EXPECT_FALSE(stream.IsInMemory()); |
215 ASSERT_TRUE(stream.get()); | 248 EXPECT_EQ(kStreamSize, stream.size()); |
216 EXPECT_EQ(kStreamSize, stream->size()); | 249 EXPECT_EQ(0U, stream.position()); |
217 EXPECT_EQ(0U, stream->position()); | 250 EXPECT_FALSE(stream.IsEOF()); |
218 EXPECT_FALSE(stream->IsEOF()); | |
219 scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize); | 251 scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize); |
220 while (!stream->IsEOF()) { | 252 while (!stream.IsEOF()) { |
221 int bytes_read = stream->Read(buf, kTestBufferSize); | 253 int bytes_read = stream.ReadSync(buf, kTestBufferSize); |
222 ASSERT_LE(0, bytes_read); // Not an error. | 254 ASSERT_LE(0, bytes_read); // Not an error. |
223 } | 255 } |
224 EXPECT_EQ(kStreamSize, stream->position()); | 256 EXPECT_EQ(kStreamSize, stream.position()); |
225 ASSERT_TRUE(stream->IsEOF()); | 257 ASSERT_TRUE(stream.IsEOF()); |
226 | 258 |
227 file_util::Delete(temp_file_path, false); | 259 file_util::Delete(temp_file_path, false); |
228 } | 260 } |
229 | 261 |
230 TEST_F(UploadDataStreamTest, Chunk) { | 262 TEST_F(UploadDataStreamTest, Chunk) { |
231 upload_data_->set_is_chunked(true); | 263 upload_data_->set_is_chunked(true); |
232 upload_data_->AppendChunk(kTestData, kTestDataSize, false); | 264 upload_data_->AppendChunk(kTestData, kTestDataSize, false); |
233 upload_data_->AppendChunk(kTestData, kTestDataSize, true); | 265 upload_data_->AppendChunk(kTestData, kTestDataSize, true); |
234 | 266 |
235 const uint64 kStreamSize = kTestDataSize*2; | 267 const uint64 kStreamSize = kTestDataSize*2; |
236 scoped_ptr<UploadDataStream> stream(new UploadDataStream(upload_data_)); | 268 UploadDataStream stream(upload_data_); |
237 ASSERT_EQ(OK, stream->InitSync()); | 269 ASSERT_EQ(OK, stream.InitSync()); |
238 EXPECT_FALSE(stream->IsInMemory()); | 270 EXPECT_FALSE(stream.IsInMemory()); |
239 ASSERT_TRUE(stream.get()); | 271 EXPECT_EQ(0U, stream.size()); // Content-Length is 0 for chunked data. |
240 EXPECT_EQ(0U, stream->size()); // Content-Length is 0 for chunked data. | 272 EXPECT_EQ(0U, stream.position()); |
241 EXPECT_EQ(0U, stream->position()); | 273 EXPECT_FALSE(stream.IsEOF()); |
242 EXPECT_FALSE(stream->IsEOF()); | |
243 scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize); | 274 scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize); |
244 while (!stream->IsEOF()) { | 275 while (!stream.IsEOF()) { |
245 int bytes_read = stream->Read(buf, kTestBufferSize); | 276 int bytes_read = stream.ReadSync(buf, kTestBufferSize); |
246 ASSERT_LE(0, bytes_read); // Not an error. | 277 ASSERT_LE(0, bytes_read); // Not an error. |
247 } | 278 } |
248 EXPECT_EQ(kStreamSize, stream->position()); | 279 EXPECT_EQ(kStreamSize, stream.position()); |
249 ASSERT_TRUE(stream->IsEOF()); | 280 ASSERT_TRUE(stream.IsEOF()); |
250 } | 281 } |
251 | 282 |
252 // Init() with on-memory and not-on-memory readers. | 283 // Init() with on-memory and not-on-memory readers. |
253 TEST_F(UploadDataStreamTest, InitAsync) { | 284 TEST_F(UploadDataStreamTest, InitAsync) { |
254 // Create stream without element readers. | 285 // Create stream without element readers. |
255 UploadDataStream stream(upload_data_); | 286 UploadDataStream stream(upload_data_); |
256 | 287 |
257 // Set mock readers to the stream. | 288 // Set mock readers to the stream. |
258 MockUploadElementReader* reader = NULL; | 289 MockUploadElementReader* reader = NULL; |
259 | 290 |
260 reader = new MockUploadElementReader; | 291 reader = new MockUploadElementReader(kTestDataSize, true); |
261 EXPECT_CALL(*reader, Init(_)).WillOnce(Return(OK)); | 292 EXPECT_CALL(*reader, Init(_)).WillOnce(Return(OK)); |
262 EXPECT_CALL(*reader, GetContentLength()).WillRepeatedly(Return(0)); | |
263 EXPECT_CALL(*reader, IsInMemory()).WillRepeatedly(Return(true)); | |
264 stream.element_readers_.push_back(reader); | 293 stream.element_readers_.push_back(reader); |
265 | 294 |
266 reader = new MockUploadElementReader; | 295 reader = new MockUploadElementReader(kTestDataSize, true); |
267 EXPECT_CALL(*reader, Init(_)).WillOnce(Return(OK)); | 296 EXPECT_CALL(*reader, Init(_)).WillOnce(Return(OK)); |
268 EXPECT_CALL(*reader, GetContentLength()).WillRepeatedly(Return(0)); | |
269 EXPECT_CALL(*reader, IsInMemory()).WillRepeatedly(Return(true)); | |
270 stream.element_readers_.push_back(reader); | 297 stream.element_readers_.push_back(reader); |
271 | 298 |
272 reader = new MockUploadElementReader; | 299 reader = new MockUploadElementReader(kTestDataSize, false); |
273 reader->SetAsyncInitExpectation(OK); | 300 reader->SetAsyncInitExpectation(OK); |
274 EXPECT_CALL(*reader, GetContentLength()).WillRepeatedly(Return(0)); | |
275 EXPECT_CALL(*reader, IsInMemory()).WillRepeatedly(Return(false)); | |
276 stream.element_readers_.push_back(reader); | 301 stream.element_readers_.push_back(reader); |
277 | 302 |
278 reader = new MockUploadElementReader; | 303 reader = new MockUploadElementReader(kTestDataSize, false); |
279 reader->SetAsyncInitExpectation(OK); | 304 reader->SetAsyncInitExpectation(OK); |
280 EXPECT_CALL(*reader, GetContentLength()).WillRepeatedly(Return(0)); | |
281 EXPECT_CALL(*reader, IsInMemory()).WillRepeatedly(Return(false)); | |
282 stream.element_readers_.push_back(reader); | 305 stream.element_readers_.push_back(reader); |
283 | 306 |
284 reader = new MockUploadElementReader; | 307 reader = new MockUploadElementReader(kTestDataSize, true); |
285 EXPECT_CALL(*reader, Init(_)).WillOnce(Return(OK)); | 308 EXPECT_CALL(*reader, Init(_)).WillOnce(Return(OK)); |
286 EXPECT_CALL(*reader, GetContentLength()).WillRepeatedly(Return(0)); | |
287 EXPECT_CALL(*reader, IsInMemory()).WillRepeatedly(Return(true)); | |
288 stream.element_readers_.push_back(reader); | 309 stream.element_readers_.push_back(reader); |
289 | 310 |
290 // Run Init(). | 311 // Run Init(). |
291 MockCompletionCallback mock_callback; | 312 MockCompletionCallback mock_callback; |
292 EXPECT_CALL(mock_callback, Run(OK)).Times(1); | 313 EXPECT_CALL(mock_callback, Run(OK)).Times(1); |
293 EXPECT_EQ(stream.Init(mock_callback.CreateCallback()), ERR_IO_PENDING); | 314 EXPECT_EQ(stream.Init(mock_callback.CreateCallback()), ERR_IO_PENDING); |
294 MessageLoop::current()->RunAllPending(); | 315 MessageLoop::current()->RunAllPending(); |
295 } | 316 } |
296 | 317 |
297 // Init() of a reader fails asynchronously. | 318 // Init() of a reader fails asynchronously. |
298 TEST_F(UploadDataStreamTest, InitAsyncFailureAsync) { | 319 TEST_F(UploadDataStreamTest, InitAsyncFailureAsync) { |
299 // Create stream without element readers. | 320 // Create stream without element readers. |
300 UploadDataStream stream(upload_data_); | 321 UploadDataStream stream(upload_data_); |
301 | 322 |
302 // Set a mock reader to the stream. | 323 // Set a mock reader to the stream. |
303 MockUploadElementReader* reader = NULL; | 324 MockUploadElementReader* reader = NULL; |
304 | 325 |
305 reader = new MockUploadElementReader; | 326 reader = new MockUploadElementReader(kTestDataSize, false); |
306 reader->SetAsyncInitExpectation(ERR_FAILED); | 327 reader->SetAsyncInitExpectation(ERR_FAILED); |
307 EXPECT_CALL(*reader, IsInMemory()).WillRepeatedly(Return(false)); | |
308 stream.element_readers_.push_back(reader); | 328 stream.element_readers_.push_back(reader); |
309 | 329 |
310 // Run Init(). | 330 // Run Init(). |
311 MockCompletionCallback mock_callback; | 331 MockCompletionCallback mock_callback; |
312 EXPECT_CALL(mock_callback, Run(ERR_FAILED)).Times(1); | 332 EXPECT_CALL(mock_callback, Run(ERR_FAILED)).Times(1); |
313 EXPECT_EQ(stream.Init(mock_callback.CreateCallback()), ERR_IO_PENDING); | 333 EXPECT_EQ(stream.Init(mock_callback.CreateCallback()), ERR_IO_PENDING); |
314 MessageLoop::current()->RunAllPending(); | 334 MessageLoop::current()->RunAllPending(); |
315 } | 335 } |
316 | 336 |
317 // Init() of a reader fails synchronously. | 337 // Init() of a reader fails synchronously. |
318 TEST_F(UploadDataStreamTest, InitAsyncFailureSync) { | 338 TEST_F(UploadDataStreamTest, InitAsyncFailureSync) { |
319 // Create stream without element readers. | 339 // Create stream without element readers. |
320 UploadDataStream stream(upload_data_); | 340 UploadDataStream stream(upload_data_); |
321 | 341 |
322 // Set mock readers to the stream. | 342 // Set mock readers to the stream. |
323 MockUploadElementReader* reader = NULL; | 343 MockUploadElementReader* reader = NULL; |
324 | 344 |
325 reader = new MockUploadElementReader; | 345 reader = new MockUploadElementReader(kTestDataSize, false); |
326 reader->SetAsyncInitExpectation(OK); | 346 reader->SetAsyncInitExpectation(OK); |
327 EXPECT_CALL(*reader, IsInMemory()).WillRepeatedly(Return(false)); | |
328 stream.element_readers_.push_back(reader); | 347 stream.element_readers_.push_back(reader); |
329 | 348 |
330 reader = new MockUploadElementReader; | 349 reader = new MockUploadElementReader(kTestDataSize, true); |
331 EXPECT_CALL(*reader, Init(_)).WillOnce(Return(ERR_FAILED)); | 350 EXPECT_CALL(*reader, Init(_)).WillOnce(Return(ERR_FAILED)); |
332 EXPECT_CALL(*reader, IsInMemory()).WillRepeatedly(Return(true)); | |
333 stream.element_readers_.push_back(reader); | 351 stream.element_readers_.push_back(reader); |
334 | 352 |
335 // Run Init(). | 353 // Run Init(). |
336 MockCompletionCallback mock_callback; | 354 MockCompletionCallback mock_callback; |
337 EXPECT_CALL(mock_callback, Run(ERR_FAILED)).Times(1); | 355 EXPECT_CALL(mock_callback, Run(ERR_FAILED)).Times(1); |
338 EXPECT_EQ(stream.Init(mock_callback.CreateCallback()), ERR_IO_PENDING); | 356 EXPECT_EQ(stream.Init(mock_callback.CreateCallback()), ERR_IO_PENDING); |
339 MessageLoop::current()->RunAllPending(); | 357 MessageLoop::current()->RunAllPending(); |
340 } | 358 } |
341 | 359 |
| 360 // Read with a buffer whose size is same as the data. |
| 361 TEST_F(UploadDataStreamTest, ReadAsyncWithExactSizeBuffer) { |
| 362 upload_data_->AppendBytes(kTestData, kTestDataSize); |
| 363 UploadDataStream stream(upload_data_); |
| 364 |
| 365 MockCompletionCallback mock_callback; |
| 366 EXPECT_CALL(mock_callback, Run(_)).Times(0); |
| 367 |
| 368 ASSERT_EQ(OK, stream.Init(mock_callback.CreateCallback())); |
| 369 EXPECT_TRUE(stream.IsInMemory()); |
| 370 EXPECT_EQ(kTestDataSize, stream.size()); |
| 371 EXPECT_EQ(0U, stream.position()); |
| 372 EXPECT_FALSE(stream.IsEOF()); |
| 373 scoped_refptr<IOBuffer> buf = new IOBuffer(kTestDataSize); |
| 374 int bytes_read = stream.Read(buf, kTestDataSize, |
| 375 mock_callback.CreateCallback()); |
| 376 ASSERT_EQ(static_cast<int>(kTestDataSize), bytes_read); // Not an error. |
| 377 EXPECT_EQ(kTestDataSize, stream.position()); |
| 378 ASSERT_TRUE(stream.IsEOF()); |
| 379 } |
| 380 |
| 381 // Async Read() with on-memory and not-on-memory readers. |
| 382 TEST_F(UploadDataStreamTest, ReadAsync) { |
| 383 // Create stream without element readers. |
| 384 UploadDataStream stream(upload_data_); |
| 385 |
| 386 // Set mock readers to the stream. |
| 387 MockUploadElementReader* reader = NULL; |
| 388 |
| 389 reader = new MockUploadElementReader(kTestDataSize, true); |
| 390 EXPECT_CALL(*reader, Init(_)).WillOnce(Return(OK)); |
| 391 reader->SetReadExpectation(kTestDataSize); |
| 392 stream.element_readers_.push_back(reader); |
| 393 |
| 394 reader = new MockUploadElementReader(kTestDataSize, false); |
| 395 reader->SetAsyncInitExpectation(OK); |
| 396 reader->SetReadExpectation(kTestDataSize); |
| 397 stream.element_readers_.push_back(reader); |
| 398 |
| 399 reader = new MockUploadElementReader(kTestDataSize, true); |
| 400 EXPECT_CALL(*reader, Init(_)).WillOnce(Return(OK)); |
| 401 reader->SetReadExpectation(kTestDataSize); |
| 402 stream.element_readers_.push_back(reader); |
| 403 |
| 404 reader = new MockUploadElementReader(kTestDataSize, false); |
| 405 reader->SetAsyncInitExpectation(OK); |
| 406 reader->SetReadExpectation(kTestDataSize); |
| 407 stream.element_readers_.push_back(reader); |
| 408 |
| 409 // Run Init(). |
| 410 MockCompletionCallback mock_callback; |
| 411 EXPECT_CALL(mock_callback, Run(OK)).Times(1); |
| 412 EXPECT_EQ(ERR_IO_PENDING, stream.Init(mock_callback.CreateCallback())); |
| 413 MessageLoop::current()->RunAllPending(); |
| 414 |
| 415 scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize); |
| 416 |
| 417 // Consume the first element. |
| 418 EXPECT_CALL(mock_callback, Run(kTestDataSize)).Times(0); |
| 419 EXPECT_EQ(static_cast<int>(kTestDataSize), |
| 420 stream.Read(buf, kTestDataSize, mock_callback.CreateCallback())); |
| 421 MessageLoop::current()->RunAllPending(); |
| 422 |
| 423 // Consume the second element. |
| 424 EXPECT_CALL(mock_callback, Run(kTestDataSize)).Times(1); |
| 425 EXPECT_EQ(ERR_IO_PENDING, |
| 426 stream.Read(buf, kTestDataSize, mock_callback.CreateCallback())); |
| 427 MessageLoop::current()->RunAllPending(); |
| 428 |
| 429 // Consume the third and the fourth elements. |
| 430 EXPECT_CALL(mock_callback, Run(kTestDataSize*2)).Times(1); |
| 431 EXPECT_EQ(ERR_IO_PENDING, |
| 432 stream.Read(buf, kTestDataSize*2, mock_callback.CreateCallback())); |
| 433 MessageLoop::current()->RunAllPending(); |
| 434 } |
| 435 |
342 void UploadDataStreamTest::FileChangedHelper(const FilePath& file_path, | 436 void UploadDataStreamTest::FileChangedHelper(const FilePath& file_path, |
343 const base::Time& time, | 437 const base::Time& time, |
344 bool error_expected) { | 438 bool error_expected) { |
345 std::vector<UploadElement> elements; | 439 std::vector<UploadElement> elements; |
346 UploadElement element; | 440 UploadElement element; |
347 element.SetToFilePathRange(file_path, 1, 2, time); | 441 element.SetToFilePathRange(file_path, 1, 2, time); |
348 elements.push_back(element); | 442 elements.push_back(element); |
349 // Don't use upload_data_ here, as this function is called twice, and | 443 // Don't use upload_data_ here, as this function is called twice, and |
350 // reusing upload_data_ is wrong. | 444 // reusing upload_data_ is wrong. |
351 scoped_refptr<UploadData> upload_data(new UploadData); | 445 scoped_refptr<UploadData> upload_data(new UploadData); |
352 upload_data->SetElements(elements); | 446 upload_data->SetElements(elements); |
353 | 447 |
354 scoped_ptr<UploadDataStream> stream(new UploadDataStream(upload_data)); | 448 UploadDataStream stream(upload_data); |
355 int error_code = stream->InitSync(); | 449 int error_code = stream.InitSync(); |
356 if (error_expected) | 450 if (error_expected) |
357 ASSERT_EQ(ERR_UPLOAD_FILE_CHANGED, error_code); | 451 ASSERT_EQ(ERR_UPLOAD_FILE_CHANGED, error_code); |
358 else | 452 else |
359 ASSERT_EQ(OK, error_code); | 453 ASSERT_EQ(OK, error_code); |
360 } | 454 } |
361 | 455 |
362 TEST_F(UploadDataStreamTest, FileChanged) { | 456 TEST_F(UploadDataStreamTest, FileChanged) { |
363 FilePath temp_file_path; | 457 FilePath temp_file_path; |
364 ASSERT_TRUE(file_util::CreateTemporaryFile(&temp_file_path)); | 458 ASSERT_TRUE(file_util::CreateTemporaryFile(&temp_file_path)); |
365 ASSERT_EQ(static_cast<int>(kTestDataSize), | 459 ASSERT_EQ(static_cast<int>(kTestDataSize), |
(...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
406 UploadDataStream stream(upload_data_); | 500 UploadDataStream stream(upload_data_); |
407 ASSERT_EQ(OK, stream.InitSync()); | 501 ASSERT_EQ(OK, stream.InitSync()); |
408 EXPECT_EQ(kTestDataSize, stream.size()); | 502 EXPECT_EQ(kTestDataSize, stream.size()); |
409 EXPECT_EQ(kTestData, ReadFromUploadDataStream(&stream)); | 503 EXPECT_EQ(kTestData, ReadFromUploadDataStream(&stream)); |
410 } | 504 } |
411 | 505 |
412 file_util::Delete(temp_file_path, false); | 506 file_util::Delete(temp_file_path, false); |
413 } | 507 } |
414 | 508 |
415 } // namespace net | 509 } // namespace net |
OLD | NEW |