| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 "chrome/browser/chromeos/drive/file_system/truncate_operation.h" | 5 #include "chrome/browser/chromeos/drive/file_system/truncate_operation.h" |
| 6 | 6 |
| 7 #include "base/file_util.h" | 7 #include "base/file_util.h" |
| 8 #include "base/files/file_path.h" | 8 #include "base/files/file_path.h" |
| 9 #include "base/task_runner_util.h" | 9 #include "base/task_runner_util.h" |
| 10 #include "chrome/browser/chromeos/drive/drive.pb.h" | 10 #include "chrome/browser/chromeos/drive/drive.pb.h" |
| 11 #include "chrome/browser/chromeos/drive/fake_free_disk_space_getter.h" | 11 #include "chrome/browser/chromeos/drive/fake_free_disk_space_getter.h" |
| 12 #include "chrome/browser/chromeos/drive/file_system/operation_test_base.h" | 12 #include "chrome/browser/chromeos/drive/file_system/operation_test_base.h" |
| 13 #include "google_apis/drive/test_util.h" | 13 #include "content/public/test/test_utils.h" |
| 14 #include "testing/gtest/include/gtest/gtest.h" | 14 #include "testing/gtest/include/gtest/gtest.h" |
| 15 | 15 |
| 16 namespace drive { | 16 namespace drive { |
| 17 namespace file_system { | 17 namespace file_system { |
| 18 | 18 |
| 19 class TruncateOperationTest : public OperationTestBase { | 19 class TruncateOperationTest : public OperationTestBase { |
| 20 protected: | 20 protected: |
| 21 virtual void SetUp() OVERRIDE { | 21 virtual void SetUp() OVERRIDE { |
| 22 OperationTestBase::SetUp(); | 22 OperationTestBase::SetUp(); |
| 23 | 23 |
| (...skipping 12 matching lines...) Expand all Loading... |
| 36 const int64 file_size = src_entry.file_info().size(); | 36 const int64 file_size = src_entry.file_info().size(); |
| 37 | 37 |
| 38 // Make sure the file has at least 2 bytes. | 38 // Make sure the file has at least 2 bytes. |
| 39 ASSERT_GE(file_size, 2); | 39 ASSERT_GE(file_size, 2); |
| 40 | 40 |
| 41 FileError error = FILE_ERROR_FAILED; | 41 FileError error = FILE_ERROR_FAILED; |
| 42 operation_->Truncate( | 42 operation_->Truncate( |
| 43 file_in_root, | 43 file_in_root, |
| 44 1, // Truncate to 1 byte. | 44 1, // Truncate to 1 byte. |
| 45 google_apis::test_util::CreateCopyResultCallback(&error)); | 45 google_apis::test_util::CreateCopyResultCallback(&error)); |
| 46 test_util::RunBlockingPoolTask(); | 46 content::RunAllBlockingPoolTasksUntilIdle(); |
| 47 EXPECT_EQ(FILE_ERROR_OK, error); | 47 EXPECT_EQ(FILE_ERROR_OK, error); |
| 48 | 48 |
| 49 base::FilePath local_path; | 49 base::FilePath local_path; |
| 50 error = FILE_ERROR_FAILED; | 50 error = FILE_ERROR_FAILED; |
| 51 base::PostTaskAndReplyWithResult( | 51 base::PostTaskAndReplyWithResult( |
| 52 blocking_task_runner(), | 52 blocking_task_runner(), |
| 53 FROM_HERE, | 53 FROM_HERE, |
| 54 base::Bind(&internal::FileCache::GetFile, | 54 base::Bind(&internal::FileCache::GetFile, |
| 55 base::Unretained(cache()), | 55 base::Unretained(cache()), |
| 56 GetLocalId(file_in_root), &local_path), | 56 GetLocalId(file_in_root), &local_path), |
| 57 google_apis::test_util::CreateCopyResultCallback(&error)); | 57 google_apis::test_util::CreateCopyResultCallback(&error)); |
| 58 test_util::RunBlockingPoolTask(); | 58 content::RunAllBlockingPoolTasksUntilIdle(); |
| 59 ASSERT_EQ(FILE_ERROR_OK, error); | 59 ASSERT_EQ(FILE_ERROR_OK, error); |
| 60 | 60 |
| 61 // The local file should be truncated. | 61 // The local file should be truncated. |
| 62 int64 local_file_size = 0; | 62 int64 local_file_size = 0; |
| 63 base::GetFileSize(local_path, &local_file_size); | 63 base::GetFileSize(local_path, &local_file_size); |
| 64 EXPECT_EQ(1, local_file_size); | 64 EXPECT_EQ(1, local_file_size); |
| 65 } | 65 } |
| 66 | 66 |
| 67 TEST_F(TruncateOperationTest, NegativeSize) { | 67 TEST_F(TruncateOperationTest, NegativeSize) { |
| 68 base::FilePath file_in_root(FILE_PATH_LITERAL("drive/root/File 1.txt")); | 68 base::FilePath file_in_root(FILE_PATH_LITERAL("drive/root/File 1.txt")); |
| 69 ResourceEntry src_entry; | 69 ResourceEntry src_entry; |
| 70 ASSERT_EQ(FILE_ERROR_OK, GetLocalResourceEntry(file_in_root, &src_entry)); | 70 ASSERT_EQ(FILE_ERROR_OK, GetLocalResourceEntry(file_in_root, &src_entry)); |
| 71 const int64 file_size = src_entry.file_info().size(); | 71 const int64 file_size = src_entry.file_info().size(); |
| 72 | 72 |
| 73 // Make sure the file has at least 2 bytes. | 73 // Make sure the file has at least 2 bytes. |
| 74 ASSERT_GE(file_size, 2); | 74 ASSERT_GE(file_size, 2); |
| 75 | 75 |
| 76 FileError error = FILE_ERROR_FAILED; | 76 FileError error = FILE_ERROR_FAILED; |
| 77 operation_->Truncate( | 77 operation_->Truncate( |
| 78 file_in_root, | 78 file_in_root, |
| 79 -1, // Truncate to "-1" byte. | 79 -1, // Truncate to "-1" byte. |
| 80 google_apis::test_util::CreateCopyResultCallback(&error)); | 80 google_apis::test_util::CreateCopyResultCallback(&error)); |
| 81 test_util::RunBlockingPoolTask(); | 81 content::RunAllBlockingPoolTasksUntilIdle(); |
| 82 EXPECT_EQ(FILE_ERROR_INVALID_OPERATION, error); | 82 EXPECT_EQ(FILE_ERROR_INVALID_OPERATION, error); |
| 83 } | 83 } |
| 84 | 84 |
| 85 TEST_F(TruncateOperationTest, HostedDocument) { | 85 TEST_F(TruncateOperationTest, HostedDocument) { |
| 86 base::FilePath file_in_root(FILE_PATH_LITERAL( | 86 base::FilePath file_in_root(FILE_PATH_LITERAL( |
| 87 "drive/root/Document 1 excludeDir-test.gdoc")); | 87 "drive/root/Document 1 excludeDir-test.gdoc")); |
| 88 | 88 |
| 89 FileError error = FILE_ERROR_FAILED; | 89 FileError error = FILE_ERROR_FAILED; |
| 90 operation_->Truncate( | 90 operation_->Truncate( |
| 91 file_in_root, | 91 file_in_root, |
| 92 1, // Truncate to 1 byte. | 92 1, // Truncate to 1 byte. |
| 93 google_apis::test_util::CreateCopyResultCallback(&error)); | 93 google_apis::test_util::CreateCopyResultCallback(&error)); |
| 94 test_util::RunBlockingPoolTask(); | 94 content::RunAllBlockingPoolTasksUntilIdle(); |
| 95 EXPECT_EQ(FILE_ERROR_INVALID_OPERATION, error); | 95 EXPECT_EQ(FILE_ERROR_INVALID_OPERATION, error); |
| 96 } | 96 } |
| 97 | 97 |
| 98 TEST_F(TruncateOperationTest, Extend) { | 98 TEST_F(TruncateOperationTest, Extend) { |
| 99 base::FilePath file_in_root(FILE_PATH_LITERAL("drive/root/File 1.txt")); | 99 base::FilePath file_in_root(FILE_PATH_LITERAL("drive/root/File 1.txt")); |
| 100 ResourceEntry src_entry; | 100 ResourceEntry src_entry; |
| 101 ASSERT_EQ(FILE_ERROR_OK, GetLocalResourceEntry(file_in_root, &src_entry)); | 101 ASSERT_EQ(FILE_ERROR_OK, GetLocalResourceEntry(file_in_root, &src_entry)); |
| 102 const int64 file_size = src_entry.file_info().size(); | 102 const int64 file_size = src_entry.file_info().size(); |
| 103 | 103 |
| 104 FileError error = FILE_ERROR_FAILED; | 104 FileError error = FILE_ERROR_FAILED; |
| 105 operation_->Truncate( | 105 operation_->Truncate( |
| 106 file_in_root, | 106 file_in_root, |
| 107 file_size + 10, // Extend to 10 bytes. | 107 file_size + 10, // Extend to 10 bytes. |
| 108 google_apis::test_util::CreateCopyResultCallback(&error)); | 108 google_apis::test_util::CreateCopyResultCallback(&error)); |
| 109 test_util::RunBlockingPoolTask(); | 109 content::RunAllBlockingPoolTasksUntilIdle(); |
| 110 EXPECT_EQ(FILE_ERROR_OK, error); | 110 EXPECT_EQ(FILE_ERROR_OK, error); |
| 111 | 111 |
| 112 base::FilePath local_path; | 112 base::FilePath local_path; |
| 113 error = FILE_ERROR_FAILED; | 113 error = FILE_ERROR_FAILED; |
| 114 base::PostTaskAndReplyWithResult( | 114 base::PostTaskAndReplyWithResult( |
| 115 blocking_task_runner(), | 115 blocking_task_runner(), |
| 116 FROM_HERE, | 116 FROM_HERE, |
| 117 base::Bind(&internal::FileCache::GetFile, | 117 base::Bind(&internal::FileCache::GetFile, |
| 118 base::Unretained(cache()), | 118 base::Unretained(cache()), |
| 119 GetLocalId(file_in_root), &local_path), | 119 GetLocalId(file_in_root), &local_path), |
| 120 google_apis::test_util::CreateCopyResultCallback(&error)); | 120 google_apis::test_util::CreateCopyResultCallback(&error)); |
| 121 test_util::RunBlockingPoolTask(); | 121 content::RunAllBlockingPoolTasksUntilIdle(); |
| 122 ASSERT_EQ(FILE_ERROR_OK, error); | 122 ASSERT_EQ(FILE_ERROR_OK, error); |
| 123 | 123 |
| 124 // The local file should be truncated. | 124 // The local file should be truncated. |
| 125 std::string content; | 125 std::string content; |
| 126 ASSERT_TRUE(base::ReadFileToString(local_path, &content)); | 126 ASSERT_TRUE(base::ReadFileToString(local_path, &content)); |
| 127 | 127 |
| 128 EXPECT_EQ(file_size + 10, static_cast<int64>(content.size())); | 128 EXPECT_EQ(file_size + 10, static_cast<int64>(content.size())); |
| 129 // All trailing 10 bytes should be '\0'. | 129 // All trailing 10 bytes should be '\0'. |
| 130 EXPECT_EQ(std::string(10, '\0'), content.substr(file_size)); | 130 EXPECT_EQ(std::string(10, '\0'), content.substr(file_size)); |
| 131 } | 131 } |
| 132 | 132 |
| 133 } // namespace file_system | 133 } // namespace file_system |
| 134 } // namespace drive | 134 } // namespace drive |
| OLD | NEW |