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..8f70ee73b173f0cbc88818b2e3012472260ad5e5 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), |
+ will_be_intercepted_on_next_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; |
} |
@@ -406,9 +409,11 @@ void TestNetworkDelegate::OnSendHeaders( |
event_order_[req_id] += "OnSendHeaders\n"; |
EXPECT_TRUE(next_states_[req_id] & kStageSendHeaders) << |
event_order_[req_id]; |
- next_states_[req_id] = |
- kStageHeadersReceived | |
- kStageCompletedError; |
+ if (!will_be_intercepted_on_next_error_) |
+ next_states_[req_id] = kStageHeadersReceived | kStageCompletedError; |
+ else |
+ next_states_[req_id] = kStageResponseStarted; |
+ will_be_intercepted_on_next_error_ = false; |
} |
int TestNetworkDelegate::OnHeadersReceived( |
@@ -445,7 +450,7 @@ int TestNetworkDelegate::OnHeadersReceived( |
if (!allowed_unsafe_redirect_url_.is_empty()) |
*allowed_unsafe_redirect_url = allowed_unsafe_redirect_url_; |
} |
- |
+ headers_received_count_++; |
return OK; |
} |