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 "base/file_util_proxy.h" | |
6 | |
7 #include <map> | |
8 | |
9 #include "base/bind.h" | |
10 #include "base/files/scoped_temp_dir.h" | |
11 #include "base/logging.h" | |
12 #include "base/memory/weak_ptr.h" | |
13 #include "base/message_loop.h" | |
14 #include "base/platform_file.h" | |
15 #include "base/threading/thread.h" | |
16 #include "testing/gtest/include/gtest/gtest.h" | |
17 | |
18 namespace base { | |
19 | |
20 class FileUtilProxyTest : public testing::Test { | |
21 public: | |
22 FileUtilProxyTest() | |
23 : message_loop_(MessageLoop::TYPE_IO), | |
24 file_thread_("FileUtilProxyTestFileThread"), | |
25 error_(PLATFORM_FILE_OK), | |
26 created_(false), | |
27 file_(kInvalidPlatformFileValue), | |
28 bytes_written_(-1), | |
29 weak_factory_(ALLOW_THIS_IN_INITIALIZER_LIST(this)) {} | |
30 | |
31 virtual void SetUp() OVERRIDE { | |
32 ASSERT_TRUE(dir_.CreateUniqueTempDir()); | |
33 ASSERT_TRUE(file_thread_.Start()); | |
34 } | |
35 | |
36 virtual void TearDown() OVERRIDE { | |
37 if (file_ != kInvalidPlatformFileValue) | |
38 ClosePlatformFile(file_); | |
39 } | |
40 | |
41 void DidFinish(PlatformFileError error) { | |
42 error_ = error; | |
43 MessageLoop::current()->QuitWhenIdle(); | |
44 } | |
45 | |
46 void DidCreateOrOpen(PlatformFileError error, | |
47 PassPlatformFile file, | |
48 bool created) { | |
49 error_ = error; | |
50 file_ = file.ReleaseValue(); | |
51 created_ = created; | |
52 MessageLoop::current()->QuitWhenIdle(); | |
53 } | |
54 | |
55 void DidCreateTemporary(PlatformFileError error, | |
56 PassPlatformFile file, | |
57 const FilePath& path) { | |
58 error_ = error; | |
59 file_ = file.ReleaseValue(); | |
60 path_ = path; | |
61 MessageLoop::current()->QuitWhenIdle(); | |
62 } | |
63 | |
64 void DidGetFileInfo(PlatformFileError error, | |
65 const PlatformFileInfo& file_info) { | |
66 error_ = error; | |
67 file_info_ = file_info; | |
68 MessageLoop::current()->QuitWhenIdle(); | |
69 } | |
70 | |
71 void DidRead(PlatformFileError error, | |
72 const char* data, | |
73 int bytes_read) { | |
74 error_ = error; | |
75 buffer_.resize(bytes_read); | |
76 memcpy(&buffer_[0], data, bytes_read); | |
77 MessageLoop::current()->QuitWhenIdle(); | |
78 } | |
79 | |
80 void DidWrite(PlatformFileError error, | |
81 int bytes_written) { | |
82 error_ = error; | |
83 bytes_written_ = bytes_written; | |
84 MessageLoop::current()->QuitWhenIdle(); | |
85 } | |
86 | |
87 protected: | |
88 PlatformFile GetTestPlatformFile(int flags) { | |
89 if (file_ != kInvalidPlatformFileValue) | |
90 return file_; | |
91 bool created; | |
92 PlatformFileError error; | |
93 file_ = CreatePlatformFile(test_path(), flags, &created, &error); | |
94 EXPECT_EQ(PLATFORM_FILE_OK, error); | |
95 EXPECT_NE(kInvalidPlatformFileValue, file_); | |
96 return file_; | |
97 } | |
98 | |
99 TaskRunner* file_task_runner() const { | |
100 return file_thread_.message_loop_proxy().get(); | |
101 } | |
102 const FilePath& test_dir_path() const { return dir_.path(); } | |
103 const FilePath test_path() const { return dir_.path().AppendASCII("test"); } | |
104 | |
105 MessageLoop message_loop_; | |
106 Thread file_thread_; | |
107 | |
108 ScopedTempDir dir_; | |
109 PlatformFileError error_; | |
110 bool created_; | |
111 PlatformFile file_; | |
112 FilePath path_; | |
113 PlatformFileInfo file_info_; | |
114 std::vector<char> buffer_; | |
115 int bytes_written_; | |
116 WeakPtrFactory<FileUtilProxyTest> weak_factory_; | |
117 }; | |
118 | |
119 TEST_F(FileUtilProxyTest, CreateOrOpen_Create) { | |
120 FileUtilProxy::CreateOrOpen( | |
121 file_task_runner(), | |
122 test_path(), | |
123 PLATFORM_FILE_CREATE | PLATFORM_FILE_READ, | |
124 Bind(&FileUtilProxyTest::DidCreateOrOpen, weak_factory_.GetWeakPtr())); | |
125 MessageLoop::current()->Run(); | |
126 | |
127 EXPECT_EQ(PLATFORM_FILE_OK, error_); | |
128 EXPECT_TRUE(created_); | |
129 EXPECT_NE(kInvalidPlatformFileValue, file_); | |
130 EXPECT_TRUE(file_util::PathExists(test_path())); | |
131 } | |
132 | |
133 TEST_F(FileUtilProxyTest, CreateOrOpen_Open) { | |
134 // Creates a file. | |
135 file_util::WriteFile(test_path(), NULL, 0); | |
136 ASSERT_TRUE(file_util::PathExists(test_path())); | |
137 | |
138 // Opens the created file. | |
139 FileUtilProxy::CreateOrOpen( | |
140 file_task_runner(), | |
141 test_path(), | |
142 PLATFORM_FILE_OPEN | PLATFORM_FILE_READ, | |
143 Bind(&FileUtilProxyTest::DidCreateOrOpen, weak_factory_.GetWeakPtr())); | |
144 MessageLoop::current()->Run(); | |
145 | |
146 EXPECT_EQ(PLATFORM_FILE_OK, error_); | |
147 EXPECT_FALSE(created_); | |
148 EXPECT_NE(kInvalidPlatformFileValue, file_); | |
149 } | |
150 | |
151 TEST_F(FileUtilProxyTest, CreateOrOpen_OpenNonExistent) { | |
152 FileUtilProxy::CreateOrOpen( | |
153 file_task_runner(), | |
154 test_path(), | |
155 PLATFORM_FILE_OPEN | PLATFORM_FILE_READ, | |
156 Bind(&FileUtilProxyTest::DidCreateOrOpen, weak_factory_.GetWeakPtr())); | |
157 MessageLoop::current()->Run(); | |
158 EXPECT_EQ(PLATFORM_FILE_ERROR_NOT_FOUND, error_); | |
159 EXPECT_FALSE(created_); | |
160 EXPECT_EQ(kInvalidPlatformFileValue, file_); | |
161 EXPECT_FALSE(file_util::PathExists(test_path())); | |
162 } | |
163 | |
164 TEST_F(FileUtilProxyTest, Close) { | |
165 // Creates a file. | |
166 PlatformFile file = GetTestPlatformFile( | |
167 PLATFORM_FILE_CREATE | PLATFORM_FILE_WRITE); | |
168 | |
169 #if defined(OS_WIN) | |
170 // This fails on Windows if the file is not closed. | |
171 EXPECT_FALSE(file_util::Move(test_path(), | |
172 test_dir_path().AppendASCII("new"))); | |
173 #endif | |
174 | |
175 FileUtilProxy::Close( | |
176 file_task_runner(), | |
177 file, | |
178 Bind(&FileUtilProxyTest::DidFinish, weak_factory_.GetWeakPtr())); | |
179 MessageLoop::current()->Run(); | |
180 EXPECT_EQ(PLATFORM_FILE_OK, error_); | |
181 | |
182 // Now it should pass on all platforms. | |
183 EXPECT_TRUE(file_util::Move(test_path(), test_dir_path().AppendASCII("new"))); | |
184 } | |
185 | |
186 TEST_F(FileUtilProxyTest, CreateTemporary) { | |
187 FileUtilProxy::CreateTemporary( | |
188 file_task_runner(), 0 /* additional_file_flags */, | |
189 Bind(&FileUtilProxyTest::DidCreateTemporary, weak_factory_.GetWeakPtr())); | |
190 MessageLoop::current()->Run(); | |
191 EXPECT_EQ(PLATFORM_FILE_OK, error_); | |
192 EXPECT_TRUE(file_util::PathExists(path_)); | |
193 EXPECT_NE(kInvalidPlatformFileValue, file_); | |
194 | |
195 // The file should be writable. | |
196 #if defined(OS_WIN) | |
197 HANDLE hEvent = CreateEvent(NULL, FALSE, FALSE, NULL); | |
198 OVERLAPPED overlapped = {0}; | |
199 overlapped.hEvent = hEvent; | |
200 DWORD bytes_written; | |
201 if (!::WriteFile(file_, "test", 4, &bytes_written, &overlapped)) { | |
202 // Temporary file is created with ASYNC flag, so WriteFile may return 0 | |
203 // with ERROR_IO_PENDING. | |
204 EXPECT_EQ(ERROR_IO_PENDING, GetLastError()); | |
205 GetOverlappedResult(file_, &overlapped, &bytes_written, TRUE); | |
206 } | |
207 EXPECT_EQ(4, bytes_written); | |
208 #else | |
209 // On POSIX ASYNC flag does not affect synchronous read/write behavior. | |
210 EXPECT_EQ(4, WritePlatformFile(file_, 0, "test", 4)); | |
211 #endif | |
212 EXPECT_TRUE(ClosePlatformFile(file_)); | |
213 file_ = kInvalidPlatformFileValue; | |
214 | |
215 // Make sure the written data can be read from the returned path. | |
216 std::string data; | |
217 EXPECT_TRUE(file_util::ReadFileToString(path_, &data)); | |
218 EXPECT_EQ("test", data); | |
219 | |
220 // Make sure we can & do delete the created file to prevent leaks on the bots. | |
221 EXPECT_TRUE(file_util::Delete(path_, false)); | |
222 } | |
223 | |
224 TEST_F(FileUtilProxyTest, GetFileInfo_File) { | |
225 // Setup. | |
226 ASSERT_EQ(4, file_util::WriteFile(test_path(), "test", 4)); | |
227 PlatformFileInfo expected_info; | |
228 file_util::GetFileInfo(test_path(), &expected_info); | |
229 | |
230 // Run. | |
231 FileUtilProxy::GetFileInfo( | |
232 file_task_runner(), | |
233 test_path(), | |
234 Bind(&FileUtilProxyTest::DidGetFileInfo, weak_factory_.GetWeakPtr())); | |
235 MessageLoop::current()->Run(); | |
236 | |
237 // Verify. | |
238 EXPECT_EQ(PLATFORM_FILE_OK, error_); | |
239 EXPECT_EQ(expected_info.size, file_info_.size); | |
240 EXPECT_EQ(expected_info.is_directory, file_info_.is_directory); | |
241 EXPECT_EQ(expected_info.is_symbolic_link, file_info_.is_symbolic_link); | |
242 EXPECT_EQ(expected_info.last_modified, file_info_.last_modified); | |
243 EXPECT_EQ(expected_info.last_accessed, file_info_.last_accessed); | |
244 EXPECT_EQ(expected_info.creation_time, file_info_.creation_time); | |
245 } | |
246 | |
247 TEST_F(FileUtilProxyTest, GetFileInfo_Directory) { | |
248 // Setup. | |
249 ASSERT_TRUE(file_util::CreateDirectory(test_path())); | |
250 PlatformFileInfo expected_info; | |
251 file_util::GetFileInfo(test_path(), &expected_info); | |
252 | |
253 // Run. | |
254 FileUtilProxy::GetFileInfo( | |
255 file_task_runner(), | |
256 test_path(), | |
257 Bind(&FileUtilProxyTest::DidGetFileInfo, weak_factory_.GetWeakPtr())); | |
258 MessageLoop::current()->Run(); | |
259 | |
260 // Verify. | |
261 EXPECT_EQ(PLATFORM_FILE_OK, error_); | |
262 EXPECT_EQ(expected_info.size, file_info_.size); | |
263 EXPECT_EQ(expected_info.is_directory, file_info_.is_directory); | |
264 EXPECT_EQ(expected_info.is_symbolic_link, file_info_.is_symbolic_link); | |
265 EXPECT_EQ(expected_info.last_modified, file_info_.last_modified); | |
266 EXPECT_EQ(expected_info.last_accessed, file_info_.last_accessed); | |
267 EXPECT_EQ(expected_info.creation_time, file_info_.creation_time); | |
268 } | |
269 | |
270 TEST_F(FileUtilProxyTest, Read) { | |
271 // Setup. | |
272 const char expected_data[] = "bleh"; | |
273 int expected_bytes = arraysize(expected_data); | |
274 ASSERT_EQ(expected_bytes, | |
275 file_util::WriteFile(test_path(), expected_data, expected_bytes)); | |
276 | |
277 // Run. | |
278 FileUtilProxy::Read( | |
279 file_task_runner(), | |
280 GetTestPlatformFile(PLATFORM_FILE_OPEN | PLATFORM_FILE_READ), | |
281 0, // offset | |
282 128, | |
283 Bind(&FileUtilProxyTest::DidRead, weak_factory_.GetWeakPtr())); | |
284 MessageLoop::current()->Run(); | |
285 | |
286 // Verify. | |
287 EXPECT_EQ(PLATFORM_FILE_OK, error_); | |
288 EXPECT_EQ(expected_bytes, static_cast<int>(buffer_.size())); | |
289 for (size_t i = 0; i < buffer_.size(); ++i) { | |
290 EXPECT_EQ(expected_data[i], buffer_[i]); | |
291 } | |
292 } | |
293 | |
294 TEST_F(FileUtilProxyTest, WriteAndFlush) { | |
295 const char data[] = "foo!"; | |
296 int data_bytes = ARRAYSIZE_UNSAFE(data); | |
297 PlatformFile file = GetTestPlatformFile( | |
298 PLATFORM_FILE_CREATE | PLATFORM_FILE_WRITE); | |
299 | |
300 FileUtilProxy::Write( | |
301 file_task_runner(), | |
302 file, | |
303 0, // offset | |
304 data, | |
305 data_bytes, | |
306 Bind(&FileUtilProxyTest::DidWrite, weak_factory_.GetWeakPtr())); | |
307 MessageLoop::current()->Run(); | |
308 EXPECT_EQ(PLATFORM_FILE_OK, error_); | |
309 EXPECT_EQ(data_bytes, bytes_written_); | |
310 | |
311 // Flush the written data. (So that the following read should always | |
312 // succeed. On some platforms it may work with or without this flush.) | |
313 FileUtilProxy::Flush( | |
314 file_task_runner(), | |
315 file, | |
316 Bind(&FileUtilProxyTest::DidFinish, weak_factory_.GetWeakPtr())); | |
317 MessageLoop::current()->Run(); | |
318 EXPECT_EQ(PLATFORM_FILE_OK, error_); | |
319 | |
320 // Verify the written data. | |
321 char buffer[10]; | |
322 EXPECT_EQ(data_bytes, file_util::ReadFile(test_path(), buffer, data_bytes)); | |
323 for (int i = 0; i < data_bytes; ++i) { | |
324 EXPECT_EQ(data[i], buffer[i]); | |
325 } | |
326 } | |
327 | |
328 TEST_F(FileUtilProxyTest, Touch) { | |
329 Time last_accessed_time = Time::Now() - TimeDelta::FromDays(12345); | |
330 Time last_modified_time = Time::Now() - TimeDelta::FromHours(98765); | |
331 | |
332 FileUtilProxy::Touch( | |
333 file_task_runner(), | |
334 GetTestPlatformFile(PLATFORM_FILE_CREATE | | |
335 PLATFORM_FILE_WRITE | | |
336 PLATFORM_FILE_WRITE_ATTRIBUTES), | |
337 last_accessed_time, | |
338 last_modified_time, | |
339 Bind(&FileUtilProxyTest::DidFinish, weak_factory_.GetWeakPtr())); | |
340 MessageLoop::current()->Run(); | |
341 EXPECT_EQ(PLATFORM_FILE_OK, error_); | |
342 | |
343 PlatformFileInfo info; | |
344 file_util::GetFileInfo(test_path(), &info); | |
345 | |
346 // The returned values may only have the seconds precision, so we cast | |
347 // the double values to int here. | |
348 EXPECT_EQ(static_cast<int>(last_modified_time.ToDoubleT()), | |
349 static_cast<int>(info.last_modified.ToDoubleT())); | |
350 EXPECT_EQ(static_cast<int>(last_accessed_time.ToDoubleT()), | |
351 static_cast<int>(info.last_accessed.ToDoubleT())); | |
352 } | |
353 | |
354 TEST_F(FileUtilProxyTest, Truncate_Shrink) { | |
355 // Setup. | |
356 const char kTestData[] = "0123456789"; | |
357 ASSERT_EQ(10, file_util::WriteFile(test_path(), kTestData, 10)); | |
358 PlatformFileInfo info; | |
359 file_util::GetFileInfo(test_path(), &info); | |
360 ASSERT_EQ(10, info.size); | |
361 | |
362 // Run. | |
363 FileUtilProxy::Truncate( | |
364 file_task_runner(), | |
365 GetTestPlatformFile(PLATFORM_FILE_OPEN | PLATFORM_FILE_WRITE), | |
366 7, | |
367 Bind(&FileUtilProxyTest::DidFinish, weak_factory_.GetWeakPtr())); | |
368 MessageLoop::current()->Run(); | |
369 | |
370 // Verify. | |
371 file_util::GetFileInfo(test_path(), &info); | |
372 ASSERT_EQ(7, info.size); | |
373 | |
374 char buffer[7]; | |
375 EXPECT_EQ(7, file_util::ReadFile(test_path(), buffer, 7)); | |
376 int i = 0; | |
377 for (; i < 7; ++i) | |
378 EXPECT_EQ(kTestData[i], buffer[i]); | |
379 } | |
380 | |
381 TEST_F(FileUtilProxyTest, Truncate_Expand) { | |
382 // Setup. | |
383 const char kTestData[] = "9876543210"; | |
384 ASSERT_EQ(10, file_util::WriteFile(test_path(), kTestData, 10)); | |
385 PlatformFileInfo info; | |
386 file_util::GetFileInfo(test_path(), &info); | |
387 ASSERT_EQ(10, info.size); | |
388 | |
389 // Run. | |
390 FileUtilProxy::Truncate( | |
391 file_task_runner(), | |
392 GetTestPlatformFile(PLATFORM_FILE_OPEN | PLATFORM_FILE_WRITE), | |
393 53, | |
394 Bind(&FileUtilProxyTest::DidFinish, weak_factory_.GetWeakPtr())); | |
395 MessageLoop::current()->Run(); | |
396 | |
397 // Verify. | |
398 file_util::GetFileInfo(test_path(), &info); | |
399 ASSERT_EQ(53, info.size); | |
400 | |
401 char buffer[53]; | |
402 EXPECT_EQ(53, file_util::ReadFile(test_path(), buffer, 53)); | |
403 int i = 0; | |
404 for (; i < 10; ++i) | |
405 EXPECT_EQ(kTestData[i], buffer[i]); | |
406 for (; i < 53; ++i) | |
407 EXPECT_EQ(0, buffer[i]); | |
408 } | |
409 | |
410 } // namespace base | |
OLD | NEW |