Index: chrome/browser/drive/drive_uploader_unittest.cc |
diff --git a/chrome/browser/drive/drive_uploader_unittest.cc b/chrome/browser/drive/drive_uploader_unittest.cc |
index 9efc58c532e87233c2ec3c29e0ceffcc877dee56..1e2a62e4ce6a1f8e5d70fd000531b0c6bed6b426 100644 |
--- a/chrome/browser/drive/drive_uploader_unittest.cc |
+++ b/chrome/browser/drive/drive_uploader_unittest.cc |
@@ -10,8 +10,8 @@ |
#include "base/bind.h" |
#include "base/files/scoped_temp_dir.h" |
#include "base/memory/scoped_ptr.h" |
-#include "base/message_loop/message_loop.h" |
#include "base/run_loop.h" |
+#include "base/thread_task_runner_handle.h" |
#include "base/values.h" |
#include "chrome/browser/drive/dummy_drive_service.h" |
#include "google_apis/drive/drive_api_parser.h" |
@@ -92,7 +92,7 @@ class MockDriveServiceWithUploadExpectation : public DummyDriveService { |
// Calls back the upload URL for subsequent ResumeUpload requests. |
// InitiateUpload is an asynchronous function, so don't callback directly. |
- base::MessageLoop::current()->PostTask(FROM_HERE, |
+ base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, |
base::Bind(callback, HTTP_SUCCESS, GURL(kTestUploadNewFileURL))); |
return CancelCallback(); |
} |
@@ -108,14 +108,14 @@ class MockDriveServiceWithUploadExpectation : public DummyDriveService { |
EXPECT_EQ(kTestInitiateUploadResourceId, resource_id); |
if (!options.etag.empty() && options.etag != kTestETag) { |
- base::MessageLoop::current()->PostTask(FROM_HERE, |
+ base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, |
base::Bind(callback, HTTP_PRECONDITION, GURL())); |
return CancelCallback(); |
} |
// Calls back the upload URL for subsequent ResumeUpload requests. |
// InitiateUpload is an asynchronous function, so don't callback directly. |
- base::MessageLoop::current()->PostTask(FROM_HERE, |
+ base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, |
base::Bind(callback, HTTP_SUCCESS, GURL(kTestUploadExistingFileURL))); |
return CancelCallback(); |
} |
@@ -156,7 +156,7 @@ class MockDriveServiceWithUploadExpectation : public DummyDriveService { |
// For the testing purpose, it always notifies the progress at the end of |
// each chunk uploading. |
int64 chunk_size = end_position - start_position; |
- base::MessageLoop::current()->PostTask(FROM_HERE, |
+ base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, |
base::Bind(progress_callback, chunk_size, chunk_size)); |
} |
@@ -196,7 +196,7 @@ class MockDriveServiceWithUploadExpectation : public DummyDriveService { |
HTTP_RESUME_INCOMPLETE, 0, received_bytes_); |
} |
// ResumeUpload is an asynchronous function, so don't callback directly. |
- base::MessageLoop::current()->PostTask(FROM_HERE, |
+ base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, |
base::Bind(callback, response, base::Passed(&entry))); |
} |
@@ -233,7 +233,7 @@ class MockDriveServiceWithUploadExpectation : public DummyDriveService { |
EXPECT_EQ(expected_upload_file_, local_file_path); |
if (!options.etag.empty() && options.etag != kTestETag) { |
- base::MessageLoop::current()->PostTask( |
+ base::ThreadTaskRunnerHandle::Get()->PostTask( |
FROM_HERE, |
base::Bind(callback, HTTP_PRECONDITION, |
base::Passed(make_scoped_ptr<FileResource>(NULL)))); |
@@ -256,7 +256,7 @@ class MockDriveServiceWithUploadExpectation : public DummyDriveService { |
if (!progress_callback.is_null()) { |
// For the testing purpose, it always notifies the progress at the end of |
// whole file uploading. |
- base::MessageLoop::current()->PostTask( |
+ base::ThreadTaskRunnerHandle::Get()->PostTask( |
FROM_HERE, |
base::Bind(progress_callback, content_length, content_length)); |
} |
@@ -266,7 +266,7 @@ class MockDriveServiceWithUploadExpectation : public DummyDriveService { |
scoped_ptr<FileResource> entry; |
entry.reset(new FileResource); |
entry->set_md5_checksum(kTestDummyMd5); |
- base::MessageLoop::current()->PostTask( |
+ base::ThreadTaskRunnerHandle::Get()->PostTask( |
FROM_HERE, base::Bind(callback, response_code, base::Passed(&entry))); |
return CancelCallback(); |
} |
@@ -288,7 +288,7 @@ class MockDriveServiceNoConnectionAtInitiate : public DummyDriveService { |
const std::string& title, |
const UploadNewFileOptions& options, |
const InitiateUploadCallback& callback) override { |
- base::MessageLoop::current()->PostTask(FROM_HERE, |
+ base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, |
base::Bind(callback, DRIVE_NO_CONNECTION, GURL())); |
return CancelCallback(); |
} |
@@ -299,7 +299,7 @@ class MockDriveServiceNoConnectionAtInitiate : public DummyDriveService { |
const std::string& resource_id, |
const UploadExistingFileOptions& options, |
const InitiateUploadCallback& callback) override { |
- base::MessageLoop::current()->PostTask(FROM_HERE, |
+ base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, |
base::Bind(callback, DRIVE_NO_CONNECTION, GURL())); |
return CancelCallback(); |
} |
@@ -327,7 +327,7 @@ class MockDriveServiceNoConnectionAtInitiate : public DummyDriveService { |
const UploadNewFileOptions& options, |
const google_apis::FileResourceCallback& callback, |
const google_apis::ProgressCallback& progress_callback) override { |
- base::MessageLoop::current()->PostTask( |
+ base::ThreadTaskRunnerHandle::Get()->PostTask( |
FROM_HERE, |
base::Bind(callback, DRIVE_NO_CONNECTION, |
base::Passed(make_scoped_ptr<FileResource>(NULL)))); |
@@ -342,7 +342,7 @@ class MockDriveServiceNoConnectionAtInitiate : public DummyDriveService { |
const UploadExistingFileOptions& options, |
const google_apis::FileResourceCallback& callback, |
const google_apis::ProgressCallback& progress_callback) override { |
- base::MessageLoop::current()->PostTask( |
+ base::ThreadTaskRunnerHandle::Get()->PostTask( |
FROM_HERE, |
base::Bind(callback, DRIVE_NO_CONNECTION, |
base::Passed(make_scoped_ptr<FileResource>(NULL)))); |
@@ -360,7 +360,7 @@ class MockDriveServiceNoConnectionAtResume : public DummyDriveService { |
const std::string& title, |
const UploadNewFileOptions& options, |
const InitiateUploadCallback& callback) override { |
- base::MessageLoop::current()->PostTask(FROM_HERE, |
+ base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, |
base::Bind(callback, HTTP_SUCCESS, GURL(kTestUploadNewFileURL))); |
return CancelCallback(); |
} |
@@ -371,7 +371,7 @@ class MockDriveServiceNoConnectionAtResume : public DummyDriveService { |
const std::string& resource_id, |
const UploadExistingFileOptions& options, |
const InitiateUploadCallback& callback) override { |
- base::MessageLoop::current()->PostTask(FROM_HERE, |
+ base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, |
base::Bind(callback, HTTP_SUCCESS, GURL(kTestUploadExistingFileURL))); |
return CancelCallback(); |
} |
@@ -386,7 +386,7 @@ class MockDriveServiceNoConnectionAtResume : public DummyDriveService { |
const base::FilePath& local_file_path, |
const UploadRangeCallback& callback, |
const ProgressCallback& progress_callback) override { |
- base::MessageLoop::current()->PostTask(FROM_HERE, |
+ base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, |
base::Bind(callback, |
UploadRangeResponse(DRIVE_NO_CONNECTION, -1, -1), |
base::Passed(scoped_ptr<FileResource>()))); |
@@ -400,7 +400,7 @@ class MockDriveServiceNoConnectionAtGetUploadStatus : public DummyDriveService { |
CancelCallback GetUploadStatus(const GURL& upload_url, |
int64 content_length, |
const UploadRangeCallback& callback) override { |
- base::MessageLoop::current()->PostTask(FROM_HERE, |
+ base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, |
base::Bind(callback, |
UploadRangeResponse(DRIVE_NO_CONNECTION, -1, -1), |
base::Passed(scoped_ptr<FileResource>()))); |
@@ -413,7 +413,6 @@ class DriveUploaderTest : public testing::Test { |
void SetUp() override { ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); } |
protected: |
hashimoto
2015/05/07 09:49:41
Sorry for being unclear, message_loop.h include ca
Pranay
2015/05/07 10:01:08
Done.
|
- base::MessageLoop message_loop_; |
base::ScopedTempDir temp_dir_; |
}; |
@@ -431,7 +430,7 @@ TEST_F(DriveUploaderTest, UploadExisting0KB) { |
MockDriveServiceWithUploadExpectation mock_service(local_path, data.size()); |
DriveUploader uploader(&mock_service, |
- base::MessageLoopProxy::current().get()); |
+ base::ThreadTaskRunnerHandle::Get().get()); |
std::vector<test_util::ProgressInfo> upload_progress_values; |
uploader.UploadExistingFile( |
kTestInitiateUploadResourceId, local_path, kTestMimeType, |
@@ -464,7 +463,7 @@ TEST_F(DriveUploaderTest, UploadExisting512KB) { |
MockDriveServiceWithUploadExpectation mock_service(local_path, data.size()); |
DriveUploader uploader(&mock_service, |
- base::MessageLoopProxy::current().get()); |
+ base::ThreadTaskRunnerHandle::Get().get()); |
std::vector<test_util::ProgressInfo> upload_progress_values; |
uploader.UploadExistingFile( |
kTestInitiateUploadResourceId, local_path, kTestMimeType, |
@@ -499,7 +498,7 @@ TEST_F(DriveUploaderTest, UploadExisting2MB) { |
MockDriveServiceWithUploadExpectation mock_service(local_path, data.size()); |
DriveUploader uploader(&mock_service, |
- base::MessageLoopProxy::current().get()); |
+ base::ThreadTaskRunnerHandle::Get().get()); |
std::vector<test_util::ProgressInfo> upload_progress_values; |
uploader.UploadExistingFile( |
kTestInitiateUploadResourceId, local_path, kTestMimeType, |
@@ -534,7 +533,7 @@ TEST_F(DriveUploaderTest, InitiateUploadFail) { |
MockDriveServiceNoConnectionAtInitiate mock_service; |
DriveUploader uploader(&mock_service, |
- base::MessageLoopProxy::current().get()); |
+ base::ThreadTaskRunnerHandle::Get().get()); |
uploader.UploadExistingFile( |
kTestInitiateUploadResourceId, local_path, kTestMimeType, |
UploadExistingFileOptions(), |
@@ -559,7 +558,7 @@ TEST_F(DriveUploaderTest, MultipartUploadFail) { |
MockDriveServiceNoConnectionAtInitiate mock_service; |
DriveUploader uploader(&mock_service, |
- base::MessageLoopProxy::current().get()); |
+ base::ThreadTaskRunnerHandle::Get().get()); |
uploader.UploadExistingFile( |
kTestInitiateUploadResourceId, local_path, kTestMimeType, |
UploadExistingFileOptions(), |
@@ -584,7 +583,7 @@ TEST_F(DriveUploaderTest, InitiateUploadNoConflict) { |
MockDriveServiceWithUploadExpectation mock_service(local_path, data.size()); |
DriveUploader uploader(&mock_service, |
- base::MessageLoopProxy::current().get()); |
+ base::ThreadTaskRunnerHandle::Get().get()); |
UploadExistingFileOptions options; |
options.etag = kTestETag; |
uploader.UploadExistingFile(kTestInitiateUploadResourceId, |
@@ -613,7 +612,7 @@ TEST_F(DriveUploaderTest, MultipartUploadConflict) { |
MockDriveServiceWithUploadExpectation mock_service(local_path, data.size()); |
DriveUploader uploader(&mock_service, |
- base::MessageLoopProxy::current().get()); |
+ base::ThreadTaskRunnerHandle::Get().get()); |
UploadExistingFileOptions options; |
options.etag = kDestinationETag; |
uploader.UploadExistingFile(kTestInitiateUploadResourceId, |
@@ -642,7 +641,7 @@ TEST_F(DriveUploaderTest, InitiateUploadConflict) { |
MockDriveServiceWithUploadExpectation mock_service(local_path, data.size()); |
DriveUploader uploader(&mock_service, |
- base::MessageLoopProxy::current().get()); |
+ base::ThreadTaskRunnerHandle::Get().get()); |
UploadExistingFileOptions options; |
options.etag = kDestinationETag; |
uploader.UploadExistingFile( |
@@ -667,7 +666,7 @@ TEST_F(DriveUploaderTest, ResumeUploadFail) { |
MockDriveServiceNoConnectionAtResume mock_service; |
DriveUploader uploader(&mock_service, |
- base::MessageLoopProxy::current().get()); |
+ base::ThreadTaskRunnerHandle::Get().get()); |
uploader.UploadExistingFile( |
kTestInitiateUploadResourceId, local_path, kTestMimeType, |
UploadExistingFileOptions(), |
@@ -691,7 +690,7 @@ TEST_F(DriveUploaderTest, GetUploadStatusFail) { |
MockDriveServiceNoConnectionAtGetUploadStatus mock_service; |
DriveUploader uploader(&mock_service, |
- base::MessageLoopProxy::current().get()); |
+ base::ThreadTaskRunnerHandle::Get().get()); |
uploader.ResumeUploadFile(GURL(kTestUploadExistingFileURL), |
local_path, |
kTestMimeType, |
@@ -710,7 +709,7 @@ TEST_F(DriveUploaderTest, NonExistingSourceFile) { |
scoped_ptr<FileResource> entry; |
DriveUploader uploader(NULL, // NULL, the service won't be used. |
- base::MessageLoopProxy::current().get()); |
+ base::ThreadTaskRunnerHandle::Get().get()); |
uploader.UploadExistingFile( |
kTestInitiateUploadResourceId, |
temp_dir_.path().AppendASCII("_this_path_should_not_exist_"), |
@@ -736,7 +735,7 @@ TEST_F(DriveUploaderTest, ResumeUpload) { |
MockDriveServiceWithUploadExpectation mock_service(local_path, data.size()); |
DriveUploader uploader(&mock_service, |
- base::MessageLoopProxy::current().get()); |
+ base::ThreadTaskRunnerHandle::Get().get()); |
// Emulate the situation that the only first part is successfully uploaded, |
// but not the latter half. |
mock_service.set_received_bytes(512 * 1024); |