content/browser/storage_partition_impl_map_unittest.cc

// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "content/browser/storage_partition_impl_map.h"

#include <unordered_set>
#include <utility>

#include "base/files/file_util.h"
#include "base/run_loop.h"
#include "base/test/scoped_feature_list.h"
#include "content/public/browser/browser_thread.h"
#include "content/public/common/content_constants.h"
#include "content/public/test/browser_task_environment.h"
#include "content/public/test/test_browser_context.h"
#include "content/public/test/test_utils.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/blink/public/common/features.h"

namespace content {

TEST(StoragePartitionImplMapTest, GarbageCollect) {
  BrowserTaskEnvironment task_environment;
  TestBrowserContext browser_context;
  StoragePartitionImplMap storage_partition_impl_map(&browser_context);

  std::unordered_set<base::FilePath> active_paths;

  base::FilePath active_path = browser_context.GetPath().Append(
      StoragePartitionImplMap::GetStoragePartitionPath(
          "active", std::string()));
  ASSERT_TRUE(base::CreateDirectory(active_path));
  active_paths.insert(active_path);

  base::FilePath inactive_path = browser_context.GetPath().Append(
      StoragePartitionImplMap::GetStoragePartitionPath(
          "inactive", std::string()));
  ASSERT_TRUE(base::CreateDirectory(inactive_path));

  base::RunLoop run_loop;
  storage_partition_impl_map.GarbageCollect(std::move(active_paths),
                                            run_loop.QuitClosure());
  run_loop.Run();
  content::RunAllPendingInMessageLoop(content::BrowserThread::IO);

  EXPECT_TRUE(base::PathExists(active_path));
  EXPECT_FALSE(base::PathExists(inactive_path));
}

TEST(StoragePartitionImplMapTest, AppCacheCleanup) {
  BrowserTaskEnvironment task_environment;
  TestBrowserContext browser_context;
  base::FilePath appcache_path;

  const auto kOnDiskConfig = content::StoragePartitionConfig::Create(
      &browser_context, "foo", /*partition_name=*/"", /*in_memory=*/false);

  {
    // Creating the partition in the map also does the deletion, so
    // create it once, so we can find out what path the partition
    // with this name is.
    StoragePartitionImplMap map(&browser_context);

    auto* partition = map.Get(kOnDiskConfig, true);
    appcache_path = partition->GetPath().Append(kAppCacheDirname);

    task_environment.RunUntilIdle();
  }

  // Create an AppCache directory that would have existed.
  EXPECT_FALSE(base::PathExists(appcache_path));
  EXPECT_TRUE(base::CreateDirectory(appcache_path));

  {
    StoragePartitionImplMap map(&browser_context);
    auto* partition = map.Get(kOnDiskConfig, true);

    ASSERT_EQ(appcache_path, partition->GetPath().Append(kAppCacheDirname));

    task_environment.RunUntilIdle();

    // Verify that creating this partition deletes any AppCache directory it may
    // have had.
    EXPECT_FALSE(base::PathExists(appcache_path));
  }
}

TEST(StoragePartitionImplMapTest, Dispose) {
  BrowserTaskEnvironment task_environment;
  TestBrowserContext browser_context;
  StoragePartitionImplMap map(&browser_context);

  // Start with no partitions.
  ASSERT_EQ(map.size(), 0u);

  // Create 1 in-memory partition.
  const auto kInMemoryConfig = content::StoragePartitionConfig::Create(
      &browser_context, "foo", /*partition_name=*/"", /*in_memory=*/true);
  auto* partition = map.Get(kInMemoryConfig, /*can_create=*/true);
  ASSERT_TRUE(partition);

  // Dispose the in-memory partition.
  map.DisposeInMemory(partition);
  EXPECT_EQ(map.size(), 0u);

  // No op to dispose the already disposed partition.
  map.DisposeInMemory(partition);
  EXPECT_EQ(map.size(), 0u);

  // No op to dispose nullptr.
  map.DisposeInMemory(nullptr);
  EXPECT_EQ(map.size(), 0u);

#if !BUILDFLAG(IS_ANDROID) && DCHECK_IS_ON()
  // Death test for non-android and when DCHECK is on.
  // Disposing an on-disk storage partition is not supported.
  const auto kOnDiskConfig = content::StoragePartitionConfig::Create(
      &browser_context, "foo", /*partition_name=*/"", /*in_memory=*/false);
  auto* on_disk_partition = map.Get(kOnDiskConfig, /*can_create=*/true);
  EXPECT_DEATH_IF_SUPPORTED(map.DisposeInMemory(on_disk_partition), "");
#endif
}

}  // namespace content