blob: 52a74f156a0f55e960b55a9ff5814c28755b58cd [file] [log] [blame]
// 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.
#ifndef SERVICES_NETWORK_TEST_TEST_URL_LOADER_CLIENT_H_
#define SERVICES_NETWORK_TEST_TEST_URL_LOADER_CLIENT_H_
#include <stdint.h>
#include <vector>
#include "base/callback.h"
#include "base/macros.h"
#include "mojo/public/c/system/data_pipe.h"
#include "mojo/public/cpp/bindings/binding.h"
#include "net/url_request/redirect_info.h"
#include "services/network/public/cpp/resource_response.h"
#include "services/network/public/cpp/url_loader_completion_status.h"
#include "services/network/public/mojom/url_loader.mojom.h"
#include "services/network/public/mojom/url_loader_factory.mojom.h"
namespace network {
// TestURLLoaderClient records URLLoaderClient function calls.
//
// Example usage:
// TestURLLoaderClient client;
// factory_->CreateLoaderAndStart(..., client.CreateInterfacePtr(), ...);
// client.RunUntilComplete();
// EXPECT_EQ(net::OK, client.completion_status().error_code);
// ...
class TestURLLoaderClient final : public mojom::URLLoaderClient {
public:
TestURLLoaderClient();
~TestURLLoaderClient() override;
void OnReceiveResponse(const ResourceResponseHead& response_head) override;
void OnReceiveRedirect(const net::RedirectInfo& redirect_info,
const ResourceResponseHead& response_head) override;
void OnReceiveCachedMetadata(mojo_base::BigBuffer data) override;
void OnTransferSizeUpdated(int32_t transfer_size_diff) override;
void OnUploadProgress(int64_t current_position,
int64_t total_size,
OnUploadProgressCallback ack_callback) override;
void OnStartLoadingResponseBody(
mojo::ScopedDataPipeConsumerHandle body) override;
void OnComplete(const URLLoaderCompletionStatus& status) override;
bool has_received_response() const { return has_received_response_; }
bool has_received_redirect() const { return has_received_redirect_; }
bool has_received_upload_progress() const {
return has_received_upload_progress_;
}
bool has_received_cached_metadata() const {
return has_received_cached_metadata_;
}
bool has_received_completion() const { return has_received_completion_; }
bool has_received_connection_error() const {
return has_received_connection_error_;
}
const ResourceResponseHead& response_head() const { return response_head_; }
const base::Optional<net::SSLInfo>& ssl_info() const {
return response_head_.ssl_info;
}
const net::RedirectInfo& redirect_info() const { return redirect_info_; }
const std::string& cached_metadata() const { return cached_metadata_; }
mojo::DataPipeConsumerHandle response_body() { return response_body_.get(); }
mojo::ScopedDataPipeConsumerHandle response_body_release() {
return std::move(response_body_);
}
const URLLoaderCompletionStatus& completion_status() const {
return completion_status_;
}
int64_t body_transfer_size() const { return body_transfer_size_; }
int64_t current_upload_position() const { return current_upload_position_; }
int64_t total_upload_size() const { return total_upload_size_; }
void reset_has_received_upload_progress() {
has_received_upload_progress_ = false;
}
void ClearHasReceivedRedirect();
// Creates an InterfacePtr, binds it to |*this| and returns it.
mojom::URLLoaderClientPtr CreateInterfacePtr();
void Unbind();
void RunUntilResponseReceived();
void RunUntilRedirectReceived();
void RunUntilCachedMetadataReceived();
void RunUntilResponseBodyArrived();
void RunUntilComplete();
void RunUntilConnectionError();
void RunUntilTransferSizeUpdated();
private:
void OnConnectionError();
mojo::Binding<mojom::URLLoaderClient> binding_;
ResourceResponseHead response_head_;
net::RedirectInfo redirect_info_;
std::string cached_metadata_;
mojo::ScopedDataPipeConsumerHandle response_body_;
URLLoaderCompletionStatus completion_status_;
bool has_received_response_ = false;
bool has_received_redirect_ = false;
bool has_received_upload_progress_ = false;
bool has_received_cached_metadata_ = false;
bool has_received_completion_ = false;
bool has_received_connection_error_ = false;
base::OnceClosure quit_closure_for_on_receive_response_;
base::OnceClosure quit_closure_for_on_receive_redirect_;
base::OnceClosure quit_closure_for_on_receive_cached_metadata_;
base::OnceClosure quit_closure_for_on_start_loading_response_body_;
base::OnceClosure quit_closure_for_on_complete_;
base::OnceClosure quit_closure_for_on_connection_error_;
base::OnceClosure quit_closure_for_on_transfer_size_updated_;
mojom::URLLoaderFactoryPtr url_loader_factory_;
int64_t body_transfer_size_ = 0;
int64_t current_upload_position_ = 0;
int64_t total_upload_size_ = 0;
DISALLOW_COPY_AND_ASSIGN(TestURLLoaderClient);
};
} // namespace network
#endif // SERVICES_NETWORK_TEST_TEST_URL_LOADER_CLIENT_H_