content/browser/shareable_file_reference_unittest.cc

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