| 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/drive/drive_scheduler.h" | 5 #include "chrome/browser/chromeos/drive/drive_scheduler.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/threading/sequenced_worker_pool.h" | 8 #include "base/threading/sequenced_worker_pool.h" |
| 9 #include "chrome/browser/chromeos/drive/drive_test_util.h" | 9 #include "chrome/browser/chromeos/drive/drive_test_util.h" |
| 10 #include "chrome/browser/chromeos/drive/file_system/remove_operation.h" | 10 #include "chrome/browser/chromeos/drive/file_system/remove_operation.h" |
| 11 #include "chrome/browser/prefs/pref_service.h" | 11 #include "chrome/browser/prefs/pref_service.h" |
| 12 #include "chrome/common/pref_names.h" | 12 #include "chrome/common/pref_names.h" |
| 13 #include "chrome/test/base/testing_profile.h" | 13 #include "chrome/test/base/testing_profile.h" |
| 14 #include "content/public/test/test_browser_thread.h" | 14 #include "content/public/test/test_browser_thread.h" |
| 15 #include "testing/gmock/include/gmock/gmock.h" | 15 #include "testing/gmock/include/gmock/gmock.h" |
| 16 #include "testing/gtest/include/gtest/gtest.h" | 16 #include "testing/gtest/include/gtest/gtest.h" |
| 17 | 17 |
| 18 using ::testing::AnyNumber; | 18 using ::testing::AnyNumber; |
| 19 using ::testing::DoAll; | 19 using ::testing::DoAll; |
| 20 using ::testing::Return; | 20 using ::testing::Return; |
| 21 using ::testing::StrictMock; | 21 using ::testing::StrictMock; |
| 22 using ::testing::_; | 22 using ::testing::_; |
| 23 | 23 |
| 24 namespace gdata { | 24 namespace drive { |
| 25 | 25 |
| 26 namespace { | 26 namespace { |
| 27 | 27 |
| 28 class MockNetworkChangeNotifier : public net::NetworkChangeNotifier { | 28 class MockNetworkChangeNotifier : public net::NetworkChangeNotifier { |
| 29 public: | 29 public: |
| 30 MOCK_CONST_METHOD0(GetCurrentConnectionType, | 30 MOCK_CONST_METHOD0(GetCurrentConnectionType, |
| 31 net::NetworkChangeNotifier::ConnectionType()); | 31 net::NetworkChangeNotifier::ConnectionType()); |
| 32 }; | 32 }; |
| 33 | 33 |
| 34 class MockRemoveOperation : public file_system::RemoveOperation { | 34 class MockRemoveOperation : public file_system::RemoveOperation { |
| (...skipping 28 matching lines...) Expand all Loading... |
| 63 &mock_remove_operation_)); | 63 &mock_remove_operation_)); |
| 64 | 64 |
| 65 scheduler_->Initialize(); | 65 scheduler_->Initialize(); |
| 66 scheduler_->SetDisableThrottling(true); | 66 scheduler_->SetDisableThrottling(true); |
| 67 } | 67 } |
| 68 | 68 |
| 69 virtual void TearDown() OVERRIDE { | 69 virtual void TearDown() OVERRIDE { |
| 70 // The scheduler should be deleted before NetworkLibrary, as it | 70 // The scheduler should be deleted before NetworkLibrary, as it |
| 71 // registers itself as observer of NetworkLibrary. | 71 // registers itself as observer of NetworkLibrary. |
| 72 scheduler_.reset(); | 72 scheduler_.reset(); |
| 73 test_util::RunBlockingPoolTask(); | 73 gdata::test_util::RunBlockingPoolTask(); |
| 74 mock_network_change_notifier_.reset(); | 74 mock_network_change_notifier_.reset(); |
| 75 } | 75 } |
| 76 | 76 |
| 77 // Sets up MockNetworkChangeNotifier as if it's connected to a network with | 77 // Sets up MockNetworkChangeNotifier as if it's connected to a network with |
| 78 // the specified connection type. | 78 // the specified connection type. |
| 79 void ChangeConnectionType(net::NetworkChangeNotifier::ConnectionType type) { | 79 void ChangeConnectionType(net::NetworkChangeNotifier::ConnectionType type) { |
| 80 EXPECT_CALL(*mock_network_change_notifier_, GetCurrentConnectionType()) | 80 EXPECT_CALL(*mock_network_change_notifier_, GetCurrentConnectionType()) |
| 81 .WillRepeatedly(Return(type)); | 81 .WillRepeatedly(Return(type)); |
| 82 // Notify the sync client that the network is changed. This is done via | 82 // Notify the sync client that the network is changed. This is done via |
| 83 // NetworkChangeNotifier in production, but here, we simulate the behavior | 83 // NetworkChangeNotifier in production, but here, we simulate the behavior |
| (...skipping 35 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 119 ConnectToWifi(); | 119 ConnectToWifi(); |
| 120 | 120 |
| 121 FilePath file_in_root(FILE_PATH_LITERAL("drive/File 1.txt")); | 121 FilePath file_in_root(FILE_PATH_LITERAL("drive/File 1.txt")); |
| 122 EXPECT_CALL(mock_remove_operation_, Remove(file_in_root, _, _)) | 122 EXPECT_CALL(mock_remove_operation_, Remove(file_in_root, _, _)) |
| 123 .WillOnce(MockRemove(DRIVE_FILE_OK)); | 123 .WillOnce(MockRemove(DRIVE_FILE_OK)); |
| 124 | 124 |
| 125 DriveFileError error; | 125 DriveFileError error; |
| 126 scheduler_->Remove( | 126 scheduler_->Remove( |
| 127 file_in_root, false, | 127 file_in_root, false, |
| 128 base::Bind(&test_util::CopyErrorCodeFromFileOperationCallback, &error)); | 128 base::Bind(&test_util::CopyErrorCodeFromFileOperationCallback, &error)); |
| 129 test_util::RunBlockingPoolTask(); | 129 gdata::test_util::RunBlockingPoolTask(); |
| 130 | 130 |
| 131 ASSERT_EQ(DRIVE_FILE_OK, error); | 131 ASSERT_EQ(DRIVE_FILE_OK, error); |
| 132 } | 132 } |
| 133 | 133 |
| 134 TEST_F(DriveSchedulerTest, RemoveFileRetry) { | 134 TEST_F(DriveSchedulerTest, RemoveFileRetry) { |
| 135 ConnectToWifi(); | 135 ConnectToWifi(); |
| 136 | 136 |
| 137 FilePath file_in_root(FILE_PATH_LITERAL("drive/File 1.txt")); | 137 FilePath file_in_root(FILE_PATH_LITERAL("drive/File 1.txt")); |
| 138 EXPECT_CALL(mock_remove_operation_, Remove(file_in_root, _, _)) | 138 EXPECT_CALL(mock_remove_operation_, Remove(file_in_root, _, _)) |
| 139 .WillOnce(MockRemove(DRIVE_FILE_ERROR_FAILED)) | 139 .WillOnce(MockRemove(DRIVE_FILE_ERROR_FAILED)) |
| 140 .WillOnce(MockRemove(DRIVE_FILE_OK)); | 140 .WillOnce(MockRemove(DRIVE_FILE_OK)); |
| 141 | 141 |
| 142 DriveFileError error; | 142 DriveFileError error; |
| 143 scheduler_->Remove( | 143 scheduler_->Remove( |
| 144 file_in_root, false, | 144 file_in_root, false, |
| 145 base::Bind(&test_util::CopyErrorCodeFromFileOperationCallback, &error)); | 145 base::Bind(&test_util::CopyErrorCodeFromFileOperationCallback, &error)); |
| 146 test_util::RunBlockingPoolTask(); | 146 gdata::test_util::RunBlockingPoolTask(); |
| 147 | 147 |
| 148 ASSERT_EQ(DRIVE_FILE_OK, error); | 148 ASSERT_EQ(DRIVE_FILE_OK, error); |
| 149 } | 149 } |
| 150 | 150 |
| 151 TEST_F(DriveSchedulerTest, QueueOperation_Offline) { | 151 TEST_F(DriveSchedulerTest, QueueOperation_Offline) { |
| 152 ConnectToNone(); | 152 ConnectToNone(); |
| 153 | 153 |
| 154 // This file will not be removed, as network is not connected. | 154 // This file will not be removed, as network is not connected. |
| 155 EXPECT_CALL(mock_remove_operation_, Remove(_, _, _)).Times(0); | 155 EXPECT_CALL(mock_remove_operation_, Remove(_, _, _)).Times(0); |
| 156 | 156 |
| 157 FilePath file_in_root(FILE_PATH_LITERAL("drive/File 1.txt")); | 157 FilePath file_in_root(FILE_PATH_LITERAL("drive/File 1.txt")); |
| 158 DriveFileError error; | 158 DriveFileError error; |
| 159 scheduler_->Remove( | 159 scheduler_->Remove( |
| 160 file_in_root, false, | 160 file_in_root, false, |
| 161 base::Bind(&test_util::CopyErrorCodeFromFileOperationCallback, &error)); | 161 base::Bind(&test_util::CopyErrorCodeFromFileOperationCallback, &error)); |
| 162 test_util::RunBlockingPoolTask(); | 162 gdata::test_util::RunBlockingPoolTask(); |
| 163 } | 163 } |
| 164 | 164 |
| 165 TEST_F(DriveSchedulerTest, QueueOperation_CelluarDisabled) { | 165 TEST_F(DriveSchedulerTest, QueueOperation_CelluarDisabled) { |
| 166 ConnectToCellular(); | 166 ConnectToCellular(); |
| 167 | 167 |
| 168 // This file will not be removed, as fetching over cellular network is | 168 // This file will not be removed, as fetching over cellular network is |
| 169 // disabled by default. | 169 // disabled by default. |
| 170 EXPECT_CALL(mock_remove_operation_, Remove(_, _, _)).Times(0); | 170 EXPECT_CALL(mock_remove_operation_, Remove(_, _, _)).Times(0); |
| 171 | 171 |
| 172 FilePath file_in_root(FILE_PATH_LITERAL("drive/File 1.txt")); | 172 FilePath file_in_root(FILE_PATH_LITERAL("drive/File 1.txt")); |
| 173 DriveFileError error; | 173 DriveFileError error; |
| 174 scheduler_->Remove( | 174 scheduler_->Remove( |
| 175 file_in_root, false, | 175 file_in_root, false, |
| 176 base::Bind(&test_util::CopyErrorCodeFromFileOperationCallback, &error)); | 176 base::Bind(&test_util::CopyErrorCodeFromFileOperationCallback, &error)); |
| 177 test_util::RunBlockingPoolTask(); | 177 gdata::test_util::RunBlockingPoolTask(); |
| 178 } | 178 } |
| 179 | 179 |
| 180 TEST_F(DriveSchedulerTest, QueueOperation_CelluarEnabled) { | 180 TEST_F(DriveSchedulerTest, QueueOperation_CelluarEnabled) { |
| 181 // Enable fetching over cellular network. | 181 // Enable fetching over cellular network. |
| 182 profile_->GetPrefs()->SetBoolean(prefs::kDisableGDataOverCellular, false); | 182 profile_->GetPrefs()->SetBoolean(prefs::kDisableGDataOverCellular, false); |
| 183 | 183 |
| 184 ConnectToCellular(); | 184 ConnectToCellular(); |
| 185 | 185 |
| 186 // This file will be removed, as syncing over cellular network is explicitly | 186 // This file will be removed, as syncing over cellular network is explicitly |
| 187 // enabled. | 187 // enabled. |
| 188 EXPECT_CALL(mock_remove_operation_, Remove(_, _, _)).Times(1); | 188 EXPECT_CALL(mock_remove_operation_, Remove(_, _, _)).Times(1); |
| 189 | 189 |
| 190 FilePath file_in_root(FILE_PATH_LITERAL("drive/File 1.txt")); | 190 FilePath file_in_root(FILE_PATH_LITERAL("drive/File 1.txt")); |
| 191 DriveFileError error; | 191 DriveFileError error; |
| 192 scheduler_->Remove( | 192 scheduler_->Remove( |
| 193 file_in_root, false, | 193 file_in_root, false, |
| 194 base::Bind(&test_util::CopyErrorCodeFromFileOperationCallback, &error)); | 194 base::Bind(&test_util::CopyErrorCodeFromFileOperationCallback, &error)); |
| 195 test_util::RunBlockingPoolTask(); | 195 gdata::test_util::RunBlockingPoolTask(); |
| 196 } | 196 } |
| 197 | 197 |
| 198 TEST_F(DriveSchedulerTest, QueueOperation_WimaxDisabled) { | 198 TEST_F(DriveSchedulerTest, QueueOperation_WimaxDisabled) { |
| 199 // Then connect to wimax. This will kick off StartSyncLoop(). | 199 // Then connect to wimax. This will kick off StartSyncLoop(). |
| 200 ConnectToWimax(); | 200 ConnectToWimax(); |
| 201 | 201 |
| 202 // This file will not be removed, as syncing over wimax network is disabled | 202 // This file will not be removed, as syncing over wimax network is disabled |
| 203 // by default. | 203 // by default. |
| 204 EXPECT_CALL(mock_remove_operation_, Remove(_, _, _)).Times(0); | 204 EXPECT_CALL(mock_remove_operation_, Remove(_, _, _)).Times(0); |
| 205 | 205 |
| 206 FilePath file_in_root(FILE_PATH_LITERAL("drive/File 1.txt")); | 206 FilePath file_in_root(FILE_PATH_LITERAL("drive/File 1.txt")); |
| 207 DriveFileError error; | 207 DriveFileError error; |
| 208 scheduler_->Remove( | 208 scheduler_->Remove( |
| 209 file_in_root, false, | 209 file_in_root, false, |
| 210 base::Bind(&test_util::CopyErrorCodeFromFileOperationCallback, &error)); | 210 base::Bind(&test_util::CopyErrorCodeFromFileOperationCallback, &error)); |
| 211 test_util::RunBlockingPoolTask(); | 211 gdata::test_util::RunBlockingPoolTask(); |
| 212 } | 212 } |
| 213 | 213 |
| 214 TEST_F(DriveSchedulerTest, QueueOperation_CelluarEnabledWithWimax) { | 214 TEST_F(DriveSchedulerTest, QueueOperation_CelluarEnabledWithWimax) { |
| 215 // Enable fetching over cellular network. | 215 // Enable fetching over cellular network. |
| 216 profile_->GetPrefs()->SetBoolean(prefs::kDisableGDataOverCellular, false); | 216 profile_->GetPrefs()->SetBoolean(prefs::kDisableGDataOverCellular, false); |
| 217 | 217 |
| 218 ConnectToWimax(); | 218 ConnectToWimax(); |
| 219 | 219 |
| 220 // This file will be removed, as syncing over cellular network is explicitly | 220 // This file will be removed, as syncing over cellular network is explicitly |
| 221 // enabled. | 221 // enabled. |
| 222 EXPECT_CALL(mock_remove_operation_, Remove(_, _, _)).Times(1); | 222 EXPECT_CALL(mock_remove_operation_, Remove(_, _, _)).Times(1); |
| 223 | 223 |
| 224 FilePath file_in_root(FILE_PATH_LITERAL("drive/File 1.txt")); | 224 FilePath file_in_root(FILE_PATH_LITERAL("drive/File 1.txt")); |
| 225 DriveFileError error; | 225 DriveFileError error; |
| 226 scheduler_->Remove( | 226 scheduler_->Remove( |
| 227 file_in_root, false, | 227 file_in_root, false, |
| 228 base::Bind(&test_util::CopyErrorCodeFromFileOperationCallback, &error)); | 228 base::Bind(&test_util::CopyErrorCodeFromFileOperationCallback, &error)); |
| 229 test_util::RunBlockingPoolTask(); | 229 gdata::test_util::RunBlockingPoolTask(); |
| 230 } | 230 } |
| 231 | 231 |
| 232 TEST_F(DriveSchedulerTest, QueueOperation_DriveDisabled) { | 232 TEST_F(DriveSchedulerTest, QueueOperation_DriveDisabled) { |
| 233 // Disable the Drive feature. | 233 // Disable the Drive feature. |
| 234 profile_->GetPrefs()->SetBoolean(prefs::kDisableGData, true); | 234 profile_->GetPrefs()->SetBoolean(prefs::kDisableGData, true); |
| 235 | 235 |
| 236 // This file will not be removed, as the Drive feature is disabled. | 236 // This file will not be removed, as the Drive feature is disabled. |
| 237 EXPECT_CALL(mock_remove_operation_, Remove(_, _, _)).Times(0); | 237 EXPECT_CALL(mock_remove_operation_, Remove(_, _, _)).Times(0); |
| 238 | 238 |
| 239 FilePath file_in_root(FILE_PATH_LITERAL("drive/File 1.txt")); | 239 FilePath file_in_root(FILE_PATH_LITERAL("drive/File 1.txt")); |
| 240 DriveFileError error; | 240 DriveFileError error; |
| 241 scheduler_->Remove( | 241 scheduler_->Remove( |
| 242 file_in_root, false, | 242 file_in_root, false, |
| 243 base::Bind(&test_util::CopyErrorCodeFromFileOperationCallback, &error)); | 243 base::Bind(&test_util::CopyErrorCodeFromFileOperationCallback, &error)); |
| 244 test_util::RunBlockingPoolTask(); | 244 gdata::test_util::RunBlockingPoolTask(); |
| 245 } | 245 } |
| 246 | 246 |
| 247 } // namespace gdata | 247 } // namespace drive |
| OLD | NEW |