| 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_file_element_reader.h" | 5 #include "net/base/upload_file_element_reader.h" |
| 6 | 6 |
| 7 #include "base/file_util.h" | 7 #include "base/file_util.h" |
| 8 #include "base/files/scoped_temp_dir.h" | 8 #include "base/files/scoped_temp_dir.h" |
| 9 #include "base/message_loop/message_loop_proxy.h" | 9 #include "base/message_loop/message_loop_proxy.h" |
| 10 #include "base/run_loop.h" | 10 #include "base/run_loop.h" |
| (...skipping 216 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 227 new UploadFileElementReader(base::MessageLoopProxy::current().get(), | 227 new UploadFileElementReader(base::MessageLoopProxy::current().get(), |
| 228 wrong_path, | 228 wrong_path, |
| 229 0, | 229 0, |
| 230 kuint64max, | 230 kuint64max, |
| 231 base::Time())); | 231 base::Time())); |
| 232 TestCompletionCallback init_callback; | 232 TestCompletionCallback init_callback; |
| 233 ASSERT_EQ(ERR_IO_PENDING, reader_->Init(init_callback.callback())); | 233 ASSERT_EQ(ERR_IO_PENDING, reader_->Init(init_callback.callback())); |
| 234 EXPECT_EQ(ERR_FILE_NOT_FOUND, init_callback.WaitForResult()); | 234 EXPECT_EQ(ERR_FILE_NOT_FOUND, init_callback.WaitForResult()); |
| 235 } | 235 } |
| 236 | 236 |
| 237 | |
| 238 class UploadFileElementReaderSyncTest : public PlatformTest { | |
| 239 protected: | |
| 240 virtual void SetUp() OVERRIDE { | |
| 241 // Some tests (*.ReadPartially) rely on bytes_.size() being even. | |
| 242 const char kData[] = "123456789abcdefghi"; | |
| 243 bytes_.assign(kData, kData + arraysize(kData) - 1); | |
| 244 | |
| 245 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); | |
| 246 | |
| 247 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), | |
| 248 &temp_file_path_)); | |
| 249 ASSERT_EQ( | |
| 250 static_cast<int>(bytes_.size()), | |
| 251 file_util::WriteFile(temp_file_path_, &bytes_[0], bytes_.size())); | |
| 252 | |
| 253 reader_.reset(new UploadFileElementReaderSync( | |
| 254 temp_file_path_, 0, kuint64max, base::Time())); | |
| 255 ASSERT_EQ(OK, reader_->Init(CompletionCallback())); | |
| 256 EXPECT_EQ(bytes_.size(), reader_->GetContentLength()); | |
| 257 EXPECT_EQ(bytes_.size(), reader_->BytesRemaining()); | |
| 258 EXPECT_FALSE(reader_->IsInMemory()); | |
| 259 } | |
| 260 | |
| 261 std::vector<char> bytes_; | |
| 262 scoped_ptr<UploadElementReader> reader_; | |
| 263 base::ScopedTempDir temp_dir_; | |
| 264 base::FilePath temp_file_path_; | |
| 265 }; | |
| 266 | |
| 267 TEST_F(UploadFileElementReaderSyncTest, ReadPartially) { | |
| 268 const size_t kHalfSize = bytes_.size() / 2; | |
| 269 ASSERT_EQ(bytes_.size(), kHalfSize * 2); | |
| 270 std::vector<char> buf(kHalfSize); | |
| 271 scoped_refptr<IOBuffer> wrapped_buffer = new WrappedIOBuffer(&buf[0]); | |
| 272 EXPECT_EQ( | |
| 273 static_cast<int>(buf.size()), | |
| 274 reader_->Read(wrapped_buffer.get(), buf.size(), CompletionCallback())); | |
| 275 EXPECT_EQ(bytes_.size() - buf.size(), reader_->BytesRemaining()); | |
| 276 EXPECT_EQ(std::vector<char>(bytes_.begin(), bytes_.begin() + kHalfSize), buf); | |
| 277 | |
| 278 EXPECT_EQ( | |
| 279 static_cast<int>(buf.size()), | |
| 280 reader_->Read(wrapped_buffer.get(), buf.size(), CompletionCallback())); | |
| 281 EXPECT_EQ(0U, reader_->BytesRemaining()); | |
| 282 EXPECT_EQ(std::vector<char>(bytes_.begin() + kHalfSize, bytes_.end()), buf); | |
| 283 } | |
| 284 | |
| 285 TEST_F(UploadFileElementReaderSyncTest, ReadAll) { | |
| 286 std::vector<char> buf(bytes_.size()); | |
| 287 scoped_refptr<IOBuffer> wrapped_buffer = new WrappedIOBuffer(&buf[0]); | |
| 288 EXPECT_EQ( | |
| 289 static_cast<int>(buf.size()), | |
| 290 reader_->Read(wrapped_buffer.get(), buf.size(), CompletionCallback())); | |
| 291 EXPECT_EQ(0U, reader_->BytesRemaining()); | |
| 292 EXPECT_EQ(bytes_, buf); | |
| 293 // Try to read again. | |
| 294 EXPECT_EQ( | |
| 295 0, reader_->Read(wrapped_buffer.get(), buf.size(), CompletionCallback())); | |
| 296 } | |
| 297 | |
| 298 TEST_F(UploadFileElementReaderSyncTest, ReadTooMuch) { | |
| 299 const size_t kTooLargeSize = bytes_.size() * 2; | |
| 300 std::vector<char> buf(kTooLargeSize); | |
| 301 scoped_refptr<IOBuffer> wrapped_buffer = new WrappedIOBuffer(&buf[0]); | |
| 302 EXPECT_EQ( | |
| 303 static_cast<int>(bytes_.size()), | |
| 304 reader_->Read(wrapped_buffer.get(), buf.size(), CompletionCallback())); | |
| 305 EXPECT_EQ(0U, reader_->BytesRemaining()); | |
| 306 buf.resize(bytes_.size()); // Resize to compare. | |
| 307 EXPECT_EQ(bytes_, buf); | |
| 308 } | |
| 309 | |
| 310 TEST_F(UploadFileElementReaderSyncTest, MultipleInit) { | |
| 311 std::vector<char> buf(bytes_.size()); | |
| 312 scoped_refptr<IOBuffer> wrapped_buffer = new WrappedIOBuffer(&buf[0]); | |
| 313 | |
| 314 // Read all. | |
| 315 EXPECT_EQ( | |
| 316 static_cast<int>(buf.size()), | |
| 317 reader_->Read(wrapped_buffer.get(), buf.size(), CompletionCallback())); | |
| 318 EXPECT_EQ(0U, reader_->BytesRemaining()); | |
| 319 EXPECT_EQ(bytes_, buf); | |
| 320 | |
| 321 // Call Init() again to reset the state. | |
| 322 ASSERT_EQ(OK, reader_->Init(CompletionCallback())); | |
| 323 EXPECT_EQ(bytes_.size(), reader_->GetContentLength()); | |
| 324 EXPECT_EQ(bytes_.size(), reader_->BytesRemaining()); | |
| 325 | |
| 326 // Read again. | |
| 327 EXPECT_EQ( | |
| 328 static_cast<int>(buf.size()), | |
| 329 reader_->Read(wrapped_buffer.get(), buf.size(), CompletionCallback())); | |
| 330 EXPECT_EQ(0U, reader_->BytesRemaining()); | |
| 331 EXPECT_EQ(bytes_, buf); | |
| 332 } | |
| 333 | |
| 334 TEST_F(UploadFileElementReaderSyncTest, Range) { | |
| 335 const uint64 kOffset = 2; | |
| 336 const uint64 kLength = bytes_.size() - kOffset * 3; | |
| 337 reader_.reset(new UploadFileElementReaderSync( | |
| 338 temp_file_path_, kOffset, kLength, base::Time())); | |
| 339 ASSERT_EQ(OK, reader_->Init(CompletionCallback())); | |
| 340 EXPECT_EQ(kLength, reader_->GetContentLength()); | |
| 341 EXPECT_EQ(kLength, reader_->BytesRemaining()); | |
| 342 std::vector<char> buf(kLength); | |
| 343 scoped_refptr<IOBuffer> wrapped_buffer = new WrappedIOBuffer(&buf[0]); | |
| 344 EXPECT_EQ(static_cast<int>(kLength), | |
| 345 reader_->Read(wrapped_buffer.get(), kLength, CompletionCallback())); | |
| 346 const std::vector<char> expected(bytes_.begin() + kOffset, | |
| 347 bytes_.begin() + kOffset + kLength); | |
| 348 EXPECT_EQ(expected, buf); | |
| 349 } | |
| 350 | |
| 351 TEST_F(UploadFileElementReaderSyncTest, FileChanged) { | |
| 352 base::File::Info info; | |
| 353 ASSERT_TRUE(base::GetFileInfo(temp_file_path_, &info)); | |
| 354 | |
| 355 // Expect one second before the actual modification time to simulate change. | |
| 356 const base::Time expected_modification_time = | |
| 357 info.last_modified - base::TimeDelta::FromSeconds(1); | |
| 358 reader_.reset(new UploadFileElementReaderSync( | |
| 359 temp_file_path_, 0, kuint64max, expected_modification_time)); | |
| 360 EXPECT_EQ(ERR_UPLOAD_FILE_CHANGED, reader_->Init(CompletionCallback())); | |
| 361 } | |
| 362 | |
| 363 TEST_F(UploadFileElementReaderSyncTest, WrongPath) { | |
| 364 const base::FilePath wrong_path(FILE_PATH_LITERAL("wrong_path")); | |
| 365 reader_.reset(new UploadFileElementReaderSync( | |
| 366 wrong_path, 0, kuint64max, base::Time())); | |
| 367 ASSERT_EQ(ERR_FILE_NOT_FOUND, reader_->Init(CompletionCallback())); | |
| 368 } | |
| 369 | |
| 370 } // namespace net | 237 } // namespace net |
| OLD | NEW |