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/elements_upload_data_stream.h" | 5 #include "net/base/elements_upload_data_stream.h" |
6 | 6 |
7 #include <algorithm> | 7 #include <algorithm> |
8 #include <vector> | 8 #include <vector> |
9 | 9 |
10 #include "base/basictypes.h" | 10 #include "base/basictypes.h" |
(...skipping 124 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
135 ~ElementsUploadDataStreamTest() override { | 135 ~ElementsUploadDataStreamTest() override { |
136 element_readers_.clear(); | 136 element_readers_.clear(); |
137 base::RunLoop().RunUntilIdle(); | 137 base::RunLoop().RunUntilIdle(); |
138 } | 138 } |
139 | 139 |
140 void FileChangedHelper(const base::FilePath& file_path, | 140 void FileChangedHelper(const base::FilePath& file_path, |
141 const base::Time& time, | 141 const base::Time& time, |
142 bool error_expected); | 142 bool error_expected); |
143 | 143 |
144 base::ScopedTempDir temp_dir_; | 144 base::ScopedTempDir temp_dir_; |
145 ScopedVector<UploadElementReader> element_readers_; | 145 std::vector<scoped_ptr<UploadElementReader>> element_readers_; |
146 }; | 146 }; |
147 | 147 |
148 TEST_F(ElementsUploadDataStreamTest, EmptyUploadData) { | 148 TEST_F(ElementsUploadDataStreamTest, EmptyUploadData) { |
149 scoped_ptr<UploadDataStream> stream( | 149 scoped_ptr<UploadDataStream> stream( |
150 new ElementsUploadDataStream(element_readers_.Pass(), 0)); | 150 new ElementsUploadDataStream(std::move(element_readers_), 0)); |
151 ASSERT_EQ(OK, stream->Init(CompletionCallback())); | 151 ASSERT_EQ(OK, stream->Init(CompletionCallback())); |
152 EXPECT_TRUE(stream->IsInMemory()); | 152 EXPECT_TRUE(stream->IsInMemory()); |
153 EXPECT_EQ(0U, stream->size()); | 153 EXPECT_EQ(0U, stream->size()); |
154 EXPECT_EQ(0U, stream->position()); | 154 EXPECT_EQ(0U, stream->position()); |
155 EXPECT_TRUE(stream->IsEOF()); | 155 EXPECT_TRUE(stream->IsEOF()); |
156 } | 156 } |
157 | 157 |
158 TEST_F(ElementsUploadDataStreamTest, ConsumeAllBytes) { | 158 TEST_F(ElementsUploadDataStreamTest, ConsumeAllBytes) { |
159 element_readers_.push_back(new UploadBytesElementReader( | 159 element_readers_.push_back( |
160 kTestData, kTestDataSize)); | 160 make_scoped_ptr(new UploadBytesElementReader(kTestData, kTestDataSize))); |
161 scoped_ptr<UploadDataStream> stream( | 161 scoped_ptr<UploadDataStream> stream( |
162 new ElementsUploadDataStream(element_readers_.Pass(), 0)); | 162 new ElementsUploadDataStream(std::move(element_readers_), 0)); |
163 ASSERT_EQ(OK, stream->Init(CompletionCallback())); | 163 ASSERT_EQ(OK, stream->Init(CompletionCallback())); |
164 EXPECT_TRUE(stream->IsInMemory()); | 164 EXPECT_TRUE(stream->IsInMemory()); |
165 EXPECT_EQ(kTestDataSize, stream->size()); | 165 EXPECT_EQ(kTestDataSize, stream->size()); |
166 EXPECT_EQ(0U, stream->position()); | 166 EXPECT_EQ(0U, stream->position()); |
167 EXPECT_FALSE(stream->IsEOF()); | 167 EXPECT_FALSE(stream->IsEOF()); |
168 scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize); | 168 scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize); |
169 while (!stream->IsEOF()) { | 169 while (!stream->IsEOF()) { |
170 int bytes_read = | 170 int bytes_read = |
171 stream->Read(buf.get(), kTestBufferSize, CompletionCallback()); | 171 stream->Read(buf.get(), kTestBufferSize, CompletionCallback()); |
172 ASSERT_LE(0, bytes_read); // Not an error. | 172 ASSERT_LE(0, bytes_read); // Not an error. |
173 } | 173 } |
174 EXPECT_EQ(kTestDataSize, stream->position()); | 174 EXPECT_EQ(kTestDataSize, stream->position()); |
175 ASSERT_TRUE(stream->IsEOF()); | 175 ASSERT_TRUE(stream->IsEOF()); |
176 } | 176 } |
177 | 177 |
178 TEST_F(ElementsUploadDataStreamTest, File) { | 178 TEST_F(ElementsUploadDataStreamTest, File) { |
179 base::FilePath temp_file_path; | 179 base::FilePath temp_file_path; |
180 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), | 180 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), |
181 &temp_file_path)); | 181 &temp_file_path)); |
182 ASSERT_EQ(static_cast<int>(kTestDataSize), | 182 ASSERT_EQ(static_cast<int>(kTestDataSize), |
183 base::WriteFile(temp_file_path, kTestData, kTestDataSize)); | 183 base::WriteFile(temp_file_path, kTestData, kTestDataSize)); |
184 | 184 |
185 element_readers_.push_back( | 185 element_readers_.push_back(make_scoped_ptr(new UploadFileElementReader( |
186 new UploadFileElementReader(base::ThreadTaskRunnerHandle::Get().get(), | 186 base::ThreadTaskRunnerHandle::Get().get(), temp_file_path, 0, kuint64max, |
187 temp_file_path, 0, kuint64max, base::Time())); | 187 base::Time()))); |
188 | 188 |
189 TestCompletionCallback init_callback; | 189 TestCompletionCallback init_callback; |
190 scoped_ptr<UploadDataStream> stream( | 190 scoped_ptr<UploadDataStream> stream( |
191 new ElementsUploadDataStream(element_readers_.Pass(), 0)); | 191 new ElementsUploadDataStream(std::move(element_readers_), 0)); |
192 ASSERT_EQ(ERR_IO_PENDING, stream->Init(init_callback.callback())); | 192 ASSERT_EQ(ERR_IO_PENDING, stream->Init(init_callback.callback())); |
193 ASSERT_EQ(OK, init_callback.WaitForResult()); | 193 ASSERT_EQ(OK, init_callback.WaitForResult()); |
194 EXPECT_FALSE(stream->IsInMemory()); | 194 EXPECT_FALSE(stream->IsInMemory()); |
195 EXPECT_EQ(kTestDataSize, stream->size()); | 195 EXPECT_EQ(kTestDataSize, stream->size()); |
196 EXPECT_EQ(0U, stream->position()); | 196 EXPECT_EQ(0U, stream->position()); |
197 EXPECT_FALSE(stream->IsEOF()); | 197 EXPECT_FALSE(stream->IsEOF()); |
198 scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize); | 198 scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize); |
199 while (!stream->IsEOF()) { | 199 while (!stream->IsEOF()) { |
200 TestCompletionCallback read_callback; | 200 TestCompletionCallback read_callback; |
201 ASSERT_EQ( | 201 ASSERT_EQ( |
202 ERR_IO_PENDING, | 202 ERR_IO_PENDING, |
203 stream->Read(buf.get(), kTestBufferSize, read_callback.callback())); | 203 stream->Read(buf.get(), kTestBufferSize, read_callback.callback())); |
204 ASSERT_LE(0, read_callback.WaitForResult()); // Not an error. | 204 ASSERT_LE(0, read_callback.WaitForResult()); // Not an error. |
205 } | 205 } |
206 EXPECT_EQ(kTestDataSize, stream->position()); | 206 EXPECT_EQ(kTestDataSize, stream->position()); |
207 ASSERT_TRUE(stream->IsEOF()); | 207 ASSERT_TRUE(stream->IsEOF()); |
208 } | 208 } |
209 | 209 |
210 TEST_F(ElementsUploadDataStreamTest, FileSmallerThanLength) { | 210 TEST_F(ElementsUploadDataStreamTest, FileSmallerThanLength) { |
211 base::FilePath temp_file_path; | 211 base::FilePath temp_file_path; |
212 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), | 212 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), |
213 &temp_file_path)); | 213 &temp_file_path)); |
214 ASSERT_EQ(static_cast<int>(kTestDataSize), | 214 ASSERT_EQ(static_cast<int>(kTestDataSize), |
215 base::WriteFile(temp_file_path, kTestData, kTestDataSize)); | 215 base::WriteFile(temp_file_path, kTestData, kTestDataSize)); |
216 const uint64_t kFakeSize = kTestDataSize * 2; | 216 const uint64_t kFakeSize = kTestDataSize * 2; |
217 | 217 |
218 UploadFileElementReader::ScopedOverridingContentLengthForTests | 218 UploadFileElementReader::ScopedOverridingContentLengthForTests |
219 overriding_content_length(kFakeSize); | 219 overriding_content_length(kFakeSize); |
220 | 220 |
221 element_readers_.push_back( | 221 element_readers_.push_back(make_scoped_ptr(new UploadFileElementReader( |
222 new UploadFileElementReader(base::ThreadTaskRunnerHandle::Get().get(), | 222 base::ThreadTaskRunnerHandle::Get().get(), temp_file_path, 0, kuint64max, |
223 temp_file_path, 0, kuint64max, base::Time())); | 223 base::Time()))); |
224 | 224 |
225 TestCompletionCallback init_callback; | 225 TestCompletionCallback init_callback; |
226 scoped_ptr<UploadDataStream> stream( | 226 scoped_ptr<UploadDataStream> stream( |
227 new ElementsUploadDataStream(element_readers_.Pass(), 0)); | 227 new ElementsUploadDataStream(std::move(element_readers_), 0)); |
228 ASSERT_EQ(ERR_IO_PENDING, stream->Init(init_callback.callback())); | 228 ASSERT_EQ(ERR_IO_PENDING, stream->Init(init_callback.callback())); |
229 ASSERT_EQ(OK, init_callback.WaitForResult()); | 229 ASSERT_EQ(OK, init_callback.WaitForResult()); |
230 EXPECT_FALSE(stream->IsInMemory()); | 230 EXPECT_FALSE(stream->IsInMemory()); |
231 EXPECT_EQ(kFakeSize, stream->size()); | 231 EXPECT_EQ(kFakeSize, stream->size()); |
232 EXPECT_EQ(0U, stream->position()); | 232 EXPECT_EQ(0U, stream->position()); |
233 EXPECT_FALSE(stream->IsEOF()); | 233 EXPECT_FALSE(stream->IsEOF()); |
234 uint64_t read_counter = 0; | 234 uint64_t read_counter = 0; |
235 scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize); | 235 scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize); |
236 while (!stream->IsEOF()) { | 236 while (!stream->IsEOF()) { |
237 TestCompletionCallback read_callback; | 237 TestCompletionCallback read_callback; |
238 ASSERT_EQ( | 238 ASSERT_EQ( |
239 ERR_IO_PENDING, | 239 ERR_IO_PENDING, |
240 stream->Read(buf.get(), kTestBufferSize, read_callback.callback())); | 240 stream->Read(buf.get(), kTestBufferSize, read_callback.callback())); |
241 int bytes_read = read_callback.WaitForResult(); | 241 int bytes_read = read_callback.WaitForResult(); |
242 ASSERT_LE(0, bytes_read); // Not an error. | 242 ASSERT_LE(0, bytes_read); // Not an error. |
243 read_counter += bytes_read; | 243 read_counter += bytes_read; |
244 EXPECT_EQ(read_counter, stream->position()); | 244 EXPECT_EQ(read_counter, stream->position()); |
245 } | 245 } |
246 // UpdateDataStream will pad out the file with 0 bytes so that the HTTP | 246 // UpdateDataStream will pad out the file with 0 bytes so that the HTTP |
247 // transaction doesn't hang. Therefore we expected the full size. | 247 // transaction doesn't hang. Therefore we expected the full size. |
248 EXPECT_EQ(kFakeSize, read_counter); | 248 EXPECT_EQ(kFakeSize, read_counter); |
249 EXPECT_EQ(read_counter, stream->position()); | 249 EXPECT_EQ(read_counter, stream->position()); |
250 } | 250 } |
251 | 251 |
252 TEST_F(ElementsUploadDataStreamTest, ReadErrorSync) { | 252 TEST_F(ElementsUploadDataStreamTest, ReadErrorSync) { |
253 // This element cannot be read. | 253 // This element cannot be read. |
254 MockUploadElementReader* reader = | 254 scoped_ptr<MockUploadElementReader> reader( |
255 new MockUploadElementReader(kTestDataSize, true); | 255 new MockUploadElementReader(kTestDataSize, true)); |
256 EXPECT_CALL(*reader, Init(_)).WillOnce(Return(OK)); | 256 EXPECT_CALL(*reader, Init(_)).WillOnce(Return(OK)); |
257 reader->SetReadExpectation(ERR_FAILED); | 257 reader->SetReadExpectation(ERR_FAILED); |
258 element_readers_.push_back(reader); | 258 element_readers_.push_back(std::move(reader)); |
259 | 259 |
260 // This element is ignored because of the error from the previous reader. | 260 // This element is ignored because of the error from the previous reader. |
261 element_readers_.push_back(new UploadBytesElementReader( | 261 element_readers_.push_back( |
262 kTestData, kTestDataSize)); | 262 make_scoped_ptr(new UploadBytesElementReader(kTestData, kTestDataSize))); |
263 | 263 |
264 scoped_ptr<UploadDataStream> stream( | 264 scoped_ptr<UploadDataStream> stream( |
265 new ElementsUploadDataStream(element_readers_.Pass(), 0)); | 265 new ElementsUploadDataStream(std::move(element_readers_), 0)); |
266 | 266 |
267 // Run Init(). | 267 // Run Init(). |
268 ASSERT_EQ(OK, stream->Init(CompletionCallback())); | 268 ASSERT_EQ(OK, stream->Init(CompletionCallback())); |
269 EXPECT_EQ(kTestDataSize*2, stream->size()); | 269 EXPECT_EQ(kTestDataSize*2, stream->size()); |
270 EXPECT_EQ(0U, stream->position()); | 270 EXPECT_EQ(0U, stream->position()); |
271 EXPECT_FALSE(stream->IsEOF()); | 271 EXPECT_FALSE(stream->IsEOF()); |
272 | 272 |
273 // Prepare a buffer filled with non-zero data. | 273 // Prepare a buffer filled with non-zero data. |
274 scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize); | 274 scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize); |
275 std::fill_n(buf->data(), kTestBufferSize, -1); | 275 std::fill_n(buf->data(), kTestBufferSize, -1); |
276 | 276 |
277 // Read() results in success even when the reader returns error. | 277 // Read() results in success even when the reader returns error. |
278 EXPECT_EQ(static_cast<int>(kTestDataSize * 2), | 278 EXPECT_EQ(static_cast<int>(kTestDataSize * 2), |
279 stream->Read(buf.get(), kTestBufferSize, CompletionCallback())); | 279 stream->Read(buf.get(), kTestBufferSize, CompletionCallback())); |
280 EXPECT_EQ(kTestDataSize * 2, stream->position()); | 280 EXPECT_EQ(kTestDataSize * 2, stream->position()); |
281 EXPECT_TRUE(stream->IsEOF()); | 281 EXPECT_TRUE(stream->IsEOF()); |
282 | 282 |
283 // The buffer is filled with zero. | 283 // The buffer is filled with zero. |
284 EXPECT_EQ(static_cast<int>(kTestDataSize*2), | 284 EXPECT_EQ(static_cast<int>(kTestDataSize*2), |
285 std::count(buf->data(), buf->data() + kTestBufferSize, 0)); | 285 std::count(buf->data(), buf->data() + kTestBufferSize, 0)); |
286 } | 286 } |
287 | 287 |
288 TEST_F(ElementsUploadDataStreamTest, ReadErrorAsync) { | 288 TEST_F(ElementsUploadDataStreamTest, ReadErrorAsync) { |
289 // This element cannot be read. | 289 // This element cannot be read. |
290 MockUploadElementReader* reader = | 290 scoped_ptr<MockUploadElementReader> reader( |
291 new MockUploadElementReader(kTestDataSize, false); | 291 new MockUploadElementReader(kTestDataSize, false)); |
292 reader->SetAsyncInitExpectation(OK); | 292 reader->SetAsyncInitExpectation(OK); |
293 reader->SetReadExpectation(ERR_FAILED); | 293 reader->SetReadExpectation(ERR_FAILED); |
294 element_readers_.push_back(reader); | 294 element_readers_.push_back(std::move(reader)); |
295 | 295 |
296 // This element is ignored because of the error from the previous reader. | 296 // This element is ignored because of the error from the previous reader. |
297 element_readers_.push_back(new UploadBytesElementReader( | 297 element_readers_.push_back( |
298 kTestData, kTestDataSize)); | 298 make_scoped_ptr(new UploadBytesElementReader(kTestData, kTestDataSize))); |
299 | 299 |
300 scoped_ptr<UploadDataStream> stream( | 300 scoped_ptr<UploadDataStream> stream( |
301 new ElementsUploadDataStream(element_readers_.Pass(), 0)); | 301 new ElementsUploadDataStream(std::move(element_readers_), 0)); |
302 | 302 |
303 // Run Init(). | 303 // Run Init(). |
304 TestCompletionCallback init_callback; | 304 TestCompletionCallback init_callback; |
305 ASSERT_EQ(ERR_IO_PENDING, stream->Init(init_callback.callback())); | 305 ASSERT_EQ(ERR_IO_PENDING, stream->Init(init_callback.callback())); |
306 EXPECT_EQ(OK, init_callback.WaitForResult()); | 306 EXPECT_EQ(OK, init_callback.WaitForResult()); |
307 EXPECT_EQ(kTestDataSize*2, stream->size()); | 307 EXPECT_EQ(kTestDataSize*2, stream->size()); |
308 EXPECT_EQ(0U, stream->position()); | 308 EXPECT_EQ(0U, stream->position()); |
309 EXPECT_FALSE(stream->IsEOF()); | 309 EXPECT_FALSE(stream->IsEOF()); |
310 | 310 |
311 // Prepare a buffer filled with non-zero data. | 311 // Prepare a buffer filled with non-zero data. |
(...skipping 15 matching lines...) Expand all Loading... |
327 | 327 |
328 TEST_F(ElementsUploadDataStreamTest, FileAndBytes) { | 328 TEST_F(ElementsUploadDataStreamTest, FileAndBytes) { |
329 base::FilePath temp_file_path; | 329 base::FilePath temp_file_path; |
330 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), | 330 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), |
331 &temp_file_path)); | 331 &temp_file_path)); |
332 ASSERT_EQ(static_cast<int>(kTestDataSize), | 332 ASSERT_EQ(static_cast<int>(kTestDataSize), |
333 base::WriteFile(temp_file_path, kTestData, kTestDataSize)); | 333 base::WriteFile(temp_file_path, kTestData, kTestDataSize)); |
334 | 334 |
335 const uint64_t kFileRangeOffset = 1; | 335 const uint64_t kFileRangeOffset = 1; |
336 const uint64_t kFileRangeLength = 4; | 336 const uint64_t kFileRangeLength = 4; |
337 element_readers_.push_back(new UploadFileElementReader( | 337 element_readers_.push_back(make_scoped_ptr(new UploadFileElementReader( |
338 base::ThreadTaskRunnerHandle::Get().get(), temp_file_path, | 338 base::ThreadTaskRunnerHandle::Get().get(), temp_file_path, |
339 kFileRangeOffset, kFileRangeLength, base::Time())); | 339 kFileRangeOffset, kFileRangeLength, base::Time()))); |
340 | 340 |
341 element_readers_.push_back(new UploadBytesElementReader( | 341 element_readers_.push_back( |
342 kTestData, kTestDataSize)); | 342 make_scoped_ptr(new UploadBytesElementReader(kTestData, kTestDataSize))); |
343 | 343 |
344 const uint64_t kStreamSize = kTestDataSize + kFileRangeLength; | 344 const uint64_t kStreamSize = kTestDataSize + kFileRangeLength; |
345 TestCompletionCallback init_callback; | 345 TestCompletionCallback init_callback; |
346 scoped_ptr<UploadDataStream> stream( | 346 scoped_ptr<UploadDataStream> stream( |
347 new ElementsUploadDataStream(element_readers_.Pass(), 0)); | 347 new ElementsUploadDataStream(std::move(element_readers_), 0)); |
348 ASSERT_EQ(ERR_IO_PENDING, stream->Init(init_callback.callback())); | 348 ASSERT_EQ(ERR_IO_PENDING, stream->Init(init_callback.callback())); |
349 ASSERT_EQ(OK, init_callback.WaitForResult()); | 349 ASSERT_EQ(OK, init_callback.WaitForResult()); |
350 EXPECT_FALSE(stream->IsInMemory()); | 350 EXPECT_FALSE(stream->IsInMemory()); |
351 EXPECT_EQ(kStreamSize, stream->size()); | 351 EXPECT_EQ(kStreamSize, stream->size()); |
352 EXPECT_EQ(0U, stream->position()); | 352 EXPECT_EQ(0U, stream->position()); |
353 EXPECT_FALSE(stream->IsEOF()); | 353 EXPECT_FALSE(stream->IsEOF()); |
354 scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize); | 354 scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize); |
355 while (!stream->IsEOF()) { | 355 while (!stream->IsEOF()) { |
356 TestCompletionCallback read_callback; | 356 TestCompletionCallback read_callback; |
357 const int result = | 357 const int result = |
358 stream->Read(buf.get(), kTestBufferSize, read_callback.callback()); | 358 stream->Read(buf.get(), kTestBufferSize, read_callback.callback()); |
359 const int bytes_read = | 359 const int bytes_read = |
360 result != ERR_IO_PENDING ? result : read_callback.WaitForResult(); | 360 result != ERR_IO_PENDING ? result : read_callback.WaitForResult(); |
361 ASSERT_LE(0, bytes_read); // Not an error. | 361 ASSERT_LE(0, bytes_read); // Not an error. |
362 } | 362 } |
363 EXPECT_EQ(kStreamSize, stream->position()); | 363 EXPECT_EQ(kStreamSize, stream->position()); |
364 ASSERT_TRUE(stream->IsEOF()); | 364 ASSERT_TRUE(stream->IsEOF()); |
365 } | 365 } |
366 | 366 |
367 // Init() with on-memory and not-on-memory readers. | 367 // Init() with on-memory and not-on-memory readers. |
368 TEST_F(ElementsUploadDataStreamTest, InitAsync) { | 368 TEST_F(ElementsUploadDataStreamTest, InitAsync) { |
369 // Create UploadDataStream with mock readers. | 369 // Create UploadDataStream with mock readers. |
370 MockUploadElementReader* reader = NULL; | 370 scoped_ptr<MockUploadElementReader> reader( |
| 371 new MockUploadElementReader(kTestDataSize, true)); |
| 372 EXPECT_CALL(*reader, Init(_)).WillOnce(Return(OK)); |
| 373 element_readers_.push_back(std::move(reader)); |
371 | 374 |
372 reader = new MockUploadElementReader(kTestDataSize, true); | 375 scoped_ptr<MockUploadElementReader> reader2( |
373 EXPECT_CALL(*reader, Init(_)).WillOnce(Return(OK)); | 376 new MockUploadElementReader(kTestDataSize, true)); |
374 element_readers_.push_back(reader); | 377 EXPECT_CALL(*reader2, Init(_)).WillOnce(Return(OK)); |
| 378 element_readers_.push_back(std::move(reader2)); |
375 | 379 |
376 reader = new MockUploadElementReader(kTestDataSize, true); | 380 scoped_ptr<MockUploadElementReader> reader3( |
377 EXPECT_CALL(*reader, Init(_)).WillOnce(Return(OK)); | 381 new MockUploadElementReader(kTestDataSize, false)); |
378 element_readers_.push_back(reader); | 382 reader3->SetAsyncInitExpectation(OK); |
| 383 element_readers_.push_back(std::move(reader3)); |
379 | 384 |
380 reader = new MockUploadElementReader(kTestDataSize, false); | 385 scoped_ptr<MockUploadElementReader> reader4( |
381 reader->SetAsyncInitExpectation(OK); | 386 new MockUploadElementReader(kTestDataSize, false)); |
382 element_readers_.push_back(reader); | 387 reader4->SetAsyncInitExpectation(OK); |
| 388 element_readers_.push_back(std::move(reader4)); |
383 | 389 |
384 reader = new MockUploadElementReader(kTestDataSize, false); | 390 scoped_ptr<MockUploadElementReader> reader5( |
385 reader->SetAsyncInitExpectation(OK); | 391 new MockUploadElementReader(kTestDataSize, true)); |
386 element_readers_.push_back(reader); | 392 EXPECT_CALL(*reader5, Init(_)).WillOnce(Return(OK)); |
387 | 393 element_readers_.push_back(std::move(reader5)); |
388 reader = new MockUploadElementReader(kTestDataSize, true); | |
389 EXPECT_CALL(*reader, Init(_)).WillOnce(Return(OK)); | |
390 element_readers_.push_back(reader); | |
391 | 394 |
392 scoped_ptr<UploadDataStream> stream( | 395 scoped_ptr<UploadDataStream> stream( |
393 new ElementsUploadDataStream(element_readers_.Pass(), 0)); | 396 new ElementsUploadDataStream(std::move(element_readers_), 0)); |
394 | 397 |
395 // Run Init(). | 398 // Run Init(). |
396 TestCompletionCallback callback; | 399 TestCompletionCallback callback; |
397 ASSERT_EQ(ERR_IO_PENDING, stream->Init(callback.callback())); | 400 ASSERT_EQ(ERR_IO_PENDING, stream->Init(callback.callback())); |
398 EXPECT_EQ(OK, callback.WaitForResult()); | 401 EXPECT_EQ(OK, callback.WaitForResult()); |
399 } | 402 } |
400 | 403 |
401 // Init() of a reader fails asynchronously. | 404 // Init() of a reader fails asynchronously. |
402 TEST_F(ElementsUploadDataStreamTest, InitAsyncFailureAsync) { | 405 TEST_F(ElementsUploadDataStreamTest, InitAsyncFailureAsync) { |
403 // Create UploadDataStream with a mock reader. | 406 // Create UploadDataStream with a mock reader. |
404 MockUploadElementReader* reader = NULL; | 407 scoped_ptr<MockUploadElementReader> reader( |
405 | 408 new MockUploadElementReader(kTestDataSize, false)); |
406 reader = new MockUploadElementReader(kTestDataSize, false); | |
407 reader->SetAsyncInitExpectation(ERR_FAILED); | 409 reader->SetAsyncInitExpectation(ERR_FAILED); |
408 element_readers_.push_back(reader); | 410 element_readers_.push_back(std::move(reader)); |
409 | 411 |
410 scoped_ptr<UploadDataStream> stream( | 412 scoped_ptr<UploadDataStream> stream( |
411 new ElementsUploadDataStream(element_readers_.Pass(), 0)); | 413 new ElementsUploadDataStream(std::move(element_readers_), 0)); |
412 | 414 |
413 // Run Init(). | 415 // Run Init(). |
414 TestCompletionCallback callback; | 416 TestCompletionCallback callback; |
415 ASSERT_EQ(ERR_IO_PENDING, stream->Init(callback.callback())); | 417 ASSERT_EQ(ERR_IO_PENDING, stream->Init(callback.callback())); |
416 EXPECT_EQ(ERR_FAILED, callback.WaitForResult()); | 418 EXPECT_EQ(ERR_FAILED, callback.WaitForResult()); |
417 } | 419 } |
418 | 420 |
419 // Init() of a reader fails synchronously. | 421 // Init() of a reader fails synchronously. |
420 TEST_F(ElementsUploadDataStreamTest, InitAsyncFailureSync) { | 422 TEST_F(ElementsUploadDataStreamTest, InitAsyncFailureSync) { |
421 // Create UploadDataStream with mock readers. | 423 // Create UploadDataStream with mock readers. |
422 MockUploadElementReader* reader = NULL; | 424 scoped_ptr<MockUploadElementReader> reader( |
| 425 new MockUploadElementReader(kTestDataSize, false)); |
| 426 reader->SetAsyncInitExpectation(OK); |
| 427 element_readers_.push_back(std::move(reader)); |
423 | 428 |
424 reader = new MockUploadElementReader(kTestDataSize, false); | 429 scoped_ptr<MockUploadElementReader> reader2( |
425 reader->SetAsyncInitExpectation(OK); | 430 new MockUploadElementReader(kTestDataSize, true)); |
426 element_readers_.push_back(reader); | 431 EXPECT_CALL(*reader2, Init(_)).WillOnce(Return(ERR_FAILED)); |
427 | 432 element_readers_.push_back(std::move(reader2)); |
428 reader = new MockUploadElementReader(kTestDataSize, true); | |
429 EXPECT_CALL(*reader, Init(_)).WillOnce(Return(ERR_FAILED)); | |
430 element_readers_.push_back(reader); | |
431 | 433 |
432 scoped_ptr<UploadDataStream> stream( | 434 scoped_ptr<UploadDataStream> stream( |
433 new ElementsUploadDataStream(element_readers_.Pass(), 0)); | 435 new ElementsUploadDataStream(std::move(element_readers_), 0)); |
434 | 436 |
435 // Run Init(). | 437 // Run Init(). |
436 TestCompletionCallback callback; | 438 TestCompletionCallback callback; |
437 ASSERT_EQ(ERR_IO_PENDING, stream->Init(callback.callback())); | 439 ASSERT_EQ(ERR_IO_PENDING, stream->Init(callback.callback())); |
438 EXPECT_EQ(ERR_FAILED, callback.WaitForResult()); | 440 EXPECT_EQ(ERR_FAILED, callback.WaitForResult()); |
439 } | 441 } |
440 | 442 |
441 // Read with a buffer whose size is same as the data. | 443 // Read with a buffer whose size is same as the data. |
442 TEST_F(ElementsUploadDataStreamTest, ReadAsyncWithExactSizeBuffer) { | 444 TEST_F(ElementsUploadDataStreamTest, ReadAsyncWithExactSizeBuffer) { |
443 element_readers_.push_back(new UploadBytesElementReader( | 445 element_readers_.push_back( |
444 kTestData, kTestDataSize)); | 446 make_scoped_ptr(new UploadBytesElementReader(kTestData, kTestDataSize))); |
445 scoped_ptr<UploadDataStream> stream( | 447 scoped_ptr<UploadDataStream> stream( |
446 new ElementsUploadDataStream(element_readers_.Pass(), 0)); | 448 new ElementsUploadDataStream(std::move(element_readers_), 0)); |
447 | 449 |
448 ASSERT_EQ(OK, stream->Init(CompletionCallback())); | 450 ASSERT_EQ(OK, stream->Init(CompletionCallback())); |
449 EXPECT_TRUE(stream->IsInMemory()); | 451 EXPECT_TRUE(stream->IsInMemory()); |
450 EXPECT_EQ(kTestDataSize, stream->size()); | 452 EXPECT_EQ(kTestDataSize, stream->size()); |
451 EXPECT_EQ(0U, stream->position()); | 453 EXPECT_EQ(0U, stream->position()); |
452 EXPECT_FALSE(stream->IsEOF()); | 454 EXPECT_FALSE(stream->IsEOF()); |
453 scoped_refptr<IOBuffer> buf = new IOBuffer(kTestDataSize); | 455 scoped_refptr<IOBuffer> buf = new IOBuffer(kTestDataSize); |
454 int bytes_read = stream->Read(buf.get(), kTestDataSize, CompletionCallback()); | 456 int bytes_read = stream->Read(buf.get(), kTestDataSize, CompletionCallback()); |
455 ASSERT_EQ(static_cast<int>(kTestDataSize), bytes_read); // Not an error. | 457 ASSERT_EQ(static_cast<int>(kTestDataSize), bytes_read); // Not an error. |
456 EXPECT_EQ(kTestDataSize, stream->position()); | 458 EXPECT_EQ(kTestDataSize, stream->position()); |
457 ASSERT_TRUE(stream->IsEOF()); | 459 ASSERT_TRUE(stream->IsEOF()); |
458 } | 460 } |
459 | 461 |
460 // Async Read() with on-memory and not-on-memory readers. | 462 // Async Read() with on-memory and not-on-memory readers. |
461 TEST_F(ElementsUploadDataStreamTest, ReadAsync) { | 463 TEST_F(ElementsUploadDataStreamTest, ReadAsync) { |
462 // Create UploadDataStream with mock readers. | 464 // Create UploadDataStream with mock readers. |
463 MockUploadElementReader* reader = NULL; | 465 scoped_ptr<MockUploadElementReader> reader( |
464 | 466 new MockUploadElementReader(kTestDataSize, true)); |
465 reader = new MockUploadElementReader(kTestDataSize, true); | |
466 EXPECT_CALL(*reader, Init(_)).WillOnce(Return(OK)); | 467 EXPECT_CALL(*reader, Init(_)).WillOnce(Return(OK)); |
467 reader->SetReadExpectation(kTestDataSize); | 468 reader->SetReadExpectation(kTestDataSize); |
468 element_readers_.push_back(reader); | 469 element_readers_.push_back(std::move(reader)); |
469 | 470 |
470 reader = new MockUploadElementReader(kTestDataSize, false); | 471 scoped_ptr<MockUploadElementReader> reader2( |
471 reader->SetAsyncInitExpectation(OK); | 472 new MockUploadElementReader(kTestDataSize, false)); |
472 reader->SetReadExpectation(kTestDataSize); | 473 reader2->SetAsyncInitExpectation(OK); |
473 element_readers_.push_back(reader); | 474 reader2->SetReadExpectation(kTestDataSize); |
| 475 element_readers_.push_back(std::move(reader2)); |
474 | 476 |
475 reader = new MockUploadElementReader(kTestDataSize, true); | 477 scoped_ptr<MockUploadElementReader> reader3( |
476 EXPECT_CALL(*reader, Init(_)).WillOnce(Return(OK)); | 478 new MockUploadElementReader(kTestDataSize, true)); |
477 reader->SetReadExpectation(kTestDataSize); | 479 EXPECT_CALL(*reader3, Init(_)).WillOnce(Return(OK)); |
478 element_readers_.push_back(reader); | 480 reader3->SetReadExpectation(kTestDataSize); |
| 481 element_readers_.push_back(std::move(reader3)); |
479 | 482 |
480 reader = new MockUploadElementReader(kTestDataSize, false); | 483 scoped_ptr<MockUploadElementReader> reader4( |
481 reader->SetAsyncInitExpectation(OK); | 484 new MockUploadElementReader(kTestDataSize, false)); |
482 reader->SetReadExpectation(kTestDataSize); | 485 reader4->SetAsyncInitExpectation(OK); |
483 element_readers_.push_back(reader); | 486 reader4->SetReadExpectation(kTestDataSize); |
| 487 element_readers_.push_back(std::move(reader4)); |
484 | 488 |
485 scoped_ptr<UploadDataStream> stream( | 489 scoped_ptr<UploadDataStream> stream( |
486 new ElementsUploadDataStream(element_readers_.Pass(), 0)); | 490 new ElementsUploadDataStream(std::move(element_readers_), 0)); |
487 | 491 |
488 // Run Init(). | 492 // Run Init(). |
489 TestCompletionCallback init_callback; | 493 TestCompletionCallback init_callback; |
490 EXPECT_EQ(ERR_IO_PENDING, stream->Init(init_callback.callback())); | 494 EXPECT_EQ(ERR_IO_PENDING, stream->Init(init_callback.callback())); |
491 EXPECT_EQ(OK, init_callback.WaitForResult()); | 495 EXPECT_EQ(OK, init_callback.WaitForResult()); |
492 | 496 |
493 scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize); | 497 scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize); |
494 | 498 |
495 // Consume the first element. | 499 // Consume the first element. |
496 TestCompletionCallback read_callback1; | 500 TestCompletionCallback read_callback1; |
(...skipping 16 matching lines...) Expand all Loading... |
513 EXPECT_EQ(static_cast<int>(kTestDataSize * 2), | 517 EXPECT_EQ(static_cast<int>(kTestDataSize * 2), |
514 read_callback3.WaitForResult()); | 518 read_callback3.WaitForResult()); |
515 } | 519 } |
516 | 520 |
517 void ElementsUploadDataStreamTest::FileChangedHelper( | 521 void ElementsUploadDataStreamTest::FileChangedHelper( |
518 const base::FilePath& file_path, | 522 const base::FilePath& file_path, |
519 const base::Time& time, | 523 const base::Time& time, |
520 bool error_expected) { | 524 bool error_expected) { |
521 // Don't use element_readers_ here, as this function is called twice, and | 525 // Don't use element_readers_ here, as this function is called twice, and |
522 // reusing element_readers_ is wrong. | 526 // reusing element_readers_ is wrong. |
523 ScopedVector<UploadElementReader> element_readers; | 527 std::vector<scoped_ptr<UploadElementReader>> element_readers; |
524 element_readers.push_back(new UploadFileElementReader( | 528 element_readers.push_back(make_scoped_ptr(new UploadFileElementReader( |
525 base::ThreadTaskRunnerHandle::Get().get(), file_path, 1, 2, time)); | 529 base::ThreadTaskRunnerHandle::Get().get(), file_path, 1, 2, time))); |
526 | 530 |
527 TestCompletionCallback init_callback; | 531 TestCompletionCallback init_callback; |
528 scoped_ptr<UploadDataStream> stream( | 532 scoped_ptr<UploadDataStream> stream( |
529 new ElementsUploadDataStream(element_readers.Pass(), 0)); | 533 new ElementsUploadDataStream(std::move(element_readers), 0)); |
530 ASSERT_EQ(ERR_IO_PENDING, stream->Init(init_callback.callback())); | 534 ASSERT_EQ(ERR_IO_PENDING, stream->Init(init_callback.callback())); |
531 int error_code = init_callback.WaitForResult(); | 535 int error_code = init_callback.WaitForResult(); |
532 if (error_expected) | 536 if (error_expected) |
533 ASSERT_EQ(ERR_UPLOAD_FILE_CHANGED, error_code); | 537 ASSERT_EQ(ERR_UPLOAD_FILE_CHANGED, error_code); |
534 else | 538 else |
535 ASSERT_EQ(OK, error_code); | 539 ASSERT_EQ(OK, error_code); |
536 } | 540 } |
537 | 541 |
538 TEST_F(ElementsUploadDataStreamTest, FileChanged) { | 542 TEST_F(ElementsUploadDataStreamTest, FileChanged) { |
539 base::FilePath temp_file_path; | 543 base::FilePath temp_file_path; |
(...skipping 15 matching lines...) Expand all Loading... |
555 } | 559 } |
556 | 560 |
557 TEST_F(ElementsUploadDataStreamTest, MultipleInit) { | 561 TEST_F(ElementsUploadDataStreamTest, MultipleInit) { |
558 base::FilePath temp_file_path; | 562 base::FilePath temp_file_path; |
559 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), | 563 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), |
560 &temp_file_path)); | 564 &temp_file_path)); |
561 ASSERT_EQ(static_cast<int>(kTestDataSize), | 565 ASSERT_EQ(static_cast<int>(kTestDataSize), |
562 base::WriteFile(temp_file_path, kTestData, kTestDataSize)); | 566 base::WriteFile(temp_file_path, kTestData, kTestDataSize)); |
563 | 567 |
564 // Prepare data. | 568 // Prepare data. |
565 element_readers_.push_back(new UploadBytesElementReader( | |
566 kTestData, kTestDataSize)); | |
567 element_readers_.push_back( | 569 element_readers_.push_back( |
568 new UploadFileElementReader(base::ThreadTaskRunnerHandle::Get().get(), | 570 make_scoped_ptr(new UploadBytesElementReader(kTestData, kTestDataSize))); |
569 temp_file_path, 0, kuint64max, base::Time())); | 571 element_readers_.push_back(make_scoped_ptr(new UploadFileElementReader( |
| 572 base::ThreadTaskRunnerHandle::Get().get(), temp_file_path, 0, kuint64max, |
| 573 base::Time()))); |
570 scoped_ptr<UploadDataStream> stream( | 574 scoped_ptr<UploadDataStream> stream( |
571 new ElementsUploadDataStream(element_readers_.Pass(), 0)); | 575 new ElementsUploadDataStream(std::move(element_readers_), 0)); |
572 | 576 |
573 std::string expected_data(kTestData, kTestData + kTestDataSize); | 577 std::string expected_data(kTestData, kTestData + kTestDataSize); |
574 expected_data += expected_data; | 578 expected_data += expected_data; |
575 | 579 |
576 // Call Init(). | 580 // Call Init(). |
577 TestCompletionCallback init_callback1; | 581 TestCompletionCallback init_callback1; |
578 ASSERT_EQ(ERR_IO_PENDING, stream->Init(init_callback1.callback())); | 582 ASSERT_EQ(ERR_IO_PENDING, stream->Init(init_callback1.callback())); |
579 ASSERT_EQ(OK, init_callback1.WaitForResult()); | 583 ASSERT_EQ(OK, init_callback1.WaitForResult()); |
580 EXPECT_FALSE(stream->IsEOF()); | 584 EXPECT_FALSE(stream->IsEOF()); |
581 EXPECT_EQ(kTestDataSize*2, stream->size()); | 585 EXPECT_EQ(kTestDataSize*2, stream->size()); |
(...skipping 16 matching lines...) Expand all Loading... |
598 | 602 |
599 TEST_F(ElementsUploadDataStreamTest, MultipleInitAsync) { | 603 TEST_F(ElementsUploadDataStreamTest, MultipleInitAsync) { |
600 base::FilePath temp_file_path; | 604 base::FilePath temp_file_path; |
601 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), | 605 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), |
602 &temp_file_path)); | 606 &temp_file_path)); |
603 ASSERT_EQ(static_cast<int>(kTestDataSize), | 607 ASSERT_EQ(static_cast<int>(kTestDataSize), |
604 base::WriteFile(temp_file_path, kTestData, kTestDataSize)); | 608 base::WriteFile(temp_file_path, kTestData, kTestDataSize)); |
605 TestCompletionCallback test_callback; | 609 TestCompletionCallback test_callback; |
606 | 610 |
607 // Prepare data. | 611 // Prepare data. |
608 element_readers_.push_back(new UploadBytesElementReader( | |
609 kTestData, kTestDataSize)); | |
610 element_readers_.push_back( | 612 element_readers_.push_back( |
611 new UploadFileElementReader(base::ThreadTaskRunnerHandle::Get().get(), | 613 make_scoped_ptr(new UploadBytesElementReader(kTestData, kTestDataSize))); |
612 temp_file_path, 0, kuint64max, base::Time())); | 614 element_readers_.push_back(make_scoped_ptr(new UploadFileElementReader( |
| 615 base::ThreadTaskRunnerHandle::Get().get(), temp_file_path, 0, kuint64max, |
| 616 base::Time()))); |
613 scoped_ptr<UploadDataStream> stream( | 617 scoped_ptr<UploadDataStream> stream( |
614 new ElementsUploadDataStream(element_readers_.Pass(), 0)); | 618 new ElementsUploadDataStream(std::move(element_readers_), 0)); |
615 | 619 |
616 std::string expected_data(kTestData, kTestData + kTestDataSize); | 620 std::string expected_data(kTestData, kTestData + kTestDataSize); |
617 expected_data += expected_data; | 621 expected_data += expected_data; |
618 | 622 |
619 // Call Init(). | 623 // Call Init(). |
620 ASSERT_EQ(ERR_IO_PENDING, stream->Init(test_callback.callback())); | 624 ASSERT_EQ(ERR_IO_PENDING, stream->Init(test_callback.callback())); |
621 EXPECT_EQ(OK, test_callback.WaitForResult()); | 625 EXPECT_EQ(OK, test_callback.WaitForResult()); |
622 EXPECT_FALSE(stream->IsEOF()); | 626 EXPECT_FALSE(stream->IsEOF()); |
623 EXPECT_EQ(kTestDataSize*2, stream->size()); | 627 EXPECT_EQ(kTestDataSize*2, stream->size()); |
624 | 628 |
(...skipping 13 matching lines...) Expand all Loading... |
638 } | 642 } |
639 | 643 |
640 TEST_F(ElementsUploadDataStreamTest, InitToReset) { | 644 TEST_F(ElementsUploadDataStreamTest, InitToReset) { |
641 base::FilePath temp_file_path; | 645 base::FilePath temp_file_path; |
642 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), | 646 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), |
643 &temp_file_path)); | 647 &temp_file_path)); |
644 ASSERT_EQ(static_cast<int>(kTestDataSize), | 648 ASSERT_EQ(static_cast<int>(kTestDataSize), |
645 base::WriteFile(temp_file_path, kTestData, kTestDataSize)); | 649 base::WriteFile(temp_file_path, kTestData, kTestDataSize)); |
646 | 650 |
647 // Prepare data. | 651 // Prepare data. |
648 element_readers_.push_back(new UploadBytesElementReader( | |
649 kTestData, kTestDataSize)); | |
650 element_readers_.push_back( | 652 element_readers_.push_back( |
651 new UploadFileElementReader(base::ThreadTaskRunnerHandle::Get().get(), | 653 make_scoped_ptr(new UploadBytesElementReader(kTestData, kTestDataSize))); |
652 temp_file_path, 0, kuint64max, base::Time())); | 654 element_readers_.push_back(make_scoped_ptr(new UploadFileElementReader( |
| 655 base::ThreadTaskRunnerHandle::Get().get(), temp_file_path, 0, kuint64max, |
| 656 base::Time()))); |
653 scoped_ptr<UploadDataStream> stream( | 657 scoped_ptr<UploadDataStream> stream( |
654 new ElementsUploadDataStream(element_readers_.Pass(), 0)); | 658 new ElementsUploadDataStream(std::move(element_readers_), 0)); |
655 | 659 |
656 std::vector<char> expected_data(kTestData, kTestData + kTestDataSize); | 660 std::vector<char> expected_data(kTestData, kTestData + kTestDataSize); |
657 expected_data.insert(expected_data.end(), kTestData, | 661 expected_data.insert(expected_data.end(), kTestData, |
658 kTestData + kTestDataSize); | 662 kTestData + kTestDataSize); |
659 | 663 |
660 // Call Init(). | 664 // Call Init(). |
661 TestCompletionCallback init_callback1; | 665 TestCompletionCallback init_callback1; |
662 ASSERT_EQ(ERR_IO_PENDING, stream->Init(init_callback1.callback())); | 666 ASSERT_EQ(ERR_IO_PENDING, stream->Init(init_callback1.callback())); |
663 EXPECT_EQ(OK, init_callback1.WaitForResult()); | 667 EXPECT_EQ(OK, init_callback1.WaitForResult()); |
664 EXPECT_FALSE(stream->IsEOF()); | 668 EXPECT_FALSE(stream->IsEOF()); |
(...skipping 29 matching lines...) Expand all Loading... |
694 } | 698 } |
695 | 699 |
696 TEST_F(ElementsUploadDataStreamTest, InitDuringAsyncInit) { | 700 TEST_F(ElementsUploadDataStreamTest, InitDuringAsyncInit) { |
697 base::FilePath temp_file_path; | 701 base::FilePath temp_file_path; |
698 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), | 702 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), |
699 &temp_file_path)); | 703 &temp_file_path)); |
700 ASSERT_EQ(static_cast<int>(kTestDataSize), | 704 ASSERT_EQ(static_cast<int>(kTestDataSize), |
701 base::WriteFile(temp_file_path, kTestData, kTestDataSize)); | 705 base::WriteFile(temp_file_path, kTestData, kTestDataSize)); |
702 | 706 |
703 // Prepare data. | 707 // Prepare data. |
704 element_readers_.push_back(new UploadBytesElementReader( | |
705 kTestData, kTestDataSize)); | |
706 element_readers_.push_back( | 708 element_readers_.push_back( |
707 new UploadFileElementReader(base::ThreadTaskRunnerHandle::Get().get(), | 709 make_scoped_ptr(new UploadBytesElementReader(kTestData, kTestDataSize))); |
708 temp_file_path, 0, kuint64max, base::Time())); | 710 element_readers_.push_back(make_scoped_ptr(new UploadFileElementReader( |
| 711 base::ThreadTaskRunnerHandle::Get().get(), temp_file_path, 0, kuint64max, |
| 712 base::Time()))); |
709 scoped_ptr<UploadDataStream> stream( | 713 scoped_ptr<UploadDataStream> stream( |
710 new ElementsUploadDataStream(element_readers_.Pass(), 0)); | 714 new ElementsUploadDataStream(std::move(element_readers_), 0)); |
711 | 715 |
712 std::vector<char> expected_data(kTestData, kTestData + kTestDataSize); | 716 std::vector<char> expected_data(kTestData, kTestData + kTestDataSize); |
713 expected_data.insert(expected_data.end(), kTestData, | 717 expected_data.insert(expected_data.end(), kTestData, |
714 kTestData + kTestDataSize); | 718 kTestData + kTestDataSize); |
715 | 719 |
716 // Start Init. | 720 // Start Init. |
717 TestCompletionCallback init_callback1; | 721 TestCompletionCallback init_callback1; |
718 EXPECT_EQ(ERR_IO_PENDING, stream->Init(init_callback1.callback())); | 722 EXPECT_EQ(ERR_IO_PENDING, stream->Init(init_callback1.callback())); |
719 | 723 |
720 // Call Init again to cancel the previous init. | 724 // Call Init again to cancel the previous init. |
(...skipping 19 matching lines...) Expand all Loading... |
740 } | 744 } |
741 | 745 |
742 TEST_F(ElementsUploadDataStreamTest, InitDuringAsyncRead) { | 746 TEST_F(ElementsUploadDataStreamTest, InitDuringAsyncRead) { |
743 base::FilePath temp_file_path; | 747 base::FilePath temp_file_path; |
744 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), | 748 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), |
745 &temp_file_path)); | 749 &temp_file_path)); |
746 ASSERT_EQ(static_cast<int>(kTestDataSize), | 750 ASSERT_EQ(static_cast<int>(kTestDataSize), |
747 base::WriteFile(temp_file_path, kTestData, kTestDataSize)); | 751 base::WriteFile(temp_file_path, kTestData, kTestDataSize)); |
748 | 752 |
749 // Prepare data. | 753 // Prepare data. |
750 element_readers_.push_back(new UploadBytesElementReader( | |
751 kTestData, kTestDataSize)); | |
752 element_readers_.push_back( | 754 element_readers_.push_back( |
753 new UploadFileElementReader(base::ThreadTaskRunnerHandle::Get().get(), | 755 make_scoped_ptr(new UploadBytesElementReader(kTestData, kTestDataSize))); |
754 temp_file_path, 0, kuint64max, base::Time())); | 756 element_readers_.push_back(make_scoped_ptr(new UploadFileElementReader( |
| 757 base::ThreadTaskRunnerHandle::Get().get(), temp_file_path, 0, kuint64max, |
| 758 base::Time()))); |
755 scoped_ptr<UploadDataStream> stream( | 759 scoped_ptr<UploadDataStream> stream( |
756 new ElementsUploadDataStream(element_readers_.Pass(), 0)); | 760 new ElementsUploadDataStream(std::move(element_readers_), 0)); |
757 | 761 |
758 std::vector<char> expected_data(kTestData, kTestData + kTestDataSize); | 762 std::vector<char> expected_data(kTestData, kTestData + kTestDataSize); |
759 expected_data.insert(expected_data.end(), kTestData, | 763 expected_data.insert(expected_data.end(), kTestData, |
760 kTestData + kTestDataSize); | 764 kTestData + kTestDataSize); |
761 | 765 |
762 // Call Init(). | 766 // Call Init(). |
763 TestCompletionCallback init_callback1; | 767 TestCompletionCallback init_callback1; |
764 ASSERT_EQ(ERR_IO_PENDING, stream->Init(init_callback1.callback())); | 768 ASSERT_EQ(ERR_IO_PENDING, stream->Init(init_callback1.callback())); |
765 EXPECT_EQ(OK, init_callback1.WaitForResult()); | 769 EXPECT_EQ(OK, init_callback1.WaitForResult()); |
766 EXPECT_FALSE(stream->IsEOF()); | 770 EXPECT_FALSE(stream->IsEOF()); |
(...skipping 24 matching lines...) Expand all Loading... |
791 wrapped_buffer2.get(), buf2.size(), read_callback2.callback())); | 795 wrapped_buffer2.get(), buf2.size(), read_callback2.callback())); |
792 EXPECT_EQ(static_cast<int>(buf2.size()), read_callback2.WaitForResult()); | 796 EXPECT_EQ(static_cast<int>(buf2.size()), read_callback2.WaitForResult()); |
793 EXPECT_EQ(expected_data, buf2); | 797 EXPECT_EQ(expected_data, buf2); |
794 EXPECT_TRUE(stream->IsEOF()); | 798 EXPECT_TRUE(stream->IsEOF()); |
795 | 799 |
796 // Make sure callbacks are not called for cancelled operations. | 800 // Make sure callbacks are not called for cancelled operations. |
797 EXPECT_FALSE(read_callback1.have_result()); | 801 EXPECT_FALSE(read_callback1.have_result()); |
798 } | 802 } |
799 | 803 |
800 } // namespace net | 804 } // namespace net |
OLD | NEW |