syncable_file_system_unittest.cc revision 116680a4aac90f2aa7413d9095a592090648e557 
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 "base/stl_util.h"
6#include "base/thread_task_runner_handle.h"
7#include "chrome/browser/sync_file_system/local/canned_syncable_file_system.h"
8#include "chrome/browser/sync_file_system/local/local_file_change_tracker.h"
9#include "chrome/browser/sync_file_system/local/local_file_sync_context.h"
10#include "chrome/browser/sync_file_system/local/sync_file_system_backend.h"
11#include "chrome/browser/sync_file_system/syncable_file_system_util.h"
12#include "content/public/test/async_file_test_helper.h"
13#include "content/public/test/sandbox_file_system_test_helper.h"
14#include "content/public/test/test_browser_thread_bundle.h"
15#include "testing/gtest/include/gtest/gtest.h"
16#include "third_party/leveldatabase/src/helpers/memenv/memenv.h"
17#include "third_party/leveldatabase/src/include/leveldb/env.h"
18#include "webkit/browser/fileapi/file_system_context.h"
19#include "webkit/browser/fileapi/file_system_operation_context.h"
20#include "webkit/browser/fileapi/isolated_context.h"
21#include "webkit/browser/quota/quota_manager.h"
22#include "webkit/common/fileapi/file_system_types.h"
23#include "webkit/common/quota/quota_types.h"
24
25using content::SandboxFileSystemTestHelper;
26using fileapi::FileSystemContext;
27using fileapi::FileSystemOperationContext;
28using fileapi::FileSystemURL;
29using fileapi::FileSystemURLSet;
30using quota::QuotaManager;
31using quota::QuotaStatusCode;
32
33namespace sync_file_system {
34
35class SyncableFileSystemTest : public testing::Test {
36 public:
37  SyncableFileSystemTest()
38      : in_memory_env_(leveldb::NewMemEnv(leveldb::Env::Default())),
39        file_system_(GURL("http://example.com/"),
40                     in_memory_env_.get(),
41                     base::ThreadTaskRunnerHandle::Get().get(),
42                     base::ThreadTaskRunnerHandle::Get().get()),
43        weak_factory_(this) {}
44
45  virtual void SetUp() {
46    ASSERT_TRUE(data_dir_.CreateUniqueTempDir());
47    file_system_.SetUp(CannedSyncableFileSystem::QUOTA_ENABLED);
48
49    sync_context_ =
50        new LocalFileSyncContext(data_dir_.path(),
51                                 in_memory_env_.get(),
52                                 base::ThreadTaskRunnerHandle::Get().get(),
53                                 base::ThreadTaskRunnerHandle::Get().get());
54    ASSERT_EQ(
55        sync_file_system::SYNC_STATUS_OK,
56        file_system_.MaybeInitializeFileSystemContext(sync_context_.get()));
57  }
58
59  virtual void TearDown() {
60    if (sync_context_.get())
61      sync_context_->ShutdownOnUIThread();
62    sync_context_ = NULL;
63
64    file_system_.TearDown();
65
66    // Make sure we don't leave the external filesystem.
67    // (CannedSyncableFileSystem::TearDown does not do this as there may be
68    // multiple syncable file systems registered for the name)
69    RevokeSyncableFileSystem();
70  }
71
72 protected:
73  void VerifyAndClearChange(const FileSystemURL& url,
74                            const FileChange& expected_change) {
75    SCOPED_TRACE(testing::Message() << url.DebugString() <<
76                 " expecting:" << expected_change.DebugString());
77    // Get the changes for URL and verify.
78    FileChangeList changes;
79    change_tracker()->GetChangesForURL(url, &changes);
80    ASSERT_EQ(1U, changes.size());
81    SCOPED_TRACE(testing::Message() << url.DebugString() <<
82                 " actual:" << changes.DebugString());
83    EXPECT_EQ(expected_change, changes.front());
84
85    // Clear the URL from the change tracker.
86    change_tracker()->ClearChangesForURL(url);
87  }
88
89  FileSystemURL URL(const std::string& path) {
90    return file_system_.URL(path);
91  }
92
93  FileSystemContext* file_system_context() {
94    return file_system_.file_system_context();
95  }
96
97  LocalFileChangeTracker* change_tracker() {
98    return file_system_.backend()->change_tracker();
99  }
100
101  base::ScopedTempDir data_dir_;
102  content::TestBrowserThreadBundle thread_bundle_;
103  scoped_ptr<leveldb::Env> in_memory_env_;
104  CannedSyncableFileSystem file_system_;
105
106 private:
107  scoped_refptr<LocalFileSyncContext> sync_context_;
108
109  base::WeakPtrFactory<SyncableFileSystemTest> weak_factory_;
110
111  DISALLOW_COPY_AND_ASSIGN(SyncableFileSystemTest);
112};
113
114// Brief combined testing. Just see if all the sandbox feature works.
115TEST_F(SyncableFileSystemTest, SyncableLocalSandboxCombined) {
116  // Opens a syncable file system.
117  EXPECT_EQ(base::File::FILE_OK,
118            file_system_.OpenFileSystem());
119
120  // Do some operations.
121  EXPECT_EQ(base::File::FILE_OK,
122            file_system_.CreateDirectory(URL("dir")));
123  EXPECT_EQ(base::File::FILE_OK,
124            file_system_.CreateFile(URL("dir/foo")));
125
126  const int64 kOriginalQuota = QuotaManager::kSyncableStorageDefaultHostQuota;
127
128  const int64 kQuota = 12345 * 1024;
129  QuotaManager::kSyncableStorageDefaultHostQuota = kQuota;
130  int64 usage, quota;
131  EXPECT_EQ(quota::kQuotaStatusOk,
132            file_system_.GetUsageAndQuota(&usage, &quota));
133
134  // Returned quota must be what we overrode. Usage must be greater than 0
135  // as creating a file or directory consumes some space.
136  EXPECT_EQ(kQuota, quota);
137  EXPECT_GT(usage, 0);
138
139  // Truncate to extend an existing file and see if the usage reflects it.
140  const int64 kFileSizeToExtend = 333;
141  EXPECT_EQ(base::File::FILE_OK,
142            file_system_.CreateFile(URL("dir/foo")));
143
144  EXPECT_EQ(base::File::FILE_OK,
145            file_system_.TruncateFile(URL("dir/foo"), kFileSizeToExtend));
146
147  int64 new_usage;
148  EXPECT_EQ(quota::kQuotaStatusOk,
149            file_system_.GetUsageAndQuota(&new_usage, &quota));
150  EXPECT_EQ(kFileSizeToExtend, new_usage - usage);
151
152  // Shrink the quota to the current usage, try to extend the file further
153  // and see if it fails.
154  QuotaManager::kSyncableStorageDefaultHostQuota = new_usage;
155  EXPECT_EQ(base::File::FILE_ERROR_NO_SPACE,
156            file_system_.TruncateFile(URL("dir/foo"), kFileSizeToExtend + 1));
157
158  usage = new_usage;
159  EXPECT_EQ(quota::kQuotaStatusOk,
160            file_system_.GetUsageAndQuota(&new_usage, &quota));
161  EXPECT_EQ(usage, new_usage);
162
163  // Deletes the file system.
164  EXPECT_EQ(base::File::FILE_OK,
165            file_system_.DeleteFileSystem());
166
167  // Now the usage must be zero.
168  EXPECT_EQ(quota::kQuotaStatusOk,
169            file_system_.GetUsageAndQuota(&usage, &quota));
170  EXPECT_EQ(0, usage);
171
172  // Restore the system default quota.
173  QuotaManager::kSyncableStorageDefaultHostQuota = kOriginalQuota;
174}
175
176// Combined testing with LocalFileChangeTracker.
177TEST_F(SyncableFileSystemTest, ChangeTrackerSimple) {
178  EXPECT_EQ(base::File::FILE_OK,
179            file_system_.OpenFileSystem());
180
181  const char kPath0[] = "dir a";
182  const char kPath1[] = "dir a/dir";   // child of kPath0
183  const char kPath2[] = "dir a/file";  // child of kPath0
184  const char kPath3[] = "dir b";
185
186  // Do some operations.
187  EXPECT_EQ(base::File::FILE_OK,
188            file_system_.CreateDirectory(URL(kPath0)));  // Creates a dir.
189  EXPECT_EQ(base::File::FILE_OK,
190            file_system_.CreateDirectory(URL(kPath1)));  // Creates another.
191  EXPECT_EQ(base::File::FILE_OK,
192            file_system_.CreateFile(URL(kPath2)));       // Creates a file.
193  EXPECT_EQ(base::File::FILE_OK,
194            file_system_.TruncateFile(URL(kPath2), 1));  // Modifies the file.
195  EXPECT_EQ(base::File::FILE_OK,
196            file_system_.TruncateFile(URL(kPath2), 2));  // Modifies it again.
197
198  FileSystemURLSet urls;
199  file_system_.GetChangedURLsInTracker(&urls);
200
201  EXPECT_EQ(3U, urls.size());
202  EXPECT_TRUE(ContainsKey(urls, URL(kPath0)));
203  EXPECT_TRUE(ContainsKey(urls, URL(kPath1)));
204  EXPECT_TRUE(ContainsKey(urls, URL(kPath2)));
205
206  VerifyAndClearChange(URL(kPath0),
207                       FileChange(FileChange::FILE_CHANGE_ADD_OR_UPDATE,
208                                  sync_file_system::SYNC_FILE_TYPE_DIRECTORY));
209  VerifyAndClearChange(URL(kPath1),
210                       FileChange(FileChange::FILE_CHANGE_ADD_OR_UPDATE,
211                                  sync_file_system::SYNC_FILE_TYPE_DIRECTORY));
212  VerifyAndClearChange(URL(kPath2),
213                       FileChange(FileChange::FILE_CHANGE_ADD_OR_UPDATE,
214                                  sync_file_system::SYNC_FILE_TYPE_FILE));
215
216  // Creates and removes a same directory.
217  EXPECT_EQ(base::File::FILE_OK,
218            file_system_.CreateDirectory(URL(kPath3)));
219  EXPECT_EQ(base::File::FILE_OK,
220            file_system_.Remove(URL(kPath3), false /* recursive */));
221
222  // The changes will be offset.
223  urls.clear();
224  file_system_.GetChangedURLsInTracker(&urls);
225  EXPECT_TRUE(urls.empty());
226
227  // Recursively removes the kPath0 directory.
228  EXPECT_EQ(base::File::FILE_OK,
229            file_system_.Remove(URL(kPath0), true /* recursive */));
230
231  urls.clear();
232  file_system_.GetChangedURLsInTracker(&urls);
233
234  // kPath0 and its all chidren (kPath1 and kPath2) must have been deleted.
235  EXPECT_EQ(3U, urls.size());
236  EXPECT_TRUE(ContainsKey(urls, URL(kPath0)));
237  EXPECT_TRUE(ContainsKey(urls, URL(kPath1)));
238  EXPECT_TRUE(ContainsKey(urls, URL(kPath2)));
239
240  VerifyAndClearChange(URL(kPath0),
241                       FileChange(FileChange::FILE_CHANGE_DELETE,
242                                  sync_file_system::SYNC_FILE_TYPE_DIRECTORY));
243  VerifyAndClearChange(URL(kPath1),
244                       FileChange(FileChange::FILE_CHANGE_DELETE,
245                                  sync_file_system::SYNC_FILE_TYPE_DIRECTORY));
246  VerifyAndClearChange(URL(kPath2),
247                       FileChange(FileChange::FILE_CHANGE_DELETE,
248                                  sync_file_system::SYNC_FILE_TYPE_FILE));
249}
250
251// Make sure directory operation is disabled (when it's configured so).
252TEST_F(SyncableFileSystemTest, DisableDirectoryOperations) {
253  ScopedDisableSyncFSV2 scoped_disable_v2;
254
255  EXPECT_EQ(base::File::FILE_OK,
256            file_system_.OpenFileSystem());
257
258  // Try some directory operations (which should fail).
259  EXPECT_EQ(base::File::FILE_ERROR_INVALID_OPERATION,
260            file_system_.CreateDirectory(URL("dir")));
261
262  // Set up another (non-syncable) local file system.
263  SandboxFileSystemTestHelper other_file_system_(
264      GURL("http://foo.com/"), fileapi::kFileSystemTypeTemporary);
265  other_file_system_.SetUp(file_system_.file_system_context());
266
267  // Create directory '/a' and file '/a/b' in the other file system.
268  const FileSystemURL kSrcDir = other_file_system_.CreateURLFromUTF8("/a");
269  const FileSystemURL kSrcChild = other_file_system_.CreateURLFromUTF8("/a/b");
270
271  EXPECT_EQ(base::File::FILE_OK,
272            content::AsyncFileTestHelper::CreateDirectory(
273                other_file_system_.file_system_context(), kSrcDir));
274  EXPECT_EQ(base::File::FILE_OK,
275            content::AsyncFileTestHelper::CreateFile(
276                other_file_system_.file_system_context(), kSrcChild));
277
278  // Now try copying the directory into the syncable file system, which should
279  // fail if directory operation is disabled. (http://crbug.com/161442)
280  EXPECT_NE(base::File::FILE_OK,
281            file_system_.Copy(kSrcDir, URL("dest")));
282
283  other_file_system_.TearDown();
284}
285
286}  // namespace sync_file_system
287