Add EXPECT_CALLs and use NiceMock to make tls_client_handshaker_test log look cleaner. gfe-relnote: test change only. not protected. PiperOrigin-RevId: 308130693 Change-Id: I4cf48bb769632dd689a01ef62f34705c005a371c
diff --git a/quic/core/tls_client_handshaker_test.cc b/quic/core/tls_client_handshaker_test.cc index bb859c1..2ba2d91 100644 --- a/quic/core/tls_client_handshaker_test.cc +++ b/quic/core/tls_client_handshaker_test.cc
@@ -190,6 +190,8 @@ } void CompleteCryptoHandshake() { + EXPECT_CALL(*connection_, SendCryptoData(_, _, _)) + .Times(testing::AnyNumber()); stream()->CryptoConnect(); QuicConfig config; crypto_test_utils::HandshakeWithFakeServer(
diff --git a/quic/test_tools/crypto_test_utils.cc b/quic/test_tools/crypto_test_utils.cc index 21561d9..2c79f92 100644 --- a/quic/test_tools/crypto_test_utils.cc +++ b/quic/test_tools/crypto_test_utils.cc
@@ -230,7 +230,7 @@ PacketSavingConnection* client_conn, QuicCryptoClientStream* client, std::string alpn) { - PacketSavingConnection* server_conn = new PacketSavingConnection( + auto* server_conn = new testing::NiceMock<PacketSavingConnection>( helper, alarm_factory, Perspective::IS_SERVER, ParsedVersionOfIndex(client_conn->supported_versions(), 0)); @@ -249,6 +249,8 @@ .Times(testing::AnyNumber()); EXPECT_CALL(*server_conn, OnCanWrite()).Times(testing::AnyNumber()); EXPECT_CALL(*client_conn, OnCanWrite()).Times(testing::AnyNumber()); + EXPECT_CALL(*server_conn, SendCryptoData(_, _, _)) + .Times(testing::AnyNumber()); EXPECT_CALL(server_session, SelectAlpn(_)) .WillRepeatedly( [alpn](const std::vector<quiche::QuicheStringPiece>& alpns) {