query cancellation test

This commit is contained in:
Christophe Macabiau 2011-11-02 16:35:09 +01:00
parent f3c8b972fe
commit c98ebb55e2

View File

@ -0,0 +1,47 @@
var helper = require(__dirname+"/test-helper");
var pg = require(__dirname + '/../../../lib');
//before running this test make sure you run the script create-test-tables
test("cancellation of a query", function() {
var client = helper.client();
var qry = client.query("select name from person order by name");
client.on('drain', client.end.bind(client));
var rows1 = 0, rows2 = 0, rows3 = 0, rows4 = 0;
var query1 = client.query(qry);
query1.on('row', function(row) {
rows1++;
});
var query2 = client.query(qry);
query2.on('row', function(row) {
rows2++;
});
var query3 = client.query(qry);
query3.on('row', function(row) {
rows3++;
});
var query4 = client.query(qry);
query4.on('row', function(row) {
rows4++;
});
pg.cancel(helper.connectionString, client, query1);
pg.cancel(helper.connectionString, client, query2);
pg.cancel(helper.connectionString, client, query4);
setTimeout(function() {
assert.equal(rows1, 0);
assert.equal(rows2, 0);
assert.equal(rows4, 0);
}, 2000);
assert.emits(query3, 'end', function() {
test("returned right number of rows", function() {
assert.equal(rows3, 26);
});
});
});