Index: net/tools/quic/end_to_end_test.cc |
diff --git a/net/tools/quic/end_to_end_test.cc b/net/tools/quic/end_to_end_test.cc |
index 6e408156c72ad31b0834cb0360bb3bc9efcf8743..4e30c39a3f6aafddfcbe5f36e7dfd69a87b1c96d 100644 |
--- a/net/tools/quic/end_to_end_test.cc |
+++ b/net/tools/quic/end_to_end_test.cc |
@@ -811,8 +811,14 @@ TEST_P(EndToEndTest, LargePostZeroRTTFailure) { |
client_->WaitForResponseForMs(-1); |
ASSERT_TRUE(client_->client()->connected()); |
EXPECT_EQ(kFooResponseBody, client_->SendCustomSynchronousRequest(request)); |
- EXPECT_EQ(1, client_->client()->session()->GetNumSentClientHellos()); |
- EXPECT_EQ(1, client_->client()->GetNumSentClientHellos()); |
+ if (FLAGS_require_strike_register_or_server_nonce) { |
+ EXPECT_EQ(expected_num_hellos_latest_session, |
+ client_->client()->session()->GetNumSentClientHellos()); |
+ EXPECT_EQ(2, client_->client()->GetNumSentClientHellos()); |
+ } else { |
+ EXPECT_EQ(1, client_->client()->session()->GetNumSentClientHellos()); |
+ EXPECT_EQ(1, client_->client()->GetNumSentClientHellos()); |
+ } |
client_->Disconnect(); |
@@ -863,8 +869,14 @@ TEST_P(EndToEndTest, SynchronousRequestZeroRTTFailure) { |
client_->WaitForInitialResponse(); |
ASSERT_TRUE(client_->client()->connected()); |
EXPECT_EQ(kFooResponseBody, client_->SendSynchronousRequest("/foo")); |
- EXPECT_EQ(1, client_->client()->session()->GetNumSentClientHellos()); |
- EXPECT_EQ(1, client_->client()->GetNumSentClientHellos()); |
+ if (FLAGS_require_strike_register_or_server_nonce) { |
+ EXPECT_EQ(expected_num_hellos_latest_session, |
+ client_->client()->session()->GetNumSentClientHellos()); |
+ EXPECT_EQ(2, client_->client()->GetNumSentClientHellos()); |
+ } else { |
+ EXPECT_EQ(1, client_->client()->session()->GetNumSentClientHellos()); |
+ EXPECT_EQ(1, client_->client()->GetNumSentClientHellos()); |
+ } |
client_->Disconnect(); |
@@ -921,8 +933,14 @@ TEST_P(EndToEndTest, LargePostSynchronousRequest) { |
client_->WaitForInitialResponse(); |
ASSERT_TRUE(client_->client()->connected()); |
EXPECT_EQ(kFooResponseBody, client_->SendCustomSynchronousRequest(request)); |
- EXPECT_EQ(1, client_->client()->session()->GetNumSentClientHellos()); |
- EXPECT_EQ(1, client_->client()->GetNumSentClientHellos()); |
+ if (FLAGS_require_strike_register_or_server_nonce) { |
+ EXPECT_EQ(expected_num_hellos_latest_session, |
+ client_->client()->session()->GetNumSentClientHellos()); |
+ EXPECT_EQ(2, client_->client()->GetNumSentClientHellos()); |
+ } else { |
+ EXPECT_EQ(1, client_->client()->session()->GetNumSentClientHellos()); |
+ EXPECT_EQ(1, client_->client()->GetNumSentClientHellos()); |
+ } |
client_->Disconnect(); |