From ade7ab95a0c3cf41b35350b94b21ef8aad4dbe2e Mon Sep 17 00:00:00 2001
From: jeromew <jerome.wagner@m4x.org>
Date: Fri, 29 Jul 2016 23:45:01 +0000
Subject: [PATCH] Test: `end` event should not be triggered 2 times on
 copy-from

---
 test/copy-from.js | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/test/copy-from.js b/test/copy-from.js
index b8e6e0a..1765ca5 100644
--- a/test/copy-from.js
+++ b/test/copy-from.js
@@ -39,6 +39,7 @@ var testRange = function(top) {
     fromClient.query('SELECT COUNT(*) FROM numbers', function(err, res) {
       assert.ifError(err)
       assert.equal(res.rows[0].count, top, 'expected ' + top + ' rows but got ' + res.rows[0].count)
+      assert.equal(stream.rowCount, top, 'expected ' + top + ' rows but db count is ' + stream.rowCount)
       //console.log('found ', res.rows.length, 'rows')
       countDone()
       var firstRowDone = gonna('have correct result')
@@ -54,3 +55,19 @@ var testRange = function(top) {
 }
 
 testRange(1000)
+
+var testSingleEnd = function() {
+  var fromClient = client()
+  fromClient.query('CREATE TEMP TABLE numbers(num int)')
+  var txt = 'COPY numbers FROM STDIN';
+  var stream = fromClient.query(copy(txt))
+  var count = 0;
+  stream.on('end', function() {
+    count++;
+    assert(count==1, '`end` Event was triggered ' + count + ' times');
+    if (count == 1) fromClient.end();
+  })
+  stream.end(Buffer('1\n'))
+    
+}
+testSingleEnd()