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