src/base/files/file_util_proxy_unittest.cc

   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/files/file_util_proxy.h"
   6
   7 #include <map>
   8
   9 #include "base/bind.h"
  10 #include "base/file_util.h"
  11 #include "base/files/scoped_temp_dir.h"
  12 #include "base/logging.h"
  13 #include "base/memory/weak_ptr.h"
  14 #include "base/message_loop/message_loop.h"
  15 #include "base/platform_file.h"
  16 #include "base/threading/thread.h"
  17 #include "testing/gtest/include/gtest/gtest.h"
  18
  19 namespace base {
  20
  21 class FileUtilProxyTest : public testing::Test {
  22  public:
  23   FileUtilProxyTest()
  24       : file_thread_("FileUtilProxyTestFileThread"),
  25         error_(File::FILE_OK),
  26         created_(false),
  27         file_(kInvalidPlatformFileValue),
  28         bytes_written_(-1),
  29         weak_factory_(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(File::Error error) {
  42     error_ = error;
  43     MessageLoop::current()->QuitWhenIdle();
  44   }
  45
  46   void DidCreateOrOpen(File::Error 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(File::Error 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(File::Error error,
  65                       const File::Info& file_info) {
  66     error_ = error;
  67     file_info_ = file_info;
  68     MessageLoop::current()->QuitWhenIdle();
  69   }
  70
  71   void DidRead(File::Error 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(File::Error 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   MessageLoopForIO message_loop_;
 106   Thread file_thread_;
 107
 108   ScopedTempDir dir_;
 109   File::Error error_;
 110   bool created_;
 111   PlatformFile file_;
 112   FilePath path_;
 113   File::Info 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(File::FILE_OK, error_);
 128   EXPECT_TRUE(created_);
 129   EXPECT_NE(kInvalidPlatformFileValue, file_);
 130   EXPECT_TRUE(PathExists(test_path()));
 131 }
 132
 133 TEST_F(FileUtilProxyTest, CreateOrOpen_Open) {
 134   // Creates a file.
 135   WriteFile(test_path(), NULL, 0);
 136   ASSERT_TRUE(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(File::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(File::FILE_ERROR_NOT_FOUND, error_);
 159   EXPECT_FALSE(created_);
 160   EXPECT_EQ(kInvalidPlatformFileValue, file_);
 161   EXPECT_FALSE(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(base::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(File::FILE_OK, error_);
 181
 182   // Now it should pass on all platforms.
 183   EXPECT_TRUE(base::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(File::FILE_OK, error_);
 192   EXPECT_TRUE(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(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(base::DeleteFile(path_, false));
 222 }
 223
 224 TEST_F(FileUtilProxyTest, GetFileInfo_File) {
 225   // Setup.
 226   ASSERT_EQ(4, WriteFile(test_path(), "test", 4));
 227   File::Info expected_info;
 228   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(File::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(base::CreateDirectory(test_path()));
 250   File::Info expected_info;
 251   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(File::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             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(File::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(File::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(File::FILE_OK, error_);
 319
 320   // Verify the written data.
 321   char buffer[10];
 322   EXPECT_EQ(data_bytes, base::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(File::FILE_OK, error_);
 342
 343   File::Info info;
 344   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, WriteFile(test_path(), kTestData, 10));
 358   File::Info info;
 359   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   GetFileInfo(test_path(), &info);
 372   ASSERT_EQ(7, info.size);
 373
 374   char buffer[7];
 375   EXPECT_EQ(7, base::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, WriteFile(test_path(), kTestData, 10));
 385   File::Info info;
 386   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   GetFileInfo(test_path(), &info);
 399   ASSERT_EQ(53, info.size);
 400
 401   char buffer[53];
 402   EXPECT_EQ(53, base::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