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
|
// Copyright 2016 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.
#ifndef MockFetchContext_h
#define MockFetchContext_h
#include "core/fetch/FetchContext.h"
#include "core/fetch/FetchRequest.h"
#include "platform/network/ResourceTimingInfo.h"
#include "platform/scheduler/test/fake_web_task_runner.h"
#include "wtf/PtrUtil.h"
#include <memory>
namespace blink {
class KURL;
class ResourceRequest;
class WebTaskRunner;
struct ResourceLoaderOptions;
// Mocked FetchContext for testing.
// TODO(toyoshim): Use this class by other unit tests that currently have own
// mocked FetchContext respectively.
class MockFetchContext : public FetchContext {
public:
enum LoadPolicy {
kShouldLoadNewResource,
kShouldNotLoadNewResource,
};
static MockFetchContext* create(LoadPolicy loadPolicy) {
return new MockFetchContext(loadPolicy);
}
~MockFetchContext() override {}
bool allowImage(bool imagesEnabled, const KURL&) const override {
return true;
}
ResourceRequestBlockedReason canRequest(
Resource::Type,
const ResourceRequest&,
const KURL&,
const ResourceLoaderOptions&,
bool forPreload,
FetchRequest::OriginRestriction) const override {
return ResourceRequestBlockedReason::None;
}
bool shouldLoadNewResource(Resource::Type) const override {
return m_loadPolicy == kShouldLoadNewResource;
}
RefPtr<WebTaskRunner> loadingTaskRunner() const override { return m_runner; }
void setCachePolicy(CachePolicy policy) { m_policy = policy; }
CachePolicy getCachePolicy() const override { return m_policy; }
void setLoadComplete(bool complete) { m_complete = complete; }
bool isLoadComplete() const override { return m_complete; }
void addResourceTiming(
const ResourceTimingInfo& resourceTimingInfo) override {
m_transferSize = resourceTimingInfo.transferSize();
}
long long getTransferSize() const { return m_transferSize; }
private:
MockFetchContext(LoadPolicy loadPolicy)
: m_loadPolicy(loadPolicy),
m_policy(CachePolicyVerify),
m_runner(adoptRef(new scheduler::FakeWebTaskRunner)),
m_complete(false),
m_transferSize(-1) {}
enum LoadPolicy m_loadPolicy;
CachePolicy m_policy;
RefPtr<scheduler::FakeWebTaskRunner> m_runner;
bool m_complete;
long long m_transferSize;
};
} // namespace blink
#endif
|
