| // Copyright 2016 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "content/renderer/loader/test_request_peer.h" |
| |
| #include "content/renderer/loader/resource_dispatcher.h" |
| #include "net/url_request/redirect_info.h" |
| #include "services/network/public/mojom/url_response_head.mojom.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| #include "third_party/blink/public/platform/scheduler/test/renderer_scheduler_test_support.h" |
| |
| namespace content { |
| |
| TestRequestPeer::TestRequestPeer(ResourceDispatcher* dispatcher, |
| Context* context) |
| : dispatcher_(dispatcher), context_(context) {} |
| |
| TestRequestPeer::~TestRequestPeer() = default; |
| |
| void TestRequestPeer::OnUploadProgress(uint64_t position, uint64_t size) { |
| EXPECT_FALSE(context_->complete); |
| } |
| |
| bool TestRequestPeer::OnReceivedRedirect( |
| const net::RedirectInfo& redirect_info, |
| network::mojom::URLResponseHeadPtr head) { |
| EXPECT_FALSE(context_->cancelled); |
| EXPECT_FALSE(context_->complete); |
| ++context_->seen_redirects; |
| context_->last_load_timing = head->load_timing; |
| if (context_->defer_on_redirect) |
| dispatcher_->SetDefersLoading(context_->request_id, true); |
| return context_->follow_redirects; |
| } |
| |
| void TestRequestPeer::OnReceivedResponse( |
| network::mojom::URLResponseHeadPtr head) { |
| EXPECT_FALSE(context_->cancelled); |
| EXPECT_FALSE(context_->received_response); |
| EXPECT_FALSE(context_->complete); |
| context_->received_response = true; |
| context_->last_load_timing = head->load_timing; |
| if (context_->cancel_on_receive_response) { |
| dispatcher_->Cancel( |
| context_->request_id, |
| blink::scheduler::GetSingleThreadTaskRunnerForTesting()); |
| context_->cancelled = true; |
| } |
| } |
| |
| void TestRequestPeer::OnStartLoadingResponseBody( |
| mojo::ScopedDataPipeConsumerHandle body) { |
| if (context_->cancelled) |
| return; |
| EXPECT_TRUE(context_->received_response); |
| EXPECT_FALSE(context_->complete); |
| context_->body_handle = std::move(body); |
| } |
| |
| void TestRequestPeer::OnTransferSizeUpdated(int transfer_size_diff) { |
| EXPECT_TRUE(context_->received_response); |
| EXPECT_FALSE(context_->complete); |
| if (context_->cancelled) |
| return; |
| context_->total_encoded_data_length += transfer_size_diff; |
| if (context_->defer_on_transfer_size_updated) |
| dispatcher_->SetDefersLoading(context_->request_id, true); |
| } |
| |
| void TestRequestPeer::OnReceivedCachedMetadata(mojo_base::BigBuffer data) { |
| EXPECT_TRUE(context_->received_response); |
| EXPECT_FALSE(context_->complete); |
| if (context_->cancelled) |
| return; |
| context_->cached_metadata = std::move(data); |
| } |
| |
| void TestRequestPeer::OnCompletedRequest( |
| const network::URLLoaderCompletionStatus& status) { |
| if (context_->cancelled) |
| return; |
| EXPECT_TRUE(context_->received_response); |
| EXPECT_FALSE(context_->complete); |
| context_->complete = true; |
| context_->completion_status = status; |
| } |
| |
| scoped_refptr<base::TaskRunner> TestRequestPeer::GetTaskRunner() { |
| return blink::scheduler::GetSingleThreadTaskRunnerForTesting(); |
| } |
| |
| TestRequestPeer::Context::Context() = default; |
| TestRequestPeer::Context::~Context() = default; |
| |
| } // namespace content |