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
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62
   63
   64
   65
   66
   67
   68
   69
   70
   71
   72
   73
   74
   75
   76
   77
   78
   79
   80
   81
   82
   83
   84
   85
   86
   87
   88
   89
   90
   91
   92
   93
   94
   95
   96
   97
   98
   99
  100
  101
  102
  103
  104
  105
  106
  107
  108
  109
  110
  111
  112
  113
  114
  115
  116
  117
  118


content / browser / browsing_data / shared_storage_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/shared_storage_clear_site_data_tester.h"

#include <string>
#include <vector>

#include "base/strings/strcat.h"
#include "base/strings/string_number_conversions.h"
#include "base/test/test_future.h"
#include "components/services/storage/public/mojom/storage_usage_info.mojom.h"
#include "components/services/storage/shared_storage/shared_storage_database.h"
#include "components/services/storage/shared_storage/shared_storage_manager.h"
#include "content/browser/storage_partition_impl.h"
#include "content/public/browser/storage_partition.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace content {

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

void SharedStorageClearSiteDataTester::AddSharedStorageEntry(
    url::Origin origin,
    std::u16string key,
    std::u16string value) {
  DCHECK_CURRENTLY_ON(BrowserThread::UI);

  auto* shared_storage_manager =
      storage_partition_impl_->GetSharedStorageManager();
  DCHECK(shared_storage_manager);

  base::test::TestFuture<storage::SharedStorageDatabase::OperationResult>
      future;
  shared_storage_manager->Set(
      std::move(origin), std::move(key), std::move(value), future.GetCallback(),
      storage::SharedStorageDatabase::SetBehavior::kDefault);
  EXPECT_EQ(storage::SharedStorageDatabase::OperationResult::kSet,
            future.Get());
}

void SharedStorageClearSiteDataTester::AddConsecutiveSharedStorageEntries(
    url::Origin origin,
    std::u16string key,
    std::u16string value,
    size_t total_to_add) {
  for (size_t i = 0; i < total_to_add; ++i) {
    AddSharedStorageEntry(origin,
                          base::StrCat({key, base::NumberToString16(i)}),
                          base::StrCat({value, base::NumberToString16(i)}));
  }
}

std::vector<url::Origin>
SharedStorageClearSiteDataTester::GetSharedStorageOrigins() {
  DCHECK_CURRENTLY_ON(BrowserThread::UI);

  auto* shared_storage_manager =
      storage_partition_impl_->GetSharedStorageManager();
  DCHECK(shared_storage_manager);

  base::test::TestFuture<std::vector<storage::mojom::StorageUsageInfoPtr>>
      future;
  shared_storage_manager->FetchOrigins(future.GetCallback());
  auto infos = future.Take();

  std::vector<url::Origin> origins;
  for (const auto& info : infos)
    origins.push_back(info->storage_key.origin());

  return origins;
}

int SharedStorageClearSiteDataTester::GetSharedStorageNumBytesForOrigin(
    url::Origin origin) {
  DCHECK_CURRENTLY_ON(BrowserThread::UI);

  auto* shared_storage_manager =
      storage_partition_impl_->GetSharedStorageManager();
  DCHECK(shared_storage_manager);

  base::test::TestFuture<std::vector<storage::mojom::StorageUsageInfoPtr>>
      future;
  shared_storage_manager->FetchOrigins(future.GetCallback());
  auto infos = future.Take();

  for (const auto& info : infos) {
    if (info->storage_key.origin() == origin)
      return info->total_size_bytes;
  }

  return 0;
}

int SharedStorageClearSiteDataTester::GetSharedStorageTotalBytes() {
  DCHECK_CURRENTLY_ON(BrowserThread::UI);

  auto* shared_storage_manager =
      storage_partition_impl_->GetSharedStorageManager();
  DCHECK(shared_storage_manager);

  base::test::TestFuture<std::vector<storage::mojom::StorageUsageInfoPtr>>
      future;
  shared_storage_manager->FetchOrigins(future.GetCallback());
  auto infos = future.Take();

  int num_bytes = 0;
  for (const auto& info : infos)
    num_bytes += info->total_size_bytes;

  return num_bytes;
}

}  // namespace content