content/common/shareable_file_reference_unittest.cc

   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
  13 using storage::ShareableFileReference;
  14
  15 namespace content {
  16
  17 TEST(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