Index: net/url_request/url_request_test_util.cc |
diff --git a/net/url_request/url_request_test_util.cc b/net/url_request/url_request_test_util.cc |
index 82d8579f54775a09015c345dd223022e9b1f8bb9..35e947ee2900299a0e4f677bb21c5b137c8a365d 100644 |
--- a/net/url_request/url_request_test_util.cc |
+++ b/net/url_request/url_request_test_util.cc |
@@ -317,11 +317,14 @@ TestNetworkDelegate::TestNetworkDelegate() |
blocked_set_cookie_count_(0), |
set_cookie_count_(0), |
observed_before_proxy_headers_sent_callbacks_(0), |
+ before_send_headers_count_(0), |
+ headers_received_count_(0), |
has_load_timing_info_before_redirect_(false), |
has_load_timing_info_before_auth_(false), |
can_access_files_(true), |
can_throttle_requests_(true), |
- cancel_request_with_policy_violating_referrer_(false) { |
+ cancel_request_with_policy_violating_referrer_(false), |
+ can_be_intercepted_on_error_(false) { |
} |
TestNetworkDelegate::~TestNetworkDelegate() { |
@@ -386,7 +389,7 @@ int TestNetworkDelegate::OnBeforeSendHeaders( |
next_states_[req_id] = |
kStageSendHeaders | |
kStageCompletedError; // request canceled by delegate |
- |
+ before_send_headers_count_++; |
return OK; |
} |
@@ -408,7 +411,8 @@ void TestNetworkDelegate::OnSendHeaders( |
event_order_[req_id]; |
next_states_[req_id] = |
kStageHeadersReceived | |
- kStageCompletedError; |
+ kStageCompletedError | |
+ (can_be_intercepted_on_error_ ? kStageResponseStarted : 0); |
} |
int TestNetworkDelegate::OnHeadersReceived( |
@@ -445,7 +449,7 @@ int TestNetworkDelegate::OnHeadersReceived( |
if (!allowed_unsafe_redirect_url_.is_empty()) |
*allowed_unsafe_redirect_url = allowed_unsafe_redirect_url_; |
} |
- |
+ headers_received_count_++; |
return OK; |
} |