1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49

content / browser / browsing_data / storage_bucket_clear_site_data_tester.cc [blame]

// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "content/browser/browsing_data/storage_bucket_clear_site_data_tester.h"

#include <set>
#include <string>
#include <utility>

#include "base/functional/callback.h"
#include "base/task/sequenced_task_runner.h"
#include "components/services/storage/public/cpp/buckets/bucket_info.h"
#include "components/services/storage/public/cpp/quota_error_or.h"
#include "content/browser/storage_partition_impl.h"
#include "content/public/browser/storage_partition.h"
#include "storage/browser/quota/quota_manager_proxy.h"
#include "third_party/blink/public/common/storage_key/storage_key.h"
#include "third_party/blink/public/mojom/quota/quota_types.mojom.h"

namespace content {

StorageBucketClearSiteDataTester::StorageBucketClearSiteDataTester(
    StoragePartition* storage_partition)
    : storage_partition_impl_(
          static_cast<StoragePartitionImpl*>(storage_partition)) {}

void StorageBucketClearSiteDataTester::CreateBucketForTesting(
    const blink::StorageKey& storage_key,
    const std::string& bucket_name,
    base::OnceCallback<void(storage::QuotaErrorOr<storage::BucketInfo>)>
        callback) {
  storage_partition_impl_->GetQuotaManagerProxy()
      ->CreateBucketForTesting(  // IN-TEST
          storage_key, bucket_name, blink::mojom::StorageType::kTemporary,
          base::SequencedTaskRunner::GetCurrentDefault(), std::move(callback));
}

void StorageBucketClearSiteDataTester::GetBucketsForStorageKey(
    const blink::StorageKey& storage_key,
    base::OnceCallback<
        void(storage::QuotaErrorOr<std::set<storage::BucketInfo>>)> callback) {
  storage_partition_impl_->GetQuotaManagerProxy()->GetBucketsForStorageKey(
      storage_key, blink::mojom::StorageType::kTemporary,
      /*delete_expired=*/false, base::SequencedTaskRunner::GetCurrentDefault(),
      std::move(callback));
}

}  // namespace content