| OLD | NEW |
|---|
| (Empty) |
| 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 | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "webkit/fileapi/file_system_file_stream_reader.h" | |
| 6 | |
| 7 #include <limits> | |
| 8 #include <string> | |
| 9 | |
| 10 #include "base/files/scoped_temp_dir.h" | |
| 11 #include "base/memory/scoped_ptr.h" | |
| 12 #include "base/message_loop.h" | |
| 13 #include "base/platform_file.h" | |
| 14 #include "net/base/io_buffer.h" | |
| 15 #include "net/base/net_errors.h" | |
| 16 #include "net/base/test_completion_callback.h" | |
| 17 #include "testing/gtest/include/gtest/gtest.h" | |
| 18 #include "webkit/browser/fileapi/external_mount_points.h" | |
| 19 #include "webkit/browser/fileapi/file_system_file_util.h" | |
| 20 #include "webkit/browser/fileapi/mock_file_system_context.h" | |
| 21 #include "webkit/browser/fileapi/sandbox_mount_point_provider.h" | |
| 22 #include "webkit/fileapi/file_system_context.h" | |
| 23 #include "webkit/fileapi/file_system_operation_context.h" | |
| 24 | |
| 25 namespace fileapi { | |
| 26 | |
| 27 namespace { | |
| 28 | |
| 29 const char kURLOrigin[] = "http://remote/"; | |
| 30 const char kTestFileName[] = "test.dat"; | |
| 31 const char kTestData[] = "0123456789"; | |
| 32 const int kTestDataSize = arraysize(kTestData) - 1; | |
| 33 | |
| 34 void ReadFromReader(FileSystemFileStreamReader* reader, | |
| 35 std::string* data, | |
| 36 size_t size, | |
| 37 int* result) { | |
| 38 ASSERT_TRUE(reader != NULL); | |
| 39 ASSERT_TRUE(result != NULL); | |
| 40 *result = net::OK; | |
| 41 net::TestCompletionCallback callback; | |
| 42 size_t total_bytes_read = 0; | |
| 43 while (total_bytes_read < size) { | |
| 44 scoped_refptr<net::IOBufferWithSize> buf( | |
| 45 new net::IOBufferWithSize(size - total_bytes_read)); | |
| 46 int rv = reader->Read(buf, buf->size(), callback.callback()); | |
| 47 if (rv == net::ERR_IO_PENDING) | |
| 48 rv = callback.WaitForResult(); | |
| 49 if (rv < 0) | |
| 50 *result = rv; | |
| 51 if (rv <= 0) | |
| 52 break; | |
| 53 total_bytes_read += rv; | |
| 54 data->append(buf->data(), rv); | |
| 55 } | |
| 56 } | |
| 57 | |
| 58 void NeverCalled(int unused) { ADD_FAILURE(); } | |
| 59 | |
| 60 } // namespace | |
| 61 | |
| 62 class FileSystemFileStreamReaderTest : public testing::Test { | |
| 63 public: | |
| 64 FileSystemFileStreamReaderTest() | |
| 65 : message_loop_(base::MessageLoop::TYPE_IO) {} | |
| 66 | |
| 67 virtual void SetUp() OVERRIDE { | |
| 68 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); | |
| 69 | |
| 70 file_system_context_ = CreateFileSystemContextForTesting( | |
| 71 NULL, temp_dir_.path()); | |
| 72 | |
| 73 file_system_context_->sandbox_provider()->ValidateFileSystemRoot( | |
| 74 GURL(kURLOrigin), kFileSystemTypeTemporary, true, // create | |
| 75 base::Bind(&OnValidateFileSystem)); | |
| 76 base::MessageLoop::current()->RunUntilIdle(); | |
| 77 | |
| 78 WriteFile(kTestFileName, kTestData, kTestDataSize, | |
| 79 &test_file_modification_time_); | |
| 80 } | |
| 81 | |
| 82 virtual void TearDown() OVERRIDE { | |
| 83 base::MessageLoop::current()->RunUntilIdle(); | |
| 84 } | |
| 85 | |
| 86 protected: | |
| 87 FileSystemFileStreamReader* CreateFileReader( | |
| 88 const std::string& file_name, | |
| 89 int64 initial_offset, | |
| 90 const base::Time& expected_modification_time) { | |
| 91 return new FileSystemFileStreamReader(file_system_context_, | |
| 92 GetFileSystemURL(file_name), | |
| 93 initial_offset, | |
| 94 expected_modification_time); | |
| 95 } | |
| 96 | |
| 97 base::Time test_file_modification_time() const { | |
| 98 return test_file_modification_time_; | |
| 99 } | |
| 100 | |
| 101 void WriteFile(const std::string& file_name, | |
| 102 const char* buf, | |
| 103 int buf_size, | |
| 104 base::Time* modification_time) { | |
| 105 FileSystemFileUtil* file_util = file_system_context_-> | |
| 106 sandbox_provider()->GetFileUtil(kFileSystemTypeTemporary); | |
| 107 FileSystemURL url = GetFileSystemURL(file_name); | |
| 108 | |
| 109 FileSystemOperationContext context(file_system_context_); | |
| 110 context.set_allowed_bytes_growth(1024); | |
| 111 | |
| 112 base::PlatformFile handle = base::kInvalidPlatformFileValue; | |
| 113 bool created = false; | |
| 114 ASSERT_EQ(base::PLATFORM_FILE_OK, file_util->CreateOrOpen( | |
| 115 &context, | |
| 116 url, | |
| 117 base::PLATFORM_FILE_CREATE | base::PLATFORM_FILE_WRITE, | |
| 118 &handle, | |
| 119 &created)); | |
| 120 EXPECT_TRUE(created); | |
| 121 ASSERT_NE(base::kInvalidPlatformFileValue, handle); | |
| 122 ASSERT_EQ(buf_size, | |
| 123 base::WritePlatformFile(handle, 0 /* offset */, buf, buf_size)); | |
| 124 base::ClosePlatformFile(handle); | |
| 125 | |
| 126 base::PlatformFileInfo file_info; | |
| 127 base::FilePath platform_path; | |
| 128 ASSERT_EQ(base::PLATFORM_FILE_OK, | |
| 129 file_util->GetFileInfo(&context, url, &file_info, | |
| 130 &platform_path)); | |
| 131 if (modification_time) | |
| 132 *modification_time = file_info.last_modified; | |
| 133 } | |
| 134 | |
| 135 private: | |
| 136 static void OnValidateFileSystem(base::PlatformFileError result) { | |
| 137 ASSERT_EQ(base::PLATFORM_FILE_OK, result); | |
| 138 } | |
| 139 | |
| 140 FileSystemURL GetFileSystemURL(const std::string& file_name) { | |
| 141 return file_system_context_->CreateCrackedFileSystemURL( | |
| 142 GURL(kURLOrigin), | |
| 143 kFileSystemTypeTemporary, | |
| 144 base::FilePath().AppendASCII(file_name)); | |
| 145 } | |
| 146 | |
| 147 base::MessageLoop message_loop_; | |
| 148 base::ScopedTempDir temp_dir_; | |
| 149 scoped_refptr<FileSystemContext> file_system_context_; | |
| 150 base::Time test_file_modification_time_; | |
| 151 }; | |
| 152 | |
| 153 TEST_F(FileSystemFileStreamReaderTest, NonExistent) { | |
| 154 const char kFileName[] = "nonexistent"; | |
| 155 scoped_ptr<FileSystemFileStreamReader> reader( | |
| 156 CreateFileReader(kFileName, 0, base::Time())); | |
| 157 int result = 0; | |
| 158 std::string data; | |
| 159 ReadFromReader(reader.get(), &data, 10, &result); | |
| 160 ASSERT_EQ(net::ERR_FILE_NOT_FOUND, result); | |
| 161 ASSERT_EQ(0U, data.size()); | |
| 162 } | |
| 163 | |
| 164 TEST_F(FileSystemFileStreamReaderTest, Empty) { | |
| 165 const char kFileName[] = "empty"; | |
| 166 WriteFile(kFileName, NULL, 0, NULL); | |
| 167 | |
| 168 scoped_ptr<FileSystemFileStreamReader> reader( | |
| 169 CreateFileReader(kFileName, 0, base::Time())); | |
| 170 int result = 0; | |
| 171 std::string data; | |
| 172 ReadFromReader(reader.get(), &data, 10, &result); | |
| 173 ASSERT_EQ(net::OK, result); | |
| 174 ASSERT_EQ(0U, data.size()); | |
| 175 | |
| 176 net::TestInt64CompletionCallback callback; | |
| 177 int64 length_result = reader->GetLength(callback.callback()); | |
| 178 if (length_result == net::ERR_IO_PENDING) | |
| 179 length_result = callback.WaitForResult(); | |
| 180 ASSERT_EQ(0, length_result); | |
| 181 } | |
| 182 | |
| 183 TEST_F(FileSystemFileStreamReaderTest, GetLengthNormal) { | |
| 184 scoped_ptr<FileSystemFileStreamReader> reader( | |
| 185 CreateFileReader(kTestFileName, 0, test_file_modification_time())); | |
| 186 net::TestInt64CompletionCallback callback; | |
| 187 int64 result = reader->GetLength(callback.callback()); | |
| 188 if (result == net::ERR_IO_PENDING) | |
| 189 result = callback.WaitForResult(); | |
| 190 ASSERT_EQ(kTestDataSize, result); | |
| 191 } | |
| 192 | |
| 193 TEST_F(FileSystemFileStreamReaderTest, GetLengthAfterModified) { | |
| 194 // Pass a fake expected modifictaion time so that the expectation fails. | |
| 195 base::Time fake_expected_modification_time = | |
| 196 test_file_modification_time() - base::TimeDelta::FromSeconds(10); | |
| 197 | |
| 198 scoped_ptr<FileSystemFileStreamReader> reader( | |
| 199 CreateFileReader(kTestFileName, 0, fake_expected_modification_time)); | |
| 200 net::TestInt64CompletionCallback callback; | |
| 201 int64 result = reader->GetLength(callback.callback()); | |
| 202 if (result == net::ERR_IO_PENDING) | |
| 203 result = callback.WaitForResult(); | |
| 204 ASSERT_EQ(net::ERR_UPLOAD_FILE_CHANGED, result); | |
| 205 | |
| 206 // With NULL expected modification time this should work. | |
| 207 reader.reset(CreateFileReader(kTestFileName, 0, base::Time())); | |
| 208 result = reader->GetLength(callback.callback()); | |
| 209 if (result == net::ERR_IO_PENDING) | |
| 210 result = callback.WaitForResult(); | |
| 211 ASSERT_EQ(kTestDataSize, result); | |
| 212 } | |
| 213 | |
| 214 TEST_F(FileSystemFileStreamReaderTest, GetLengthWithOffset) { | |
| 215 scoped_ptr<FileSystemFileStreamReader> reader( | |
| 216 CreateFileReader(kTestFileName, 3, base::Time())); | |
| 217 net::TestInt64CompletionCallback callback; | |
| 218 int64 result = reader->GetLength(callback.callback()); | |
| 219 if (result == net::ERR_IO_PENDING) | |
| 220 result = callback.WaitForResult(); | |
| 221 // Initial offset does not affect the result of GetLength. | |
| 222 ASSERT_EQ(kTestDataSize, result); | |
| 223 } | |
| 224 | |
| 225 TEST_F(FileSystemFileStreamReaderTest, ReadNormal) { | |
| 226 scoped_ptr<FileSystemFileStreamReader> reader( | |
| 227 CreateFileReader(kTestFileName, 0, test_file_modification_time())); | |
| 228 int result = 0; | |
| 229 std::string data; | |
| 230 ReadFromReader(reader.get(), &data, kTestDataSize, &result); | |
| 231 ASSERT_EQ(net::OK, result); | |
| 232 ASSERT_EQ(kTestData, data); | |
| 233 } | |
| 234 | |
| 235 TEST_F(FileSystemFileStreamReaderTest, ReadAfterModified) { | |
| 236 // Pass a fake expected modifictaion time so that the expectation fails. | |
| 237 base::Time fake_expected_modification_time = | |
| 238 test_file_modification_time() - base::TimeDelta::FromSeconds(10); | |
| 239 | |
| 240 scoped_ptr<FileSystemFileStreamReader> reader( | |
| 241 CreateFileReader(kTestFileName, 0, fake_expected_modification_time)); | |
| 242 int result = 0; | |
| 243 std::string data; | |
| 244 ReadFromReader(reader.get(), &data, kTestDataSize, &result); | |
| 245 ASSERT_EQ(net::ERR_UPLOAD_FILE_CHANGED, result); | |
| 246 ASSERT_EQ(0U, data.size()); | |
| 247 | |
| 248 // With NULL expected modification time this should work. | |
| 249 data.clear(); | |
| 250 reader.reset(CreateFileReader(kTestFileName, 0, base::Time())); | |
| 251 ReadFromReader(reader.get(), &data, kTestDataSize, &result); | |
| 252 ASSERT_EQ(net::OK, result); | |
| 253 ASSERT_EQ(kTestData, data); | |
| 254 } | |
| 255 | |
| 256 TEST_F(FileSystemFileStreamReaderTest, ReadWithOffset) { | |
| 257 scoped_ptr<FileSystemFileStreamReader> reader( | |
| 258 CreateFileReader(kTestFileName, 3, base::Time())); | |
| 259 int result = 0; | |
| 260 std::string data; | |
| 261 ReadFromReader(reader.get(), &data, kTestDataSize, &result); | |
| 262 ASSERT_EQ(net::OK, result); | |
| 263 ASSERT_EQ(&kTestData[3], data); | |
| 264 } | |
| 265 | |
| 266 TEST_F(FileSystemFileStreamReaderTest, DeleteWithUnfinishedRead) { | |
| 267 scoped_ptr<FileSystemFileStreamReader> reader( | |
| 268 CreateFileReader(kTestFileName, 0, base::Time())); | |
| 269 | |
| 270 net::TestCompletionCallback callback; | |
| 271 scoped_refptr<net::IOBufferWithSize> buf( | |
| 272 new net::IOBufferWithSize(kTestDataSize)); | |
| 273 int rv = reader->Read(buf, buf->size(), base::Bind(&NeverCalled)); | |
| 274 ASSERT_TRUE(rv == net::ERR_IO_PENDING || rv >= 0); | |
| 275 | |
| 276 // Delete immediately. | |
| 277 // Should not crash; nor should NeverCalled be callback. | |
| 278 reader.reset(); | |
| 279 } | |
| 280 | |
| 281 } // namespace fileapi | |
| OLD | NEW |
|---|
Chromium Code Reviews
Issue 15859007:
Move browser-specific FileAPI code from webkit/fileapi to webkit/browser/fileapi (Closed)
Base URL: svn://svn.chromium.org/chrome/trunk/src