src/webkit/common/blob/shareable_file_reference_unittest.cc

   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
  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