1 files changed, 119 insertions, 0 deletions
diff --git a/chrome/browser/chromeos/drive/file_write_watcher_unittest.cc b/chrome/browser/chromeos/drive/file_write_watcher_unittest.cc
new file mode 100644
index 0000000..23be2a8
--- /dev/null
+++ b/chrome/browser/chromeos/drive/file_write_watcher_unittest.cc
@@ -0,0 +1,119 @@
+// Copyright 2013 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "chrome/browser/chromeos/drive/file_write_watcher.h"
+
+#include <set>
+
+#include "base/bind.h"
+#include "base/file_util.h"
+#include "base/files/scoped_temp_dir.h"
+#include "base/run_loop.h"
+#include "chrome/browser/chromeos/drive/file_system/operation_observer.h"
+#include "content/public/test/test_browser_thread_bundle.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace drive {
+namespace internal {
+
+namespace {
+
+class TestObserver : public file_system::OperationObserver {
+ public:
+  // After all the resource_ids in |expected_upload| are notified for the
+  // need of uploading, runs |quit_closure|. Also checks that each id is
+  // notified only once.
+  TestObserver(const std::set<std::string>& expected_upload,
+               const base::Closure& quit_closure)
+      : expected_upload_(expected_upload),
+        quit_closure_(quit_closure) {
+  }
+
+  virtual void OnDirectoryChangedByOperation(
+      const base::FilePath& path) OVERRIDE {}
+
+  virtual void OnCacheFileUploadNeededByOperation(
+      const std::string& resource_id) OVERRIDE {
+    EXPECT_EQ(1U, expected_upload_.count(resource_id)) << resource_id;
+    expected_upload_.erase(resource_id);
+    if (expected_upload_.empty())
+      quit_closure_.Run();
+  }
+
+ private:
+  std::set<std::string> expected_upload_;
+  base::Closure quit_closure_;
+};
+
+// Writes something on the file at |path|.
+void WriteSomethingAfterStartWatch(const base::FilePath& path,
+                                   bool watch_success) {
+  EXPECT_TRUE(watch_success) << path.value();
+
+  const char kDummy[] = "hello";
+  ASSERT_TRUE(file_util::WriteFile(path, kDummy, arraysize(kDummy)));
+}
+
+class FileWriteWatcherTest : public testing::Test {
+ protected:
+  // The test requires UI thread (FileWriteWatcher DCHECKs that its public
+  // interface is running on UI thread) and FILE thread (Linux version of
+  // base::FilePathWatcher needs to live on an IOAllowed thread with TYPE_IO,
+  // which is FILE thread in the production environment).
+  //
+  // By using the IO_MAINLOOP test thread bundle, the main thread is used
+  // both as UI and FILE thread, with TYPE_IO message loop.
+  FileWriteWatcherTest()
+      : thread_bundle_(content::TestBrowserThreadBundle::IO_MAINLOOP) {
+  }
+
+  virtual void SetUp() OVERRIDE {
+    ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
+  }
+
+  base::FilePath GetTempPath(const std::string& name) {
+    return temp_dir_.path().Append(name);
+  }
+
+ private:
+  content::TestBrowserThreadBundle thread_bundle_;
+  base::ScopedTempDir temp_dir_;
+};
+
+}  // namespace
+
+TEST_F(FileWriteWatcherTest, WatchThreeFiles) {
+  std::set<std::string> expected;
+  expected.insert("1");
+  expected.insert("2");
+  expected.insert("3");
+
+  base::RunLoop loop;
+  TestObserver observer(expected, loop.QuitClosure());
+
+  // Set up the watcher.
+  FileWriteWatcher watcher(&observer);
+  watcher.DisableDelayForTesting();
+
+  // Start watching and running.
+  base::FilePath path1 = GetTempPath("foo.txt");
+  base::FilePath path2 = GetTempPath("bar.png");
+  base::FilePath path3 = GetTempPath("buz.doc");
+  base::FilePath path4 = GetTempPath("mya.mp3");
+  watcher.StartWatch(path1, "1",
+                     base::Bind(&WriteSomethingAfterStartWatch, path1));
+  watcher.StartWatch(path2, "2",
+                     base::Bind(&WriteSomethingAfterStartWatch, path2));
+  watcher.StartWatch(path3, "3",
+                     base::Bind(&WriteSomethingAfterStartWatch, path3));
+
+  // Unwatched write. It shouldn't be notified.
+  WriteSomethingAfterStartWatch(path4, true);
+
+  // The loop should quit if all the three paths are notified to be written.
+  loop.Run();
+}
+
+}  // namespace internal
+}  // namespace drive