1 files changed, 61 insertions, 0 deletions
diff --git a/chromium/content/common/shareable_file_reference_unittest.cc b/chromium/content/common/shareable_file_reference_unittest.cc
new file mode 100644
index 00000000000..ca6218a8d1d
--- /dev/null
+++ b/chromium/content/common/shareable_file_reference_unittest.cc
@@ -0,0 +1,61 @@
+// Copyright 2014 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 "webkit/common/blob/shareable_file_reference.h"
+
+#include "base/file_util.h"
+#include "base/files/scoped_temp_dir.h"
+#include "base/message_loop/message_loop_proxy.h"
+#include "base/run_loop.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+using webkit_blob::ShareableFileReference;
+
+namespace content {
+
+TEST(ShareableFileReferenceTest, TestReferences) {
+  base::MessageLoop message_loop;
+  scoped_refptr<base::MessageLoopProxy> loop_proxy =
+      base::MessageLoopProxy::current();
+  base::ScopedTempDir temp_dir;
+  ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
+
+  // Create a file.
+  base::FilePath file;
+  base::CreateTemporaryFileInDir(temp_dir.path(), &file);
+  EXPECT_TRUE(base::PathExists(file));
+
+  // Create a first reference to that file.
+  scoped_refptr<ShareableFileReference> reference1;
+  reference1 = ShareableFileReference::Get(file);
+  EXPECT_FALSE(reference1.get());
+  reference1 = ShareableFileReference::GetOrCreate(
+      file, ShareableFileReference::DELETE_ON_FINAL_RELEASE, loop_proxy.get());
+  EXPECT_TRUE(reference1.get());
+  EXPECT_TRUE(file == reference1->path());
+
+  // Get a second reference to that file.
+  scoped_refptr<ShareableFileReference> reference2;
+  reference2 = ShareableFileReference::Get(file);
+  EXPECT_EQ(reference1.get(), reference2.get());
+  reference2 = ShareableFileReference::GetOrCreate(
+      file, ShareableFileReference::DELETE_ON_FINAL_RELEASE, loop_proxy.get());
+  EXPECT_EQ(reference1.get(), reference2.get());
+
+  // Drop the first reference, the file and reference should still be there.
+  reference1 = NULL;
+  EXPECT_TRUE(ShareableFileReference::Get(file).get());
+  base::RunLoop().RunUntilIdle();
+  EXPECT_TRUE(base::PathExists(file));
+
+  // Drop the second reference, the file and reference should get deleted.
+  reference2 = NULL;
+  EXPECT_FALSE(ShareableFileReference::Get(file).get());
+  base::RunLoop().RunUntilIdle();
+  EXPECT_FALSE(base::PathExists(file));
+
+  // TODO(michaeln): add a test for files that aren't deletable behavior.
+}
+
+}  // namespace content