diff --git a/test/brpc_streaming_rpc_unittest.cpp b/test/brpc_streaming_rpc_unittest.cpp index 3e3d0f7c8e..5a49de69f0 100644 --- a/test/brpc_streaming_rpc_unittest.cpp +++ b/test/brpc_streaming_rpc_unittest.cpp @@ -1366,12 +1366,13 @@ class MyServiceWithMismatchedExtraStreamIds : public test::EchoService { settings->add_extra_stream_ids(settings->extra_stream_ids(0)); } - brpc::StreamIds response_streams; ASSERT_EQ(0, brpc::StreamAccept(response_streams, *cntl, nullptr)); ASSERT_EQ((int)_stream_count + _adjustment, (int)response_streams.size()); } + brpc::StreamIds response_streams; + private: size_t _stream_count; int _adjustment; @@ -1486,6 +1487,9 @@ TEST_F(StreamingRpcTest, reject_mismatched_returned_stream_identifiers) { brpc::StreamUniquePtr stream; ASSERT_NE(0, brpc::Stream::Address(stream_id, &stream)); } + for (brpc::StreamId stream_id : service.response_streams) { + ASSERT_EQ(0, brpc::StreamClose(stream_id)); + } server.Stop(0); server.Join();