From a8ad13614bca3ed0aabc0024ad0f413571b7fe32 Mon Sep 17 00:00:00 2001
From: xicilion <lion@9465.net>
Date: Sun, 24 Jun 2018 14:34:07 +0800
Subject: [PATCH] zlib, feat: reset dataSize when attaching to new stream.

---
 fibjs/include/ZlibStream.h |  1 +
 test/ws_test.js            | 31 ++++++++++++++++++++++++++++++-
 2 files changed, 31 insertions(+), 1 deletion(-)

diff --git a/fibjs/include/ZlibStream.h b/fibjs/include/ZlibStream.h
index 0490fbcab1..f8e9134a64 100644
--- a/fibjs/include/ZlibStream.h
+++ b/fibjs/include/ZlibStream.h
@@ -104,6 +104,7 @@ class ZlibStream : public Stream_base {
     void attach(Stream_base* stm)
     {
         m_stm = stm;
+        m_dataSize = 0;
     }
 
 public:
diff --git a/test/ws_test.js b/test/ws_test.js
index eb30a9f753..74fc2a9187 100644
--- a/test/ws_test.js
+++ b/test/ws_test.js
@@ -485,7 +485,10 @@ describe('ws', () => {
                             msg.stream.close();
                         else if (msg.data === "close")
                             this.close(3000, "remote");
-                        else
+                        else if (msg.data === "many") {
+                            for (var i = 0; i < 1025; i++)
+                                s.send(new Buffer(1024 * 64));
+                        } else
                             this.send(msg.data);
                     };
                 })
@@ -556,6 +559,32 @@ describe('ws', () => {
             s.close();
         });
 
+        it('many compressed message', () => {
+            var cnt = 0;
+            var sz = 0;
+            var msg;
+            var s = new ws.Socket("ws://127.0.0.1:" + (8814 + base_port) + "/ws", "test");
+            s.onopen = () => {
+                s.send('many');
+            };
+
+            s.onmessage = m => {
+                cnt++;
+                sz += m.data.length;
+            };
+
+            s.onerror = e => {
+                console.error(e);
+            };
+
+            for (var i = 0; i < 1000 && cnt < 1025; i++)
+                coroutine.sleep(1);
+
+            s.close();
+
+            assert.equal(sz, 1024 * 1025 * 64);
+        });
+
         it('send/on("message")', () => {
             var httpd = new http.Server(8815 + base_port, {
                 "/ws": ws.upgrade((s) => {