From 17222654bff8172f95cbb0b364fdc5a873477db6 Mon Sep 17 00:00:00 2001
From: senhuang42 <senhuang96@fb.com>
Date: Thu, 7 Jan 2021 12:07:35 -0500
Subject: [PATCH] Add streaming decompression to unit test

---
 tests/fuzzer.c | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/tests/fuzzer.c b/tests/fuzzer.c
index a67113edd..b808ad2d2 100644
--- a/tests/fuzzer.c
+++ b/tests/fuzzer.c
@@ -2382,6 +2382,14 @@ static int basicUnitTests(U32 const seed, double compressibility)
                 CHECK_Z( ZSTD_DCtx_refDDict(dctx, ddictTable[i]));
             }
             CHECK_Z( ZSTD_decompressDCtx(dctx, decodedBuffer, CNBuffSize, compressedBuffer, cSize) );
+            /* Streaming decompression should also work */
+            {
+                ZSTD_inBuffer in = {compressedBuffer, cSize, 0};
+                ZSTD_outBuffer out = {decodedBuffer, CNBuffSize, 0};
+                while (in.pos < in.size) {
+                    CHECK_Z(ZSTD_decompressStream(dctx, &out, &in));
+                }
+            }
             ZSTD_freeDCtx(dctx);
             for (i = 0; i < numDicts; ++i) {
                 ZSTD_freeCDict(cdictTable[i]);
-- 
2.47.2