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 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
56 } | 56 } |
57 bool ShouldLoadNewResource(Resource::Type) const override { | 57 bool ShouldLoadNewResource(Resource::Type) const override { |
58 return load_policy_ == kShouldLoadNewResource; | 58 return load_policy_ == kShouldLoadNewResource; |
59 } | 59 } |
60 RefPtr<WebTaskRunner> LoadingTaskRunner() const override { return runner_; } | 60 RefPtr<WebTaskRunner> LoadingTaskRunner() const override { return runner_; } |
61 bool IsLoadComplete() const override { return complete_; } | 61 bool IsLoadComplete() const override { return complete_; } |
62 void AddResourceTiming( | 62 void AddResourceTiming( |
63 const ResourceTimingInfo& resource_timing_info) override { | 63 const ResourceTimingInfo& resource_timing_info) override { |
64 transfer_size_ = resource_timing_info.TransferSize(); | 64 transfer_size_ = resource_timing_info.TransferSize(); |
65 } | 65 } |
| 66 WebURLLoader* CreateURLLoader() override { |
| 67 return Platform::Current()->CreateURLLoader(); |
| 68 } |
66 | 69 |
67 private: | 70 private: |
68 MockFetchContext(LoadPolicy load_policy, RefPtr<WebTaskRunner> task_runner) | 71 MockFetchContext(LoadPolicy load_policy, RefPtr<WebTaskRunner> task_runner) |
69 : load_policy_(load_policy), | 72 : load_policy_(load_policy), |
70 runner_(task_runner ? std::move(task_runner) | 73 runner_(task_runner ? std::move(task_runner) |
71 : AdoptRef(new scheduler::FakeWebTaskRunner)), | 74 : AdoptRef(new scheduler::FakeWebTaskRunner)), |
72 complete_(false), | 75 complete_(false), |
73 transfer_size_(-1) {} | 76 transfer_size_(-1) {} |
74 | 77 |
75 enum LoadPolicy load_policy_; | 78 enum LoadPolicy load_policy_; |
76 RefPtr<WebTaskRunner> runner_; | 79 RefPtr<WebTaskRunner> runner_; |
77 bool complete_; | 80 bool complete_; |
78 long long transfer_size_; | 81 long long transfer_size_; |
79 }; | 82 }; |
80 | 83 |
81 } // namespace blink | 84 } // namespace blink |
82 | 85 |
83 #endif | 86 #endif |
OLD | NEW |