diff --git a/test/integration/client/simple-query-tests.js b/test/integration/client/simple-query-tests.js index 59972584..45aa846d 100644 --- a/test/integration/client/simple-query-tests.js +++ b/test/integration/client/simple-query-tests.js @@ -6,6 +6,8 @@ test("simple query interface", function() { var query = client.query("select name from person"); + client.on('drain', client.end.bind(client)); + var rows = []; query.on('row', function(row) { rows.push(row.fields[0]) @@ -19,8 +21,20 @@ test("simple query interface", function() { assert.equal(rows[0], "Aaron"); assert.equal(rows[25], "Zanzabar"); }); - client.end(); }); - +}); + +test("multiple simple queries", function() { + var client = helper.client(); + client.query("create temp table bang(id serial, name varchar(5));insert into bang(name) VALUES('boom');") + client.query("insert into bang(name) VALUES ('yes');"); + var query = client.query("select name from bang"); + assert.raises(query, 'row', function(row) { + assert.equal(row.fields[0], 'boom'); + assert.raises(query, 'row', function(row) { + assert.equal(row.fields[0],'yes'); + }); + }); + client.on('drain', client.end.bind(client)); });