| 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 <memory> |
| 8 #include "platform/loader/fetch/FetchContext.h" | 9 #include "platform/loader/fetch/FetchContext.h" |
| 9 #include "platform/loader/fetch/FetchParameters.h" | 10 #include "platform/loader/fetch/FetchParameters.h" |
| 10 #include "platform/loader/fetch/ResourceTimingInfo.h" | 11 #include "platform/loader/fetch/ResourceTimingInfo.h" |
| 11 #include "platform/scheduler/test/fake_web_task_runner.h" | 12 #include "platform/scheduler/test/fake_web_task_runner.h" |
| 12 #include "platform/wtf/PtrUtil.h" | 13 #include "platform/wtf/PtrUtil.h" |
| 13 | 14 #include "public/platform/Platform.h" |
| 14 #include <memory> | |
| 15 | 15 |
| 16 namespace blink { | 16 namespace blink { |
| 17 | 17 |
| 18 class KURL; | 18 class KURL; |
| 19 class ResourceRequest; | 19 class ResourceRequest; |
| 20 class WebTaskRunner; | 20 class WebTaskRunner; |
| 21 struct ResourceLoaderOptions; | 21 struct ResourceLoaderOptions; |
| 22 | 22 |
| 23 // Mocked FetchContext for testing. | 23 // Mocked FetchContext for testing. |
| 24 class MockFetchContext : public FetchContext { | 24 class MockFetchContext : public FetchContext { |
| (...skipping 41 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 66 } | 66 } |
| 67 bool ShouldLoadNewResource(Resource::Type) const override { | 67 bool ShouldLoadNewResource(Resource::Type) const override { |
| 68 return load_policy_ == kShouldLoadNewResource; | 68 return load_policy_ == kShouldLoadNewResource; |
| 69 } | 69 } |
| 70 RefPtr<WebTaskRunner> LoadingTaskRunner() const override { return runner_; } | 70 RefPtr<WebTaskRunner> LoadingTaskRunner() const override { return runner_; } |
| 71 bool IsLoadComplete() const override { return complete_; } | 71 bool IsLoadComplete() const override { return complete_; } |
| 72 void AddResourceTiming( | 72 void AddResourceTiming( |
| 73 const ResourceTimingInfo& resource_timing_info) override { | 73 const ResourceTimingInfo& resource_timing_info) override { |
| 74 transfer_size_ = resource_timing_info.TransferSize(); | 74 transfer_size_ = resource_timing_info.TransferSize(); |
| 75 } | 75 } |
| 76 std::unique_ptr<WebURLLoader> CreateURLLoader() override { |
| 77 return Platform::Current()->CreateURLLoader(); |
| 78 } |
| 76 | 79 |
| 77 private: | 80 private: |
| 78 MockFetchContext(LoadPolicy load_policy, RefPtr<WebTaskRunner> task_runner) | 81 MockFetchContext(LoadPolicy load_policy, RefPtr<WebTaskRunner> task_runner) |
| 79 : load_policy_(load_policy), | 82 : load_policy_(load_policy), |
| 80 runner_(task_runner ? std::move(task_runner) | 83 runner_(task_runner ? std::move(task_runner) |
| 81 : AdoptRef(new scheduler::FakeWebTaskRunner)), | 84 : AdoptRef(new scheduler::FakeWebTaskRunner)), |
| 82 complete_(false), | 85 complete_(false), |
| 83 transfer_size_(-1) {} | 86 transfer_size_(-1) {} |
| 84 | 87 |
| 85 enum LoadPolicy load_policy_; | 88 enum LoadPolicy load_policy_; |
| 86 RefPtr<WebTaskRunner> runner_; | 89 RefPtr<WebTaskRunner> runner_; |
| 87 bool complete_; | 90 bool complete_; |
| 88 long long transfer_size_; | 91 long long transfer_size_; |
| 89 }; | 92 }; |
| 90 | 93 |
| 91 } // namespace blink | 94 } // namespace blink |
| 92 | 95 |
| 93 #endif | 96 #endif |
| OLD | NEW |