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

Unified Diff: content/public/test/test_download_request_handler.cc

Issue 1958383002: [Merge M51][Downloads] Set the Referrer header for resumed requests. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@2704
Patch Set: Created 4 years, 7 months 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
« no previous file with comments | « content/public/test/test_download_request_handler.h ('k') | content/test/data/download/download-link.html » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: content/public/test/test_download_request_handler.cc
diff --git a/content/public/test/test_download_request_handler.cc b/content/public/test/test_download_request_handler.cc
index 620ebcaa3d9841f7157de3fbba07f6ec1cb8305b..5e83cf797d0c2236c438ccea5250b044dc7bb5f0 100644
--- a/content/public/test/test_download_request_handler.cc
+++ b/content/public/test/test_download_request_handler.cc
@@ -5,6 +5,8 @@
#include "content/public/test/test_download_request_handler.h"
#include <inttypes.h>
+
+#include <memory>
#include <utility>
#include "base/logging.h"
@@ -61,8 +63,7 @@ class TestDownloadRequestHandler::Interceptor
// Can be called by a URLRequestJob to notify this interceptor of a completed
// request.
- void AddCompletedRequest(
- const TestDownloadRequestHandler::CompletedRequest& request);
+ void AddCompletedRequest(std::unique_ptr<CompletedRequest> request);
// Returns the task runner that should be used for invoking any client
// supplied callbacks.
@@ -199,6 +200,13 @@ scoped_refptr<net::HttpResponseHeaders> HeadersFromString(
} // namespace
+TestDownloadRequestHandler::CompletedRequest::CompletedRequest() {}
+
+TestDownloadRequestHandler::CompletedRequest::~CompletedRequest() {}
+
+TestDownloadRequestHandler::CompletedRequest::CompletedRequest(
+ CompletedRequest&&) = default;
+
// static
net::URLRequestJob* TestDownloadRequestHandler::PartialResponseJob::Factory(
const Parameters& parameters,
@@ -310,10 +318,17 @@ int TestDownloadRequestHandler::PartialResponseJob::ReadRawData(
void TestDownloadRequestHandler::PartialResponseJob::ReportCompletedRequest() {
if (interceptor_.get()) {
- TestDownloadRequestHandler::CompletedRequest completed_request;
- completed_request.transferred_byte_count = read_byte_count_;
- completed_request.request_headers = request()->extra_request_headers();
- interceptor_->AddCompletedRequest(completed_request);
+ std::unique_ptr<CompletedRequest> completed_request(new CompletedRequest);
+ completed_request->transferred_byte_count = read_byte_count_;
+ completed_request->request_headers = request()->extra_request_headers();
+ completed_request->referrer = request()->referrer();
+ completed_request->referrer_policy = request()->referrer_policy();
+ completed_request->initiator = request()->initiator();
+ completed_request->first_party_for_cookies =
+ request()->first_party_for_cookies();
+ completed_request->first_party_url_policy =
+ request()->first_party_url_policy();
+ interceptor_->AddCompletedRequest(std::move(completed_request));
}
}
@@ -511,8 +526,8 @@ void TestDownloadRequestHandler::Interceptor::GetAndResetCompletedRequests(
}
void TestDownloadRequestHandler::Interceptor::AddCompletedRequest(
- const TestDownloadRequestHandler::CompletedRequest& request) {
- completed_requests_.push_back(request);
+ std::unique_ptr<CompletedRequest> request) {
+ completed_requests_.push_back(std::move(request));
}
scoped_refptr<base::SequencedTaskRunner>
« no previous file with comments | « content/public/test/test_download_request_handler.h ('k') | content/test/data/download/download-link.html » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698