From b77e1de1d19680b9aa18e9dc03d8768c5c45a79a Mon Sep 17 00:00:00 2001 From: winlin Date: Mon, 22 Dec 2014 15:31:57 +0800 Subject: [PATCH] fix bug of utest --- trunk/src/utest/srs_utest_protocol.cpp | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/trunk/src/utest/srs_utest_protocol.cpp b/trunk/src/utest/srs_utest_protocol.cpp index 10afb87f4..f214b63c1 100644 --- a/trunk/src/utest/srs_utest_protocol.cpp +++ b/trunk/src/utest/srs_utest_protocol.cpp @@ -4635,7 +4635,10 @@ VOID TEST(ProtocolStackTest, ProtocolSendVMessage) msg->payload = new char[msg->size]; memcpy(msg->payload, data, msg->size); - EXPECT_TRUE(ERROR_SUCCESS == proto.send_and_free_message(msg, 0)); + SrsSharedPtrMessage m; + ASSERT_TRUE(ERROR_SUCCESS == m.create(msg)); + + EXPECT_TRUE(ERROR_SUCCESS == proto.send_and_free_message(m.copy(), 0)); EXPECT_EQ(16, bio.out_buffer.length()); } @@ -5346,8 +5349,11 @@ VOID TEST(ProtocolStackTest, ProtocolAckSizeFlow) msg->header.message_type = 9; EXPECT_TRUE(msg->header.is_video()); + + SrsSharedPtrMessage m; + ASSERT_TRUE(ERROR_SUCCESS == m.create(msg)); - EXPECT_TRUE(ERROR_SUCCESS == proto.send_and_free_message(msg, 1)); + EXPECT_TRUE(ERROR_SUCCESS == proto.send_and_free_message(m.copy(), 1)); } // copy output to input @@ -5392,8 +5398,11 @@ VOID TEST(ProtocolStackTest, ProtocolAckSizeFlow) msg->header.message_type = 9; EXPECT_TRUE(msg->header.is_video()); + + SrsSharedPtrMessage m; + ASSERT_TRUE(ERROR_SUCCESS == m.create(msg)); - EXPECT_TRUE(ERROR_SUCCESS == proto.send_and_free_message(msg, 1)); + EXPECT_TRUE(ERROR_SUCCESS == proto.send_and_free_message(m.copy(), 1)); } // copy output to input if (true) {