OLD | NEW |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef MockFetchContext_h | 5 #ifndef MockFetchContext_h |
6 #define MockFetchContext_h | 6 #define MockFetchContext_h |
7 | 7 |
8 #include "core/fetch/FetchContext.h" | 8 #include "core/fetch/FetchContext.h" |
9 #include "core/fetch/FetchRequest.h" | 9 #include "core/fetch/FetchRequest.h" |
10 #include "platform/network/ResourceTimingInfo.h" | 10 #include "platform/network/ResourceTimingInfo.h" |
(...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
42 const ResourceRequest&, | 42 const ResourceRequest&, |
43 const KURL&, | 43 const KURL&, |
44 const ResourceLoaderOptions&, | 44 const ResourceLoaderOptions&, |
45 bool forPreload, | 45 bool forPreload, |
46 FetchRequest::OriginRestriction) const override { | 46 FetchRequest::OriginRestriction) const override { |
47 return true; | 47 return true; |
48 } | 48 } |
49 bool shouldLoadNewResource(Resource::Type) const override { | 49 bool shouldLoadNewResource(Resource::Type) const override { |
50 return m_loadPolicy == kShouldLoadNewResource; | 50 return m_loadPolicy == kShouldLoadNewResource; |
51 } | 51 } |
52 WebTaskRunner* loadingTaskRunner() const override { return m_runner.get(); } | 52 RefPtr<WebTaskRunner> loadingTaskRunner() const override { return m_runner; } |
53 | 53 |
54 void setCachePolicy(CachePolicy policy) { m_policy = policy; } | 54 void setCachePolicy(CachePolicy policy) { m_policy = policy; } |
55 CachePolicy getCachePolicy() const override { return m_policy; } | 55 CachePolicy getCachePolicy() const override { return m_policy; } |
56 void setLoadComplete(bool complete) { m_complete = complete; } | 56 void setLoadComplete(bool complete) { m_complete = complete; } |
57 bool isLoadComplete() const override { return m_complete; } | 57 bool isLoadComplete() const override { return m_complete; } |
58 | 58 |
59 void addResourceTiming( | 59 void addResourceTiming( |
60 const ResourceTimingInfo& resourceTimingInfo) override { | 60 const ResourceTimingInfo& resourceTimingInfo) override { |
61 m_transferSize = resourceTimingInfo.transferSize(); | 61 m_transferSize = resourceTimingInfo.transferSize(); |
62 } | 62 } |
63 long long getTransferSize() const { return m_transferSize; } | 63 long long getTransferSize() const { return m_transferSize; } |
64 | 64 |
65 private: | 65 private: |
66 MockFetchContext(LoadPolicy loadPolicy) | 66 MockFetchContext(LoadPolicy loadPolicy) |
67 : m_loadPolicy(loadPolicy), | 67 : m_loadPolicy(loadPolicy), |
68 m_policy(CachePolicyVerify), | 68 m_policy(CachePolicyVerify), |
69 m_runner(wrapUnique(new scheduler::FakeWebTaskRunner)), | 69 m_runner(adoptRef(new scheduler::FakeWebTaskRunner)), |
70 m_complete(false), | 70 m_complete(false), |
71 m_transferSize(-1) {} | 71 m_transferSize(-1) {} |
72 | 72 |
73 enum LoadPolicy m_loadPolicy; | 73 enum LoadPolicy m_loadPolicy; |
74 CachePolicy m_policy; | 74 CachePolicy m_policy; |
75 std::unique_ptr<scheduler::FakeWebTaskRunner> m_runner; | 75 RefPtr<scheduler::FakeWebTaskRunner> m_runner; |
76 bool m_complete; | 76 bool m_complete; |
77 long long m_transferSize; | 77 long long m_transferSize; |
78 }; | 78 }; |
79 | 79 |
80 } // namespace blink | 80 } // namespace blink |
81 | 81 |
82 #endif | 82 #endif |
OLD | NEW |