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
3 // found in the LICENSE file.
4
5 #include "webkit/common/blob/shareable_file_reference.h"
6
7 #include "base/file_util.h"
8 #include "base/files/scoped_temp_dir.h"
9 #include "base/message_loop/message_loop_proxy.h"
10 #include "base/run_loop.h"
11 #include "testing/gtest/include/gtest/gtest.h"
12
13 namespace webkit_blob {
14
TEST(ShareableFileReferenceTest,TestReferences)15 TEST(ShareableFileReferenceTest, TestReferences) {
16 base::MessageLoop message_loop;
17 scoped_refptr<base::MessageLoopProxy> loop_proxy =
18 base::MessageLoopProxy::current();
19 base::ScopedTempDir temp_dir;
20 ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
21
22 // Create a file.
23 base::FilePath file;
24 base::CreateTemporaryFileInDir(temp_dir.path(), &file);
25 EXPECT_TRUE(base::PathExists(file));
26
27 // Create a first reference to that file.
28 scoped_refptr<ShareableFileReference> reference1;
29 reference1 = ShareableFileReference::Get(file);
30 EXPECT_FALSE(reference1.get());
31 reference1 = ShareableFileReference::GetOrCreate(
32 file, ShareableFileReference::DELETE_ON_FINAL_RELEASE, loop_proxy.get());
33 EXPECT_TRUE(reference1.get());
34 EXPECT_TRUE(file == reference1->path());
35
36 // Get a second reference to that file.
37 scoped_refptr<ShareableFileReference> reference2;
38 reference2 = ShareableFileReference::Get(file);
39 EXPECT_EQ(reference1.get(), reference2.get());
40 reference2 = ShareableFileReference::GetOrCreate(
41 file, ShareableFileReference::DELETE_ON_FINAL_RELEASE, loop_proxy.get());
42 EXPECT_EQ(reference1.get(), reference2.get());
43
44 // Drop the first reference, the file and reference should still be there.
45 reference1 = NULL;
46 EXPECT_TRUE(ShareableFileReference::Get(file).get());
47 base::RunLoop().RunUntilIdle();
48 EXPECT_TRUE(base::PathExists(file));
49
50 // Drop the second reference, the file and reference should get deleted.
51 reference2 = NULL;
52 EXPECT_FALSE(ShareableFileReference::Get(file).get());
53 base::RunLoop().RunUntilIdle();
54 EXPECT_FALSE(base::PathExists(file));
55
56 // TODO(michaeln): add a test for files that aren't deletable behavior.
57 }
58
59 } // namespace webkit_blob
60