// 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::unique_ptr<std::unordered_set<base::FilePath>> active_paths(
      new std::unordered_set<base::FilePath>);

  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) {
  const auto kOnDiskConfig = content::StoragePartitionConfig::Create(
      "foo", /*partition_name=*/"", /*in_memory=*/false);

  base::test::ScopedFeatureList f;
  f.InitAndDisableFeature(blink::features::kAppCache);
  BrowserTaskEnvironment task_environment;
  TestBrowserContext browser_context;
  base::FilePath appcache_path;

  {
    // 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));
  }
}

}  // namespace content