| Index: webkit/blob/blob_url_request_job_unittest.cc
|
| ===================================================================
|
| --- webkit/blob/blob_url_request_job_unittest.cc (revision 201961)
|
| +++ webkit/blob/blob_url_request_job_unittest.cc (working copy)
|
| @@ -1,451 +0,0 @@
|
| -// Copyright (c) 2012 The Chromium Authors. All rights reserved.
|
| -// Use of this source code is governed by a BSD-style license that can be
|
| -// found in the LICENSE file.
|
| -
|
| -#include "base/bind.h"
|
| -#include "base/file_util.h"
|
| -#include "base/files/file_path.h"
|
| -#include "base/files/scoped_temp_dir.h"
|
| -#include "base/memory/ref_counted.h"
|
| -#include "base/memory/scoped_ptr.h"
|
| -#include "base/message_loop.h"
|
| -#include "base/time.h"
|
| -#include "net/base/io_buffer.h"
|
| -#include "net/http/http_request_headers.h"
|
| -#include "net/http/http_response_headers.h"
|
| -#include "net/url_request/url_request.h"
|
| -#include "net/url_request/url_request_context.h"
|
| -#include "net/url_request/url_request_job_factory_impl.h"
|
| -#include "testing/gtest/include/gtest/gtest.h"
|
| -#include "webkit/blob/blob_data.h"
|
| -#include "webkit/blob/blob_url_request_job.h"
|
| -#include "webkit/browser/fileapi/file_system_file_util.h"
|
| -#include "webkit/fileapi/file_system_context.h"
|
| -#include "webkit/fileapi/file_system_operation_context.h"
|
| -#include "webkit/fileapi/file_system_url.h"
|
| -#include "webkit/fileapi/mock_file_system_context.h"
|
| -
|
| -namespace webkit_blob {
|
| -
|
| -namespace {
|
| -
|
| -const int kBufferSize = 1024;
|
| -const char kTestData1[] = "Hello";
|
| -const char kTestData2[] = "Here it is data.";
|
| -const char kTestFileData1[] = "0123456789";
|
| -const char kTestFileData2[] = "This is sample file.";
|
| -const char kTestFileSystemFileData1[] = "abcdefghijklmnop";
|
| -const char kTestFileSystemFileData2[] = "File system file test data.";
|
| -const char kTestContentType[] = "foo/bar";
|
| -const char kTestContentDisposition[] = "attachment; filename=foo.txt";
|
| -
|
| -const char kFileSystemURLOrigin[] = "http://remote";
|
| -const fileapi::FileSystemType kFileSystemType =
|
| - fileapi::kFileSystemTypeTemporary;
|
| -
|
| -} // namespace
|
| -
|
| -class BlobURLRequestJobTest : public testing::Test {
|
| - public:
|
| -
|
| - // Test Harness -------------------------------------------------------------
|
| - // TODO(jianli): share this test harness with AppCacheURLRequestJobTest
|
| -
|
| - class MockURLRequestDelegate : public net::URLRequest::Delegate {
|
| - public:
|
| - MockURLRequestDelegate()
|
| - : received_data_(new net::IOBuffer(kBufferSize)) {}
|
| -
|
| - virtual void OnResponseStarted(net::URLRequest* request) OVERRIDE {
|
| - if (request->status().is_success()) {
|
| - EXPECT_TRUE(request->response_headers());
|
| - ReadSome(request);
|
| - } else {
|
| - RequestComplete();
|
| - }
|
| - }
|
| -
|
| - virtual void OnReadCompleted(net::URLRequest* request,
|
| - int bytes_read) OVERRIDE {
|
| - if (bytes_read > 0)
|
| - ReceiveData(request, bytes_read);
|
| - else
|
| - RequestComplete();
|
| - }
|
| -
|
| - const std::string& response_data() const { return response_data_; }
|
| -
|
| - private:
|
| - void ReadSome(net::URLRequest* request) {
|
| - if (!request->is_pending()) {
|
| - RequestComplete();
|
| - return;
|
| - }
|
| -
|
| - int bytes_read = 0;
|
| - if (!request->Read(received_data_.get(), kBufferSize, &bytes_read)) {
|
| - if (!request->status().is_io_pending()) {
|
| - RequestComplete();
|
| - }
|
| - return;
|
| - }
|
| -
|
| - ReceiveData(request, bytes_read);
|
| - }
|
| -
|
| - void ReceiveData(net::URLRequest* request, int bytes_read) {
|
| - if (bytes_read) {
|
| - response_data_.append(received_data_->data(),
|
| - static_cast<size_t>(bytes_read));
|
| - ReadSome(request);
|
| - } else {
|
| - RequestComplete();
|
| - }
|
| - }
|
| -
|
| - void RequestComplete() {
|
| - base::MessageLoop::current()->Quit();
|
| - }
|
| -
|
| - scoped_refptr<net::IOBuffer> received_data_;
|
| - std::string response_data_;
|
| - };
|
| -
|
| - // A simple ProtocolHandler implementation to create BlobURLRequestJob.
|
| - class MockProtocolHandler :
|
| - public net::URLRequestJobFactory::ProtocolHandler {
|
| - public:
|
| - MockProtocolHandler(BlobURLRequestJobTest* test) : test_(test) {}
|
| -
|
| - // net::URLRequestJobFactory::ProtocolHandler override.
|
| - virtual net::URLRequestJob* MaybeCreateJob(
|
| - net::URLRequest* request,
|
| - net::NetworkDelegate* network_delegate) const OVERRIDE {
|
| - return new BlobURLRequestJob(request,
|
| - network_delegate,
|
| - test_->blob_data_,
|
| - test_->file_system_context_,
|
| - base::MessageLoopProxy::current());
|
| - }
|
| -
|
| - private:
|
| - BlobURLRequestJobTest* test_;
|
| - };
|
| -
|
| - BlobURLRequestJobTest()
|
| - : message_loop_(base::MessageLoop::TYPE_IO),
|
| - blob_data_(new BlobData()),
|
| - expected_status_code_(0) {}
|
| -
|
| - virtual void SetUp() {
|
| - ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
|
| -
|
| - temp_file1_ = temp_dir_.path().AppendASCII("BlobFile1.dat");
|
| - ASSERT_EQ(static_cast<int>(arraysize(kTestFileData1) - 1),
|
| - file_util::WriteFile(temp_file1_, kTestFileData1,
|
| - arraysize(kTestFileData1) - 1));
|
| - base::PlatformFileInfo file_info1;
|
| - file_util::GetFileInfo(temp_file1_, &file_info1);
|
| - temp_file_modification_time1_ = file_info1.last_modified;
|
| -
|
| - temp_file2_ = temp_dir_.path().AppendASCII("BlobFile2.dat");
|
| - ASSERT_EQ(static_cast<int>(arraysize(kTestFileData2) - 1),
|
| - file_util::WriteFile(temp_file2_, kTestFileData2,
|
| - arraysize(kTestFileData2) - 1));
|
| - base::PlatformFileInfo file_info2;
|
| - file_util::GetFileInfo(temp_file2_, &file_info2);
|
| - temp_file_modification_time2_ = file_info2.last_modified;
|
| -
|
| - url_request_job_factory_.SetProtocolHandler("blob",
|
| - new MockProtocolHandler(this));
|
| - url_request_context_.set_job_factory(&url_request_job_factory_);
|
| - }
|
| -
|
| - virtual void TearDown() {
|
| - }
|
| -
|
| - void SetUpFileSystem() {
|
| - // Prepare file system.
|
| - file_system_context_ = fileapi::CreateFileSystemContextForTesting(
|
| - NULL, temp_dir_.path());
|
| -
|
| - file_system_context_->OpenFileSystem(
|
| - GURL(kFileSystemURLOrigin),
|
| - kFileSystemType,
|
| - true, // create
|
| - base::Bind(&BlobURLRequestJobTest::OnValidateFileSystem,
|
| - base::Unretained(this)));
|
| - base::MessageLoop::current()->RunUntilIdle();
|
| - ASSERT_TRUE(file_system_root_url_.is_valid());
|
| -
|
| - // Prepare files on file system.
|
| - const char kFilename1[] = "FileSystemFile1.dat";
|
| - temp_file_system_file1_ = GetFileSystemURL(kFilename1);
|
| - WriteFileSystemFile(kFilename1, kTestFileSystemFileData1,
|
| - arraysize(kTestFileSystemFileData1) - 1,
|
| - &temp_file_system_file_modification_time1_);
|
| - const char kFilename2[] = "FileSystemFile2.dat";
|
| - temp_file_system_file2_ = GetFileSystemURL(kFilename2);
|
| - WriteFileSystemFile(kFilename2, kTestFileSystemFileData2,
|
| - arraysize(kTestFileSystemFileData2) - 1,
|
| - &temp_file_system_file_modification_time2_);
|
| - }
|
| -
|
| - GURL GetFileSystemURL(const std::string& filename) {
|
| - return GURL(file_system_root_url_.spec() + filename);
|
| - }
|
| -
|
| - void WriteFileSystemFile(const std::string& filename,
|
| - const char* buf, int buf_size,
|
| - base::Time* modification_time) {
|
| - fileapi::FileSystemURL url =
|
| - file_system_context_->CreateCrackedFileSystemURL(
|
| - GURL(kFileSystemURLOrigin),
|
| - kFileSystemType,
|
| - base::FilePath().AppendASCII(filename));
|
| -
|
| - fileapi::FileSystemFileUtil* file_util =
|
| - file_system_context_->GetFileUtil(kFileSystemType);
|
| -
|
| - fileapi::FileSystemOperationContext context(file_system_context_);
|
| - context.set_allowed_bytes_growth(1024);
|
| -
|
| - base::PlatformFile handle = base::kInvalidPlatformFileValue;
|
| - bool created = false;
|
| - ASSERT_EQ(base::PLATFORM_FILE_OK, file_util->CreateOrOpen(
|
| - &context,
|
| - url,
|
| - base::PLATFORM_FILE_CREATE | base::PLATFORM_FILE_WRITE,
|
| - &handle,
|
| - &created));
|
| - EXPECT_TRUE(created);
|
| - ASSERT_NE(base::kInvalidPlatformFileValue, handle);
|
| - ASSERT_EQ(buf_size,
|
| - base::WritePlatformFile(handle, 0 /* offset */, buf, buf_size));
|
| - base::ClosePlatformFile(handle);
|
| -
|
| - base::PlatformFileInfo file_info;
|
| - base::FilePath platform_path;
|
| - ASSERT_EQ(base::PLATFORM_FILE_OK,
|
| - file_util->GetFileInfo(&context, url, &file_info,
|
| - &platform_path));
|
| - if (modification_time)
|
| - *modification_time = file_info.last_modified;
|
| - }
|
| -
|
| - void OnValidateFileSystem(base::PlatformFileError result,
|
| - const std::string& name,
|
| - const GURL& root) {
|
| - ASSERT_EQ(base::PLATFORM_FILE_OK, result);
|
| - ASSERT_TRUE(root.is_valid());
|
| - file_system_root_url_ = root;
|
| - }
|
| -
|
| - void TestSuccessRequest(const std::string& expected_response) {
|
| - expected_status_code_ = 200;
|
| - expected_response_ = expected_response;
|
| - TestRequest("GET", net::HttpRequestHeaders());
|
| - }
|
| -
|
| - void TestErrorRequest(int expected_status_code) {
|
| - expected_status_code_ = expected_status_code;
|
| - expected_response_ = "";
|
| - TestRequest("GET", net::HttpRequestHeaders());
|
| - }
|
| -
|
| - void TestRequest(const std::string& method,
|
| - const net::HttpRequestHeaders& extra_headers) {
|
| - request_.reset(url_request_context_.CreateRequest(
|
| - GURL("blob:blah"), &url_request_delegate_));
|
| - request_->set_method(method);
|
| - if (!extra_headers.IsEmpty())
|
| - request_->SetExtraRequestHeaders(extra_headers);
|
| - request_->Start();
|
| -
|
| - base::MessageLoop::current()->Run();
|
| -
|
| - // Verify response.
|
| - EXPECT_TRUE(request_->status().is_success());
|
| - EXPECT_EQ(expected_status_code_,
|
| - request_->response_headers()->response_code());
|
| - EXPECT_EQ(expected_response_, url_request_delegate_.response_data());
|
| - }
|
| -
|
| - void BuildComplicatedData(std::string* expected_result) {
|
| - blob_data_->AppendData(kTestData1 + 1, 2);
|
| - blob_data_->AppendFile(temp_file1_, 2, 3, temp_file_modification_time1_);
|
| - blob_data_->AppendFileSystemFile(temp_file_system_file1_, 3, 4,
|
| - temp_file_system_file_modification_time1_);
|
| - blob_data_->AppendData(kTestData2 + 4, 5);
|
| - blob_data_->AppendFile(temp_file2_, 5, 6, temp_file_modification_time2_);
|
| - blob_data_->AppendFileSystemFile(temp_file_system_file2_, 6, 7,
|
| - temp_file_system_file_modification_time2_);
|
| - *expected_result = std::string(kTestData1 + 1, 2);
|
| - *expected_result += std::string(kTestFileData1 + 2, 3);
|
| - *expected_result += std::string(kTestFileSystemFileData1 + 3, 4);
|
| - *expected_result += std::string(kTestData2 + 4, 5);
|
| - *expected_result += std::string(kTestFileData2 + 5, 6);
|
| - *expected_result += std::string(kTestFileSystemFileData2 + 6, 7);
|
| - }
|
| -
|
| - protected:
|
| - base::ScopedTempDir temp_dir_;
|
| - base::FilePath temp_file1_;
|
| - base::FilePath temp_file2_;
|
| - base::Time temp_file_modification_time1_;
|
| - base::Time temp_file_modification_time2_;
|
| - GURL file_system_root_url_;
|
| - GURL temp_file_system_file1_;
|
| - GURL temp_file_system_file2_;
|
| - base::Time temp_file_system_file_modification_time1_;
|
| - base::Time temp_file_system_file_modification_time2_;
|
| -
|
| - base::MessageLoop message_loop_;
|
| - scoped_refptr<fileapi::FileSystemContext> file_system_context_;
|
| - scoped_refptr<BlobData> blob_data_;
|
| - net::URLRequestJobFactoryImpl url_request_job_factory_;
|
| - net::URLRequestContext url_request_context_;
|
| - MockURLRequestDelegate url_request_delegate_;
|
| - scoped_ptr<net::URLRequest> request_;
|
| -
|
| - int expected_status_code_;
|
| - std::string expected_response_;
|
| -};
|
| -
|
| -TEST_F(BlobURLRequestJobTest, TestGetSimpleDataRequest) {
|
| - blob_data_->AppendData(kTestData1);
|
| - TestSuccessRequest(kTestData1);
|
| -}
|
| -
|
| -TEST_F(BlobURLRequestJobTest, TestGetSimpleFileRequest) {
|
| - blob_data_->AppendFile(temp_file1_, 0, -1, base::Time());
|
| - TestSuccessRequest(kTestFileData1);
|
| -}
|
| -
|
| -TEST_F(BlobURLRequestJobTest, TestGetLargeFileRequest) {
|
| - base::FilePath large_temp_file = temp_dir_.path().AppendASCII("LargeBlob.dat");
|
| - std::string large_data;
|
| - large_data.reserve(kBufferSize * 5);
|
| - for (int i = 0; i < kBufferSize * 5; ++i)
|
| - large_data.append(1, static_cast<char>(i % 256));
|
| - ASSERT_EQ(static_cast<int>(large_data.size()),
|
| - file_util::WriteFile(large_temp_file, large_data.data(),
|
| - large_data.size()));
|
| - blob_data_->AppendFile(large_temp_file, 0, -1, base::Time());
|
| - TestSuccessRequest(large_data);
|
| -}
|
| -
|
| -TEST_F(BlobURLRequestJobTest, TestGetNonExistentFileRequest) {
|
| - base::FilePath non_existent_file =
|
| - temp_file1_.InsertBeforeExtension(FILE_PATH_LITERAL("-na"));
|
| - blob_data_->AppendFile(non_existent_file, 0, -1, base::Time());
|
| - TestErrorRequest(404);
|
| -}
|
| -
|
| -TEST_F(BlobURLRequestJobTest, TestGetChangedFileRequest) {
|
| - base::Time old_time =
|
| - temp_file_modification_time1_ - base::TimeDelta::FromSeconds(10);
|
| - blob_data_->AppendFile(temp_file1_, 0, 3, old_time);
|
| - TestErrorRequest(404);
|
| -}
|
| -
|
| -TEST_F(BlobURLRequestJobTest, TestGetSlicedFileRequest) {
|
| - blob_data_->AppendFile(temp_file1_, 2, 4, temp_file_modification_time1_);
|
| - std::string result(kTestFileData1 + 2, 4);
|
| - TestSuccessRequest(result);
|
| -}
|
| -
|
| -TEST_F(BlobURLRequestJobTest, TestGetSimpleFileSystemFileRequest) {
|
| - SetUpFileSystem();
|
| - blob_data_->AppendFileSystemFile(temp_file_system_file1_, 0, -1,
|
| - base::Time());
|
| - TestSuccessRequest(kTestFileSystemFileData1);
|
| -}
|
| -
|
| -TEST_F(BlobURLRequestJobTest, TestGetLargeFileSystemFileRequest) {
|
| - SetUpFileSystem();
|
| - std::string large_data;
|
| - large_data.reserve(kBufferSize * 5);
|
| - for (int i = 0; i < kBufferSize * 5; ++i)
|
| - large_data.append(1, static_cast<char>(i % 256));
|
| -
|
| - const char kFilename[] = "LargeBlob.dat";
|
| - WriteFileSystemFile(kFilename, large_data.data(), large_data.size(), NULL);
|
| -
|
| - blob_data_->AppendFileSystemFile(GetFileSystemURL(kFilename),
|
| - 0, -1, base::Time());
|
| - TestSuccessRequest(large_data);
|
| -}
|
| -
|
| -TEST_F(BlobURLRequestJobTest, TestGetNonExistentFileSystemFileRequest) {
|
| - SetUpFileSystem();
|
| - GURL non_existent_file = GetFileSystemURL("non-existent.dat");
|
| - blob_data_->AppendFileSystemFile(non_existent_file, 0, -1, base::Time());
|
| - TestErrorRequest(404);
|
| -}
|
| -
|
| -TEST_F(BlobURLRequestJobTest, TestGetChangedFileSystemFileRequest) {
|
| - SetUpFileSystem();
|
| - base::Time old_time =
|
| - temp_file_system_file_modification_time1_ -
|
| - base::TimeDelta::FromSeconds(10);
|
| - blob_data_->AppendFileSystemFile(temp_file_system_file1_, 0, 3, old_time);
|
| - TestErrorRequest(404);
|
| -}
|
| -
|
| -TEST_F(BlobURLRequestJobTest, TestGetSlicedFileSystemFileRequest) {
|
| - SetUpFileSystem();
|
| - blob_data_->AppendFileSystemFile(temp_file_system_file1_, 2, 4,
|
| - temp_file_system_file_modification_time1_);
|
| - std::string result(kTestFileSystemFileData1 + 2, 4);
|
| - TestSuccessRequest(result);
|
| -}
|
| -
|
| -TEST_F(BlobURLRequestJobTest, TestGetComplicatedDataAndFileRequest) {
|
| - SetUpFileSystem();
|
| - std::string result;
|
| - BuildComplicatedData(&result);
|
| - TestSuccessRequest(result);
|
| -}
|
| -
|
| -TEST_F(BlobURLRequestJobTest, TestGetRangeRequest1) {
|
| - SetUpFileSystem();
|
| - std::string result;
|
| - BuildComplicatedData(&result);
|
| - net::HttpRequestHeaders extra_headers;
|
| - extra_headers.SetHeader(net::HttpRequestHeaders::kRange, "bytes=5-10");
|
| - expected_status_code_ = 206;
|
| - expected_response_ = result.substr(5, 10 - 5 + 1);
|
| - TestRequest("GET", extra_headers);
|
| -}
|
| -
|
| -TEST_F(BlobURLRequestJobTest, TestGetRangeRequest2) {
|
| - SetUpFileSystem();
|
| - std::string result;
|
| - BuildComplicatedData(&result);
|
| - net::HttpRequestHeaders extra_headers;
|
| - extra_headers.SetHeader(net::HttpRequestHeaders::kRange, "bytes=-10");
|
| - expected_status_code_ = 206;
|
| - expected_response_ = result.substr(result.length() - 10);
|
| - TestRequest("GET", extra_headers);
|
| -}
|
| -
|
| -TEST_F(BlobURLRequestJobTest, TestExtraHeaders) {
|
| - blob_data_->set_content_type(kTestContentType);
|
| - blob_data_->set_content_disposition(kTestContentDisposition);
|
| - blob_data_->AppendData(kTestData1);
|
| - expected_status_code_ = 200;
|
| - expected_response_ = kTestData1;
|
| - TestRequest("GET", net::HttpRequestHeaders());
|
| -
|
| - std::string content_type;
|
| - EXPECT_TRUE(request_->response_headers()->GetMimeType(&content_type));
|
| - EXPECT_EQ(kTestContentType, content_type);
|
| - void* iter = NULL;
|
| - std::string content_disposition;
|
| - EXPECT_TRUE(request_->response_headers()->EnumerateHeader(
|
| - &iter, "Content-Disposition", &content_disposition));
|
| - EXPECT_EQ(kTestContentDisposition, content_disposition);
|
| -}
|
| -
|
| -} // namespace webkit_blob
|
|
|