Merge branch 'master' of github.com:brianc/node-pg-cursor

brianc · brianc · commit 71dde045eaea · 2019-01-08T08:52:25.000-06:00
diff --git a/index.js b/index.js
@@ -94,7 +94,8 @@ Cursor.prototype._sendRows = function () {
   })
 }
 
-Cursor.prototype.handleCommandComplete = function () {
+Cursor.prototype.handleCommandComplete = function (msg) {
+  this._result.addCommandComplete(msg)
   this.connection.sync()
 }
 
diff --git a/test/index.js b/test/index.js
@@ -160,4 +160,18 @@ describe('cursor', function () {
       done()
     })
   })
+
+  it('returns rowCount on insert', function (done) {
+    var pgCursor = this.pgCursor
+    this.client.query('CREATE TEMPORARY TABLE pg_cursor_test (foo VARCHAR(1), bar VARCHAR(1))')
+      .then(function () {
+        var cursor = pgCursor('insert into pg_cursor_test values($1, $2)', ['a', 'b'])
+        cursor.read(1, function (err, rows, result) {
+          assert.ifError(err)
+          assert.equal(rows.length, 0)
+          assert.equal(result.rowCount, 1)
+          done()
+        })
+      }).catch(done)
+  })
 })

Original file line number	Diff line number	Diff line change
`@@ -94,7 +94,8 @@ Cursor.prototype._sendRows = function () {`
`94`	`94`	`})`
`95`	`95`	`}`
`96`	`96`
`97`		`-Cursor.prototype.handleCommandComplete = function () {`
	`97`	`+Cursor.prototype.handleCommandComplete = function (msg) {`
	`98`	`+ this._result.addCommandComplete(msg)`
`98`	`99`	`this.connection.sync()`
`99`	`100`	`}`
`100`	`101`