Home | History | Annotate | Download | only in drive 
      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
      3 // found in the LICENSE file.
      4 
      5 #include "chrome/browser/chromeos/drive/write_on_cache_file.h"
      6 
      7 #include "base/bind.h"
      8 #include "chrome/browser/chromeos/drive/dummy_file_system.h"
      9 #include "chrome/browser/chromeos/drive/test_util.h"
     10 #include "content/public/test/test_browser_thread_bundle.h"
     11 #include "testing/gtest/include/gtest/gtest.h"
     12 
     13 namespace drive {
     14 
     15 namespace {
     16 
     17 const base::FilePath::CharType kDrivePath[] =
     18     FILE_PATH_LITERAL("drive/root/file.txt");
     19 const base::FilePath::CharType kInvalidPath[] =
     20     FILE_PATH_LITERAL("drive/invalid/path");
     21 const base::FilePath::CharType kLocalPath[] =
     22     FILE_PATH_LITERAL("/tmp/local.txt");
     23 
     24 class TestFileSystem : public DummyFileSystem {
     25  public:
     26   TestFileSystem() : num_closed_(0) {
     27   }
     28 
     29   int num_closed() const { return num_closed_; }
     30 
     31   // Mimics OpenFile. It fails if the |file_path| points to a hosted document.
     32   virtual void OpenFile(const base::FilePath& file_path,
     33                         OpenMode open_mode,
     34                         const std::string& mime_type,
     35                         const OpenFileCallback& callback) OVERRIDE {
     36     EXPECT_EQ(OPEN_OR_CREATE_FILE, open_mode);
     37 
     38     // Emulate a case of opening a hosted document.
     39     if (file_path == base::FilePath(kInvalidPath)) {
     40       callback.Run(FILE_ERROR_INVALID_OPERATION, base::FilePath(),
     41                    base::Closure());
     42       return;
     43     }
     44 
     45     callback.Run(FILE_ERROR_OK, base::FilePath(kLocalPath),
     46                  base::Bind(&TestFileSystem::CloseFile,
     47                             base::Unretained(this)));
     48   }
     49 
     50  private:
     51 
     52   void CloseFile() {
     53     ++num_closed_;
     54   }
     55 
     56   int num_closed_;
     57 };
     58 
     59 }  // namespace
     60 
     61 TEST(WriteOnCacheFileTest, PrepareFileForWritingSuccess) {
     62   content::TestBrowserThreadBundle thread_bundle;
     63   TestFileSystem test_file_system;
     64 
     65   FileError error = FILE_ERROR_FAILED;
     66   base::FilePath path;
     67   // The file should successfully be opened.
     68   WriteOnCacheFile(
     69       &test_file_system,
     70       base::FilePath(kDrivePath),
     71       std::string(),  // mime_type
     72       google_apis::test_util::CreateCopyResultCallback(&error, &path));
     73   test_util::RunBlockingPoolTask();
     74 
     75   EXPECT_EQ(FILE_ERROR_OK, error);
     76   EXPECT_EQ(kLocalPath, path.value());
     77 
     78   // Make sure that the file is actually closed.
     79   EXPECT_EQ(1, test_file_system.num_closed());
     80 }
     81 
     82 TEST(WriteOnCacheFileTest, PrepareFileForWritingCreateFail) {
     83   content::TestBrowserThreadBundle thread_bundle;
     84   TestFileSystem test_file_system;
     85 
     86   FileError error = FILE_ERROR_FAILED;
     87   base::FilePath path;
     88   // Access to kInvalidPath should fail, and FileWriteHelper should not try to
     89   // open or close the file.
     90   WriteOnCacheFile(
     91       &test_file_system,
     92       base::FilePath(kInvalidPath),
     93       std::string(),  // mime_type
     94       google_apis::test_util::CreateCopyResultCallback(&error, &path));
     95   test_util::RunBlockingPoolTask();
     96 
     97   EXPECT_EQ(FILE_ERROR_INVALID_OPERATION, error);
     98   EXPECT_TRUE(path.empty());
     99 }
    100 
    101 }   // namespace drive
    102