mirror of
https://github.com/klzgrad/naiveproxy.git
synced 2024-11-24 14:26:09 +03:00
104 lines
3.5 KiB
C++
104 lines
3.5 KiB
C++
// 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 NET_BASE_TEST_PROXY_DELEGATE_H_
|
|
#define NET_BASE_TEST_PROXY_DELEGATE_H_
|
|
|
|
#include <string>
|
|
|
|
#include "net/base/host_port_pair.h"
|
|
#include "net/base/proxy_delegate.h"
|
|
#include "net/proxy/proxy_server.h"
|
|
|
|
class GURL;
|
|
|
|
namespace net {
|
|
|
|
class HttpRequestHeaders;
|
|
class HttpResponseHeaders;
|
|
class ProxyInfo;
|
|
|
|
class TestProxyDelegate : public ProxyDelegate {
|
|
public:
|
|
TestProxyDelegate();
|
|
~TestProxyDelegate() override;
|
|
|
|
bool on_before_tunnel_request_called() const {
|
|
return on_before_tunnel_request_called_;
|
|
}
|
|
|
|
bool on_tunnel_request_completed_called() const {
|
|
return on_tunnel_request_completed_called_;
|
|
}
|
|
|
|
bool on_tunnel_headers_received_called() const {
|
|
return on_tunnel_headers_received_called_;
|
|
}
|
|
|
|
void set_trusted_spdy_proxy(const net::ProxyServer& proxy_server) {
|
|
trusted_spdy_proxy_ = proxy_server;
|
|
}
|
|
|
|
void VerifyOnTunnelRequestCompleted(const std::string& endpoint,
|
|
const std::string& proxy_server) const;
|
|
|
|
void VerifyOnTunnelHeadersReceived(const std::string& origin,
|
|
const std::string& proxy_server,
|
|
const std::string& status_line) const;
|
|
|
|
// ProxyDelegate implementation:
|
|
void OnResolveProxy(const GURL& url,
|
|
const std::string& method,
|
|
const ProxyRetryInfoMap& proxy_retry_info,
|
|
ProxyInfo* result) override;
|
|
void OnTunnelConnectCompleted(const HostPortPair& endpoint,
|
|
const HostPortPair& proxy_server,
|
|
int net_error) override;
|
|
void OnFallback(const ProxyServer& bad_proxy, int net_error) override;
|
|
void OnBeforeTunnelRequest(const HostPortPair& proxy_server,
|
|
HttpRequestHeaders* extra_headers) override;
|
|
void OnTunnelHeadersReceived(
|
|
const HostPortPair& origin,
|
|
const HostPortPair& proxy_server,
|
|
const HttpResponseHeaders& response_headers) override;
|
|
bool IsTrustedSpdyProxy(const net::ProxyServer& proxy_server) override;
|
|
void GetAlternativeProxy(
|
|
const GURL& url,
|
|
const ProxyServer& resolved_proxy_server,
|
|
ProxyServer* alternative_proxy_server) const override;
|
|
void OnAlternativeProxyBroken(
|
|
const ProxyServer& alternative_proxy_server) override;
|
|
|
|
void set_alternative_proxy_server(
|
|
const ProxyServer& alternative_proxy_server) {
|
|
alternative_proxy_server_ = alternative_proxy_server;
|
|
}
|
|
const ProxyServer& alternative_proxy_server() const {
|
|
return alternative_proxy_server_;
|
|
}
|
|
|
|
int get_alternative_proxy_invocations() const {
|
|
return get_alternative_proxy_invocations_;
|
|
}
|
|
|
|
private:
|
|
bool on_before_tunnel_request_called_;
|
|
bool on_tunnel_request_completed_called_;
|
|
bool on_tunnel_headers_received_called_;
|
|
net::ProxyServer trusted_spdy_proxy_;
|
|
HostPortPair on_tunnel_request_completed_endpoint_;
|
|
HostPortPair on_tunnel_request_completed_proxy_server_;
|
|
HostPortPair on_tunnel_headers_received_origin_;
|
|
HostPortPair on_tunnel_headers_received_proxy_server_;
|
|
std::string on_tunnel_headers_received_status_line_;
|
|
ProxyServer alternative_proxy_server_;
|
|
|
|
// Number of times GetAlternativeProxy() method has been called.
|
|
mutable int get_alternative_proxy_invocations_;
|
|
};
|
|
|
|
} // namespace net
|
|
|
|
#endif // NET_BASE_TEST_PROXY_DELEGATE_H_
|