Chromium Code Reviews| 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 17 matching lines...) Expand all Loading... | |
| 28 enum LoadPolicy { | 28 enum LoadPolicy { |
| 29 kShouldLoadNewResource, | 29 kShouldLoadNewResource, |
| 30 kShouldNotLoadNewResource, | 30 kShouldNotLoadNewResource, |
| 31 }; | 31 }; |
| 32 static MockFetchContext* create(LoadPolicy loadPolicy) { | 32 static MockFetchContext* create(LoadPolicy loadPolicy) { |
| 33 return new MockFetchContext(loadPolicy); | 33 return new MockFetchContext(loadPolicy); |
| 34 } | 34 } |
| 35 | 35 |
| 36 ~MockFetchContext() override {} | 36 ~MockFetchContext() override {} |
| 37 | 37 |
| 38 void setCachePolicy(CachePolicy policy) { m_policy = policy; } | |
| 39 void setLoadComplete(bool complete) { m_complete = complete; } | |
| 40 long long getTransferSize() const { return m_transferSize; } | |
| 41 void runLoadingTaskRunnerUntilIdle() { m_runner->runUntilIdle(); } | |
| 42 | |
| 43 // FetchContext: | |
| 38 bool allowImage(bool imagesEnabled, const KURL&) const override { | 44 bool allowImage(bool imagesEnabled, const KURL&) const override { |
| 39 return true; | 45 return true; |
| 40 } | 46 } |
| 41 ResourceRequestBlockedReason canRequest( | 47 ResourceRequestBlockedReason canRequest( |
| 42 Resource::Type, | 48 Resource::Type, |
| 43 const ResourceRequest&, | 49 const ResourceRequest&, |
| 44 const KURL&, | 50 const KURL&, |
| 45 const ResourceLoaderOptions&, | 51 const ResourceLoaderOptions&, |
| 46 bool forPreload, | 52 bool forPreload, |
| 47 FetchRequest::OriginRestriction) const override { | 53 FetchRequest::OriginRestriction) const override { |
| 48 return ResourceRequestBlockedReason::None; | 54 return ResourceRequestBlockedReason::None; |
| 49 } | 55 } |
| 50 bool shouldLoadNewResource(Resource::Type) const override { | 56 bool shouldLoadNewResource(Resource::Type) const override { |
| 51 return m_loadPolicy == kShouldLoadNewResource; | 57 return m_loadPolicy == kShouldLoadNewResource; |
| 52 } | 58 } |
| 53 RefPtr<WebTaskRunner> loadingTaskRunner() const override { return m_runner; } | 59 RefPtr<WebTaskRunner> loadingTaskRunner() const override { return m_runner; } |
|
kinuko
2017/01/19 18:40:18
Could you help me understand... so the task runner
Takashi Toyoshima
2017/01/20 04:13:16
The right approach to obtain the task runner in fe
kinuko
2017/01/20 05:00:33
When you say MemoryCache do you mean the task obse
| |
| 54 | |
| 55 void setCachePolicy(CachePolicy policy) { m_policy = policy; } | |
| 56 CachePolicy getCachePolicy() const override { return m_policy; } | 60 CachePolicy getCachePolicy() const override { return m_policy; } |
| 57 void setLoadComplete(bool complete) { m_complete = complete; } | |
| 58 bool isLoadComplete() const override { return m_complete; } | 61 bool isLoadComplete() const override { return m_complete; } |
| 59 | |
| 60 void addResourceTiming( | 62 void addResourceTiming( |
| 61 const ResourceTimingInfo& resourceTimingInfo) override { | 63 const ResourceTimingInfo& resourceTimingInfo) override { |
| 62 m_transferSize = resourceTimingInfo.transferSize(); | 64 m_transferSize = resourceTimingInfo.transferSize(); |
| 63 } | 65 } |
| 64 long long getTransferSize() const { return m_transferSize; } | |
| 65 | 66 |
| 66 private: | 67 private: |
| 67 MockFetchContext(LoadPolicy loadPolicy) | 68 MockFetchContext(LoadPolicy loadPolicy) |
| 68 : m_loadPolicy(loadPolicy), | 69 : m_loadPolicy(loadPolicy), |
| 69 m_policy(CachePolicyVerify), | 70 m_policy(CachePolicyVerify), |
| 70 m_runner(adoptRef(new scheduler::FakeWebTaskRunner)), | 71 m_runner(adoptRef(new scheduler::FakeWebTaskRunner)), |
| 71 m_complete(false), | 72 m_complete(false), |
| 72 m_transferSize(-1) {} | 73 m_transferSize(-1) {} |
| 73 | 74 |
| 74 enum LoadPolicy m_loadPolicy; | 75 enum LoadPolicy m_loadPolicy; |
| 75 CachePolicy m_policy; | 76 CachePolicy m_policy; |
| 76 RefPtr<scheduler::FakeWebTaskRunner> m_runner; | 77 RefPtr<scheduler::FakeWebTaskRunner> m_runner; |
| 77 bool m_complete; | 78 bool m_complete; |
| 78 long long m_transferSize; | 79 long long m_transferSize; |
| 79 }; | 80 }; |
| 80 | 81 |
| 81 } // namespace blink | 82 } // namespace blink |
| 82 | 83 |
| 83 #endif | 84 #endif |
| OLD | NEW |