blob: a09456b607fa0efefa68e09604a83e1bfac08c5d [file] [log] [blame]
// Copyright 2013 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/run_loop.h"
#include "base/time/time.h"
#include "content/public/test/test_file_system_context.h"
#include "net/base/io_buffer.h"
#include "net/base/request_priority.h"
#include "net/http/http_byte_range.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/browser/blob/blob_url_request_job.h"
#include "webkit/browser/fileapi/async_file_test_helper.h"
#include "webkit/browser/fileapi/file_system_context.h"
#include "webkit/browser/fileapi/file_system_operation_context.h"
#include "webkit/browser/fileapi/file_system_url.h"
#include "webkit/common/blob/blob_data.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_.get(),
test_->file_system_context_.get(),
base::MessageLoopProxy::current().get());
}
private:
BlobURLRequestJobTest* test_;
};
BlobURLRequestJobTest()
: 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;
base::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;
base::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,
fileapi::OPEN_FILE_SYSTEM_CREATE_IF_NONEXISTENT,
base::Bind(&BlobURLRequestJobTest::OnValidateFileSystem,
base::Unretained(this)));
base::RunLoop().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));
ASSERT_EQ(base::PLATFORM_FILE_OK,
fileapi::AsyncFileTestHelper::CreateFileWithData(
file_system_context_, url, buf, buf_size));
base::PlatformFileInfo file_info;
ASSERT_EQ(base::PLATFORM_FILE_OK,
fileapi::AsyncFileTestHelper::GetMetadata(
file_system_context_, url, &file_info));
if (modification_time)
*modification_time = file_info.last_modified;
}
void OnValidateFileSystem(const GURL& root,
const std::string& name,
base::PlatformFileError result) {
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_ = url_request_context_.CreateRequest(
GURL("blob:blah"), net::DEFAULT_PRIORITY, &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::MessageLoopForIO 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,
net::HttpByteRange::Bounded(5, 10).GetHeaderValue());
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,
net::HttpByteRange::Suffix(10).GetHeaderValue());
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