From 779c4e41af928521b84c91e27092e45b12bcbeda Mon Sep 17 00:00:00 2001
From: hai007 <hais@google.com>
Date: Fri, 12 Mar 2021 14:03:34 -0800
Subject: [PATCH] Internal change

PiperOrigin-RevId: 362596467
---
 cpp/core/internal/endpoint_manager_test.cc | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/cpp/core/internal/endpoint_manager_test.cc b/cpp/core/internal/endpoint_manager_test.cc
index ae865bff..10fbcd3a 100644
--- a/cpp/core/internal/endpoint_manager_test.cc
+++ b/cpp/core/internal/endpoint_manager_test.cc
@@ -258,6 +258,8 @@ TEST_F(EndpointManagerTest, SingleReadOnInvalidPayload) {
   EXPECT_CALL(*endpoint_channel, Read())
       .WillOnce(
           Return(ExceptionOr<ByteArray>(Exception::kInvalidProtocolBuffer)));
+  EXPECT_CALL(*endpoint_channel, Write(_))
+      .WillRepeatedly(Return(Exception{Exception::kSuccess}));
   EXPECT_CALL(*endpoint_channel, Close(_)).Times(1);
   RegisterEndpoint(std::move(endpoint_channel));
 }