OLD | NEW |
| (Empty) |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #ifndef NET_URL_REQUEST_URL_REQUEST_MOCK_DATA_JOB_H_ | |
6 #define NET_URL_REQUEST_URL_REQUEST_MOCK_DATA_JOB_H_ | |
7 | |
8 #include <string> | |
9 | |
10 #include "base/memory/weak_ptr.h" | |
11 #include "net/base/net_export.h" | |
12 #include "net/url_request/url_request_job.h" | |
13 | |
14 namespace net { | |
15 | |
16 class URLRequest; | |
17 | |
18 // Mock data job, which synchronously returns data repeated multiple times. | |
19 class URLRequestMockDataJob : public URLRequestJob { | |
20 public: | |
21 URLRequestMockDataJob(URLRequest* request, | |
22 NetworkDelegate* network_delegate, | |
23 const std::string& data, | |
24 int data_repeat_count); | |
25 | |
26 void Start() override; | |
27 bool ReadRawData(IOBuffer* buf, int buf_size, int* bytes_read) override; | |
28 | |
29 // Adds the testing URLs to the URLRequestFilter. | |
30 static void AddUrlHandler(); | |
31 static void AddUrlHandlerForHostname(const std::string& hostname); | |
32 | |
33 // Given data and repeat cound, constructs a mock URL that will return that | |
34 // data repeated |repeat_count| times when started. |data| must be safe for | |
35 // URL. | |
36 static GURL GetMockHttpUrl(const std::string& data, int repeat_count); | |
37 static GURL GetMockHttpsUrl(const std::string& data, int repeat_count); | |
38 | |
39 // URLRequestFailedJob must be added as a handler for |hostname| for | |
40 // the returned URL to return |net_error|. | |
41 static GURL GetMockHttpUrlForHostname(const std::string& hostname, | |
42 const std::string& data, | |
43 int repeat_count); | |
44 static GURL GetMockHttpsUrlForHostname(const std::string& hostname, | |
45 const std::string& data, | |
46 int repeat_count); | |
47 | |
48 private: | |
49 ~URLRequestMockDataJob() override; | |
50 | |
51 void StartAsync(); | |
52 | |
53 std::string data_; | |
54 size_t data_offset_; | |
55 base::WeakPtrFactory<URLRequestMockDataJob> weak_factory_; | |
56 }; | |
57 | |
58 } // namespace net | |
59 | |
60 #endif // NET_URL_REQUEST_URL_REQUEST_SIMPLE_JOB_H_ | |
OLD | NEW |