From 645616c2e0811c72e8ee8b468ab9aa83da8f942f Mon Sep 17 00:00:00 2001
From: Rafa de la Torre <rtorre@carto.com>
Date: Tue, 4 Jun 2019 15:41:59 +0200
Subject: [PATCH] Test interspersed messages emit a warning event

---
 test/copy-to.js | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/test/copy-to.js b/test/copy-to.js
index bca3d95..e885c27 100644
--- a/test/copy-to.js
+++ b/test/copy-to.js
@@ -151,6 +151,24 @@ var testInterspersedMessageDoesNotBreakCopyFlow = function() {
 };
 testInterspersedMessageDoesNotBreakCopyFlow();
 
+var testInterspersedMessageEmitsWarnign = function() {
+  var toClient = client();
+  toClient.query(warnAndReturnOne, (err, res) => {
+    var q = "COPY (SELECT * FROM pg_temp.test_warn_return_one()) TO STDOUT WITH (FORMAT 'csv', HEADER true)";
+    var stream = toClient.query(copy(q));
+    var done = gonna('got expected warning event', 1000, function() {
+      toClient.end();
+    });
+
+    stream.on('warning', function(msg) {
+      assert(msg.match(/Got an interspersed message:.*WARNING.*hey, this is returning one/),
+             'did not get expected warning for interspersed message in COPY TO');
+      done();
+    })
+  });
+};
+testInterspersedMessageEmitsWarnign();
+
 var testClientReuse = function() {
   var c = client();
   var limit = 100000;