From c4575d141946e5200d84047bfa8b78c7d05fa551 Mon Sep 17 00:00:00 2001
From: Victor Julien <victor@inliniac.net>
Date: Wed, 13 Apr 2016 10:14:35 +0200
Subject: [PATCH] stream-tcp: improve test function cleanup

---
 src/stream-tcp-util.c | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/stream-tcp-util.c b/src/stream-tcp-util.c
index 54c500bc93..b58db5c283 100644
--- a/src/stream-tcp-util.c
+++ b/src/stream-tcp-util.c
@@ -68,6 +68,7 @@ void StreamTcpUTClearSession(TcpSession *ssn)
 {
     StreamTcpUTClearStream(&ssn->client);
     StreamTcpUTClearStream(&ssn->server);
+    StreamTcpSessionCleanup(ssn);
 }
 
 void StreamTcpUTSetupStream(TcpStream *s, uint32_t isn)
@@ -80,7 +81,7 @@ void StreamTcpUTSetupStream(TcpStream *s, uint32_t isn)
 
 void StreamTcpUTClearStream(TcpStream *s)
 {
-    StreamTcpReturnStreamSegments(s);
+    StreamTcpStreamCleanup(s);
 }
 
 int StreamTcpUTAddSegmentWithPayload(ThreadVars *tv, TcpReassemblyThreadCtx *ra_ctx, TcpStream *stream, uint32_t seq, uint8_t *payload, uint16_t len)
-- 
2.47.2