| 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 "platform/loader/fetch/FetchContext.h" | 8 #include "platform/loader/fetch/FetchContext.h" |
| 9 #include "platform/loader/fetch/FetchRequest.h" | 9 #include "platform/loader/fetch/FetchRequest.h" |
| 10 #include "platform/network/ResourceTimingInfo.h" | 10 #include "platform/network/ResourceTimingInfo.h" |
| (...skipping 20 matching lines...) Expand all Loading... |
| 31 // FetchTestingPlatformSupport's WebTaskRunner. Probably, MockFetchContext | 31 // FetchTestingPlatformSupport's WebTaskRunner. Probably, MockFetchContext |
| 32 // would be available only through the FetchTestingPlatformSupport in the | 32 // would be available only through the FetchTestingPlatformSupport in the |
| 33 // future. | 33 // future. |
| 34 static MockFetchContext* create(LoadPolicy loadPolicy, | 34 static MockFetchContext* create(LoadPolicy loadPolicy, |
| 35 RefPtr<WebTaskRunner> taskRunner = nullptr) { | 35 RefPtr<WebTaskRunner> taskRunner = nullptr) { |
| 36 return new MockFetchContext(loadPolicy, std::move(taskRunner)); | 36 return new MockFetchContext(loadPolicy, std::move(taskRunner)); |
| 37 } | 37 } |
| 38 | 38 |
| 39 ~MockFetchContext() override {} | 39 ~MockFetchContext() override {} |
| 40 | 40 |
| 41 void setCachePolicy(CachePolicy policy) { m_policy = policy; } | |
| 42 void setLoadComplete(bool complete) { m_complete = complete; } | 41 void setLoadComplete(bool complete) { m_complete = complete; } |
| 43 long long getTransferSize() const { return m_transferSize; } | 42 long long getTransferSize() const { return m_transferSize; } |
| 44 | 43 |
| 45 // FetchContext: | 44 // FetchContext: |
| 46 bool allowImage(bool imagesEnabled, const KURL&) const override { | 45 bool allowImage(bool imagesEnabled, const KURL&) const override { |
| 47 return true; | 46 return true; |
| 48 } | 47 } |
| 49 ResourceRequestBlockedReason canRequest( | 48 ResourceRequestBlockedReason canRequest( |
| 50 Resource::Type, | 49 Resource::Type, |
| 51 const ResourceRequest&, | 50 const ResourceRequest&, |
| 52 const KURL&, | 51 const KURL&, |
| 53 const ResourceLoaderOptions&, | 52 const ResourceLoaderOptions&, |
| 54 SecurityViolationReportingPolicy, | 53 SecurityViolationReportingPolicy, |
| 55 FetchRequest::OriginRestriction) const override { | 54 FetchRequest::OriginRestriction) const override { |
| 56 return ResourceRequestBlockedReason::None; | 55 return ResourceRequestBlockedReason::None; |
| 57 } | 56 } |
| 58 bool shouldLoadNewResource(Resource::Type) const override { | 57 bool shouldLoadNewResource(Resource::Type) const override { |
| 59 return m_loadPolicy == kShouldLoadNewResource; | 58 return m_loadPolicy == kShouldLoadNewResource; |
| 60 } | 59 } |
| 61 RefPtr<WebTaskRunner> loadingTaskRunner() const override { return m_runner; } | 60 RefPtr<WebTaskRunner> loadingTaskRunner() const override { return m_runner; } |
| 62 CachePolicy getCachePolicy() const override { return m_policy; } | |
| 63 bool isLoadComplete() const override { return m_complete; } | 61 bool isLoadComplete() const override { return m_complete; } |
| 64 void addResourceTiming( | 62 void addResourceTiming( |
| 65 const ResourceTimingInfo& resourceTimingInfo) override { | 63 const ResourceTimingInfo& resourceTimingInfo) override { |
| 66 m_transferSize = resourceTimingInfo.transferSize(); | 64 m_transferSize = resourceTimingInfo.transferSize(); |
| 67 } | 65 } |
| 68 | 66 |
| 69 private: | 67 private: |
| 70 MockFetchContext(LoadPolicy loadPolicy, RefPtr<WebTaskRunner> taskRunner) | 68 MockFetchContext(LoadPolicy loadPolicy, RefPtr<WebTaskRunner> taskRunner) |
| 71 : m_loadPolicy(loadPolicy), | 69 : m_loadPolicy(loadPolicy), |
| 72 m_policy(CachePolicyVerify), | |
| 73 m_runner(taskRunner ? std::move(taskRunner) | 70 m_runner(taskRunner ? std::move(taskRunner) |
| 74 : adoptRef(new scheduler::FakeWebTaskRunner)), | 71 : adoptRef(new scheduler::FakeWebTaskRunner)), |
| 75 m_complete(false), | 72 m_complete(false), |
| 76 m_transferSize(-1) {} | 73 m_transferSize(-1) {} |
| 77 | 74 |
| 78 enum LoadPolicy m_loadPolicy; | 75 enum LoadPolicy m_loadPolicy; |
| 79 CachePolicy m_policy; | |
| 80 RefPtr<WebTaskRunner> m_runner; | 76 RefPtr<WebTaskRunner> m_runner; |
| 81 bool m_complete; | 77 bool m_complete; |
| 82 long long m_transferSize; | 78 long long m_transferSize; |
| 83 }; | 79 }; |
| 84 | 80 |
| 85 } // namespace blink | 81 } // namespace blink |
| 86 | 82 |
| 87 #endif | 83 #endif |
| OLD | NEW |