Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(1809)

Unified Diff: net/test/url_request/url_request_mock_data_job.h

Issue 743713002: Cronet Fix Channel Write after Close when request is canceled after success. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Added UrlRequestMockDataJob and tests that use it. Created 6 years ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: net/test/url_request/url_request_mock_data_job.h
diff --git a/net/test/url_request/url_request_mock_data_job.h b/net/test/url_request/url_request_mock_data_job.h
new file mode 100644
index 0000000000000000000000000000000000000000..c1c5106007d805ab582091dcca87c6932e6f3126
--- /dev/null
+++ b/net/test/url_request/url_request_mock_data_job.h
@@ -0,0 +1,78 @@
+// Copyright (c) 2012 The Chromium Authors. All rights reserved.
mmenke 2014/12/09 21:00:04 -(c), 2012->2014
mef 2014/12/09 22:21:00 Done.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#ifndef NET_URL_REQUEST_URL_REQUEST_MOCK_DATA_JOB_H_
+#define NET_URL_REQUEST_URL_REQUEST_MOCK_DATA_JOB_H_
+
+#include <string>
+
+#include "base/memory/ref_counted.h"
+#include "base/memory/weak_ptr.h"
+#include "base/strings/string_piece.h"
+#include "net/base/completion_callback.h"
+#include "net/base/net_export.h"
+#include "net/url_request/url_request_job.h"
+
+namespace base {
+class RefCountedMemory;
+}
+
+namespace net {
+
+class URLRequest;
+
+// Mock data job, which synchronously returns data repeated multiple times.
+class NET_EXPORT URLRequestMockDataJob : public URLRequestJob {
mmenke 2014/12/09 21:00:04 Maybe MockData -> MockSyncData (Here and elsewhere
mef 2014/12/09 22:21:00 I thought that it may come in handy for async test
+ public:
+ URLRequestMockDataJob(URLRequest* request,
+ NetworkDelegate* network_delegate,
+ const std::string& data,
+ unsigned data_repeat_count);
+
+ void Start() override;
+ bool ReadRawData(IOBuffer* buf, int buf_size, int* bytes_read) override;
+ bool GetMimeType(std::string* mime_type) const override;
+ bool GetCharset(std::string* charset) override;
+
+ // Adds the testing URLs to the URLRequestFilter.
+ static void AddUrlHandler();
+ static void AddUrlHandlerForHostname(const std::string& hostname);
+
+ // Given data and repeat cound, constructs a mock URL that will return that
+ // data repeated |repeat_count| times when started. |data| must be safe for
+ // URL.
+ static GURL GetMockHttpUrl(const std::string& data, int repeat_count);
+ static GURL GetMockHttpsUrl(const std::string& data, int repeat_count);
+
+ // URLRequestFailedJob must be added as a handler for |hostname| for
+ // the returned URL to return |net_error|.
+ static GURL GetMockHttpUrlForHostname(const std::string& hostname,
+ const std::string& data,
+ int repeat_count);
+ static GURL GetMockHttpsUrlForHostname(const std::string& hostname,
+ const std::string& data,
+ int repeat_count);
+
+ protected:
+ ~URLRequestMockDataJob() override;
+
+ void StartAsync();
mmenke 2014/12/09 21:00:04 Suggest just making these private, for now.
mef 2014/12/09 22:21:00 Done.
+
+ private:
+ static URLRequestJob* Factory(URLRequest* request,
+ NetworkDelegate* network_delegate,
+ const std::string& scheme);
+
+ void OnGetDataCompleted(int result);
+
+ std::string mime_type_;
+ std::string charset_;
mmenke 2014/12/09 21:00:04 We don't need these, as they're never initialized.
mef 2014/12/09 22:20:59 Done.
+ std::string data_;
+ int64 data_offset_;
+ base::WeakPtrFactory<URLRequestMockDataJob> weak_factory_;
+};
+
+} // namespace net
+
+#endif // NET_URL_REQUEST_URL_REQUEST_SIMPLE_JOB_H_

Powered by Google App Engine
This is Rietveld 408576698