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
|
// Copyright 2015 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/browsing_data/content/mock_cache_storage_helper.h"
#include "base/functional/callback.h"
#include "content/public/browser/storage_partition.h"
#include "content/public/browser/storage_usage_info.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/blink/public/common/storage_key/storage_key.h"
#include "url/gurl.h"
namespace browsing_data {
MockCacheStorageHelper::MockCacheStorageHelper(
content::StoragePartition* storage_partition)
: CacheStorageHelper(storage_partition) {}
MockCacheStorageHelper::~MockCacheStorageHelper() = default;
void MockCacheStorageHelper::StartFetching(FetchCallback callback) {
ASSERT_FALSE(callback.is_null());
ASSERT_TRUE(callback_.is_null());
callback_ = std::move(callback);
fetched_ = true;
}
void MockCacheStorageHelper::DeleteCacheStorage(
const blink::StorageKey& storage_key) {
ASSERT_TRUE(fetched_);
ASSERT_TRUE(storage_keys_.find(storage_key) != storage_keys_.end());
storage_keys_[storage_key] = false;
}
void MockCacheStorageHelper::AddCacheStorageSamples() {
const url::Origin kOrigin1 = url::Origin::Create(GURL("https://cshost1:1/"));
const url::Origin kOrigin2 = url::Origin::Create(GURL("https://cshost2:2/"));
content::StorageUsageInfo info1(blink::StorageKey::CreateFirstParty(kOrigin1),
1, base::Time());
response_.push_back(info1);
storage_keys_[blink::StorageKey::CreateFirstParty(kOrigin1)] = true;
content::StorageUsageInfo info2(blink::StorageKey::CreateFirstParty(kOrigin2),
2, base::Time());
response_.push_back(info2);
storage_keys_[blink::StorageKey::CreateFirstParty(kOrigin2)] = true;
}
void MockCacheStorageHelper::Notify() {
ASSERT_FALSE(callback_.is_null());
std::move(callback_).Run(response_);
}
void MockCacheStorageHelper::Reset() {
for (auto& pair : storage_keys_) {
pair.second = true;
}
}
bool MockCacheStorageHelper::AllDeleted() {
for (const auto& pair : storage_keys_) {
if (pair.second)
return false;
}
return true;
}
} // namespace browsing_data
|
