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.h" 10 #include "base/message_loop/message_loop_proxy.h" 11 #include "testing/gtest/include/gtest/gtest.h" 12 13 namespace webkit_blob { 14 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 file_util::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::MessageLoop::current()->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::MessageLoop::current()->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
