| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 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/gdata/file_write_helper.h" | 5 #include "chrome/browser/chromeos/gdata/file_write_helper.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/message_loop.h" | 8 #include "base/message_loop.h" |
| 9 #include "base/threading/thread_restrictions.h" | 9 #include "base/threading/thread_restrictions.h" |
| 10 #include "chrome/browser/chromeos/gdata/gdata_test_util.h" | 10 #include "chrome/browser/chromeos/gdata/gdata_test_util.h" |
| (...skipping 18 matching lines...) Expand all Loading... |
| 29 ACTION_P2(MockOpenFile, error, local_path) { | 29 ACTION_P2(MockOpenFile, error, local_path) { |
| 30 if (!arg1.is_null()) | 30 if (!arg1.is_null()) |
| 31 arg1.Run(error, local_path); | 31 arg1.Run(error, local_path); |
| 32 } | 32 } |
| 33 | 33 |
| 34 ACTION_P(MockCloseFile, error) { | 34 ACTION_P(MockCloseFile, error) { |
| 35 if (!arg1.is_null()) | 35 if (!arg1.is_null()) |
| 36 arg1.Run(error); | 36 arg1.Run(error); |
| 37 } | 37 } |
| 38 | 38 |
| 39 void RecordOpenFileCallbackArguments(GDataFileError* error, | 39 void RecordOpenFileCallbackArguments(DriveFileError* error, |
| 40 FilePath* path, | 40 FilePath* path, |
| 41 GDataFileError error_arg, | 41 DriveFileError error_arg, |
| 42 const FilePath& path_arg) { | 42 const FilePath& path_arg) { |
| 43 base::ThreadRestrictions::AssertIOAllowed(); | 43 base::ThreadRestrictions::AssertIOAllowed(); |
| 44 *error = error_arg; | 44 *error = error_arg; |
| 45 *path = path_arg; | 45 *path = path_arg; |
| 46 } | 46 } |
| 47 | 47 |
| 48 } // namespace | 48 } // namespace |
| 49 | 49 |
| 50 class FileWriteHelperTest : public testing::Test { | 50 class FileWriteHelperTest : public testing::Test { |
| 51 public: | 51 public: |
| 52 FileWriteHelperTest() | 52 FileWriteHelperTest() |
| 53 : ui_thread_(content::BrowserThread::UI, &message_loop_), | 53 : ui_thread_(content::BrowserThread::UI, &message_loop_), |
| 54 mock_file_system_(new StrictMock<MockGDataFileSystem>) { | 54 mock_file_system_(new StrictMock<MockGDataFileSystem>) { |
| 55 } | 55 } |
| 56 | 56 |
| 57 protected: | 57 protected: |
| 58 MessageLoopForUI message_loop_; | 58 MessageLoopForUI message_loop_; |
| 59 content::TestBrowserThread ui_thread_; | 59 content::TestBrowserThread ui_thread_; |
| 60 scoped_ptr< StrictMock<MockGDataFileSystem> > mock_file_system_; | 60 scoped_ptr< StrictMock<MockGDataFileSystem> > mock_file_system_; |
| 61 }; | 61 }; |
| 62 | 62 |
| 63 TEST_F(FileWriteHelperTest, PrepareFileForWritingSuccess) { | 63 TEST_F(FileWriteHelperTest, PrepareFileForWritingSuccess) { |
| 64 const FilePath kDrivePath("/drive/file.txt"); | 64 const FilePath kDrivePath("/drive/file.txt"); |
| 65 const FilePath kLocalPath("/tmp/dummy.txt"); | 65 const FilePath kLocalPath("/tmp/dummy.txt"); |
| 66 | 66 |
| 67 EXPECT_CALL(*mock_file_system_, CreateFile(kDrivePath, false, _)) | 67 EXPECT_CALL(*mock_file_system_, CreateFile(kDrivePath, false, _)) |
| 68 .WillOnce(MockCreateFile(GDATA_FILE_OK)); | 68 .WillOnce(MockCreateFile(DRIVE_FILE_OK)); |
| 69 EXPECT_CALL(*mock_file_system_, OpenFile(kDrivePath, _)) | 69 EXPECT_CALL(*mock_file_system_, OpenFile(kDrivePath, _)) |
| 70 .WillOnce(MockOpenFile(GDATA_FILE_OK, kLocalPath)); | 70 .WillOnce(MockOpenFile(DRIVE_FILE_OK, kLocalPath)); |
| 71 EXPECT_CALL(*mock_file_system_, CloseFile(kDrivePath, _)) | 71 EXPECT_CALL(*mock_file_system_, CloseFile(kDrivePath, _)) |
| 72 .WillOnce(MockCloseFile(GDATA_FILE_OK)); | 72 .WillOnce(MockCloseFile(DRIVE_FILE_OK)); |
| 73 | 73 |
| 74 FileWriteHelper file_write_helper(mock_file_system_.get()); | 74 FileWriteHelper file_write_helper(mock_file_system_.get()); |
| 75 GDataFileError error = GDATA_FILE_ERROR_FAILED; | 75 DriveFileError error = DRIVE_FILE_ERROR_FAILED; |
| 76 FilePath path; | 76 FilePath path; |
| 77 file_write_helper.PrepareWritableFileAndRun( | 77 file_write_helper.PrepareWritableFileAndRun( |
| 78 kDrivePath, base::Bind(&RecordOpenFileCallbackArguments, &error, &path)); | 78 kDrivePath, base::Bind(&RecordOpenFileCallbackArguments, &error, &path)); |
| 79 test_util::RunBlockingPoolTask(); | 79 test_util::RunBlockingPoolTask(); |
| 80 | 80 |
| 81 EXPECT_EQ(GDATA_FILE_OK, error); | 81 EXPECT_EQ(DRIVE_FILE_OK, error); |
| 82 EXPECT_EQ(kLocalPath, path); | 82 EXPECT_EQ(kLocalPath, path); |
| 83 } | 83 } |
| 84 | 84 |
| 85 TEST_F(FileWriteHelperTest, PrepareFileForWritingCreateFail) { | 85 TEST_F(FileWriteHelperTest, PrepareFileForWritingCreateFail) { |
| 86 const FilePath kDrivePath("/drive/file.txt"); | 86 const FilePath kDrivePath("/drive/file.txt"); |
| 87 | 87 |
| 88 EXPECT_CALL(*mock_file_system_, CreateFile(kDrivePath, false, _)) | 88 EXPECT_CALL(*mock_file_system_, CreateFile(kDrivePath, false, _)) |
| 89 .WillOnce(MockCreateFile(GDATA_FILE_ERROR_ACCESS_DENIED)); | 89 .WillOnce(MockCreateFile(DRIVE_FILE_ERROR_ACCESS_DENIED)); |
| 90 EXPECT_CALL(*mock_file_system_, OpenFile(_, _)).Times(0); | 90 EXPECT_CALL(*mock_file_system_, OpenFile(_, _)).Times(0); |
| 91 EXPECT_CALL(*mock_file_system_, CloseFile(_, _)).Times(0); | 91 EXPECT_CALL(*mock_file_system_, CloseFile(_, _)).Times(0); |
| 92 | 92 |
| 93 FileWriteHelper file_write_helper(mock_file_system_.get()); | 93 FileWriteHelper file_write_helper(mock_file_system_.get()); |
| 94 GDataFileError error = GDATA_FILE_ERROR_FAILED; | 94 DriveFileError error = DRIVE_FILE_ERROR_FAILED; |
| 95 FilePath path; | 95 FilePath path; |
| 96 file_write_helper.PrepareWritableFileAndRun( | 96 file_write_helper.PrepareWritableFileAndRun( |
| 97 kDrivePath, base::Bind(&RecordOpenFileCallbackArguments, &error, &path)); | 97 kDrivePath, base::Bind(&RecordOpenFileCallbackArguments, &error, &path)); |
| 98 test_util::RunBlockingPoolTask(); | 98 test_util::RunBlockingPoolTask(); |
| 99 | 99 |
| 100 EXPECT_EQ(GDATA_FILE_ERROR_ACCESS_DENIED, error); | 100 EXPECT_EQ(DRIVE_FILE_ERROR_ACCESS_DENIED, error); |
| 101 EXPECT_EQ(FilePath(), path); | 101 EXPECT_EQ(FilePath(), path); |
| 102 } | 102 } |
| 103 | 103 |
| 104 TEST_F(FileWriteHelperTest, PrepareFileForWritingOpenFail) { | 104 TEST_F(FileWriteHelperTest, PrepareFileForWritingOpenFail) { |
| 105 const FilePath kDrivePath("/drive/file.txt"); | 105 const FilePath kDrivePath("/drive/file.txt"); |
| 106 | 106 |
| 107 EXPECT_CALL(*mock_file_system_, CreateFile(kDrivePath, false, _)) | 107 EXPECT_CALL(*mock_file_system_, CreateFile(kDrivePath, false, _)) |
| 108 .WillOnce(MockCreateFile(GDATA_FILE_OK)); | 108 .WillOnce(MockCreateFile(DRIVE_FILE_OK)); |
| 109 EXPECT_CALL(*mock_file_system_, OpenFile(kDrivePath, _)) | 109 EXPECT_CALL(*mock_file_system_, OpenFile(kDrivePath, _)) |
| 110 .WillOnce(MockOpenFile(GDATA_FILE_ERROR_IN_USE, FilePath())); | 110 .WillOnce(MockOpenFile(DRIVE_FILE_ERROR_IN_USE, FilePath())); |
| 111 EXPECT_CALL(*mock_file_system_, CloseFile(_, _)).Times(0); | 111 EXPECT_CALL(*mock_file_system_, CloseFile(_, _)).Times(0); |
| 112 | 112 |
| 113 FileWriteHelper file_write_helper(mock_file_system_.get()); | 113 FileWriteHelper file_write_helper(mock_file_system_.get()); |
| 114 GDataFileError error = GDATA_FILE_ERROR_FAILED; | 114 DriveFileError error = DRIVE_FILE_ERROR_FAILED; |
| 115 FilePath path; | 115 FilePath path; |
| 116 file_write_helper.PrepareWritableFileAndRun( | 116 file_write_helper.PrepareWritableFileAndRun( |
| 117 kDrivePath, base::Bind(&RecordOpenFileCallbackArguments, &error, &path)); | 117 kDrivePath, base::Bind(&RecordOpenFileCallbackArguments, &error, &path)); |
| 118 test_util::RunBlockingPoolTask(); | 118 test_util::RunBlockingPoolTask(); |
| 119 | 119 |
| 120 EXPECT_EQ(GDATA_FILE_ERROR_IN_USE, error); | 120 EXPECT_EQ(DRIVE_FILE_ERROR_IN_USE, error); |
| 121 EXPECT_EQ(FilePath(), path); | 121 EXPECT_EQ(FilePath(), path); |
| 122 } | 122 } |
| 123 | 123 |
| 124 } // namespace gdata | 124 } // namespace gdata |
| OLD | NEW |