From 85ab7a3fd068698a9cff936b3439bacda9945802 Mon Sep 17 00:00:00 2001
From: Damian Jarek <damian.jarek93@gmail.com>
Date: Mon, 9 Sep 2019 02:55:39 +0200
Subject: [PATCH] Add test for partial flush in deflate stream

Signed-off-by: Damian Jarek <damian.jarek93@gmail.com>
---
 test/beast/zlib/deflate_stream.cpp | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/test/beast/zlib/deflate_stream.cpp b/test/beast/zlib/deflate_stream.cpp
index fc6a48bc..44897874 100644
--- a/test/beast/zlib/deflate_stream.cpp
+++ b/test/beast/zlib/deflate_stream.cpp
@@ -371,6 +371,26 @@ public:
         BEAST_EXPECT(ec == error::need_buffers);
     }
 
+    void
+    testFlushPartial()
+    {
+        z_params zp;
+        deflate_stream ds;
+        ds.reset();
+        std::string out;
+        out.resize(1024);
+        string_view s = "Hello";
+        zp.next_in = s.data();
+        zp.avail_in = s.size();
+        zp.next_out = &out.front();
+        zp.avail_out = out.size();
+        error_code ec;
+        ds.write(zp, Flush::none, ec);
+        BEAST_EXPECT(!ec);
+        ds.write(zp, Flush::partial, ec);
+        BEAST_EXPECT(!ec);
+    }
+
     void
     run() override
     {
@@ -381,6 +401,7 @@ public:
         testDeflate();
         testInvalidSettings();
         testWriteAfterFinish();
+        testFlushPartial();
     }
 };