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/FetchRequest.h" | 10 #include "platform/loader/fetch/FetchRequest.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 #include "wtf/PtrUtil.h" |
15 | 16 |
16 namespace blink { | 17 namespace blink { |
17 | 18 |
18 class KURL; | 19 class KURL; |
19 class ResourceRequest; | 20 class ResourceRequest; |
20 class WebTaskRunner; | 21 class WebTaskRunner; |
21 struct ResourceLoaderOptions; | 22 struct ResourceLoaderOptions; |
22 | 23 |
23 // Mocked FetchContext for testing. | 24 // Mocked FetchContext for testing. |
24 class MockFetchContext : public FetchContext { | 25 class MockFetchContext : public FetchContext { |
(...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
56 } | 57 } |
57 bool shouldLoadNewResource(Resource::Type) const override { | 58 bool shouldLoadNewResource(Resource::Type) const override { |
58 return m_loadPolicy == kShouldLoadNewResource; | 59 return m_loadPolicy == kShouldLoadNewResource; |
59 } | 60 } |
60 RefPtr<WebTaskRunner> loadingTaskRunner() const override { return m_runner; } | 61 RefPtr<WebTaskRunner> loadingTaskRunner() const override { return m_runner; } |
61 bool isLoadComplete() const override { return m_complete; } | 62 bool isLoadComplete() const override { return m_complete; } |
62 void addResourceTiming( | 63 void addResourceTiming( |
63 const ResourceTimingInfo& resourceTimingInfo) override { | 64 const ResourceTimingInfo& resourceTimingInfo) override { |
64 m_transferSize = resourceTimingInfo.transferSize(); | 65 m_transferSize = resourceTimingInfo.transferSize(); |
65 } | 66 } |
| 67 WebURLLoader* createURLLoader() override { |
| 68 return Platform::current()->createURLLoader(); |
| 69 } |
66 | 70 |
67 private: | 71 private: |
68 MockFetchContext(LoadPolicy loadPolicy, RefPtr<WebTaskRunner> taskRunner) | 72 MockFetchContext(LoadPolicy loadPolicy, RefPtr<WebTaskRunner> taskRunner) |
69 : m_loadPolicy(loadPolicy), | 73 : m_loadPolicy(loadPolicy), |
70 m_runner(taskRunner ? std::move(taskRunner) | 74 m_runner(taskRunner ? std::move(taskRunner) |
71 : adoptRef(new scheduler::FakeWebTaskRunner)), | 75 : adoptRef(new scheduler::FakeWebTaskRunner)), |
72 m_complete(false), | 76 m_complete(false), |
73 m_transferSize(-1) {} | 77 m_transferSize(-1) {} |
74 | 78 |
75 enum LoadPolicy m_loadPolicy; | 79 enum LoadPolicy m_loadPolicy; |
76 RefPtr<WebTaskRunner> m_runner; | 80 RefPtr<WebTaskRunner> m_runner; |
77 bool m_complete; | 81 bool m_complete; |
78 long long m_transferSize; | 82 long long m_transferSize; |
79 }; | 83 }; |
80 | 84 |
81 } // namespace blink | 85 } // namespace blink |
82 | 86 |
83 #endif | 87 #endif |
OLD | NEW |